#define NOMINMAX
#include <algorithm>
#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"
#include "SDK/Phoenix_classes.hpp"

using namespace SDK;

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

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

// Screen informations
static int screenWidth = GetSystemMetrics(SM_CXSCREEN);
static int screenHeight = GetSystemMetrics(SM_CYSCREEN);
static float aspectRatio = (float)screenWidth / screenHeight;

// Plugin states
static bool AOBScanDone = false;
static bool g_fix_enabled = false;
static bool g_camera_fix_enabled = false;
static bool g_HUD_fix_enabled = false;
static bool g_fov_fix_enabled = false;
static bool g_CA_fix_enabled = false;
static bool g_Vignetting_fix_enabled = false;
static bool g_Fog_fix_enabled = false;
static int g_AdditionalFOVValue = 0;
static int g_CameraOffset = 0;
static float g_TransitionCameraOffset = 0.f;
static float g_NativeCameraDistance = 80.f;
static float g_LastCameraDeltaTime = 1.f / 60.f; // fallback
static float g_HUDLeft = 0.f;
static float g_HUDRight = 1.f;
static bool g_Console = false;
static bool bIsInventoryOpen = false;
static bool bIsInCinematicMode = false;
static bool bIsInPrinterMode = false;
static bool bIsAiming = false;
static double LastCineCallTime = 0.0;
static bool user_inputs_logged = false;

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

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

// AOB Scan pointers
static uint8_t* FOVaddress = nullptr;
static uint8_t* ConstrainAspectaddress = nullptr;
static uint8_t* Cameraaddress = nullptr;
static uint8_t* CameraObjectaddress = nullptr;
static uint8_t* HUDaddress = nullptr;
static uint8_t* CAaddress = nullptr;
static uint8_t* Vignettingaddress = nullptr;
static uint8_t* Fogaddress = nullptr;

// Hooking
static SafetyHookMid FOVHook{};
static SafetyHookMid CameraHook{};
static SafetyHookMid HUDHook{};
static SafetyHookMid FogHook{};
static SafetyHookMid PEHook{};

// Prototypes
static void FOVFixEnabled();
static void UltraWideFixEnabled();
static void CameraFixEnabled();
static void HUDFixEnabled();
static void DOFFixEnabled();
static void CAFixEnabled();
static void VignettingFixEnabled();
static void FogFixEnabled();
static void EnableConsole();

// Unreal Engine camera struct
struct BlueprintModifyCamera_Params
{
    float DeltaTime;           // in
    FVector ViewLocation;      // in
    FRotator ViewRotation;     // in
    float FOV;                 // in

    // OUT parameters (Passed by refernce in Blueprint signature)
    FVector NewViewLocation;   // out
    FRotator NewViewRotation;  // out
    float NewFOV;              // out
};

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

            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 signature found at address: 0x{:X}.", reinterpret_cast<uintptr_t>(FOVaddress));
                ConstrainAspectaddress = FOVaddress + 0x25;
            }
        }

        if (!Cameraaddress) {
            constexpr auto CameraStringObfuscated = make_obfuscated<0x4A>("49 8B ?? ?? ?? ?? ?? 48 ?? ?? 74 ?? 48 8D ?? ?? ?? E8 ?? ?? ?? ?? 49 8B ?? ?? ?? ?? ?? EB");
            Cameraaddress = Memory::AOBScan("", CameraStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

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

        if (!HUDaddress) {
            constexpr auto HUDStringObfuscated = make_obfuscated<0x4A>("FF ?? 48 8B ?? ?? ?? 0F ?? ?? 48 ?? ?? 0F ?? ?? 48 83");
            HUDaddress = Memory::AOBScan("", HUDStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

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

        if (!CAaddress) {
            constexpr auto CAStringObfuscated = make_obfuscated<0x4A>("7F ?? 89 B3 ?? ?? ?? ?? 8B ?? ?? 39 05 ?? ?? ?? ?? 0F");
            CAaddress = Memory::AOBScan("", CAStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

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

        if (Vignettingaddress == nullptr) {
            constexpr auto VignettingStringObfuscated = make_obfuscated<0x4A>("8B ?? 83 ?? ?? 7D ?? 89 B3");
            Vignettingaddress = Memory::AOBScan("", VignettingStringObfuscated.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 == nullptr) {
            constexpr auto FogStringObfuscated = make_obfuscated<0x4A>("83 ?? ?? ?? 75 ?? B3 ?? EB ?? 32 ?? 48 8B ?? ?? ?? 48 ?? ?? 74 ?? E8");
            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 (FOVaddress && Cameraaddress && HUDaddress && CAaddress && Vignettingaddress && Fogaddress)
            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 GWorldStringObfuscated = make_obfuscated<0x4A>("48 8B ?? ?? ?? ?? ?? 48 ?? ?? 74 ?? 48 ?? ?? E8 ?? ?? ?? ?? 83 ?? ?? 74");
            GWorldaddress = Memory::AOBScan("", GWorldStringObfuscated.decrypt(), PAGE_EXECUTE_READ);
            constexpr auto GObjetcsStringObfuscated = make_obfuscated<0x4A>("48 8B ?? ?? ?? ?? ?? 48 8B ?? ?? 48 8D ?? ?? EB ?? 33");
            GObjectsaddress = Memory::AOBScan("", GObjetcsStringObfuscated.decrypt(), PAGE_EXECUTE_READ);
            constexpr auto AppendStringStringObfuscated = 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("", AppendStringStringObfuscated.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 ?? ?? 48 89 ?? ?? ?? ?? ?? 8B");
            ProcessEventaddress = Memory::AOBScan("", ProcessEventStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

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

            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 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);  // Get Offset from opcode
                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 && HUDaddress) HUDFixEnabled();
    if (!init && CAaddress) CAFixEnabled();
    if (!init && Vignettingaddress) VignettingFixEnabled();
    if (!init && Fogaddress) FogFixEnabled();
    if (!init && GObjectsaddress && AppendStringaddress && ProcessEventaddress) EnableConsole();
    if (ConstrainAspectaddress) UltraWideFixEnabled();
}

// Setters for Reshade addon call
extern "C" __declspec(dllexport) void SetFixesEnabled(GameFixes fix, bool enabled) {
    if (fix == GameFixes::DevConsole)           { g_Console = enabled; EnableConsole(); };
    if (fix == GameFixes::FOV)                  { g_fov_fix_enabled = enabled; FOVFixEnabled(); }
    if (fix == GameFixes::Camera)               { g_camera_fix_enabled = enabled; CameraFixEnabled(); }
    if (fix == GameFixes::HUD)                  { g_HUD_fix_enabled = enabled; HUDFixEnabled(); }
    if (fix == GameFixes::ChromaticAberrations) { g_CA_fix_enabled = enabled; CAFixEnabled(); }
    if (fix == GameFixes::Vignetting)           { g_Vignetting_fix_enabled = enabled; VignettingFixEnabled(); }
    if (fix == GameFixes::Fog)                  { g_Fog_fix_enabled = enabled; FogFixEnabled(); }
}

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

extern "C" __declspec(dllexport) void SetCameraOffset(int cameraOffset) {
    g_CameraOffset = cameraOffset + 80;
    g_TransitionCameraOffset = (float)g_CameraOffset;
}

extern "C" __declspec(dllexport) void SetHUDScale(int HUDScale) {
    g_HUDLeft = (float)HUDScale / 100;
    g_HUDRight = 1.f - ((float)HUDScale / 100);
}

// 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->CompensatedFOV = g_NativeCameraDistance;
    infos->consoleEnabled = g_Console_Enabled;
}

// Code injection functions
static void FOVFixEnabled() {
    if (g_fix_enabled && g_fov_fix_enabled && FOVaddress) {
        if (!FOVHook) { // Hook only once
            FOVHook = safetyhook::create_mid(FOVaddress + 0xa,
                [](SafetyHookContext& ctx) {
                    g_FOV_In = ctx.xmm0.f32[0];                   
                    ctx.xmm0.f32[0] += g_AdditionalFOVValue; // Additional FOV
                    g_FOV_Out = ctx.xmm0.f32[0];
                });
        }
        else FOVHook.enable();
        logger->info("FOV fix enabled");
    }
    if (!(g_fix_enabled && g_fov_fix_enabled) && FOVaddress) {
        if (FOVHook) FOVHook.disable();
        logger->info("FOV fix disabled");
    }
}

static FVector RotatorToForwardVector(const FRotator& rotator)
{
    const float DEG_TO_RAD = 3.1415926535f / 180.0f;

    float cp = cosf(rotator.Pitch * DEG_TO_RAD);
    float sp = sinf(rotator.Pitch * DEG_TO_RAD);
    float cy = cosf(rotator.Yaw * DEG_TO_RAD);
    float sy = sinf(rotator.Yaw * DEG_TO_RAD);

    return FVector(cp * cy, cp * sy, sp);
}

inline double NowSeconds()
{
    using namespace std::chrono;
    static steady_clock::time_point start = steady_clock::now();
    return duration<double>(steady_clock::now() - start).count();
}

static void CameraFixEnabled() {
    if (g_fix_enabled && g_camera_fix_enabled && ProcessEventaddress) {
        if (!PEHook) {
            PEHook = safetyhook::create_mid(ProcessEventaddress + 0xc,
                [](SafetyHookContext& ctx) {
                    UObject* object = (UObject*)ctx.rcx;
                    UFunction* func = (UFunction*)ctx.rdx;

                    if (object && func) {
                        std::string funcName = func->GetName();
                        std::string objectName = object->GetFullName();

                        static auto start = std::chrono::high_resolution_clock::now();

                        // Detect cinematics to disable temporarily camera fix
                        if (objectName.contains("CineCameraComponent") && funcName == "SetCurrentFocalLength") {
                            bIsInCinematicMode = true;
                            LastCineCallTime = NowSeconds();
                        }
                        // Detect reforge station interactions to disable temporarily camera fix
                        if (objectName.contains("BP_PrinterTabs_C")) {
                            if (funcName == "Show") bIsInPrinterMode = true;
                            if (funcName == "Hide") bIsInPrinterMode = false;
                        }
                        // Detect inventory interactions to disable temporarily camera fix
                        if (objectName.contains("BP_State_Inventory_C")) {
                            if (funcName == "OnBegin") bIsInventoryOpen = true;
                            if (funcName == "OnEnd") bIsInventoryOpen = false;
                        }
                        // Detect camera collisions to avoid world to vanish ending in character death
                        if (objectName.contains("CameraModifier_PhxCameraRig") && funcName == "BlueprintModifyCamera")
                        {
                            auto* cameraParams = reinterpret_cast<BlueprintModifyCamera_Params*>(ctx.r8);
                            if (!cameraParams) return;

                            if (cameraParams->DeltaTime > 0.f && cameraParams->DeltaTime < 0.1f)
                                g_LastCameraDeltaTime = cameraParams->DeltaTime;

                            // --- World to Pawn (player) ---
                            UWorld* world = UWorld::GetWorld();
                            if (!world) return;
                            UGameInstance* gameInstance = world->OwningGameInstance;
                            if (!gameInstance || gameInstance->LocalPlayers.Num() == 0) return;
                            ULocalPlayer* localPlayer = gameInstance->LocalPlayers[0];
                            if (!localPlayer) return;
                            APlayerController* pc = localPlayer->PlayerController;
                            if (!pc) return;
                            APawn* pawn = pc->AcknowledgedPawn;
                            if (!pawn) return;
                            // Retrieve Character class
                            void* inventoryComp = *(void**)((uintptr_t)pawn + 0x6d8); // inventoryComp offset from SDK
                            void* moveComp = *(void**)((uintptr_t)pawn + 0x908); // moveComp offset FROM SDK
                            if (inventoryComp && moveComp) {
                                APhxCharacter* phxChar = static_cast<APhxCharacter*>(pawn);
                                if (phxChar) bIsAiming = phxChar->GetIsAiming(); // Is player aiming
                            }

                            // --- Positions ---
                            FVector pawnPos = pawn->K2_GetActorLocation();
                            FVector engineCamPos = cameraParams->ViewLocation;
                            // --- Camera direction ---
                            FVector forward = RotatorToForwardVector(cameraParams->ViewRotation);
                            FVector backward = forward * -1.f;
                            // --- ENGINE OFFSET (full) ---
                            FVector engineOffset = engineCamPos - pawnPos;
                            // --- DECOMPOSE OFFSET ---
                            // Projection of engine offset onto backward axis (recul moteur)
                            float engineBackDist = engineOffset.Dot(backward);
                            g_NativeCameraDistance = engineBackDist;
                            FVector backComponent = backward * engineBackDist;

                            if (bIsInCinematicMode || bIsInPrinterMode || bIsInventoryOpen || bIsAiming) return;

                            // What we KEEP (height + lateral)
                            FVector preservedOffset = engineOffset - backComponent;

                            // --- Desired corrected distance ---
                            float maxDistance = g_TransitionCameraOffset;
                            FVector traceStart = pawnPos /*+ preservedOffset*/;
                            FVector traceEnd = traceStart + backward * maxDistance;

                            // --- Sphere Trace ---
                            TArray<AActor*> ignoreActors;
                            TArray<FHitResult> hits;
                            float radius = 12.f;

                            bool bHit = UKismetSystemLibrary::SphereTraceMulti(
                                object,
                                traceStart,
                                traceEnd,
                                radius,
                                ETraceTypeQuery::TraceTypeQuery2, // camera channel
                                true,
                                ignoreActors,
                                EDrawDebugTrace::None,
                                &hits,
                                true,
                                FLinearColor(1, 0, 0, 1),
                                FLinearColor(0, 1, 0, 1),
                                0.f);

                            // --- Compute target distance ---
                            float targetDist = maxDistance;

                            if (bHit && hits.Num() > 0)
                            {
                                float best = maxDistance;
                                float padding = 15.f;

                                for (const FHitResult& hit : hits)
                                {
                                    if (hit.bStartPenetrating) continue;
                                    if (hit.PenetrationDepth > 2.f) continue;

                                    float dist = (hit.Location - traceStart).Magnitude();
                                    dist -= padding;
                                    dist = std::max(dist, 0.f);

                                    if (dist < best) best = dist;
                                }
                                targetDist = best;
                            }

                            // --- Smoothing ---
                            static float lastSafeDistance = -1.f;
                            if (lastSafeDistance < 0.f) lastSafeDistance = targetDist;

                            float interpSpeed = (targetDist < lastSafeDistance) ? 14.f : 6.f;
                            float alpha = std::clamp(interpSpeed * cameraParams->DeltaTime, 0.f, 1.f);
                            lastSafeDistance = UKismetMathLibrary::Lerp(lastSafeDistance, targetDist, alpha);
                            lastSafeDistance = std::clamp(lastSafeDistance, 0.f, maxDistance);

                            // --- FINAL CAMERA POSITION ---
                            FVector finalCamPos = pawnPos + preservedOffset +
                                                  backward * lastSafeDistance;

                            if (!world->K2_GetWorldSettings()) return;
                            // --- APPLY ---
                            cameraParams->NewViewLocation = finalCamPos;
                            cameraParams->NewViewRotation = cameraParams->ViewRotation;
                            cameraParams->NewFOV = cameraParams->FOV;
                        }

                    }
                });
        }
        else PEHook.enable();
    }
    if (!(g_fix_enabled && g_camera_fix_enabled) && ProcessEventaddress) {
        if (PEHook) PEHook.disable();
    }

    if (g_fix_enabled && g_camera_fix_enabled && Cameraaddress) {
        if (!CameraHook) {
            CameraHook = safetyhook::create_mid(Cameraaddress,
                [](SafetyHookContext& ctx) {
                    if (bIsInCinematicMode) {
                        double now = NowSeconds();
                        if ((now - LastCineCallTime) > 1.0) // 1 seconde sans appel
                            bIsInCinematicMode = false;
                    }

                    float dt = g_LastCameraDeltaTime;
                    float fadeInSpeed = 3.5f;  // vitesse quand on RAPPROCHE la caméra
                    float fadeOutSpeed = 3.5f;  // vitesse quand on RECULE la caméra
                    // sécurité
                    if (dt <= 0.f || dt > 0.1f) dt = 1.f / 60.f;

                    float targetOffset = (bIsInventoryOpen || bIsInPrinterMode || bIsInCinematicMode || bIsAiming) ?
                                         g_NativeCameraDistance :   // Minimum offset
                                         (float)g_CameraOffset;     // User offset
                    float interpSpeed = (targetOffset < g_TransitionCameraOffset) ? fadeInSpeed : fadeOutSpeed;

                    g_TransitionCameraOffset = UKismetMathLibrary::FInterpTo(g_TransitionCameraOffset,
                                               targetOffset,dt,interpSpeed);
                });
        }
        else CameraHook.enable();
        logger->info("Camera fix enabled");
    }
    if (!(g_fix_enabled && g_camera_fix_enabled) && Cameraaddress) {
        if (CameraHook) CameraHook.disable();
        logger->info("Camera fix disabled");
    }
}

static void HUDFixEnabled() {
    if (g_fix_enabled && g_HUD_fix_enabled && HUDaddress) {
        if (!HUDHook) { // Hook only once
            HUDHook = safetyhook::create_mid(HUDaddress + 0xa,
                [](SafetyHookContext& ctx) {
                    ctx.xmm0.f32[0] = g_HUDLeft;
                    ctx.xmm0.f32[1] = 0.f;
                    ctx.xmm0.f32[2] = g_HUDRight;
                    ctx.xmm0.f32[3] = 1.f;
                });
        }
        else HUDHook.enable();
        logger->info("HUD fix enabled");
    }
    if (!(g_fix_enabled && g_HUD_fix_enabled) && HUDaddress) {
        if (HUDHook) HUDHook.disable();
        logger->info("HUD fix disabled");
    }
}

// Memory patch fixes
static void UltraWideFixEnabled() {
    if (ConstrainAspectaddress)
        Memory::PatchBytes(ConstrainAspectaddress, "\x31\xC0\x90\x90\x90\x90\x90", 7); // bConstrainAspectRatio = 0
}

static void CAFixEnabled() {
    if (g_fix_enabled && g_CA_fix_enabled && CAaddress) {
        Memory::PatchBytes(CAaddress, "\x90\x90", 2); // NOP jg "HellIsUs-Win64-Shipping.exe"+3D73BD5 r.SceneColorFringeQuality = 0
        logger->info("Chromatic aberrations fix enabled");
    }
    if (!(g_fix_enabled && g_CA_fix_enabled) && CAaddress) {
        Memory::RestoreBytes(CAaddress);
        logger->info("Chromatic aberrations 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 FogFixEnabled() {
    if (g_fix_enabled && g_Fog_fix_enabled && Fogaddress) {
        if (!FogHook) {
            FogHook = safetyhook::create_mid(Fogaddress,
                [](SafetyHookContext& ctx) {
                    if (!ctx.rax) return;
                    *reinterpret_cast<uint8_t*>(ctx.rax + 0x4) = g_fix_enabled && g_Fog_fix_enabled ? 0 : 1; // r.Fog = 0
                });
        }
        logger->info("Fog fix enabled");
    }
    if (!(g_fix_enabled && g_Fog_fix_enabled) && Fogaddress)
        logger->info("Fog fix disabled");
}

// UE Console creation
static void EnableConsole()
{
    if (g_Console_Enabled || !g_Console || !GObjectsaddress || !AppendStringaddress || !ProcessEventaddress) {
        if (!g_Console && !user_inputs_logged) {
            logger->info("------------------ User inputs ------------------");
            user_inputs_logged = true;
        }
        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;

        for (int i = 0; i < 10; ++i) { // gives 10 seconds to find UE Engine
            std::this_thread::sleep_for(std::chrono::milliseconds(1000));
            Engine = UEngine::GetEngine();
            if (Engine && Engine->ConsoleClass && Engine->GameViewport)
                break;
        }

        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);
            // Compute console enabling duration
            auto end = std::chrono::high_resolution_clock::now();
            std::chrono::duration<double> elapsed = end - start;

            // Set the all the console shortkey to F2
            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");
            logger->info("------------------ User inputs ------------------");
        }
    }).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>("The Callisto Protocol", 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;
}