thunderbrew/src/ui/ScriptFunctionsCharCreate.cpp

#include "ui/ScriptFunctions.hpp"
#include "ui/CSimpleModelFFX.hpp"
#include "ui/FrameScript.hpp"
#include "ui/Types.hpp"
#include "util/Lua.hpp"
#include "util/Unimplemented.hpp"
#include "util/SFile.hpp"
#include "db/Db.hpp"
#include "clientobject/Unit_C.hpp"
#include "glue/CCharacterCreation.hpp"
#include "glue/CCharacterComponent.hpp"
#include "glue/CCharacterSelection.hpp"
#include "client/ClientServices.hpp"
#include <cstdint>

int32_t Script_SetCharCustomizeFrame(lua_State* L) {
    if (!lua_isstring(L, 1)) {
        return luaL_error(L, "Usage: SetCharCustomizeFrame(\"frameName\")");
    }

    auto type = CSimpleModel::GetObjectType();
    auto name = lua_tolstring(L, 1, nullptr);
    auto frame = CScriptObject::GetScriptObjectByName(name, type);

    if (frame) {
        CCharacterCreation::SetCharCustomizeFrame(static_cast<CSimpleModelFFX*>(frame));
    }

    return 0;
}

int32_t Script_SetCharCustomizeBackground(lua_State* L) {
    if (!lua_isstring(L, 1)) {
        return luaL_error(L, "Usage: SetCharCustomizeBackground(\"filename\")");
    }

    auto filename = lua_tolstring(L, 1, nullptr);
    CCharacterCreation::SetCharCustomizeModel(filename);

    return 0;
}

int32_t Script_ResetCharCustomize(lua_State*) {
    CCharacterCreation::ResetCharCustomizeInfo();
    return 0;
}

int32_t Script_GetNameForRace(lua_State* L) {
    auto raceID = CCharacterCreation::m_character->m_data.m_info.raceID;
    auto sexID = CCharacterCreation::m_character->m_data.m_info.sexID;
    auto record = g_chrRacesDB.GetRecord(raceID);
    auto raceName = CGUnit_C::GetDisplayRaceNameFromRecord(record, sexID);
    if (record && raceName) {
        lua_pushstring(L, raceName);
        lua_pushstring(L, record->m_clientFileString);
    } else {
        lua_pushnil(L);
        lua_pushnil(L);
    }

    return 2;
}

int32_t Script_GetFactionForRace(lua_State* L) {
    if (!lua_isnumber(L, 1)) {
        return luaL_error(L, "Usage: GetFactionForRace(index)");
    }

    uint32_t index = static_cast<uint32_t>(lua_tonumber(L, 1)) - 1;
    int32_t raceID = 0;
    if (index < CCharacterCreation::m_races.Count()) {
        raceID = CCharacterCreation::m_races[index];
    }

    auto raceRecord = g_chrRacesDB.GetRecord(raceID);
    if (raceRecord) {
        auto factionTemplateRecord = g_factionTemplateDB.GetRecord(raceRecord->m_factionID);
        if (factionTemplateRecord) {
            for (int32_t i = 0; i < g_factionGroupDB.GetNumRecords(); ++i) {
                auto factionGroupRecord = g_factionGroupDB.GetRecordByIndex(i);
                if (!factionGroupRecord || factionGroupRecord->m_maskID == 0) {
                    continue;
                }

                if (((1 << factionGroupRecord->m_maskID) & factionTemplateRecord->m_factionGroup) != 0) {
                    lua_pushstring(L, factionGroupRecord->m_name);
                    lua_pushstring(L, factionGroupRecord->m_internalName);
                    return 2;
                }
            }
        }
    }

    lua_pushnil(L);
    lua_pushnil(L);
    return 2;
}

int32_t Script_GetAvailableRaces(lua_State* L) {
    auto sexID = CCharacterCreation::m_character->m_data.m_info.sexID;

    for (uint32_t i = 0; i < CCharacterCreation::m_races.Count(); ++i) {
        auto raceRecord = g_chrRacesDB.GetRecord(CCharacterCreation::m_races[i]);
        auto raceName = CGUnit_C::GetDisplayRaceNameFromRecord(raceRecord, sexID);

        lua_pushstring(L, raceName);
        if (raceRecord) {
            lua_pushstring(L, raceRecord->m_clientFileString);
            bool available = ClientServices::GetExpansionLevel() >= raceRecord->m_requiredExpansion;
            lua_pushnumber(L, available ? 1.0 : 0.0);
        } else {
            lua_pushstring(L, nullptr);
            lua_pushnumber(L, 0.0);
        }
    }
    return CCharacterCreation::m_races.Count() * 3;
}

int32_t Script_GetAvailableClasses(lua_State* L) {
    auto sexID = CCharacterCreation::m_character->m_data.m_info.sexID;

    for (int32_t i = 0; i < g_chrClassesDB.GetNumRecords(); ++i) {
        auto record = g_chrClassesDB.GetRecordByIndex(i);
        auto className = CGUnit_C::GetDisplayClassNameFromRecord(record, sexID);

        if (className) {
            lua_pushstring(L, className);
            lua_pushstring(L, record->m_filename);
            bool available = ClientServices::GetExpansionLevel() >= record->m_requiredExpansion;
            lua_pushnumber(L, available ? 1.0 : 0.0);
        } else {
            lua_pushnil(L);
            lua_pushnil(L);
            lua_pushnil(L);
        }
    }
    return g_chrClassesDB.GetNumRecords() * 3;
}

int32_t Script_GetClassesForRace(lua_State* L) {
    auto sexID = CCharacterCreation::m_character->m_data.m_info.sexID;

    for (uint32_t i = 0; i < CCharacterCreation::m_classes.Count(); ++i) {
        auto record = CCharacterCreation::m_classes[i];
        auto className = CGUnit_C::GetDisplayClassNameFromRecord(record, sexID);

        if (className) {
            lua_pushstring(L, className);
            lua_pushstring(L, record->m_filename);
            bool available = ClientServices::GetExpansionLevel() >= record->m_requiredExpansion;
            lua_pushnumber(L, available ? 1.0 : 0.0);
        } else {
            lua_pushnil(L);
            lua_pushnil(L);
            lua_pushnil(L);
        }
    }
    return CCharacterCreation::m_classes.Count() * 3;
}

int32_t Script_GetHairCustomization(lua_State* L) {
    int32_t raceID = 0;

    if (CCharacterCreation::m_raceIndex >= 0 && CCharacterCreation::m_raceIndex < CCharacterCreation::m_races.Count()) {
        raceID = CCharacterCreation::m_races[CCharacterCreation::m_raceIndex];
    }

    auto record = g_chrRacesDB.GetRecord(raceID);
    lua_pushstring(L, record ? record->m_hairCustomization : "NORMAL");
    return 1;
}

int32_t Script_GetFacialHairCustomization(lua_State* L) {
    int32_t raceID = 0;

    if (CCharacterCreation::m_raceIndex >= 0 && CCharacterCreation::m_raceIndex < CCharacterCreation::m_races.Count()) {
        raceID = CCharacterCreation::m_races[CCharacterCreation::m_raceIndex];
    }

    auto record = g_chrRacesDB.GetRecord(raceID);
    if (record) {
        auto sexID = CCharacterCreation::m_character->m_data.m_info.sexID;
        lua_pushstring(L, record->m_facialHairCustomization[sexID]);
    } else {
        lua_pushstring(L, "NORMAL");
    }
    return 1;
}

int32_t Script_GetSelectedRace(lua_State* L) {
    lua_pushnumber(L, CCharacterCreation::m_raceIndex + 1.0);
    return 1;
}

int32_t Script_GetSelectedSex(lua_State* L) {
    auto sexID = CCharacterCreation::m_character->m_data.m_info.sexID;
    lua_pushnumber(L, g_glueFrameScriptGenders[sexID]);
    return 1;
}

int32_t Script_GetSelectedClass(lua_State* L) {
    auto record = g_chrClassesDB.GetRecord(CCharacterCreation::m_selectedClassID);
    if (!record) {
        return 0;
    }
    auto className = CGUnit_C::GetDisplayClassNameFromRecord(
        record,
        CCharacterCreation::m_character->m_data.m_info.sexID);

    lua_pushstring(L, className);
    lua_pushstring(L, record->m_filename);

    int32_t index = 0;

    for (int32_t i = 0; i < g_chrClassesDB.GetNumRecords(); ++i) {
        record = g_chrClassesDB.GetRecordByIndex(i);
        if (record && record->m_ID == CCharacterCreation::m_selectedClassID) {
            index = i;
        }
    }

    lua_pushnumber(L, index + 1);
    // TODO: uint8_t roles = CGLookingForGroup::GetClassRoles(CCharacterCreation::m_selectedClassID);
    uint8_t roles = 0;
    lua_pushboolean(L, roles & 2);
    lua_pushboolean(L, roles & 4);
    lua_pushboolean(L, roles & 8);
    return 6;
}

int32_t Script_SetSelectedRace(lua_State* L) {
    if (!lua_isnumber(L, 1)) {
        return luaL_error(L, "Usage: SetSelectedRace(index)");
    }

    int32_t raceID = static_cast<int32_t>(lua_tonumber(L, 1)) - 1;
    CCharacterCreation::SetSelectedRace(raceID);
    return 0;
}

int32_t Script_SetSelectedSex(lua_State* L) {
    if (!lua_isnumber(L, 1)) {
        return luaL_error(L, "Usage: SetSelectedSex(index)");
    }

    int32_t sexID = static_cast<int32_t>(lua_tonumber(L, 1));
    for (int32_t i = 0; i < 3; ++i) {
        if (g_glueFrameScriptGenders[i] == sexID) {
            CCharacterCreation::SetSelectedSex(i);
        }
    }
    return 0;
}

int32_t Script_SetSelectedClass(lua_State* L) {
    if (!lua_isnumber(L, 1)) {
        return luaL_error(L, "Usage: SetSelectedSex(index)");
    }

    int32_t index = static_cast<int32_t>(lua_tonumber(L, 1)) - 1;

    // NOTICE: Original client has access violation issue in this method
    auto record = g_chrClassesDB.GetRecordByIndex(index);
    CCharacterCreation::SetSelectedClass(record ? record->m_ID : 0);
    return 0;
}

int32_t Script_UpdateCustomizationBackground(lua_State*) {
    CCharacterCreation::SetSelectedRace(CCharacterCreation::m_raceIndex);
    return 0;
}

int32_t Script_UpdateCustomizationScene(lua_State*) {
    CCharacterCreation::m_character->RenderPrep(0);
    return 0;
}

int32_t Script_CycleCharCustomization(lua_State* L) {
    if (!lua_isnumber(L, 1) || !lua_isnumber(L, 2)) {
        return luaL_error(L, "Usage: CycleCharCustomization(index, delta)");
    }

    using CHAR_CUSTOMIZATION_TYPE = CCharacterCreation::CHAR_CUSTOMIZATION_TYPE;

    auto customization = static_cast<CHAR_CUSTOMIZATION_TYPE>(lua_tonumber(L, 1) - 1.0);
    int32_t delta = static_cast<int32_t>(lua_tonumber(L, 2));

    CCharacterCreation::CycleCharCustomization(customization, delta);
    return 0;
}

int32_t Script_RandomizeCharCustomization(lua_State*) {
    CCharacterCreation::RandomizeCharCustomization();
    return 0;
}

int32_t Script_GetCharacterCreateFacing(lua_State* L) {
    lua_pushnumber(L, CCharacterCreation::m_charFacing * 57.29578);
    return 1;
}

int32_t Script_SetCharacterCreateFacing(lua_State* L) {
    if (!lua_isnumber(L, 1)) {
        return luaL_error(L, "Usage: SetCharacterCreateFacing(degrees)");
    }

    // Degree to Radian
    float facing = lua_tonumber(L, 1) * 0.017453292;
    CCharacterCreation::SetCharFacing(facing);
    return 1;
}

int32_t Script_GetRandomName(lua_State* L) {
    // TODO: Proper implementation
    // WORKAROUND:
    lua_pushstring(L, "RandomName");
    return 1;
}

int32_t Script_CreateCharacter(lua_State* L) {
    if (lua_isstring(L, 1)) {
        lua_tolstring(L, 1, nullptr);
    }

    CCharacterCreation::CreateCharacter(lua_tolstring(L, 1, nullptr));
    return 0;
}

int32_t Script_CustomizeExistingCharacter(lua_State* L) {
    if (!lua_isnumber(L, 1)) {
        return luaL_error(L, "Usage: CustomizeExistingCharacter(index)");
    }

    uint32_t index = static_cast<uint32_t>(lua_tonumber(L, 1)) - 1;
    CCharacterCreation::SetToExistingCharacter(index);
    return 0;
}

int32_t Script_PaidChange_GetPreviousRaceIndex(lua_State* L) {
    WHOA_UNIMPLEMENTED(0);
}

int32_t Script_PaidChange_GetCurrentRaceIndex(lua_State* L) {
    WHOA_UNIMPLEMENTED(0);
}

int32_t Script_PaidChange_GetCurrentClassIndex(lua_State* L) {
    WHOA_UNIMPLEMENTED(0);
}

int32_t Script_PaidChange_GetName(lua_State* L) {
    WHOA_UNIMPLEMENTED(0);
}

int32_t Script_IsRaceClassValid(lua_State* L) {
    if (!lua_isnumber(L, 1) || !lua_isnumber(L, 2)) {
        return luaL_error(L, "Usage: IsRaceClassValid(raceIndex, classIndex)");
    }

    uint32_t raceIndex = static_cast<uint32_t>(lua_tonumber(L, 1)) - 1;
    int32_t classIndex = static_cast<int32_t>(lua_tonumber(L, 2)) - 1;

    auto classRecord = g_chrClassesDB.GetRecordByIndex(classIndex);
    if (!classRecord) {
        return 0;
    }

    int32_t raceID = 0;
    if (raceIndex < CCharacterCreation::m_races.Count()) {
        raceID = CCharacterCreation::m_races[raceIndex];
    }

    if (CCharacterCreation::IsRaceClassValid(raceID, classRecord->m_ID)) {
        lua_pushnumber(L, 1.0);
    } else {
        lua_pushnil(L);
    }
    return 1;
}

int32_t Script_IsRaceClassRestricted(lua_State* L) {
    if (!lua_isnumber(L, 1) || !lua_isnumber(L, 2)) {
        return luaL_error(L, "Usage: Script_IsRaceClassRestricted(raceID, classID)");
    }

    int32_t raceID = static_cast<int32_t>(lua_tonumber(L, 1));
    int32_t classID = static_cast<int32_t>(lua_tonumber(L, 2));

    classID = (1 << classID);

    bool restricted = false;
    switch (raceID) {
    case 1:
        restricted = (CCharacterSelection::m_restrictHuman & classID) != 0;
        break;
    case 2:
        restricted = (CCharacterSelection::m_restrictOrc & classID) != 0;
        break;
    case 3:
        restricted = (CCharacterSelection::m_restrictDwarf & classID) != 0;
        break;
    case 4:
        restricted = (CCharacterSelection::m_restrictNightElf & classID) != 0;
        break;
    case 5:
        restricted = (CCharacterSelection::m_restrictUndead & classID) != 0;
        break;
    case 6:
        restricted = (CCharacterSelection::m_restrictTauren & classID) != 0;
        break;
    case 7:
        restricted = (CCharacterSelection::m_restrictGnome & classID) != 0;
        break;
    case 8:
        restricted = (CCharacterSelection::m_restrictTroll & classID) != 0;
        break;
    case 10:
        restricted = (CCharacterSelection::m_restrictBloodElf & classID) != 0;
        break;
    case 11:
        restricted = (CCharacterSelection::m_restrictDraenei & classID) != 0;
        break;
    default:
        luaL_error(L, "Script_IsRaceClassRestricted: unsupported race ID(%d)", raceID);
        break;
    }

    lua_pushnumber(L, restricted ? 1.0 : 0.0);
    return 1;
}

int32_t Script_GetCreateBackgroundModel(lua_State* L) {
    if (SFile::IsTrial()) {
        lua_pushstring(L, "CharacterSelect");
        return 1;
    }

    if (CCharacterCreation::m_selectedClassID == 6) {
        auto record = g_chrClassesDB.GetRecord(6);
        if (record) {
            lua_pushstring(L, record->m_filename);
            return 1;
        }
    } else {
        int32_t raceID = CCharacterCreation::m_character->m_data.m_info.raceID;
        if (raceID == 7) {
            raceID = 3;
        } else if (raceID == 8) {
            raceID = 2;
        }

        auto record = g_chrRacesDB.GetRecord(raceID);
        if (record) {
            lua_pushstring(L, record->m_clientFileString);
            return 1;
        }
    }
    lua_pushstring(L, "");
    return 1;
}

FrameScript_Method FrameScript::s_ScriptFunctions_CharCreate[NUM_SCRIPT_FUNCTIONS_CHAR_CREATE] = {
    { "SetCharCustomizeFrame",          &Script_SetCharCustomizeFrame },
    { "SetCharCustomizeBackground",     &Script_SetCharCustomizeBackground },
    { "ResetCharCustomize",             &Script_ResetCharCustomize },
    { "GetNameForRace",                 &Script_GetNameForRace },
    { "GetFactionForRace",              &Script_GetFactionForRace },
    { "GetAvailableRaces",              &Script_GetAvailableRaces },
    { "GetAvailableClasses",            &Script_GetAvailableClasses },
    { "GetClassesForRace",              &Script_GetClassesForRace },
    { "GetHairCustomization",           &Script_GetHairCustomization },
    { "GetFacialHairCustomization",     &Script_GetFacialHairCustomization },
    { "GetSelectedRace",                &Script_GetSelectedRace },
    { "GetSelectedSex",                 &Script_GetSelectedSex },
    { "GetSelectedClass",               &Script_GetSelectedClass },
    { "SetSelectedRace",                &Script_SetSelectedRace },
    { "SetSelectedSex",                 &Script_SetSelectedSex },
    { "SetSelectedClass",               &Script_SetSelectedClass },
    { "UpdateCustomizationBackground",  &Script_UpdateCustomizationBackground },
    { "UpdateCustomizationScene",       &Script_UpdateCustomizationScene },
    { "CycleCharCustomization",         &Script_CycleCharCustomization },
    { "RandomizeCharCustomization",     &Script_RandomizeCharCustomization },
    { "GetCharacterCreateFacing",       &Script_GetCharacterCreateFacing },
    { "SetCharacterCreateFacing",       &Script_SetCharacterCreateFacing },
    { "GetRandomName",                  &Script_GetRandomName },
    { "CreateCharacter",                &Script_CreateCharacter },
    { "CustomizeExistingCharacter",     &Script_CustomizeExistingCharacter },
    { "PaidChange_GetPreviousRaceIndex", &Script_PaidChange_GetPreviousRaceIndex },
    { "PaidChange_GetCurrentRaceIndex", &Script_PaidChange_GetCurrentRaceIndex },
    { "PaidChange_GetCurrentClassIndex", &Script_PaidChange_GetCurrentClassIndex },
    { "PaidChange_GetName",             &Script_PaidChange_GetName },
    { "IsRaceClassValid",               &Script_IsRaceClassValid },
    { "IsRaceClassRestricted",          &Script_IsRaceClassRestricted },
    { "GetCreateBackgroundModel",       &Script_GetCreateBackgroundModel }
};