#include "Control.h"
#include "D2Ptrs.h"
#include "D2Helpers.h"
#include "Helpers.h"
#include "Constants.h"

Control* findControl(int Type, int LocaleID, int Disabled, int PosX, int PosY, int SizeX, int SizeY) {
    if (ClientState() != ClientStateMenu)
        return NULL;

    wchar_t* localeStr = NULL;
    if (LocaleID >= 0) {
        localeStr = _wcsdup(D2LANG_GetLocaleText((WORD)LocaleID));
        if (!localeStr)
            return NULL;
        Control* res = findControl(Type, localeStr, Disabled, PosX, PosY, SizeX, SizeY);
        free(localeStr);
        return res;
    }

    return NULL;
}

Control* findControl(int Type, const wchar_t* Text, int Disabled, int PosX, int PosY, int SizeX, int SizeY) {
    if (ClientState() != ClientStateMenu)
        return NULL;

    if (Type == -1 && Text == NULL && Disabled == -1 && PosX == -1 && PosY == -1 && SizeX == -1 && SizeY == -1)
        return *p_D2WIN_FirstControl;

    BOOL bFound = FALSE;

    for (Control* pControl = *p_D2WIN_FirstControl; pControl; pControl = pControl->pNext) {
        if (Type >= 0 && static_cast<int>(pControl->dwType) == Type)
            bFound = TRUE;
        else if (Type >= 0 && static_cast<int>(pControl->dwType) != Type) {
            bFound = FALSE;
            continue;
        }

        if (Disabled >= 0 && static_cast<int>(pControl->dwDisabled) == Disabled) {
            if (pControl->dwType == CONTROL_BUTTON && pControl->unkState == 1) {
                bFound = FALSE;
                continue;
            }
            bFound = TRUE;
        } else if (Disabled >= 0 && static_cast<int>(pControl->dwDisabled) != Disabled) {
            bFound = FALSE;
            continue;
        }

        if (PosX >= 0 && static_cast<int>(pControl->dwPosX) == PosX)
            bFound = TRUE;
        else if (PosX >= 0 && static_cast<int>(pControl->dwPosX) != PosX) {
            bFound = FALSE;
            continue;
        }

        if (PosY >= 0 && static_cast<int>(pControl->dwPosY) == PosY)
            bFound = TRUE;
        else if (PosY >= 0 && static_cast<int>(pControl->dwPosY) != PosY) {
            bFound = FALSE;
            continue;
        }

        if (SizeX >= 0 && static_cast<int>(pControl->dwSizeX) == SizeX)
            bFound = TRUE;
        else if (SizeX >= 0 && static_cast<int>(pControl->dwSizeX) != SizeX) {
            bFound = FALSE;
            continue;
        }

        if (SizeY >= 0 && static_cast<int>(pControl->dwSizeY) == SizeY)
            bFound = TRUE;
        else if (SizeY >= 0 && static_cast<int>(pControl->dwSizeY) != SizeY) {
            bFound = FALSE;
            continue;
        }

        if (Text && pControl->dwType == CONTROL_BUTTON) {
            if (!pControl->wText2)
                return NULL;
            if (wcscmp(pControl->wText2, Text) == 0) {
                bFound = TRUE;
            } else {
                bFound = FALSE;
                continue;
            }
        }

        if (Text && pControl->dwType == CONTROL_TEXTBOX) {
            if (pControl->pFirstText != NULL && pControl->pFirstText->wText[0] != NULL) {
                if (!pControl->pFirstText->wText[0])
                    return NULL;
                if (wcsstr(Text, pControl->pFirstText->wText[0]) != 0) {
                    bFound = TRUE;
                } else {
                    bFound = FALSE;
                    continue;
                }
            } else {
                bFound = FALSE;
                continue;
            }
        }
        if (bFound)
            return pControl;
    }

    return NULL;
}

bool clickControl(Control* pControl, int x, int y) {
    if (ClientState() != ClientStateMenu)
        return false;

    if (pControl) {
        if (x == -1)
            x = pControl->dwPosX + (pControl->dwSizeX / 2);
        if (y == -1)
            y = pControl->dwPosY - (pControl->dwSizeY / 2);

        Sleep(100);
        SendMouseClick(x, y, 0);
        Sleep(100);
        SendMouseClick(x, y, 1);
        Sleep(100);

        return true;
    }
    return false;
}

BOOL OOG_CreateCharacter(const wchar_t* szCharacter, int type, bool hardcore, bool ladder) {
    if (OOG_GetLocation() != OOG_CHAR_SELECT || wcslen(szCharacter) > 15 || type > 6 || type < 0)
        return FALSE;

    // click the create character button
    Control* ctrl = findControl(CONTROL_BUTTON, (const wchar_t*)NULL, -1, 33, 528, 168, 60);
    if (ctrl)
        clickControl(ctrl);

    // TODO: replace with real time checking code
    int i = 0;
    while ((OOG_GetLocation() != OOG_CHARACTER_CREATE) && i++ < 30)
        Sleep(100);

    int locs[7][5] = {
        {100, 337, 80, 330, 4011},  // amazon
        {626, 353, 600, 300, 4010}, // sorceress
        {301, 333, 300, 330, 4009}, // necromancer
        {521, 339, 500, 330, 4008}, // paladin
        {400, 330, 390, 330, 4007}, // barbarian
        {720, 370, 700, 370, 4012}, // druid
        {232, 364, 200, 300, 4013}, // assassin
    };

    ctrl = findControl(CONTROL_IMAGE, (const wchar_t*)NULL, -1, locs[type][0], locs[type][1], 88, 184);
    if (ctrl) {
        clickControl(ctrl, locs[type][2], locs[type][3]);
        Sleep(500);
        clickControl(ctrl, locs[type][2], locs[type][3]);
    }

    // verify that the correct type got selected
    ctrl = findControl(CONTROL_TEXTBOX, (const wchar_t*)NULL, -1, 0, 180, 800, 100);
    wchar_t* name = D2LANG_GetLocaleText((WORD)locs[type][4]);
    if (_wcsicmp(name, ctrl->pFirstText->wText[0]) != 0)
        return FALSE; // something bad happened?

    // set the name
    //	ctrl = findControl();
    // still need to find the name editbox, set the name, and click the various
    // checkboxes and the ok button

    return FALSE;
}

BOOL OOG_SelectCharacter(const wchar_t* szCharacter) {
    if (ClientState() != ClientStateMenu)
        return NULL;

    // Select the first control on the character selection screen.

    Control* pControl = findControl(CONTROL_TEXTBOX, (const wchar_t*)NULL, -1, 237, 178, 72, 93);
    ControlText* cText;

    while (pControl != NULL) {
        if (pControl->dwType == CONTROL_TEXTBOX && pControl->pFirstText != NULL && pControl->pFirstText->pNext != NULL)
            cText = pControl->pFirstText->pNext;
        else
            cText = NULL;

        if (cText != NULL) {
            if (!cText->wText[0])
                return FALSE;

            wchar_t* cLine = _wcsdup(cText->wText[0]);
            wchar_t* cCharacter = _wcsdup(szCharacter);
            StringToLower(cLine);
            StringToLower(cCharacter);

            if (wcslen(cLine) == wcslen(cCharacter) && wcsstr(cLine, cCharacter) != NULL) {
                free(cLine);
                free(cCharacter);
                if (!clickControl(pControl))
                    return FALSE;

                // OK Button
                // Bobode Sleep(7000);
                pControl = findControl(CONTROL_BUTTON, (const wchar_t*)NULL, -1, 627, 572, 128, 35);
                if (pControl) {
                    if (!clickControl(pControl))
                        return FALSE;

                    return TRUE;
                } else
                    return FALSE;

            } else {
                free(cLine);
                free(cCharacter);
            }
        }
        pControl = pControl->pNext;
    }
    return FALSE;
}

void SetControlText(Control* pControl, const wchar_t* Text) {
    if (ClientState() != ClientStateMenu)
        return;

    if (pControl && Text) {
        D2WIN_SetControlText(pControl, Text);
    }
}

BOOL OOG_SelectGateway(const wchar_t* szGateway, size_t strSize) {
    if (ClientState() != ClientStateMenu)
        return FALSE;

    if (wcsstr(szGateway, L"ERROR"))
        return FALSE;
    // Select the gateway control.
    Control* pControl = findControl(CONTROL_BUTTON, (const wchar_t*)NULL, -1, 264, 391, 272, 25);

    // if the control exists and has the text label, check if it matches the selected gateway
    if (pControl && pControl->wText2) {
        wchar_t* wzLine = _wcsdup(pControl->wText2);
        wchar_t* wzGate = _wcsdup(szGateway);
        StringToLower(wzLine);
        StringToLower(wzGate);

        if (wcsstr(wzLine, wzGate)) {
            // gateway is correct, do nothing and return true
            free(wzLine);
            free(wzGate);
            return TRUE;
        } else {
            free(wzLine);
            // gateway is NOT correct, change gateway to selected gateway if it exists
            // open the gateway select screen
            if (!clickControl(pControl))
                return FALSE;

            int index = 0;
            bool gatefound = false;

            // loop here till we find the right gateway if we can
            pControl = findControl(CONTROL_TEXTBOX, (const wchar_t*)NULL, -1, 257, 500, 292, 160);
            ControlText* cText;
            if (pControl && pControl->pFirstText) {
                cText = pControl->pFirstText;
                while (cText) {
                    wchar_t* wzGatelist = _wcsdup(cText->wText[0]);
                    if (!wzGatelist) {
                        free(wzGate);
                        return FALSE;
                    }

                    StringToLower(wzGatelist);
                    if (wcsstr(wzGatelist, wzGate)) {
                        // chosen gateway IS in the list and matches, cleanup and break the loop
                        free(wzGatelist);
                        free(wzGate);
                        gatefound = true;
                        break;
                    }
                    free(wzGatelist);
                    index++;
                    cText = cText->pNext;
                }
                if (gatefound) {
                    // click the correct gateway using the control plus a default x and a y based on (index*24)+12
                    if (!clickControl(pControl, -1, 344 + ((index * 24) + 12))) {
                        free(wzGate);
                        return FALSE;
                    }
                }
            }

            free(wzGate);

            // OK Button, gateway select screen
            pControl = findControl(CONTROL_BUTTON, (const wchar_t*)NULL, -1, 281, 538, 96, 32);
            if (pControl) {
                if (!clickControl(pControl))
                    return FALSE;
            } else
                return FALSE;

            return TRUE;
        }
    }
    return FALSE;
}

OOG_Location OOG_GetLocation(void) {
    if (ClientState() != ClientStateMenu)
        return OOG_NONE;

    if (findControl(CONTROL_BUTTON, 5103, -1, 330, 416, 128, 35))
        return OOG_MAIN_MENU_CONNECTING; // 21 Connecting to Battle.net
    else if (findControl(CONTROL_BUTTON, 5102, -1, 335, 412, 128, 35))
        return OOG_LOGIN_ERROR;                                       // 10 Login Error
    else if (findControl(CONTROL_BUTTON, 5102, -1, 351, 337, 96, 32)) // 5102 =OK
    {
        if (findControl(CONTROL_TEXTBOX, 5351, -1, 268, 320, 264, 120))
            return OOG_LOST_CONNECTION; // 17 lost connection
        else if (findControl(CONTROL_TEXTBOX, 5347, -1, 268, 320, 264, 120))
            return OOG_DISCONNECTED; // 14  Disconnected
        else if (findControl(6, (const wchar_t*)NULL, -1, 265, 206, 272, 35))
            return OOG_UNABLE_TO_CONNECT_TCPIP;
        else
            return OOG_CHARACTER_CREATE_ALREADY_EXISTS;                 // 30 Character Create - Dupe Name
    } else if (findControl(CONTROL_BUTTON, 5103, -1, 351, 337, 96, 32)) // 5103 = CANCEL
    {
        if (findControl(CONTROL_TEXTBOX, 5243, -1, 268, 300, 264, 100))
            return OOG_CHARACTER_SELECT_PLEASE_WAIT; // 16 char select please wait...
        if (findControl(CONTROL_TEXTBOX, (const wchar_t*)NULL, -1, 268, 320, 264, 120))
            return OOG_PLEASE_WAIT; // 25 "Please Wait..."single player already exists also
    } else if (findControl(CONTROL_BUTTON, 5103, -1, 433, 433, 96, 32)) {
        if (findControl(CONTROL_TEXTBOX, (const wchar_t*)NULL, -1, 427, 234, 300, 100))
            return OOG_INLINE; // 2 waiting in line
        else if (findControl(CONTROL_TEXTBOX, 10018, -1, 459, 380, 150, 12))
            return OOG_CREATE; // 4 Create game
        else if (findControl(CONTROL_BUTTON, 5119, -1, 594, 433, 172, 32))
            return OOG_JOIN; // 5 Join Game
        else if (findControl(CONTROL_BUTTON, 5102, -1, 671, 433, 96, 32))
            return OOG_CHANNEL; // 7 "Channel List"
        else
            return OOG_LADDER;                                          // 6 "Ladder"
    } else if (findControl(CONTROL_BUTTON, 5101, -1, 33, 572, 128, 35)) // 5101 = EXIT
    {
        if (findControl(CONTROL_BUTTON, 5288, -1, 264, 484, 272, 35))
            return OOG_LOGIN; // 9 Login
        if (findControl(CONTROL_BUTTON, 5102, -1, 495, 438, 96, 32))
            return OOG_CHARACTER_SELECT_CHANGE_REALM;                                                                             // 43 char select change realm
        if (findControl(CONTROL_BUTTON, 5102, -1, 627, 572, 128, 35) && findControl(CONTROL_BUTTON, 10832, -1, 33, 528, 168, 60)) // 10832=create new
        {
            if (findControl(CONTROL_BUTTON, 10018, -1, 264, 297, 272, 35)) // NORMAL
                return OOG_DIFFICULTY;                                     // 20 single char Difficulty
            Control* pControl = findControl(CONTROL_TEXTBOX, (const wchar_t*)NULL, -1, 37, 178, 200, 92);
            if (pControl && pControl->pFirstText && pControl->pFirstText->pNext)
                return OOG_CHAR_SELECT; // 12 char select
            else {
                if (findControl(CONTROL_TEXTBOX, 11162, -1, 45, 318, 531, 140) || findControl(CONTROL_TEXTBOX, 11066, -1, 45, 318, 531, 140))
                    return OOG_REALM_DOWN;
                // Look for CONNECTING... string
                else if (findControl(CONTROL_TEXTBOX, 11065, -1, 0x2D, 0x13E, 0x213, 0x8C))
                    return OOG_CONNECTING;
                else
                    return OOG_CHARACTER_SELECT_NO_CHARS; // 42 char info not loaded
            }
        }
        if (findControl(CONTROL_BUTTON, 5101, -1, 33, 572, 128, 35)) // 5101=Exit
        {
            if (findControl(CONTROL_BUTTON, 5102, 0, 627, 572, 128, 35)) // 5102=ok
                return OOG_CHARACTER_CREATE;                             // 29 char create screen with char selected
            else {
                if (findControl(CONTROL_TEXTBOX, 5226, -1, 321, 448, 300, 32))
                    return OOG_NEW_ACCOUNT; // 32 create new bnet account
                else
                    return OOG_NEW_CHARACTER; // 15 char create screen no char selected
            }
        }
    }
    if (findControl(CONTROL_BUTTON, 5102, -1, 335, 450, 128, 35)) {
        if (findControl(CONTROL_TEXTBOX, 5200, -1, 162, 270, 477, 50))
            return OOG_CDKEY_IN_USE;                                         // 19 CD-KEY in use
        else if (findControl(CONTROL_TEXTBOX, 5190, -1, 162, 420, 477, 100)) // 5190="If using a modem"
            return OOG_UNABLE_TO_CONNECT;                                    // 11 unable to connect
        else
            return OOG_INVALID_CDKEY; // 22 invalid CD-KEY
    } else if (findControl(CONTROL_TEXTBOX, 5159, -1, 438, 300, 326, 150))
        return OOG_GAME_DOES_NOT_EXIST; // 28 game doesn't exist
    else if (findControl(CONTROL_TEXTBOX, 5161, -1, 438, 300, 326, 150))
        return OOG_GAME_IS_FULL; // 38 Game is full
    else if (findControl(CONTROL_TEXTBOX, 5138, -1, 438, 300, 326, 150))
        return OOG_GAME_EXIST; // 26 Game already exists
    else if (findControl(CONTROL_TEXTBOX, 5139, -1, 438, 300, 326, 150))
        return OOG_SERVER_DOWN;                                        // 24 server down
    else if (findControl(CONTROL_BUTTON, 5106, -1, 264, 324, 272, 35)) // 5106="SINGLE PLAYER"
        return OOG_MAIN_MENU;                                          // 8 Main Menu
    else if (findControl(CONTROL_BUTTON, 11126, -1, 27, 480, 120, 20)) // 11126=ENTER CHAT
        return OOG_LOBBY;                                              // 1 base bnet
    else if (findControl(CONTROL_BUTTON, 5308, -1, 187, 470, 80, 20))  // 5308="HELP"
        return OOG_CHAT;                                               // 3 chat bnet
    else if (findControl(CONTROL_TEXTBOX, 21882, -1, 100, 580, 600, 80))
        return OOG_D2SPLASH; // 18 Spash
    else if (findControl(CONTROL_BUTTON, 5102, -1, 281, 538, 96, 32))
        return OOG_GATEWAY; // 27 select gateway
    else if (findControl(CONTROL_BUTTON, 5181, -1, 525, 513, 128, 35))
        return OOG_AGREE_TO_TERMS; // 31 agree to terms
    else if (findControl(CONTROL_BUTTON, 5102, -1, 525, 513, 128, 35))
        return OOG_PLEASE_READ; // 33 please read
    else if (findControl(CONTROL_BUTTON, 11097, -1, 265, 527, 272, 35))
        return OOG_REGISTER_EMAIL; // 34 register email
    else if (findControl(CONTROL_BUTTON, 5101, -1, 33, 578, 128, 35))
        return OOG_CREDITS; // 35 Credits
    else if (findControl(CONTROL_BUTTON, 5103, -1, 334, 488, 128, 35))
        return OOG_CINEMATICS; // 36 Cinematics
    else if (findControl(CONTROL_BUTTON, 5116, -1, 264, 350, 272, 35))
        return OOG_OTHER_MULTIPLAYER; // 39 other multi player
    else if (findControl(CONTROL_BUTTON, 5103, -1, 281, 337, 96, 32))
        return OOG_ENTER_IP_ADDRESS; // 41 enter ip
    else if (findControl(CONTROL_BUTTON, 5118, -1, 265, 206, 272, 35))
        return OOG_TCP_IP; // 40 tcp-ip

    return OOG_NONE;
}

bool OOG_CreateGame(const wchar_t* name, const wchar_t* pass, int difficulty) {
    if (ClientState() != ClientStateMenu)
        return FALSE;

    // reject name/password combinations over 15 characters
    if (!name || !pass || wcslen(name) > 15 || wcslen(pass) > 15)
        return FALSE;

    Control* pControl = NULL;

    // Battle.net/open game creation
    OOG_Location loc = OOG_GetLocation();
    if (!(loc == OOG_LOBBY || loc == OOG_CHAT || loc == OOG_DIFFICULTY || loc == OOG_CREATE))
        return FALSE;

    if (loc == OOG_DIFFICULTY) {
        // just click the difficulty button
        Control *normal = findControl(CONTROL_BUTTON, (const wchar_t*)NULL, -1, 264, 297, 272, 35),
                *nightmare = findControl(CONTROL_BUTTON, (const wchar_t*)NULL, -1, 264, 340, 272, 35),
                *hell = findControl(CONTROL_BUTTON, (const wchar_t*)NULL, -1, 264, 383, 272, 35);

        switch (difficulty) {
        case 0: // normal button
            if (normal->dwDisabled != 0x0d || !clickControl(normal))
                return FALSE;
            break;
        case 1: // nightmare button
            if (nightmare->dwDisabled != 0x0d || !clickControl(nightmare))
                return FALSE;
            break;
        case 2: // hell button
            if (hell->dwDisabled != 0x0d || !clickControl(hell))
                return FALSE;
            break;
        case 3: // hardest difficulty available
            if (hell->dwDisabled != 0x0d) {
                if (!clickControl(hell))
                    return FALSE;
            } else if (nightmare->dwDisabled != 0x0d) {
                if (!clickControl(nightmare))
                    return FALSE;
            } else if (normal->dwDisabled != 0x0d) {
                if (!clickControl(normal))
                    return FALSE;
            }
            break;
        default:
            return FALSE;
        }
    } else {
        // Create button
        if (loc != OOG_CREATE) {
            pControl = findControl(CONTROL_BUTTON, (const wchar_t*)NULL, -1, 533, 469, 120, 20);
            if (!pControl || !clickControl(pControl))
                return FALSE;
            Sleep(100);
        }
        if (OOG_GetLocation() == OOG_CREATE) {
            // Game name edit box
            if (name)
                SetControlText(findControl(1, (const wchar_t*)NULL, -1, 432, 162, 158, 20), name);
            else
                return FALSE;
            Sleep(100);

            // Password edit box
            if (pass)
                SetControlText(findControl(1, (const wchar_t*)NULL, -1, 432, 217, 158, 20), pass);
            else
                return FALSE;
            Sleep(100);
            Control *normal = findControl(CONTROL_BUTTON, (const wchar_t*)NULL, -1, 430, 381, 16, 16),
                    *nightmare = findControl(CONTROL_BUTTON, (const wchar_t*)NULL, -1, 555, 381, 16, 16),
                    *hell = findControl(CONTROL_BUTTON, (const wchar_t*)NULL, -1, 698, 381, 16, 16);

            switch (difficulty) {
            case 0: // normal button
                if (normal->dwDisabled == 0x4 || !clickControl(normal))
                    return FALSE;
                break;
            case 1: // nightmare button
                if (nightmare->dwDisabled == 0x4 || !clickControl(nightmare))
                    return FALSE;
                break;
            case 2: // hell button
                if (hell->dwDisabled == 0x4 || !clickControl(hell))
                    return FALSE;
                break;
            case 3: // hardest difficulty available
                if (hell->dwDisabled != 0x4) {
                    if (!clickControl(hell))
                        return FALSE;
                } else if (nightmare->dwDisabled != 0x4) {
                    if (!clickControl(nightmare))
                        return FALSE;
                } else if (normal->dwDisabled != 0x4) {
                    if (!clickControl(normal))
                        return FALSE;
                }
                break;
            default:
                return FALSE;
            }

            // Create Game Button
            pControl = findControl(CONTROL_BUTTON, (const wchar_t*)NULL, -1, 594, 433, 172, 32);
            if (!pControl || !clickControl(pControl))
                return FALSE;
        }
    }

    return TRUE;
}

bool OOG_JoinGame(const wchar_t* name, const wchar_t* pass) {
    if (ClientState() != ClientStateMenu)
        return FALSE;

    // reject name/password combinations over 15 characters
    if (wcslen(name) > 15 || wcslen(pass) > 15)
        return FALSE;

    Control* pControl = NULL;

    // Battle.net/open lobby/chat area
    if (!(OOG_GetLocation() == OOG_LOBBY || OOG_GetLocation() == OOG_CHAT || OOG_GetLocation() == OOG_JOIN))
        return FALSE;

    // JOIN button
    if (OOG_GetLocation() != OOG_JOIN) {
        pControl = findControl(CONTROL_BUTTON, (const wchar_t*)NULL, -1, 652, 469, 120, 20);
        if (!pControl || !clickControl(pControl))
            return FALSE;
        Sleep(100);
    }
    if (OOG_GetLocation() == OOG_JOIN) {
        // Game name edit box
        if (name)
            SetControlText(findControl(1, (const wchar_t*)NULL, -1, 432, 148, 155, 20), name);
        else
            return FALSE;
        // Password edit box
        if (pass)
            SetControlText(findControl(1, (const wchar_t*)NULL, -1, 606, 148, 155, 20), pass);
        else
            return FALSE;

        // Join Game Button
        pControl = findControl(CONTROL_BUTTON, (const wchar_t*)NULL, -1, 594, 433, 172, 32);
        if (!pControl || !clickControl(pControl))
            return FALSE;
    }

    return TRUE;
}