#include "CommonHeaders.h"
#include "UEngine.hpp";
#include "SDK/Basic.hpp"
#include "SDK/Engine_classes.hpp"
#include "SDK/WBP_Cutscene_classes.hpp"
#include "SDK/WBP_TitleFlow_classes.hpp"
#include "SDK/WBP_TitleFlow_Logo_classes.hpp"
#include "SDK/WBP_TitleFlow_Warning_classes.hpp"

using namespace SDK;
// Unreal Engine variables
static UWorld* previousWorld = nullptr;
static ULevel* previousLevel = nullptr;
static AExponentialHeightFog* previousFogActor = nullptr;

// Constants
const std::string PLUGIN_NAME = "SilentHillf";
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_DOF_fix_enabled = false;
static bool g_CA_fix_enabled = false;
static bool g_Vignetting_fix_enabled = false;
static bool g_Volumetric_Fog_fix_enabled = false;
static bool g_Fog_fix_enabled = false;
static bool g_Fog_Density_fix_enabled = false;
static bool g_Fog_Max_Opacity_fix_enabled = false;
static bool g_Camera_fix_enabled = false;
static bool g_Cutscenes_fix_enabled = false;
static bool g_Cutscenes_FPS_fix_enabled = false;
static bool g_SkipIntro_fix_enabled = false;
static bool g_TimeDilation_fix_enabled = false;
static bool g_StealthMode_fix_enabled = false;
static int g_AdditionalFOVValue = 0;
static float g_CameraDistance = 1.f;
static float g_FogDensity = 0.4f;
static float g_FogMaxOpacity = 0.4f;
static float g_WorldTimeDilationValue = 1.f;
static float g_AITimeDilationValue = 1.f;
static bool g_Console = false;
static bool user_inputs_logged = false;

// Shared values
static float g_FOV_In = 65.f;
static float g_FOV_Out = 65.f;
static float g_Camera_In = 230.f;
static float g_Camera_Out = 230.f;
static float g_DefaultFogDensity = -1.f;
static float g_DefaultFogMaxOpacity = -1.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* DOFaddress = nullptr;
static uint8_t* CAaddress = nullptr;
static uint8_t* Vignettingaddress = nullptr;
static uint8_t* Fogaddress = nullptr;
static uint8_t* VolumetricFogaddress = nullptr;
static uint8_t* FogDensityEngineaddress = nullptr;
static uint8_t* FogOpacityEngineaddress = nullptr;
static uint8_t* Cameraaddress = nullptr;
static uint8_t* CutscenesFPSaddress = nullptr;
static uint8_t* ConstrainAspectRatioaddress = nullptr;
static uint8_t* AspectRatioAxisConstraintaddress = nullptr;
static uint8_t* WorldTimedilationaddress = nullptr;

// Hooking
static SafetyHookMid FOVHook{};
static SafetyHookMid PEHook{};
static SafetyHookMid FogHook{};
static SafetyHookMid VolumetricFogHook{};
static SafetyHookMid FogDensityHook{};
static SafetyHookMid FogMaxOpacityHook{};
static SafetyHookMid CameraHook{};
static SafetyHookMid CutscenesHook{};
static SafetyHookMid CutscenesFPSHook{};
static SafetyHookMid ConstrainAspectRatioHook{};
static SafetyHookMid AspectRatioAxisConstraintHook{};
static SafetyHookMid WorldTimeDilationHook{};

// Prototypes
static void FOVFixEnabled();
static void DOFFixEnabled();
static void CAFixEnabled();
static void VignettingFixEnabled();
static void FogFixEnabled();
static void VolumetricFogFixEnabled();
static void CameraFixEnabled();
static void EnableConsole();
static void CutscenesUltrawideFixEnabled();
static void CutscenesFPSFixEnabled();
static void ProcessEvent();
static void EnableCheats(Cheat cheat);

extern "C" __declspec(dllexport) void SetFixEnabled(bool enabled, bool init) {
    g_fix_enabled = enabled;
    if (!AOBScanDone) { // Unreal Engine 5.4.2
        uint8_t* baseModule = reinterpret_cast<uint8_t*>(GetModuleHandleA(nullptr)); // Get game base address
        logger->info("--------------- AOB scan started ---------------");

        using AOBScan::Make;
        using OffsetScan::Make;
        constexpr auto FOVStringObfuscated = make_obfuscated<0x7C>("F3 0F ?? ?? 0F 28 ?? ?? ?? F3 0F ?? ?? ?? F3 0F 11 ?? ?? ?? ?? ?? ?? 45");
        constexpr auto DOFStringObfuscated = make_obfuscated<0x4A>("8B ?? ?? 48 ?? ?? E8 ?? ?? ?? ?? 0F ?? ?? 48 6B ?? ?? 48 8D");
        constexpr auto CAStringObfuscated = make_obfuscated<0xD4>("7F ?? 44 89 ?? ?? ?? ?? ?? 43 ?? ?? ?? 39 05");
        constexpr auto VignettingStringObfuscated = make_obfuscated<0x8E>("8B ?? 83 ?? ?? 7D ?? 44 89");
        constexpr auto FogStringObfuscated = make_obfuscated<0x35>("40 ?? 48 83 ?? ?? 48 8B ?? ?? ?? ?? ?? 48 ?? ?? 48 89 ?? ?? ?? 48 8D ?? ?? 48 8D ?? ?? ?? E8");
        constexpr auto VolumetricFogStringObfuscated = make_obfuscated<0xBF>("83 3D ?? ?? ?? ?? ?? 0F 84 ?? ?? ?? ?? F6 46 ?? ?? 0F 84 ?? ?? ?? ?? 83 BF");
        constexpr auto FogDensityStringObfuscated = make_obfuscated<0xC2>("0F 2E ?? 30 02 ?? ?? 74 ?? F3 0F ?? ?? ?? ?? ?? ?? E9 ?? ?? ?? ?? C3");
        constexpr auto FogOpacityStringObfuscated = make_obfuscated<0xBB>("0F 2E ?? C4 02 ?? ?? 74 ?? F3 0F ?? ?? ?? ?? ?? ?? E9 ?? ?? ?? ?? C3");
        constexpr auto CameraStringObfuscated = make_obfuscated<0x7E>("41 0F ?? ?? 48 8D ?? ?? ?? 48 8D ?? ?? ?? ?? ?? F3 0F ?? ?? ?? ?? F3 0F");
        constexpr auto CutscenesFPSStringObfuscated = make_obfuscated<0x0F>("EB ?? 0F ?? ?? 48 8B ?? ?? ?? 0F 28 ?? ?? ?? 44 0F ?? ?? ?? ?? 44 0F");
        constexpr auto ConstrainAspectRatioaddressStringObfuscated = make_obfuscated<0xB6>("89 ?? ?? 0F B6 ?? ?? ?? ?? ?? D1 ?? 33 ?? 83 ?? 02 33 ?? 89");
        constexpr auto AspectRatioAxisConstraintaddressStringObfuscated = make_obfuscated<0xE8>("48 ?? ?? ?? ?? 00 00 4C ?? ?? 4D ?? ?? E8 ?? ?? ?? ?? 48 8B");
        constexpr auto WorldTimeDilationStringObfuscated = make_obfuscated<0xC0>("F3 0F ?? ?? ?? ?? ?? ?? F3 0F ?? ?? C0 03 ?? ?? F3 0F ?? ?? ?? ?? ?? ?? C3");
        // Prepare all data for scanning
        std::vector<AOBScanEntry> signatures = {
            Make(&FOVaddress, FOVStringObfuscated, "FOV"),
            Make(&Cameraaddress, CameraStringObfuscated, "Camera"),
            Make(&ConstrainAspectRatioaddress, ConstrainAspectRatioaddressStringObfuscated, "Constrain aspect ratio"),
            Make(&AspectRatioAxisConstraintaddress, AspectRatioAxisConstraintaddressStringObfuscated, "Aspect Ratio Axis Constraint"),
            Make(&DOFaddress, DOFStringObfuscated, "DOF"),
            Make(&CAaddress, CAStringObfuscated, "Chromatic aberrations"),
            Make(&Vignettingaddress, VignettingStringObfuscated, "Vignetting"),
            Make(&Fogaddress, FogStringObfuscated, "Fog"),
            Make(&FogDensityEngineaddress, FogDensityStringObfuscated, "Fog density"),
            Make(&FogOpacityEngineaddress, FogOpacityStringObfuscated, "Fog opacity"),
            Make(&VolumetricFogaddress, VolumetricFogStringObfuscated, "Volumetric fog"),
            Make(&CutscenesFPSaddress, CutscenesFPSStringObfuscated, "Cutscenes framerate"),
            Make(&WorldTimedilationaddress, WorldTimeDilationStringObfuscated, "World time dilation"),
        };
        // Scan all signature in a batch
        Memory::AOBScanBatch(signatures, logger);

        if (FOVaddress && Cameraaddress && DOFaddress && CAaddress && Vignettingaddress && Fogaddress &&
            FogDensityEngineaddress && FogOpacityEngineaddress && ConstrainAspectRatioaddress && VolumetricFogaddress &&
            AspectRatioAxisConstraintaddress && CutscenesFPSaddress && WorldTimedilationaddress)
            logger->info("All AOB signatures found. Ready to patch...");


        if (!GObjectsaddress || !AppendStringaddress || !ProcessEventaddress) {
            logger->info("------------ UEngine offsets search ------------");

            constexpr auto GObjetcsStringObfuscated = make_obfuscated<0x4A>("48 8B ?? ?? ?? ?? ?? 48 8B ?? ?? 48 8D ?? ?? EB ?? 33");
            constexpr auto AppendStringStringObfuscated = make_obfuscated<0x4A>("48 89 ?? ?? ?? 48 89 ?? ?? ?? 57 48 83 ?? ?? 80 3D ?? ?? ?? ?? ?? 48 8B F2 8B ?? 48");
            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 ?? ?? ?? ?? ?? 4D ?? ?? 48");
            // Prepare all data for scanning
            std::vector<OffsetScanEntry> UEoffsetsScans = {
                Make(&GObjectsaddress, GObjetcsStringObfuscated, "GObjects", OffsetCalcType::GetOffsetFromOpcode, &Offsets::GObjects, 0x3),
                Make(&AppendStringaddress, AppendStringStringObfuscated, "AppendString", OffsetCalcType::UE_CalculateOffset, &Offsets::AppendString),
                Make(&ProcessEventaddress, ProcessEventStringObfuscated, "ProcessEvent", OffsetCalcType::UE_CalculateOffset, &Offsets::ProcessEvent)
            };
            // Retrieve all Unreal Engine offsets in a batch
            Memory::OffsetScanBatch(UEoffsetsScans, baseModule, logger, "");

            logger->info("-------------- Fixes initialisation -------------");
        }
        AOBScanDone = true;
    }

    if (!init && FOVaddress) FOVFixEnabled();
    if (!init && Cameraaddress) CameraFixEnabled();
    if (!init && DOFaddress) DOFFixEnabled();
    if (!init && CAaddress) CAFixEnabled();
    if (!init && Vignettingaddress) VignettingFixEnabled();
    if (!init && WorldTimedilationaddress) {
        EnableCheats(Cheat::TimeDilation);
        EnableCheats(Cheat::Stealth);
    }
    if (!init && Fogaddress && FogDensityEngineaddress && FogOpacityEngineaddress) FogFixEnabled();
    if (!init && VolumetricFogaddress) VolumetricFogFixEnabled();
    if (!init) CutscenesUltrawideFixEnabled();
    if (!init && CutscenesFPSaddress) CutscenesFPSFixEnabled();
    if (!init && GObjectsaddress && AppendStringaddress && ProcessEventaddress) EnableConsole();
}

// 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::DOF)                  { g_DOF_fix_enabled = enabled; DOFFixEnabled(); }
    if (fix == GameFixes::ChromaticAberrations) { g_CA_fix_enabled = enabled; CAFixEnabled(); }
    if (fix == GameFixes::Vignetting)           { g_Vignetting_fix_enabled = enabled; VignettingFixEnabled(); }
    if (fix == GameFixes::VolumetricFog)        { g_Volumetric_Fog_fix_enabled = enabled; VolumetricFogFixEnabled(); }
    if (fix == GameFixes::Camera)               { g_Camera_fix_enabled = enabled; CameraFixEnabled(); }
    if (fix == GameFixes::Cutscenes)            { g_Cutscenes_fix_enabled = enabled; ProcessEvent(); CutscenesUltrawideFixEnabled(); }
    if (fix == GameFixes::SkipIntro)            { g_SkipIntro_fix_enabled = enabled; }
    if (fix == GameFixes::Framerate)            { g_Cutscenes_FPS_fix_enabled = enabled; CutscenesFPSFixEnabled(); }
    if (fix == GameFixes::Fog)                  { g_Fog_fix_enabled = enabled; FogFixEnabled(); }
    if (fix == GameFixes::TimeDilation)         { g_TimeDilation_fix_enabled = enabled; EnableCheats(Cheat::TimeDilation); }
    if (fix == GameFixes::Stealth)              { g_StealthMode_fix_enabled = enabled; EnableCheats(Cheat::Stealth); }
}

// Setters for Reshade addon call
extern "C" __declspec(dllexport) void SetFOV(int fov) {
    g_AdditionalFOVValue = fov;
}

extern "C" __declspec(dllexport) void SetCameraDistance(float multiplier) {
    g_CameraDistance = multiplier;
}

extern "C" __declspec(dllexport) void SetFogDensity(float fogDensity) {
    g_FogDensity = fogDensity;
}

extern "C" __declspec(dllexport) void SetFogMaxOpacity(float fogMaxOpacity) {
    g_FogMaxOpacity = fogMaxOpacity;
}

extern "C" __declspec(dllexport) void SetWorldTimeDilation(float timeDilation) {
    g_WorldTimeDilationValue = timeDilation;
}

extern "C" __declspec(dllexport) void SetAITimeDilation(float timeDilation) {
    g_AITimeDilationValue = timeDilation;
}

// 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->cameraIn = g_Camera_In;
    infos->cameraOut = g_Camera_Out;
    infos->defaultFogDensity = g_DefaultFogDensity;
    infos->fogMaxOpacity = g_DefaultFogMaxOpacity;
    infos->consoleEnabled = g_Console_Enabled;
}

// Code injection functions
static void FOVFixEnabled() {
    if (g_fix_enabled && g_fov_fix_enabled && FOVaddress) {
        if (!FOVHook) FOVHook = safetyhook::create_mid(FOVaddress, // Hook only once
            [](SafetyHookContext& ctx) { // SDK based FOV -> AdvancedThirdPersonCamera.ATPCFOVSettings:CameraFOV
                                         // ATPC.CameraMode.FOVSettings.CameraFOV
                g_FOV_In = ctx.xmm6.f32[0];
                ctx.xmm6.f32[0] += (g_fix_enabled && g_fov_fix_enabled ? g_AdditionalFOVValue : g_FOV_In);
                g_FOV_Out = ctx.xmm6.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 void CameraFixEnabled() {
    if (g_fix_enabled && g_Camera_fix_enabled && Cameraaddress) {
        if (!CameraHook) CameraHook = safetyhook::create_mid(Cameraaddress,
            [](SafetyHookContext& ctx) {
                g_Camera_In = ctx.xmm0.f32[0];
                ctx.xmm0.f32[0] *= g_CameraDistance;
                g_Camera_Out = ctx.xmm0.f32[0];
            });
        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 FogFixEnabled() {
    if (g_fix_enabled && g_Fog_fix_enabled && FogDensityEngineaddress && FogOpacityEngineaddress) {
        if (!FogDensityHook) FogDensityHook = safetyhook::create_mid(FogDensityEngineaddress + 0x7, // Hook for fog density
            [](SafetyHookContext& ctx) {
                ctx.xmm1.f32[0] = g_FogDensity;
                ctx.rflags &= ~(1ULL << 6);
            });
        else FogDensityHook.enable();

        if (!FogMaxOpacityHook) FogMaxOpacityHook = safetyhook::create_mid(FogOpacityEngineaddress + 0x7, // Hook for fog opacity
            [](SafetyHookContext& ctx) {
                ctx.xmm1.f32[0] = g_FogMaxOpacity;
                ctx.rflags &= ~(1ULL << 6);
            });
        else FogMaxOpacityHook.enable();
        logger->info("Fog fix enabled");
    }
    if (!(g_fix_enabled && g_Fog_fix_enabled)) {
        if (FogDensityHook) FogDensityHook.disable();
        if (FogMaxOpacityHook) FogMaxOpacityHook.disable();
        logger->info("Fog fix disabled");
    }

    if (g_fix_enabled && g_Fog_fix_enabled && Fogaddress) {
        if (!FogHook) FogHook = safetyhook::create_mid(Fogaddress + 0x5f, // Generic hook for fogs fixes
            [](SafetyHookContext& ctx) {
                UWorld* World = UWorld::GetWorld();
                if (!World) return;

                if (World != previousWorld) {
                    previousWorld = World;
                    previousLevel = World->PersistentLevel;
                    previousFogActor = nullptr;
                    g_DefaultFogDensity = -1.f;
                    g_DefaultFogMaxOpacity = -1.f;
                }

                if (!previousFogActor || !previousFogActor->Component) { // Dynamic search for fog actor if none is valid
                    if (previousLevel) {
                        for (AActor* Actor : previousLevel->Actors) {
                            if (!Actor || !Actor->Class) continue;
                            if (Actor->IsA(AExponentialHeightFog::StaticClass())) {
                                previousFogActor = reinterpret_cast<AExponentialHeightFog*>(Actor);
                                // Save default values
                                g_DefaultFogDensity = previousFogActor->Component->FogDensity;
                                g_DefaultFogMaxOpacity = previousFogActor->Component->FogMaxOpacity;
                                break;
                            }
                        }
                    }
                }
                if (previousFogActor && previousFogActor->Component) {
                    // Get default engine values
                    if (g_DefaultFogDensity == -1.f) g_DefaultFogDensity = previousFogActor->Component->FogDensity;
                    if (g_DefaultFogMaxOpacity == -1.f) g_DefaultFogMaxOpacity = previousFogActor->Component->FogMaxOpacity;
                    // Force fix values
                    previousFogActor->Component->SetFogDensity(g_fix_enabled && g_Fog_fix_enabled ? g_FogDensity : g_DefaultFogDensity);
                    previousFogActor->Component->SetFogMaxOpacity(g_fix_enabled && g_Fog_fix_enabled ? g_FogMaxOpacity : g_DefaultFogMaxOpacity);
                }
            });
    }
}

static void VolumetricFogFixEnabled() {
    static uint8_t* bVolumetricFog = nullptr;

    if (g_fix_enabled && g_Volumetric_Fog_fix_enabled && VolumetricFogaddress) {
        bVolumetricFog = Memory::GetAddressFromOpcode(VolumetricFogaddress, 2, 7); // Get adress where is stored r.VolumetricFog
        if (!VolumetricFogHook) VolumetricFogHook = safetyhook::create_mid(VolumetricFogaddress +0x7,
            [](SafetyHookContext& ctx) { *bVolumetricFog = 0;}); // r.VolumetricFog = 0
        else VolumetricFogHook.enable();
        logger->info("Volumetric fog fix enabled");
    }
    if (!(g_fix_enabled && g_Volumetric_Fog_fix_enabled) && VolumetricFogaddress) {
        if (VolumetricFogHook) VolumetricFogHook.disable();
        if (bVolumetricFog) *bVolumetricFog = 1; // r.VolumetricFog = 1
        logger->info("Volumetric fog fix disabled");
    }
}

static APawn* g_PlayerPawn = nullptr;
static void ProcessEvent() {
    if (!PEHook && ProcessEventaddress) {
        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->GetName();
                    static bool bLastState = false;

                    if (object->GetName().contains("TitleFlow")) { // Intro Skip
                        UWBP_TitleFlow_C* titleFlow = static_cast<UWBP_TitleFlow_C*>(object);
                        if (titleFlow && titleFlow->IsA(UWBP_TitleFlow_C::StaticClass()) && g_SkipIntro_fix_enabled) {
                            if (titleFlow->WBP_DevTitleFlow_Logo) titleFlow->WBP_DevTitleFlow_Logo->ShowNextLogo();
                            if (titleFlow->WBP_TitleFlow_Warning) titleFlow->WBP_TitleFlow_Warning->ShowNextWarning();
                        }
                    }

                    bool bEnable = g_fix_enabled && g_Cutscenes_fix_enabled;
                    if (object->GetName().contains("WBP_Cutscene_C") && (funcName == "Destruct" || funcName == "Construct")) bLastState = false;
                    if (object->GetName().contains("WBP_Cutscene_C")) { // Black bars widgets removal
                        auto cutscene = static_cast<UWBP_Cutscene_C*>(object);
                        if (bEnable != bLastState) {
                            float opacity = bEnable ? 0.f : 1.f;
                            cutscene->LeftGroup_1->SetRenderOpacity(opacity);
                            cutscene->LeftGroup_2->SetRenderOpacity(opacity);
                            cutscene->RightGroup_1->SetRenderOpacity(opacity);
                            cutscene->RightGroup_2->SetRenderOpacity(opacity);
                            bLastState = bEnable;
                        }
                        // Unlock FPS in cinematics ALevelSequenceActor -> ULevelSequencePlayer -> UMovieSceneSequencePlayer -> SetFrameRate
                        if (cutscene->SequenceActor && cutscene->SequenceActor->SequencePlayer) {
                            FFrameRate framerate = { g_Cutscenes_FPS_fix_enabled ? 240 : 30 , 1 };
                            cutscene->SequenceActor->SequencePlayer->SetFrameRate(framerate);
                        }
                    }
                }
        });
    }

    if (g_fix_enabled && g_Cutscenes_fix_enabled && !ConstrainAspectRatioHook) {
        if (!ConstrainAspectRatioHook) ConstrainAspectRatioHook = safetyhook::create_mid(ConstrainAspectRatioaddress,
            [](SafetyHookContext& ctx) {
                ctx.rcx = 0;  // bConstrainAspectRatio
            });
    }
    if (g_fix_enabled && g_Cutscenes_fix_enabled && !AspectRatioAxisConstraintHook)
        if(!AspectRatioAxisConstraintHook) AspectRatioAxisConstraintHook = safetyhook::create_mid(AspectRatioAxisConstraintaddress,
            [](SafetyHookContext& ctx) {
                ctx.rdx = 0; // AspectRatioAxisConstraint Y
            });
}

static void CutscenesUltrawideFixEnabled() {
    logger->info("Cutscenes ultrawide fix {}", g_fix_enabled && g_Cutscenes_fix_enabled ? "enabled" : "disabled");
}

static void CutscenesFPSFixEnabled() {
    if (g_fix_enabled && g_Cutscenes_FPS_fix_enabled && CutscenesFPSaddress) {
        if (!CutscenesFPSHook) CutscenesFPSHook = safetyhook::create_mid(CutscenesFPSaddress + 0xa,
            [](SafetyHookContext& ctx) {
                ctx.xmm0.f32[0] = 0.f;
            });
        else CutscenesFPSHook.enable();
        logger->info("Cutscenes FPS unlocker fix enabled");
    }
    if (!(g_fix_enabled && g_Cutscenes_FPS_fix_enabled) && CutscenesFPSaddress)
        logger->info("Cutscenes FPS unlocker fix disabled");
}

constexpr int ScanEveryNFrames = 120;
static std::unordered_map<ANoceEnemyCharacter*, float> CachedAlertness;
// Cheats
static void EnableCheats(Cheat cheat) {
    if (WorldTimedilationaddress && !WorldTimeDilationHook) {
        WorldTimeDilationHook = safetyhook::create_mid(WorldTimedilationaddress + 0x10,
            [](SafetyHookContext& ctx) {
                // From AWorldSettings retrieved from world->K2_GetWorldSettings()
                ctx.xmm0.f32[0] *= g_TimeDilation_fix_enabled ? g_WorldTimeDilationValue : 1.f;

                UWorld* world = UWorld::GetWorld();
                if (world && world->OwningGameInstance->LocalPlayers[0]->PlayerController->AcknowledgedPawn)
                    g_PlayerPawn = world->OwningGameInstance->LocalPlayers[0]->PlayerController->AcknowledgedPawn;

                // Scan levels and actors every 120 frames
                static uint64_t FrameCounter = 0;
                if (++FrameCounter < ScanEveryNFrames) return;
                FrameCounter = 0;
                // Enemies time dilation & stealth
                for (int i = 0; i < world->Levels.Num(); i++) { // Loop through level to find actors
                    ULevel* Level = world->Levels[i];
                    if (!Level) continue;

                    for (int j = 0; j < Level->Actors.Num(); j++) { // Loop through actors
                        AActor* actor = Level->Actors[j];
                        if (!actor || !g_PlayerPawn || actor == g_PlayerPawn) continue; // We don't want to affect Hinako
                        if (actor->IsA(ANoceEnemyCharacter::StaticClass())) { // actor is enemy
                            actor->CustomTimeDilation = g_TimeDilation_fix_enabled ? g_AITimeDilationValue : 1.f; // Enemy time dilation
                            // Set enemy alert
                            ANoceEnemyCharacter* enemy = static_cast<ANoceEnemyCharacter*>(actor);
                            if (enemy && enemy->EnemyAIController) {
                                if (g_StealthMode_fix_enabled) {
                                    if (CachedAlertness.find(enemy) == CachedAlertness.end())
                                        CachedAlertness[enemy] = enemy->EnemyAIController->GetAlertness();

                                    enemy->EnemyAIController->SetAlertness(0.f, true); // Set enemy alert to 0
                                }
                                else {
                                    auto it = CachedAlertness.find(enemy);
                                    if (it != CachedAlertness.end()) {
                                        enemy->EnemyAIController->SetAlertness(it->second, true); // Restore original enemy alert
                                        CachedAlertness.erase(it);
                                    }
                                }
                            }
                        }
                    }
                }
            });
    }
    if (cheat == Cheat::TimeDilation) logger->info("Time dilation cheat {}",g_TimeDilation_fix_enabled ? "enabled" : "disabled");
    if (cheat == Cheat::Stealth) logger->info("Stealth mode cheat {}", g_StealthMode_fix_enabled ? "enabled" : "disabled");
}

// Memory patch fixes
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 CAFixEnabled() {
    if (g_fix_enabled && g_CA_fix_enabled && CAaddress) {
        Memory::PatchBytes(CAaddress, "\x90\x90", 2); // NOP jg "HellIsUs-Win64-Shipping.exe"+3D73BD5
        logger->info("Chromatics aberrations fix enabled");
    }
    if (!(g_fix_enabled && g_CA_fix_enabled) && CAaddress) {
        Memory::RestoreBytes(CAaddress);
        logger->info("Chromatics 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");
    }
}

// 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>("Silent Hill f", 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;
}