#pragma once

#define WIN32_LEAN_AND_MEAN
#include <windows.h>
#include "vftable.h"
#include "Common.h"

#ifdef USE_ST6
#include "st6.h"
#else
#include <vector>
namespace st6 = std;
#endif

struct InputChar
{
    WCHAR c;
    DWORD flags; // I don't know whether this field actually represents flags; it's just an educated guess.
    DWORD unk; // Allocated but never assigned.
};

struct KeyMapInfo
{
    BYTE x00[0x8];
    DWORD controlCharacterFlags; // 0x8
    DWORD x0C;
    WCHAR enteredKey; // 0x10

    inline bool IsCtrlPressed() const
    {
        return (controlCharacterFlags & 4) == 4;
    }
};

struct InputBoxWindow
{
    BYTE x04[0x498];
    int pos; // 0x49C
    BYTE x4A0[0x24];
    st6::vector<InputChar> chars; // 0x4C4

    BYTE x4D4[0x3C];
    int maxCharsLength; // 0x510
    BYTE x514[0x14];
    WCHAR forbiddenChar; // 0x528
    bool noForbiddenChar; // 0x52A
    PDWORD ime; // 0x52C

    FILL_VFTABLE(0)
    FILL_VFTABLE(1)
    FILL_VFTABLE(2)
    FILL_VFTABLE(3)
    FILL_VFTABLE(4)
    FILL_VFTABLE(5)
    FILL_VFTABLE(6)
    FILL_VFTABLE(7)
    FILL_VFTABLE(8)
    FILL_VFTABLE(9)
    FILL_VFTABLE(A)
    FILL_VFTABLE(B)
    virtual void Vftable_xC0();
    virtual void Vftable_xC4();
    virtual bool WriteTypedKey(const KeyMapInfo& kmi);

    void HandleCopyPaste(const KeyMapInfo& kmi);
    void CopyToClipboard();
    void CopyFromClipboard();
    void WriteString(LPCWSTR str);
};

void HandleDefaultInputKey_Hook();
void InitCopyPasteFeature();