#pragma once

struct VIRTUAL_KEY
{
    LPCTSTR Name;
    int Key;
    LPCTSTR KeyName;
};

class CMenuShortCutKey
{
public:
    enum RUNNING_STATE
    {
        RUNNING_STATE_NOT_RUNNING = 1,
        RUNNING_STATE_WINDOWED = 2,
        RUNNING_STATE_FULLSCREEN = 3,
    };

    enum ACCESS_MODE
    {
        ACCESS_NONE = 0,
        ACCESS_GAME_NOT_RUNNING = 1,
        ACCESS_GAME_RUNNING_WINDOW = 2,
        ACCESS_NOT_IN_FULLSCREEN = 3,
        ACCESS_GAME_RUNNING_FULLSCREEN = 4,
        ACCESS_GAME_RUNNING = 6,
        ACCESS_ANYTIME = 7,
    };

public:
    CMenuShortCutKey(void);
    CMenuShortCutKey(WORD key, bool bCtrl, bool bAlt, bool bShift, ACCESS_MODE AccessMode, bool bUserAdded, bool bInactive);
    bool Match(WORD key, bool bCtrl, bool bAlt, bool bShift, RUNNING_STATE RunningState) const;
    bool Active(RUNNING_STATE RunningState) const;

    static VIRTUAL_KEY * VirtualKeyList(int & Size);
    static RUNNING_STATE RunningState(void);

    inline const std::wstring & Name(void) const
    {
        return m_ShortCutName;
    }
    inline WORD Key(void) const
    {
        return m_key;
    }
    inline bool Ctrl(void) const
    {
        return m_bCtrl;
    }
    inline bool Alt(void) const
    {
        return m_bAlt;
    }
    inline bool Shift(void) const
    {
        return m_bShift;
    }
    inline bool UserAdded(void) const
    {
        return m_bUserAdded;
    }
    inline bool Inactive(void) const
    {
        return m_bInactive;
    }
    inline ACCESS_MODE AccessMode(void) const
    {
        return m_AccessMode;
    }

    inline void SetInactive(bool Inactive)
    {
        m_bInactive = Inactive;
    }

private:
    static VIRTUAL_KEY m_VirtualKeyList[];

    std::wstring m_ShortCutName;
    WORD m_key;
    bool m_bCtrl;
    bool m_bAlt;
    bool m_bShift;
    ACCESS_MODE m_AccessMode;
    bool m_bUserAdded;
    bool m_bInactive;
};

class CShortCutItem
{
public:
    typedef std::list<CMenuShortCutKey> SHORTCUT_KEY_LIST;

private:
    typedef CMenuShortCutKey::ACCESS_MODE ACCESS_MODE;
    typedef CMenuShortCutKey::RUNNING_STATE RUNNING_STATE;

    ACCESS_MODE m_Access;
    LanguageStringID m_Section;
    LanguageStringID m_Title;
    SHORTCUT_KEY_LIST m_AccelList;

public:
    CShortCutItem(LanguageStringID Section, LanguageStringID Title, ACCESS_MODE Access);
    void Reset(LanguageStringID Section, LanguageStringID Title, ACCESS_MODE Access);
    void AddShortCut(WORD key, bool bCtrl, bool bAlt, bool bShift, ACCESS_MODE AccessMode, bool bUserAdded = false, bool bInactive = false);
    void RemoveItem(CMenuShortCutKey * ShortCut);
    bool Avaliable(RUNNING_STATE RunningState) const;

    inline const SHORTCUT_KEY_LIST & GetAccelItems(void) const
    {
        return m_AccelList;
    }
    inline LanguageStringID Section(void) const
    {
        return m_Section;
    }
    inline LanguageStringID Title(void) const
    {
        return m_Title;
    }
    inline ACCESS_MODE AccessMode(void) const
    {
        return m_Access;
    }
};

typedef std::map<int, CShortCutItem> MSC_MAP;

class CShortCuts
{
    typedef CShortCutItem::SHORTCUT_KEY_LIST SHORTCUT_KEY_LIST;
    typedef CMenuShortCutKey::ACCESS_MODE ACCESS_MODE;
    typedef CMenuShortCutKey::RUNNING_STATE RUNNING_STATE;
    typedef LanguageStringID LangStr;

public:
    CShortCuts(void);
    ~CShortCuts(void);

    std::wstring ShortCutString(int MenuID, RUNNING_STATE RunningState);
    LangStr GetMenuItemName(WORD key, bool bCtrl, bool bAlt, bool bShift, RUNNING_STATE RunningState);
    HACCEL GetAcceleratorTable(void);
    const MSC_MAP & GetShortCuts(void) const
    {
        return m_ShortCuts;
    }

    void Load(bool InitialValues = false);
    void Save(void);

private:
    CriticalSection m_CS;
    MSC_MAP m_ShortCuts;

    void AddShortCut(WORD ID, LangStr Section, LangStr LangID, CMenuShortCutKey::ACCESS_MODE AccessMode);
};