#include "CommonHeaders.h"
#include "GameFixes.h"
#include "GameInformations.h"
#include "ObfuscateString.h"
#include "Memory.hpp"
#include "Maths.hpp"
#include "UEngine.hpp"
#include "SDK/Basic.hpp"
#include "SDK/Engine_classes.hpp"

using namespace SDK;

// Constants
const std::string PLUGIN_NAME = "JediSurvivor";
const std::string PLUGIN_LOG = PLUGIN_NAME + ".log";

// Logger
std::shared_ptr<spdlog::logger> logger;

// Plugin states
static bool AOBScanDone = false;
static bool g_fix_enabled = false;
static bool g_fov_fix_enabled = false;
static bool g_ultrawide_fix_enabled = false;
static bool g_camera_fix_enabled = false;
static bool g_DOF_fix_enabled = false;
static bool g_Vignetting_fix_enabled = false;
static bool g_Fog_fix_enabled = false;
static bool g_VolumetricFog_fix_enabled = false;
static bool g_sharpening_fix_enabled = false;
static bool g_Console = false;
static int g_AdditionalFOVValue = 0;
static int g_CinematicsFOVValue = 0;
static float g_CameraMultiplier = 1.f;
static float g_sharpeningValue = 1.f;

// Shared values
static float g_FOV_In = 60.f;
static float g_FOV_Out = 60.f;
static bool g_Console_Enabled = false;

// AOB Unreal Engine offsets addresses
static uint8_t* GObjectsaddress = nullptr;
static uint8_t* AppendStringaddress = nullptr;
static uint8_t* ProcessEventaddress = nullptr;

// AOB Scan pointers
static uint8_t* FOVaddress = nullptr;
static uint8_t* CinematicsFOVaddress = nullptr;
static uint8_t* Cameraaddress = nullptr;
static uint8_t* DOFaddress = nullptr;
static uint8_t* CAaddress = nullptr;
static uint8_t* Vignettingaddress = nullptr;
static uint8_t* VolumetricFogaddress = nullptr;
static uint8_t* Fogaddress = nullptr;
static uint8_t* CameraComponentaddress = nullptr;
static uint8_t* ConstrainAspectRatioaddress = nullptr;
static uint8_t* Sharpeningaddress = nullptr;

// Hooking
static SafetyHookMid FOVHook{};
static SafetyHookMid CinematicsFOVHook{};
static SafetyHookMid FogHook{};
static SafetyHookMid SharpeningHook{};

// VEH debugger handles
PVOID vehCameraHandle = nullptr;

// Prototypes
static void FOVFixEnabled();
static void UltraWideFixEnabled();
static void CameraFixEnabled();
static void DOFFixEnabled();
static void VignettingFixEnabled();
static void FogFixEnabled();
static void VolumetricFogFixEnabled();
static void SharpeningFixEnabled();
static void EnableConsole();

extern "C" __declspec(dllexport) void SetFixEnabled(bool enabled, bool init)
{
    g_fix_enabled = enabled;
    if (g_fix_enabled && !AOBScanDone) {
        logger->info("--------------- AOB scan started ---------------");
        if (CameraComponentaddress == nullptr) { // Unreal Engine 4.26.2
            constexpr auto CameraComponentStringObfuscated = make_obfuscated<0x4A>("EB ?? F3 0F ?? ?? ?? ?? ?? ?? F3 0F ?? ?? ?? 8B 83");
            CameraComponentaddress = Memory::AOBScan("", CameraComponentStringObfuscated.decrypt(), PAGE_EXECUTE_READ, logger);

            if (!CameraComponentaddress)
                logger->warn("Camera component signature not found. Maybe your game has been updated and is no more compatible with this plugin.");
            else {
                logger->info("Camera component found at address: 0x{:X}.", reinterpret_cast<uintptr_t>(CameraComponentaddress));
                ConstrainAspectRatioaddress = CameraComponentaddress +0x21;
            }
        }

        if (!FOVaddress) {
            constexpr auto FOVStringObfuscated = make_obfuscated<0x4A>("77 ?? 48 ?? ?? FF 90 ?? ?? ?? ?? F3 0F ?? ?? ?? 48");
            FOVaddress = Memory::AOBScan("", FOVStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

            if (!FOVaddress)
                logger->warn("FOV signature not found. Maybe your game has been updated and is no more compatible with this plugin.");
            else
                logger->info("FOV found at address: 0x{:X}.", reinterpret_cast<uintptr_t>(FOVaddress));
        }

        if (!CinematicsFOVaddress) {
            constexpr auto CinematicsFOVStringObfuscated = make_obfuscated<0x4A>("F3 0F ?? ?? ?? ?? ?? ?? EB ?? 0F ?? ?? F3 0F ?? ?? ?? ?? ?? ?? 0F ?? ?? F3 0F ?? ?? ?? ?? ?? ?? 76");
            CinematicsFOVaddress = Memory::AOBScan("", CinematicsFOVStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

            if (!CinematicsFOVaddress)
                logger->warn("Cinematics FOV signature not found. Maybe your game has been updated and is no more compatible with this plugin.");
            else
                logger->info("Cinematics FOV found at address: 0x{:X}.", reinterpret_cast<uintptr_t>(CinematicsFOVaddress));
        }

        if (!Cameraaddress) {
            constexpr auto CameraStringObfuscated = make_obfuscated<0x4A>("48 ?? ?? 48 ?? ?? ?? 48 ?? ?? ?? 48 ?? ?? ?? 57 48 81 ?? ?? ?? ?? ?? F3 0F ?? ?? ?? 49");
            Cameraaddress = Memory::AOBScan("", CameraStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

            if (!Cameraaddress)
                logger->warn("Camera not found. Maybe your game has been updated and is no more compatible with this plugin.");
            else
                logger->info("Camera found at address: 0x{:X}.", reinterpret_cast<uintptr_t>(Cameraaddress));
        }

        if (!DOFaddress) {
            constexpr auto DOFStringObfuscated = make_obfuscated<0x4A>("8B ?? ?? E8 ?? ?? ?? ?? 8B ?? E8 ?? ?? ?? ?? 84 ?? 74 ?? 48");
            DOFaddress = Memory::AOBScan("", DOFStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

            if (!DOFaddress)
                logger->warn("DOF signature not found. Maybe your game has been updated and is no more compatible with this plugin.");
            else
                logger->info("DOF signature found at address: 0x{:X}.", reinterpret_cast<uintptr_t>(DOFaddress));
        }

        if (!Vignettingaddress) {
            constexpr auto CAStringObfuscated = make_obfuscated<0x4A>("8B ?? 83 ?? ?? 7D ?? 89 AB ?? ?? ?? ?? 89 AB ?? ?? ?? ?? 89 AB");
            Vignettingaddress = Memory::AOBScan("", CAStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

            if (!Vignettingaddress)
                logger->warn("Vignetting signature not found. Maybe your game has been updated and is no more compatible with this plugin.");
            else
                logger->info("Vignetting signature found at address: 0x{:X}.", reinterpret_cast<uintptr_t>(Vignettingaddress));
        }

        if (!Fogaddress) {
            constexpr auto FogStringObfuscated = make_obfuscated<0x4A>("83 ?? ?? ?? 75 ?? B3 ?? EB ?? 32 ?? 48 8B ?? ?? ?? 48 ?? ?? 74 ?? E8 ?? ?? ?? ?? 84");
            Fogaddress = Memory::AOBScan("", FogStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

            if (!Fogaddress)
                logger->warn("Fog signature not found. Maybe your game has been updated and is no more compatible with this plugin.");
            else
                logger->info("Fog signature found at address: 0x{:X}.", reinterpret_cast<uintptr_t>(Fogaddress));
        }

        if (!VolumetricFogaddress){
            constexpr auto VolumetricFogStringObfuscated = make_obfuscated<0x4A>("74 ?? F7 47 ?? ?? ?? ?? ?? 74 ?? 83 BB");
            VolumetricFogaddress = Memory::AOBScan("", VolumetricFogStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

            if (!VolumetricFogaddress)
                logger->warn("Volumetric fog signature not found. Maybe your game has been updated and is no more compatible with this plugin.");
            else
                logger->info("Volumetric fog signature found at address: 0x{:X}.", reinterpret_cast<uintptr_t>(VolumetricFogaddress));
        }

        if (!Sharpeningaddress) {
            constexpr auto VolumetricFogStringObfuscated = make_obfuscated<0x4A>("F3 44 ?? ?? ?? ?? 45 0F ?? ?? 73 ?? 45 0F ?? ?? EB ?? F3 44 0F ?? ?? ?? ?? ?? ?? F3");
            Sharpeningaddress = Memory::AOBScan("", VolumetricFogStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

            if (!Sharpeningaddress)
                logger->warn("Sharpening signature not found. Maybe your game has been updated and is no more compatible with this plugin.");
            else
                logger->info("Sharpening signature found at address: 0x{:X}.", reinterpret_cast<uintptr_t>(Sharpeningaddress));
        }

        if (CameraComponentaddress && FOVaddress && CinematicsFOVaddress && Cameraaddress && DOFaddress && 
            Vignettingaddress && Fogaddress && VolumetricFogaddress && Sharpeningaddress && ConstrainAspectRatioaddress)
            logger->info("All AOB signatures found. Ready to patch...");

        AOBScanDone = true;

        if (!GObjectsaddress || !AppendStringaddress || !ProcessEventaddress) {
            logger->info("------------ UEngine offsets search ------------");
            uint8_t* baseModule = reinterpret_cast<uint8_t*>(GetModuleHandleA(nullptr)); // Get game base address

            constexpr auto GObjetcsStringObfuscated = make_obfuscated<0x4A>("48 8B ?? ?? ?? ?? ?? 48 8B ?? ?? 48 8D ?? ?? EB ?? 33");
            GObjectsaddress = Memory::AOBScan("", GObjetcsStringObfuscated.decrypt(), PAGE_EXECUTE_READ);
            constexpr auto GAppendStringStringObfuscated = make_obfuscated<0x4A>("48 89 ?? ?? ?? 48 89 ?? ?? ?? 57 48 ?? ?? ?? 8B ?? 48 ?? ?? 8B ?? 44 ?? ?? ?? C1 ?? ?? 48 ?? ?? 80 3D ?? ?? ?? ?? ?? 89 ?? ?? ?? 44 89 ?? ?? ?? 74 ?? 4C 8D ?? ?? ?? ?? ?? EB ?? 48 8D ?? ?? ?? ?? ?? E8 ?? ?? ?? ?? 4C ?? ?? C6 05 ?? ?? ?? ?? ?? 48 8B ?? ?? ?? 48 ?? ?? 48 ?? ?? ?? 8D ?? ?? 49 03 ?? ?? ?? E8 ?? ?? ?? ?? 83");
            AppendStringaddress = Memory::AOBScan("", GAppendStringStringObfuscated.decrypt(), PAGE_EXECUTE_READ);
            constexpr auto ProcessEventStringObfuscated = make_obfuscated<0x4A>("40 ?? 56 57 41 ?? 41 ?? 41 ?? 41 ?? 48 81 ?? ?? ?? ?? ?? 48 8D ?? ?? ?? 48 89 ?? ?? ?? ?? ?? 48 8B ?? ?? ?? ?? ?? 48 33 ?? 48 89 ?? ?? ?? ?? ?? 8B 41");
            ProcessEventaddress = Memory::AOBScan("", ProcessEventStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

            if (!GObjectsaddress)
                logger->warn("GObjects signature not found. Maybe your game has been updated and is no more compatible with this plugin.");
            else {
                uint32_t gObjectsOffset = static_cast<uint32_t>(Memory::GetOffsetFromOpcode(GObjectsaddress + 0x3) - baseModule);
                logger->info("GObjects offset is: 0x{:X}.", gObjectsOffset);
                Offsets::GObjects = static_cast<UC::uint32>(gObjectsOffset); // Update GObjects offset
            }

            if (!AppendStringaddress)
                logger->warn("AppendString Event signature not found. Maybe your game has been updated and is no more compatible with this plugin.");
            else {
                std::optional<uint32_t> gAppendStringOffsetOpt = UE::CalculateOffset("", AppendStringaddress);
                uint32_t gAppendStringOffset = *gAppendStringOffsetOpt;
                logger->info("AppendString offset is: 0x{:X}.", gAppendStringOffset);
                Offsets::AppendString = static_cast<UC::uint32>(gAppendStringOffset);// Update AppendString offset
            }

            if (!ProcessEventaddress)
                logger->warn("Process Event signature not found. Maybe your game has been updated and is no more compatible with this plugin.");
            else {
                std::optional<uint32_t> gProcessEventOffsetOpt = UE::CalculateOffset("", ProcessEventaddress);
                uint32_t gProcessEventOffset = *gProcessEventOffsetOpt;
                logger->info("Process Event offset is: 0x{:X}.", gProcessEventOffset);
                Offsets::ProcessEvent = static_cast<UC::uint32>(gProcessEventOffset);// Update ProcessEvent offset
            }

            if (!init)
                logger->info("-------------- Fixes initialisation -------------");
        }
    }

    if (init && g_fix_enabled)
        logger->info("-------------- Fixes initialisation -------------");
    if (!init && FOVaddress) FOVFixEnabled();
    if (!init && Cameraaddress) CameraFixEnabled();
    if (!init && DOFaddress) DOFFixEnabled();
    if (!init && Vignettingaddress) VignettingFixEnabled();
    if (!init && Fogaddress) FogFixEnabled();
    if (!init && VolumetricFogaddress) VolumetricFogFixEnabled();
    if (!init && Sharpeningaddress) SharpeningFixEnabled();
    if (!init && ConstrainAspectRatioaddress) UltraWideFixEnabled();
    if (!init) EnableConsole();
}

// Setters for Reshade addon call
extern "C" __declspec(dllexport) void SetFixesEnabled(GameFixes fix, bool enabled)
{ // Set each fix individually
    if (fix == GameFixes::DevConsole)           { g_Console = enabled; EnableConsole(); }
    if (fix == GameFixes::FOV)                  { g_fov_fix_enabled = enabled; FOVFixEnabled(); }
    if (fix == GameFixes::UltraWide)            { g_ultrawide_fix_enabled = enabled; UltraWideFixEnabled(); }
    if (fix == GameFixes::Camera)               { g_camera_fix_enabled = enabled; CameraFixEnabled(); }
    if (fix == GameFixes::DOF)                  { g_DOF_fix_enabled = enabled; DOFFixEnabled(); }
    if (fix == GameFixes::Vignetting)           { g_Vignetting_fix_enabled = enabled; VignettingFixEnabled(); }
    if (fix == GameFixes::Fog)                  { g_Fog_fix_enabled = enabled; FogFixEnabled(); }
    if (fix == GameFixes::VolumetricFog)        { g_VolumetricFog_fix_enabled = enabled; VolumetricFogFixEnabled(); }
    if (fix == GameFixes::Sharpening)           { g_sharpening_fix_enabled = enabled; SharpeningFixEnabled(); }
}

extern "C" __declspec(dllexport) void SetFOV(int fov) {
    g_AdditionalFOVValue = fov;
}

extern "C" __declspec(dllexport) void SetCinematicsFOV(int fov) {
    g_CinematicsFOVValue = fov;
}

extern "C" __declspec(dllexport) void SetCameraDistance(float cameraDistance) {
    g_CameraMultiplier = cameraDistance;
}

extern "C" __declspec(dllexport) void SetSharpening(float sharpening) {
    g_sharpeningValue = sharpening;
}

// Getters for Reshade addon call
extern "C" __declspec(dllexport) void GetGameInfos(GameInfos* infos) {
    if (!infos) return;

    infos->FOVIn = g_FOV_In;
    infos->FOVOut = g_FOV_Out;
    infos->consoleEnabled = g_Console_Enabled;
}

// Code injection functions
static void FOVFixEnabled() {
    if (g_fix_enabled && g_fov_fix_enabled && FOVaddress && CinematicsFOVaddress) {
        if (!FOVHook) { // Hook only once
            FOVHook = safetyhook::create_mid(FOVaddress + 0x10,
                [](SafetyHookContext& ctx) {
                    g_FOV_In = ctx.xmm0.f32[0];
                    ctx.xmm0.f32[0] += g_AdditionalFOVValue;
                    g_FOV_Out = ctx.xmm0.f32[0];
                });
        }
        else FOVHook.enable();

        if (!CinematicsFOVHook) { // Hook only once
            CinematicsFOVHook = safetyhook::create_mid(CinematicsFOVaddress + 0xd,
                [](SafetyHookContext& ctx) {
                    ctx.xmm0.f32[0] += g_CinematicsFOVValue;
                });
        }
        else CinematicsFOVHook.enable();
        logger->info("FOV fix enabled");
    }
    if (!(g_fix_enabled && g_fov_fix_enabled) && FOVaddress && CinematicsFOVaddress) {
        if (FOVHook) FOVHook.disable();
        if (CinematicsFOVHook) CinematicsFOVHook.disable();
        logger->info("FOV fix disabled");
    }
}

LONG WINAPI CameraVehHandler(PEXCEPTION_POINTERS ep) {
    if (!ep || !ep->ContextRecord || !ep->ExceptionRecord) return EXCEPTION_CONTINUE_SEARCH;
    if (ep->ExceptionRecord->ExceptionCode != EXCEPTION_SINGLE_STEP) return EXCEPTION_CONTINUE_SEARCH;

    CONTEXT* ctx = ep->ContextRecord; // Check if instruction breakpoint is the one targeted
    if (ctx->Rip == reinterpret_cast<uintptr_t>(Cameraaddress + 0x17)) {
        float value = *reinterpret_cast<float*>(ctx->Rcx + 0x4C); // Recreate original opcode skipped
        reinterpret_cast<float*>(&ctx->Xmm0)[0] = value;        
        reinterpret_cast<float*>(&ctx->Xmm6)[0] = g_CameraMultiplier; // Set new cam multiplier
        
        ctx->Rip += 5; // Skip original opcode size in bytes otherwise the game will crash

        return EXCEPTION_CONTINUE_EXECUTION;
    }
}

static void CameraFixEnabled() {
    if (g_fix_enabled && g_camera_fix_enabled && Cameraaddress) {
        vehCameraHandle = Memory::SetupOrClearHardwareBreakPointForAllThreads(reinterpret_cast<uintptr_t>(Cameraaddress + 0x17), vehCameraHandle, true, CameraVehHandler, 0);
        if (vehCameraHandle)
            logger->info("Camera fix enabled");
    }
    if (!(g_fix_enabled && g_camera_fix_enabled) && Cameraaddress) {
        if (vehCameraHandle)
            vehCameraHandle = Memory::SetupOrClearHardwareBreakPointForAllThreads(reinterpret_cast<uintptr_t>(Cameraaddress + 0x17), vehCameraHandle, false, nullptr, 0);
        if (!vehCameraHandle)
            logger->info("Camera fix disabled");
    }
}

static void FogFixEnabled() {
    if (g_fix_enabled && g_Fog_fix_enabled && Fogaddress) {
        if (!FogHook) {
            FogHook = safetyhook::create_mid(Fogaddress,
                [](SafetyHookContext& ctx) {
                    if (!ctx.rax) return;
                    uintptr_t* rax = reinterpret_cast<uintptr_t*>(ctx.rax);
                    bool* bFog = reinterpret_cast<bool*>(reinterpret_cast<uint8_t*>(rax) + 0x04);
                    *bFog = g_fix_enabled && g_Fog_fix_enabled ? false : true;
                });
        }
        logger->info("Fog fix enabled");
    }
    if (!(g_fix_enabled && g_Fog_fix_enabled) && Fogaddress)
        logger->info("Fog fix disabled");
}

static void SharpeningFixEnabled() {
    if (g_fix_enabled && g_sharpening_fix_enabled && Sharpeningaddress) {
        if (!SharpeningHook) { // Hook only once
            SharpeningHook = safetyhook::create_mid(Sharpeningaddress,
                [](SafetyHookContext& ctx) {
                    if (!ctx.rax) return;
                    float* sharpeningValue = reinterpret_cast<float*>(ctx.rax + 0x04);
                    *sharpeningValue = g_sharpeningValue;
                });
        }
        else SharpeningHook.enable();
        logger->info("Sharpening fix enabled");
    }
    if (!(g_fix_enabled && g_sharpening_fix_enabled) && Sharpeningaddress) {
        if (SharpeningHook) SharpeningHook.disable();
        logger->info("Sharpening fix disabled");
    }
}

// Memory patch fixes
static void UltraWideFixEnabled() {
    if (g_fix_enabled && g_ultrawide_fix_enabled && ConstrainAspectRatioaddress) {
        Memory::PatchBytes(ConstrainAspectRatioaddress, "\x31\xC0\x90\x90\x90\x90\x90", 7); // bConstrainAspectRatio = 0
        logger->info("Ultra wide fix enabled");
    }
    if (!(g_fix_enabled && g_ultrawide_fix_enabled) && ConstrainAspectRatioaddress) {
        Memory::RestoreBytes(ConstrainAspectRatioaddress);
        logger->info("Ultra wide fix disabled");
    }
}

static void DOFFixEnabled() {
    if (g_fix_enabled && g_DOF_fix_enabled && DOFaddress) {
        Memory::PatchBytes(DOFaddress, "\x31\xFF\x90", 3); // xor edi,edi r.DepthOfFieldQuality = 0
        logger->info("Depth of field fix enabled");
    }
    if (!(g_fix_enabled && g_DOF_fix_enabled) && DOFaddress) {
        Memory::RestoreBytes(DOFaddress);
        logger->info("Depth of field fix disabled");
    }
}

static void VignettingFixEnabled() {
    if (g_fix_enabled && g_Vignetting_fix_enabled && Vignettingaddress) {
        Memory::PatchBytes(Vignettingaddress, "\x31\xC9", 2); // xor ecx,ecx r.Tonemapper.Quality=0
        logger->info("Vignetting fix enabled");
    }
    if (!(g_fix_enabled && g_Vignetting_fix_enabled) && Vignettingaddress) {
        Memory::RestoreBytes(Vignettingaddress);
        logger->info("Vignetting fix disabled");
    }
}

static void VolumetricFogFixEnabled() {
    if (g_fix_enabled && g_VolumetricFog_fix_enabled && VolumetricFogaddress) {
        Memory::PatchBytes(VolumetricFogaddress, "\xEB", 1); // je-> jmp r.Fog=0
        logger->info("Volumetric fog fix enabled");
    }
    if (!(g_fix_enabled && g_VolumetricFog_fix_enabled) && VolumetricFogaddress) {
        Memory::RestoreBytes(VolumetricFogaddress);
        logger->info("Volumetric fog fix disabled");
    }
}

// UE Console creation
static void EnableConsole()
{
    if (g_Console_Enabled || !g_Console || !GObjectsaddress || !AppendStringaddress || !ProcessEventaddress) {
        if (!g_Console)
            logger->info("------------------ User inputs ------------------");
        return;
    }

    logger->info("-------------- Console re-enabling --------------");
    std::thread([&]() {
        auto start = std::chrono::high_resolution_clock::now(); // Measure the time to renable console
        UEngine* Engine = nullptr;
        // Usual UEngine::GetEngine() is too long to retrieve Engine instance with hundred of thousands objects to scan
        // This approach is based on signature (limited to 4.26.2 ?) and pointers to final instance (working in 8 secs usually).
        uint8_t* GEngineaddress = nullptr;
        constexpr auto GEngineStringObfuscated = make_obfuscated<0x4A>("48 8B ?? ?? ?? ?? ?? 48 8B ?? ?? ?? ?? ?? 48 ?? ?? 74 ?? 48 ?? ?? ?? 48 ?? ?? 74 ?? 48 ?? ?? 33");
        GEngineaddress = Memory::AOBScan("", GEngineStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

        if (GEngineaddress) {
            uint8_t* GEnginePtraddress = Memory::GetOffsetFromOpcode(GEngineaddress + 0x3);

            if (GEnginePtraddress) {
                for (int i = 0; i < 200; i++) {
                    Engine = *reinterpret_cast<UEngine**>(GEnginePtraddress);
                    if (Engine && Engine->GameViewport && Engine->ConsoleClass) break;

                    std::this_thread::sleep_for(std::chrono::milliseconds(100));
                }
            }
        }

        if (!Engine || !Engine->ConsoleClass || !Engine->GameViewport) {
            logger->error("Console could not be found in engine.");
            return;
        }
        logger->info("Console found in engine");

        /* Creates a new UObject of class-type specified by Engine->ConsoleClass */
        UObject* NewObject = UGameplayStatics::SpawnObject(Engine->ConsoleClass, Engine->GameViewport);
        if (NewObject)
        {
            logger->info("Successfully spawned console object");
            // Set the console viewport so that it will be displayed
            Engine->GameViewport->ViewportConsole = static_cast<UConsole*>(NewObject);
            auto end = std::chrono::high_resolution_clock::now();
            std::chrono::duration<double> elapsed = end - start;

            for (int i = 0; i < UInputSettings::GetDefaultObj()->ConsoleKeys.Num(); i++)
            {
                UInputSettings::GetDefaultObj()->ConsoleKeys[i].KeyName = UKismetStringLibrary::Conv_StringToName(L"F2");
            }
            logger->info("Console fully reactivated in {:.3f}s and bound to key F2", elapsed.count());
            logger->info("------------------ User inputs ------------------");

            g_Console_Enabled = true;
        }
        else {
            logger->error("Could not spawn console object");
        }
    }).detach();
}

static void InitializeLogger()
{
    try
    {
        std::filesystem::path log_path = std::filesystem::absolute(PLUGIN_LOG);
        if (std::filesystem::exists(log_path))
            std::filesystem::remove(log_path);
        logger = std::make_shared<spdlog::logger>("Jedi Survivor", std::make_shared<spdlog::sinks::rotating_file_sink_st>(PLUGIN_LOG, 10 * 1024 * 1024, 1));
        logger->set_level(spdlog::level::debug);
        logger->flush_on(spdlog::level::debug); // Flush automatically
    }
    catch (const spdlog::spdlog_ex& ex)
    {
        std::string plugin_error_message = "Could not open " + PLUGIN_LOG;
        MessageBoxA(nullptr, plugin_error_message.c_str(), "Logger Error", MB_ICONERROR | MB_OK);
    }
}

// Standard dll entry
BOOL APIENTRY DllMain(HMODULE hModule, DWORD reason, LPVOID)
{
    if (reason == DLL_PROCESS_ATTACH)
    {
        InitializeLogger();
        logger->info("Plugin {} loaded.", PLUGIN_NAME);
    }
    else if (reason == DLL_PROCESS_DETACH)
    {
        logger->info("Plugin {} unloaded.", PLUGIN_NAME);
        spdlog::drop_all();
    }
    return TRUE;
}