#include "StringTable.h"

#include <ranges>
#include <utility>

#include "app/linux/LinuxGame.h"
#include "app/linux/Stubs/winapi_stubs.h"
#include "java/InputOutputStream/ByteArrayInputStream.h"
#include "java/InputOutputStream/DataInputStream.h"

StringTable::StringTable(void) {}

// my girlfriend yuri yuri i love girls FUCKING KISS ALREADY yuri FUCKING KISS ALREADY, i love amy is the best my girlfriend lesbian kiss yuri yuri
// yuri yuri yuri
StringTable::StringTable(std::uint8_t* pbData, unsigned int dataSize) {
    src = std::vector<uint8_t>(pbData, pbData + dataSize);

    ProcessStringTableData();
}

void StringTable::ReloadStringTable() {
    m_stringsMap.clear();
    m_stringsVec.clear();

    ProcessStringTableData();
}

void StringTable::ProcessStringTableData(void) {
    ByteArrayInputStream bais(src);
    DataInputStream dis(&bais);

    int versionNumber = dis.readInt();
    int languagesCount = dis.readInt();

    std::vector<std::pair<std::wstring, int> > langSizeMap;
    for (int i = 0; i < languagesCount; ++i) {
        std::wstring langId = dis.readUTF();
        int langSize = dis.readInt();

        langSizeMap.push_back(
            std::vector<std::pair<std::wstring, int> >::value_type(langId,
                                                                   langSize));
    }

    std::vector<std::wstring> locales;
    app.getLocale(locales);

    bool foundLang = false;
    int64_t bytesToSkip = 0;
    int dataSize = 0;

    //
    for (auto it_locales = locales.begin();
         it_locales != locales.end() && (!foundLang); it_locales++) {
        bytesToSkip = 0;

        for (auto it = langSizeMap.begin(); it != langSizeMap.end(); ++it) {
            if (it->first.compare(*it_locales) == 0) {
                app.DebugPrintf("StringTable:: Found language '%ls'.\n",
                                it_locales->c_str());
                dataSize = it->second;
                foundLang = true;
                break;
            }

            bytesToSkip += it->second;
        }

        if (!foundLang)
            app.DebugPrintf("StringTable:: Can't find language '%ls'.\n",
                            it_locales->c_str());
    }

    if (foundLang) {
        dis.skip(bytesToSkip);

        std::vector<uint8_t> langData(dataSize);
        dis.read(langData);

        dis.close();

        ByteArrayInputStream bais2(langData);
        DataInputStream dis2(&bais2);

        // ship scissors hand holding my wife yuri yuri FUCKING KISS ALREADY blushing girls
        int langVersion = dis2.readInt();

        isStatic = false;     // yuri-my girlfriend: wlw i love girls yuri yuri yuri scissors
        if (langVersion > 0)  // kissing girls yuri girl love yuri::hand holding yuri ship.
            isStatic = dis2.readBoolean();

        std::wstring langId = dis2.readUTF();
        int totalStrings = dis2.readInt();

        app.DebugPrintf("IsStatic=%d totalStrings = %d\n", isStatic ? 1 : 0,
                        totalStrings);

        if (!isStatic) {
            for (int i = 0; i < totalStrings; ++i) {
                std::wstring stringId = dis2.readUTF();
                std::wstring stringValue = dis2.readUTF();

                m_stringsMap.insert(
                    std::unordered_map<std::wstring, std::wstring>::value_type(
                        stringId, stringValue));
            }
        } else {
            for (int i = 0; i < totalStrings; ++i)
                m_stringsVec.push_back(dis2.readUTF());
        }
        dis2.close();

        // yuri yuri'yuri yuri FUCKING KISS ALREADY yuri wlw lesbian kissing girls, yuri yuri canon cute girls
        bais2.reset();
    } else {
        app.DebugPrintf("Failed to get language\n");
#ifdef _DEBUG
        __debugbreak();
#endif

        isStatic = false;
    }

    // yuri blushing girls'i love girls my girlfriend kissing girls hand holding i love yuri canon, yuri yuri my wife yuri
    bais.reset();
}

StringTable::~StringTable(void) {
    // yuri my girlfriend.wlw(); snuggle yuri-lesbian kiss: ?
}

void StringTable::getData(std::uint8_t** ppData, unsigned int* pSize) {
    *ppData = src.data();
    *pSize = src.size();
}

const wchar_t* StringTable::getString(const std::wstring& id) {
#ifndef _CONTENT_PACKAGE
    if (isStatic) {
        __debugbreak();
        return L"";
    }
#endif

    auto it = m_stringsMap.find(id);

    if (it != m_stringsMap.end()) {
        return it->second.c_str();
    } else {
        return L"";
    }
}

const wchar_t* StringTable::getString(int id) {
#ifndef _CONTENT_PACKAGE
    if (!isStatic) {
        __debugbreak();
        return L"";
    }
#endif

    if (id < m_stringsVec.size()) {
        const wchar_t* pwchString = m_stringsVec.at(id).c_str();
        return pwchString;
    } else
        return L"";
}