#include "CommonHeaders.h"
#include "UEngine.hpp"
#include "UETools.hpp"
#include "UEvars.hpp"
#include "Logger.hpp"
#include "SDK/Basic.hpp"
#include "SDK/Engine_classes.hpp"
#include "SDK/SHCharacterPlay_BP_classes.hpp"
#include "SDK/UMG_parameters.hpp"

using namespace SDK;
// Constants
const std::string PLUGIN_NAME = "SilentHill22024";
const std::string PLUGIN_LOG = PLUGIN_NAME + ".log";
constexpr ULONGLONG DEFAULT_ACTORS_SCAN_BETWEEN_TICKS = 300; // Used for enemies time dilation

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

// Screen informations
static int screenWidth = GetSystemMetrics(SM_CXSCREEN);
static int screenHeight = GetSystemMetrics(SM_CYSCREEN);
const float baseAspect = 1.7777778; // Game base aspect ratio
static float aspectRatio = (float)screenWidth / screenHeight;

// Plugin states
static bool AOBScanDone = false;
static bool g_Console = false;
static bool g_fix_enabled = false;
static bool g_fov_fix_enabled = false;
static bool g_ultrawide_fix_enabled = false;
static bool g_CutscenesFPS_fix_enabled = false;
static bool g_camera_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_Fog_fix_enabled = false;
static bool g_TimeDilation_fix_enabled = false;
static bool g_GodMode_fix_enabled = false;
static bool g_IgnoreHits_fix_enabled = false;
static bool g_Stealth_fix_enabled = false;
static bool g_SkipIntro_fix_enabled = false;
static float g_FogDensity = 0.15f;
static float g_FogOpacity = 1.f;
static int g_AdditionalFOVValue = 0;
static float g_CameraDistanceValue = 1.f;
static float g_WorldTimeDilationValue = 0.5f;
static float g_AITimeDilationValue = 0.2f;
static float g_Health = 0.f;
static bool user_inputs_logged = false;

// Shared values
static float g_FOV_In = 60.f;
static float g_FOV_Out = 60.f;
static float g_CompensatedFOV = 0.f;
static float g_CameraIn = 156.25f;
static float g_CameraOut = 156.25f;

// AOB Scan pointers
static uint8_t* FOVaddress = nullptr;
static uint8_t* UltraWideaddress = nullptr;
static uint8_t* Cameraaddress = nullptr;
static uint8_t* CutscenesFPSaddress = nullptr;
static uint8_t* DOFaddress = nullptr;
static uint8_t* CAaddress = nullptr;
static uint8_t* Vignettingaddress = nullptr;
static uint8_t* FogDensityEngineaddress = nullptr;
static uint8_t* FogOpacityEngineaddress = nullptr;
static uint8_t* WorldTimedilationaddress = nullptr;
static uint8_t* GodModeaddress = nullptr;

// Hooking
static SafetyHookMid FOVHook{};
static SafetyHookMid FogDensityHook{};
static SafetyHookMid FogOpacityHook{};
static SafetyHookMid CameraHook{};
static SafetyHookMid CutscenesFPSHook{};
static SafetyHookMid DOFHook{};
static SafetyHookMid PEHook{};
static SafetyHookMid WorldTimeDilationHook{};

// Prototypes
static void FOVFixEnabled();
static void UltraWideFixEnabled();
static void CutsceneFPSFixEnabled();
static void CameraFixEnabled();
static void DOFFixEnabled();
static void CAFixEnabled();
static void VignettingFixEnabled();
static void FogFixEnabled();
static void EnableConsole();
static void EnableCheats(Cheat cheat);
static void ProcessEvent();

extern "C" __declspec(dllexport) void SetFixEnabled(bool enabled, bool init) {
    g_fix_enabled = enabled;
    if (!AOBScanDone) { // Unreal Engine 5.1.1
        uint8_t* baseModule = reinterpret_cast<uint8_t*>(GetModuleHandleA(nullptr)); // Get game base address
        logger->info("--------------- AOB scan started ---------------");
        constexpr auto FOVStringObfuscated = make_obfuscated<0xF3>("77 ?? 48 8B ?? FF 90 ?? ?? ?? ?? F3 0F ?? ?? ?? 48");
        constexpr auto UltraWideStringObfuscated = make_obfuscated<0x4A>("C7 47 ?? ?? ?? ?? ?? C7 83 ?? ?? ?? ?? ?? ?? ?? ?? 48 8B ?? ?? ?? 48 83");
        constexpr auto CameraStringObfuscated = make_obfuscated<0x8D>("E8 ?? ?? ?? ?? F2 0F ?? ?? ?? ?? 48 8D ?? ?? F3 0F ?? ?? ?? ?? ?? ?? 48 8D");
        constexpr auto CutscenesFPSStringObfuscated = make_obfuscated<0x9F>("FF 95 ?? ?? ?? ?? 48 8B ?? ?? ?? 48 8B ?? ?? ?? 48 89 9F ?? ?? ?? ?? 48 8B ?? ?? ?? 48 8B ?? ?? ?? 48 83 ?? ?? 41 ?? C3");
        constexpr auto DOFStringObfuscated = make_obfuscated<0x83>("83 ?? ?? ?? 0F 8E ?? ?? ?? ?? 0F 2F ?? ?? ?? ?? ?? 0F 83 ?? ?? ?? ?? 0F 2F ?? ?? ?? ?? ?? 0F 83");
        constexpr auto CAStringObfuscated = make_obfuscated<0x4A>("7F ?? 89 B3 ?? ?? ?? ?? 8B ?? ?? 39 05 ?? ?? ?? ?? 7E");
        constexpr auto VignettingStringObfuscated = make_obfuscated<0xB3>("8B ?? 83 ?? ?? 7D ?? 89 B3 ?? ?? ?? ?? EB ?? 83");
        constexpr auto FogDensityStringObfuscated = make_obfuscated<0xA0>("0F 2E ?? A0 02 ?? ?? 74 ?? F3 0F ?? ?? ?? ?? ?? ?? E9 ?? ?? ?? ?? C3");
        constexpr auto FogOpacityStringObfuscated = make_obfuscated<0xBB>("0F 2E ?? 34 03 ?? ?? 74 ?? F3 0F ?? ?? ?? ?? ?? ?? E9 ?? ?? ?? ?? C3");
        constexpr auto WorldTimeDilationStringObfuscated = make_obfuscated<0x59>("F3 0F 10 ?? ?? ?? ?? ?? F3 0F 59 ?? E8 03 ?? ?? F3 0F 59 ?? ?? ?? ?? ?? C3");
        constexpr auto GodModeStringObfuscated = make_obfuscated<0XC1>("F3 0F 11 ?? ?? ?? ?? ?? 48 ?? ?? 48 81 C1 ?? ?? ?? ?? E8 ?? ?? ?? ?? F3 0F 10 83");
        // Prepare all data for scanning
        using AOBScan::Make;
        using OffsetScan::Make;
        // Prepare all data for scanning
        std::vector<AOBScanEntry> signatures = {
            Make(&FOVaddress, FOVStringObfuscated, "FOV"),
            Make(&UltraWideaddress, UltraWideStringObfuscated, "Ultrawide"),
            Make(&Cameraaddress, CameraStringObfuscated, "Camera"),
            Make(&CutscenesFPSaddress, CutscenesFPSStringObfuscated, "Cutscene FPS"),
            Make(&DOFaddress, DOFStringObfuscated, "DOF"),
            Make(&CAaddress, CAStringObfuscated, "Chromatic aberrations"),
            Make(&Vignettingaddress, VignettingStringObfuscated, "Vignetting"),
            Make(&FogDensityEngineaddress, FogDensityStringObfuscated, "Fog density"),
            Make(&FogOpacityEngineaddress, FogOpacityStringObfuscated, "Fog opacity"),
            Make(&WorldTimedilationaddress, WorldTimeDilationStringObfuscated, "World time dilation"),
            Make(&GodModeaddress, GodModeStringObfuscated, "God mode")
        };
        // Scan all signature in a batch
        Memory::AOBScanBatch(signatures, logger);

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

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

            constexpr auto GObjetcsStringObfuscated = make_obfuscated<0x8B>("48 8B ?? ?? ?? ?? ?? 48 8B ?? ?? 48 8D ?? ?? EB ?? 33");
            constexpr auto GWorldStringObfuscated = make_obfuscated<0x83>("48 8B 05 ?? ?? ?? ?? 48 ?? ?? 74 ?? 48 ?? ?? E8 ?? ?? ?? ?? 83");
            constexpr auto AppendStringStringObfuscated = make_obfuscated<0x3D>("48 89 ?? ?? ?? 48 89 ?? ?? ?? 57 48 83 ?? ?? 80 3D ?? ?? ?? ?? ?? 48 ?? F2 8B ?? 48 ?? ?? 74 ?? 4C 8D ?? ?? ?? ?? ?? EB ?? 48 8D ?? ?? ?? ?? ?? E8 ?? ?? ?? ?? 4C");
            constexpr auto ProcessEventStringObfuscated = make_obfuscated<0x81>("40 ?? 56 57 41 ?? 41 ?? 41 ?? 41 ?? 48 81 ?? ?? ?? ?? ?? 48 8D ?? ?? ?? 48 89 ?? ?? ?? ?? ?? 48 8B ?? ?? ?? ?? ?? 48 ?? ?? 48 89 ?? ?? ?? ?? ?? 4D");
            // Prepare all data for scanning
            std::vector<OffsetScanEntry> UEoffsetsScans = {
                Make(&GObjectsaddress, GObjetcsStringObfuscated, "GObjects", OffsetCalcType::GetOffsetFromOpcode, &Offsets::GObjects, 0x3),
                Make(&GWorldaddress, GWorldStringObfuscated, "GWorld", OffsetCalcType::GetOffsetFromOpcode, &Offsets::GWorld, 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 && UltraWideaddress) UltraWideFixEnabled();
    if (!init && CutscenesFPSaddress) CutsceneFPSFixEnabled();
    if (!init && Cameraaddress) CameraFixEnabled();
    if (!init && DOFaddress) DOFFixEnabled();
    if (!init && CAaddress) CAFixEnabled();
    if (!init && Vignettingaddress) VignettingFixEnabled();
    if (!init && FogDensityEngineaddress && FogOpacityEngineaddress) FogFixEnabled();
    if (!init && WorldTimedilationaddress) {
        EnableCheats(Cheat::TimeDilation);
        EnableCheats(Cheat::GodMode);
        EnableCheats(Cheat::IgnoreHits);
        EnableCheats(Cheat::Stealth);
    }
    ProcessEvent();
}

// 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::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(); }
    if (fix == GameFixes::Framerate)            { g_CutscenesFPS_fix_enabled = enabled; CutsceneFPSFixEnabled(); }
    if (fix == GameFixes::TimeDilation)         { g_TimeDilation_fix_enabled = enabled; EnableCheats(Cheat::TimeDilation); }
    if (fix == GameFixes::GodMode)              { g_GodMode_fix_enabled = enabled; EnableCheats(Cheat::GodMode); }
    if (fix == GameFixes::IgnoreHits)           { g_IgnoreHits_fix_enabled = enabled; EnableCheats(Cheat::IgnoreHits); }
    if (fix == GameFixes::SkipIntro)            { g_SkipIntro_fix_enabled = enabled; }
    if (fix == GameFixes::Stealth)              { g_Stealth_fix_enabled = enabled; EnableCheats(Cheat::Stealth); }
}

extern "C" __declspec(dllexport) void SetValues(GameSetting setting, float value) {
    if (setting == GameSetting::FOV) g_AdditionalFOVValue = (int)(value);
    if (setting == GameSetting::CameraDistance) g_CameraDistanceValue = value;
    if (setting == GameSetting::FogDensity) g_FogDensity = value;
    if (setting == GameSetting::FogMaxOpacity) g_FogOpacity = value;
    if (setting == GameSetting::WorldTimeDilation) g_WorldTimeDilationValue = value;
    if (setting == GameSetting::AITimeDilation) g_AITimeDilationValue = value;
}

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

    infos->FOVIn = g_FOV_In;
    infos->CompensatedFOV = g_CompensatedFOV;
    infos->FOVOut = g_FOV_Out;
    infos->cameraIn = g_CameraIn;
    infos->cameraOut = g_CameraOut;
    infos->consoleEnabled = g_Console_Enabled;
    infos->Health = g_Health;
}

// Hooks injections
static bool inContinue = false;
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 && g_DOF_fix_enabled) {
                    std::string funcName = func->GetName();
                    std::string objectName = object->GetName();
                    // Skip disclaimers
                    if (objectName.contains("StartGameMenuWidget")) {
                        auto* widget = static_cast<UStartGameWidget*>(object);
                        if (widget && widget->Class && widget->IsA(UStartGameWidget::StaticClass()) && funcName == "OnNext" && !inContinue) {
                            inContinue = true;
                            if (g_SkipIntro_fix_enabled) widget->Continue();
                            inContinue = false;
                        }
                    }
                }
            });
    }
}

static void FOVFixEnabled() {
    if (g_fix_enabled && (g_fov_fix_enabled || g_ultrawide_fix_enabled) && FOVaddress) {
        if (!FOVHook) { // Hook only once
            FOVHook = safetyhook::create_mid(FOVaddress + 0x10,
                [](SafetyHookContext& ctx) {
                    g_FOV_In = ctx.xmm0.f32[0];
                    if (g_fix_enabled && g_ultrawide_fix_enabled)
                        ctx.xmm0.f32[0] = Maths::CompensateHorizontalFOV(g_FOV_In, baseAspect, aspectRatio);
                    g_CompensatedFOV = ctx.xmm0.f32[0];
                    ctx.xmm0.f32[0] += (g_fix_enabled && g_fov_fix_enabled ? g_AdditionalFOVValue : 0);
                    g_FOV_Out = ctx.xmm0.f32[0];
                });
        }
        else FOVHook.enable();
        logger->info("FOV fix enabled");
    }
    if (!(g_fix_enabled && (g_fov_fix_enabled || g_ultrawide_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) { // Hook only once
            CameraHook = safetyhook::create_mid(Cameraaddress + 0x17,
                [](SafetyHookContext& ctx) {
                    g_CameraIn = ctx.xmm1.f32[0];
                    ctx.xmm1.f32[0] *= (g_fix_enabled && g_camera_fix_enabled ? g_CameraDistanceValue : 1.f);
                    g_CameraOut = ctx.xmm1.f32[0];
                });
        }
        else CameraHook.enable();
        logger->info("Camera distance fix enabled");
    }
    if (!(g_fix_enabled && g_camera_fix_enabled) && Cameraaddress) {
        if (CameraHook) CameraHook.disable();
        logger->info("Camera distance fix disabled");
    }
}

static void DOFFixEnabled() {
    if (g_fix_enabled && g_DOF_fix_enabled && DOFaddress) {
        if (!DOFHook) {
            DOFHook = safetyhook::create_mid(DOFaddress,
                [](SafetyHookContext& ctx) {
                    if (!ctx.rax) return;
                    uint8_t* rDepthOfFieldQuality = reinterpret_cast<uint8_t*>(ctx.rax) + 0x4;
                    if (g_DOF_fix_enabled)
                        Memory::PatchBytes(rDepthOfFieldQuality, "\x00", 1); // r.DepthOfFieldQuality = 0 <= memory patched
                    else 
                        Memory::RestoreBytes(rDepthOfFieldQuality); // r.DepthOfFieldQuality = 0 <= memory patched
                });
        }
        logger->info("Depth of field fix enabled");
    }
    if (!(g_fix_enabled && g_DOF_fix_enabled) && DOFaddress)
        logger->info("Depth of field fix disabled");
}

static void CutsceneFPSFixEnabled() {
    if (g_fix_enabled && g_CutscenesFPS_fix_enabled && CutscenesFPSaddress) {
        if (!CutscenesFPSHook) {
            CutscenesFPSHook = safetyhook::create_mid(CutscenesFPSaddress,
                [](SafetyHookContext& ctx) {
                    if (!ctx.rcx) return;
                    // Unlock FPS in cinematics ALevelSequenceActor -> ULevelSequencePlayer -> UMovieSceneSequencePlayer -> SetFrameRate
                    ACutscenePlayer* cutscene = reinterpret_cast<ACutscenePlayer*>(ctx.rcx);
                    if (cutscene && cutscene->IsA(ACutscenePlayer::StaticClass())) {
                        if (cutscene->LevelSequenceActor && cutscene->LevelSequenceActor->SequencePlayer) {
                            FFrameRate framerate = { g_CutscenesFPS_fix_enabled ? 240 : 30 , 1 };
                            cutscene->LevelSequenceActor->SequencePlayer->SetFrameRate(framerate);
                        }
                    }
                });
        }
        logger->info("Cutscenes FPS fix enabled");
    }
    if (!(g_fix_enabled && g_CutscenesFPS_fix_enabled) && CutscenesFPSaddress)
        logger->info("Cutscenes FPS fix disabled");
}

static void FogFixEnabled() {
    if (g_fix_enabled && g_Fog_fix_enabled && FogDensityEngineaddress) {
        if (!FogDensityHook) { // Hook only once
            FogDensityHook = safetyhook::create_mid(FogDensityEngineaddress,
                [](SafetyHookContext& ctx) {
                    ctx.xmm1.f32[0] = g_FogDensity;
                });
        }
        else FogDensityHook.enable();
    }
    if (!(g_fix_enabled && g_Fog_fix_enabled) && FogDensityEngineaddress)
        if (FogDensityHook) FogDensityHook.disable();

    if (g_fix_enabled && g_Fog_fix_enabled && FogOpacityEngineaddress) {
        if (!FogOpacityHook) { // Hook only once
            FogOpacityHook = safetyhook::create_mid(FogOpacityEngineaddress,
                [](SafetyHookContext& ctx) {
                    ctx.xmm1.f32[0] = g_FogOpacity;
                });
        }
        else FogOpacityHook.enable();
    }
    if (!(g_fix_enabled && g_Fog_fix_enabled) && FogOpacityEngineaddress)
        if (FogOpacityHook) FogOpacityHook.disable();

    if (g_fix_enabled && g_Fog_fix_enabled && FogDensityEngineaddress && FogOpacityEngineaddress)
        logger->info("Fog fix enabled");
    if (!(g_fix_enabled && g_Fog_fix_enabled) && FogDensityEngineaddress && FogOpacityEngineaddress)
        logger->info("Fog fix disabled");
}

// Cheats
static UWorld* LastWorld = nullptr;
static ULONGLONG lastScanTick = 0; // Last time Actors were scanned for enemies time dilation
static void EnableCheats(Cheat cheat) {
    //return;
    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) return;
                if (world != LastWorld) {
                    LastWorld = world;
                }
                APawn* playerPawn = GetPawnFromWorld(world);

                ULONGLONG now = GetTickCount64();
                if (now - lastScanTick < DEFAULT_ACTORS_SCAN_BETWEEN_TICKS) return;
                lastScanTick = now;

                if (world->Levels.Num() == 0) return;
                // 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 || !playerPawn || actor == playerPawn) continue; // We don't want to affect James
                        if (!actor->IsA(ASHAICharacter::StaticClass())) continue; // actor is enemy
                        // Set enemy alert
                        ASHAICharacter* enemy = static_cast<ASHAICharacter*>(actor);
                        if (!enemy) continue;
                        enemy->CustomTimeDilation = g_TimeDilation_fix_enabled ? g_AITimeDilationValue : 1.f; // Enemy time dilation
                    }
                }
                // ignore hits && Stealth mode
                if (!playerPawn) return;
                ASHCharacterPlay_BP_C* james = (ASHCharacterPlay_BP_C*)(playerPawn);
                if (!james || !james->IsA(ASHCharacterPlay_BP_C::StaticClass()) || !james->AIIgnoreMeStat) return;
                if (james->Health) g_Health = james->Health->HealthValue;
                james->AIIgnoreMeStat->_AlwaysMissMe = g_IgnoreHits_fix_enabled;
                james->AIIgnoreMeStat->_IgnoreMe = g_Stealth_fix_enabled;
            });
    }
    if (g_GodMode_fix_enabled && GodModeaddress)
        Memory::PatchBytes(GodModeaddress, "\x90\x90\x90\x90\x90\x90\x90\x90", 8); // subss xmm6,xmm9
    if (!g_GodMode_fix_enabled && GodModeaddress) Memory::RestoreBytes(GodModeaddress);

    if (cheat == Cheat::TimeDilation) logger->info("Time dilation cheat {}", g_TimeDilation_fix_enabled ? "enabled" : "disabled");
    if (cheat == Cheat::GodMode) logger->info("God mode cheat {}", g_GodMode_fix_enabled ? "enabled" : "disabled");
    if (cheat == Cheat::IgnoreHits) logger->info("Ignore hits cheat {}", g_IgnoreHits_fix_enabled ? "enabled" : "disabled");
    if (cheat == Cheat::Stealth) logger->info("Stealth cheat {}", g_Stealth_fix_enabled ? "enabled" : "disabled");
}

// Memory patch fixes
static void UltraWideFixEnabled() {
    if (g_fix_enabled && g_ultrawide_fix_enabled && UltraWideaddress) {
        const char* newAspectRatio = Memory::Float32ToHexBytes(aspectRatio); // Converts new apsect ratio in 4 char bytes
        Memory::PatchBytes(UltraWideaddress + 0x3, newAspectRatio, 4); // mov[rdi + 48], 3FE38E39 <== patch here
        logger->info("Ultra wide fix enabled");
    }
    if (!(g_fix_enabled && g_ultrawide_fix_enabled) && UltraWideaddress) {
        Memory::RestoreBytes(UltraWideaddress + 0x3);
        logger->info("Ultra wide fix disabled");
        if (!g_fov_fix_enabled) FOVFixEnabled();
    }
}

static void CAFixEnabled() {
    if (g_fix_enabled && g_CA_fix_enabled && CAaddress) {
        Memory::PatchBytes(CAaddress, "\x90\x90", 2); // NOP x 2 r.SceneColorFringeQuality = 0
        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 --------------");
    ReactivateDevConsole(logger);
}

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