#include "CommonHeaders.h"
#include "UEngine.hpp"
#include "UETools.hpp"
#include "UEWidgets.hpp"
#include "UEvars.hpp"
#include "Logger.hpp"
#include "SDK/Basic.hpp"
#include "SDK/Engine_classes.hpp"
#include "SDK/BP_CombatCharacter_Player_Final_classes.hpp"
#include "SDK/ExtendedStatComponent_Stamina_classes.hpp"
#include "SDK/Impl_BaseAIController_classes.hpp"
#include "SDK/WB_MainUI_classes.hpp"
#include "SDK/WB_ShowAddItemMain_New_classes.hpp"
#include "SDK/WB_NewHelpBlock_classes.hpp"
#include "SDK/WB_GameSaving_classes.hpp"
#include "SDK/WB_PoseMain_classes.hpp"
#include "SDK/WB_PlayerCoins_classes.hpp"
#include "SDK/WB_NewPlayerGuidelines_classes.hpp"

using namespace SDK;

// Constants
const std::string PLUGIN_NAME = "WuchangFF";
const std::string PLUGIN_LOG = PLUGIN_NAME + ".log";
constexpr ULONGLONG DEFAULT_DELAY_BETWEEN_TICK = 1000; // Used ProcessEvent tick
constexpr ULONGLONG DEFAULT_DELAY_BETWEEN_ACTOR_TICK = 30; // Used for enemies tick
// Logger
std::shared_ptr<spdlog::logger> logger;

// Screen informations
static int screenWidth = GetSystemMetrics(SM_CXSCREEN);
static int screenHeight = GetSystemMetrics(SM_CYSCREEN);
static float g_AspectRatio = (float)screenWidth / screenHeight;
float baseAspect = 1.777778f;

// Plugin states
static bool AOBScanDone = false;
static bool g_Console = false;
static std::atomic<bool> g_fix_enabled = false;
static std::atomic<bool> g_fov_fix_enabled = false;
static std::atomic<bool> g_ultrawide_fix_enabled = false;
static std::atomic<bool> g_HUD_fix_enabled = false;
static std::atomic<bool> g_Camera_fix_enabled = false;
static std::atomic<bool> g_DOF_fix_enabled = false;
static std::atomic<bool> g_CA_fix_enabled = false;
static std::atomic<bool> g_Vignetting_fix_enabled = false;
static std::atomic<bool> g_Fog_fix_enabled = false;
static std::atomic<bool> g_TimeDilation_fix_enabled = false;
static std::atomic<bool> g_GodMode_fix_enabled = false;
static std::atomic<bool> g_Stamina_fix_enabled = false;
static std::atomic<bool> g_Stealth_fix_enabled = false;
static int g_AdditionalFOVValue = 0;
static float g_CameraMultiplier = 1.f;
static int g_HUDOffsets = 0;
static float g_WorldTimeDilationValue = 1.f;
static float g_AITimeDilationValue = 1.f;
static bool user_inputs_logged = false;
// Player stats
double g_Health = 0.0;
double g_Stamina = 0.0;
// Shared values
static float g_FOV_In = 80.f;
static float g_CompensatedFOV = 80.f;
static float g_FOV_Out = 80.f;
static float g_CameraIn = 350.f;
static float g_CameraOut = 350.f;

// AOB Scan pointers
static uint8_t* CameraComponentaddress = nullptr;
static uint8_t* CameraDistanceaddress = nullptr;
static uint8_t* DOFaddress = nullptr;
static uint8_t* CAaddress = nullptr;
static uint8_t* Vignettingaddress = nullptr;
static uint8_t* Fogaddress = nullptr;
static uint8_t* WorldTimedilationaddress = nullptr;
static uint8_t* Timedilationaddress = nullptr;
// Hooking
static SafetyHookMid FOVHook{};
static SafetyHookMid UWHook{};
static SafetyHookMid CameraHook{};
static SafetyHookMid PEHook{};
static SafetyHookMid WorldTimeDilationHook{};
static SafetyHookMid TimeDilationHook{};

// Prototypes
static void FOVFixEnabled();
static void UltraWideFixEnabled(GameFixes fix);
static void HUDUpdate(bool writeLog);
static void CameraDistanceFixEnabled();
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
        logger->info("--------------- AOB scan started ---------------");
        constexpr auto CameraComponentStringObfuscated = make_obfuscated<0x8B>("EB ?? F3 0F ?? ?? ?? ?? ?? ?? F3 0F ?? ?? ?? 0F ?? ?? 8B");
        constexpr auto CameraDistanceStringObfuscated = make_obfuscated<0x5A>("48 8D 8F ?? ?? ?? ?? F2 0F ?? ?? ?? ?? ?? ?? 48 8D ?? ?? ?? ?? ?? 0F 5A");
        constexpr auto DOFStringObfuscated = make_obfuscated<0x6B>("48 ?? ?? 8B ?? ?? E8 ?? ?? ?? ?? 48 ?? ?? 48 6B ?? ?? 48 8D");
        constexpr auto VignettingStringObfuscated = make_obfuscated<0xB3>("8B ?? 83 ?? ?? 7D ?? 89 B3 ?? ?? ?? ?? EB");
        constexpr auto FogStringObfuscated = make_obfuscated<0x32>("75 ?? B3 ?? EB ?? 32 ?? 48 8B ?? ?? ?? 48 ?? ?? 74 ?? E8 ?? ?? ?? ?? 0F ?? ?? 48 8B");
        constexpr auto WorldTimeDilationStringObfuscated = make_obfuscated<0x59>("F3 0F 10 ?? ?? ?? ?? ?? F3 0F 59 ?? ?? ?? ?? ?? F3 0F 59 ?? ?? ?? ?? ?? C3");
        constexpr auto TimeDilationStringObfuscated = make_obfuscated<0x44>("F3 0F ?? ?? ?? ?? ?? ?? EB ?? F3 0F ?? ?? ?? ?? ?? ?? 48 8B ?? ?? 4C ?? ?? F3 0F ?? ?? 44");

        using AOBScan::Make;
        using OffsetScan::Make;
        // Prepare all data for scanning
        std::vector<AOBScanEntry> signatures = {
            Make(&CameraComponentaddress, CameraComponentStringObfuscated, "Camera component"),
            Make(&CameraDistanceaddress, CameraDistanceStringObfuscated, "Camera distance"),
            Make(&DOFaddress, DOFStringObfuscated, "Depth of field"),
            Make(&Vignettingaddress, VignettingStringObfuscated, "Vignetting"),
            Make(&Fogaddress, FogStringObfuscated, "Fog"),
            Make(&WorldTimedilationaddress, WorldTimeDilationStringObfuscated, "World time dilation"),
            Make(&Timedilationaddress, TimeDilationStringObfuscated, "Actor time dilation")
        };
        // Scan all signature in a batch
        Memory::AOBScanBatch(signatures, logger);

        if (CameraComponentaddress && DOFaddress && CameraDistanceaddress 
            && Vignettingaddress && Fogaddress && WorldTimedilationaddress && Timedilationaddress)
            logger->info("All AOB signatures found. Ready to patch...");

        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<0x8D>("48 8B ?? ?? ?? ?? ?? 48 8B ?? ?? 48 8D ?? ?? EB ?? 33");
            constexpr auto GWorldStringObfuscated = make_obfuscated<0x5B>("48 8B 05 ?? ?? ?? ?? 48 8D ?? ?? ?? ?? ?? 48 8B ?? ?? 48 89 ?? ?? 48 8D ?? ?? E8");
            constexpr auto AppendStringStringObfuscated = make_obfuscated<0x80>("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<0x56>("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 && CameraComponentaddress) {
        FOVFixEnabled();
		UltraWideFixEnabled(GameFixes::UltraWide);
    }
    if (!init && CameraDistanceaddress) CameraDistanceFixEnabled();
    if (!init && DOFaddress) DOFFixEnabled();
    if (!init) HUDUpdate(true);
    if (!init && Vignettingaddress) VignettingFixEnabled();
    if (!init && Fogaddress) FogFixEnabled();
    if (!init && WorldTimedilationaddress) {
        EnableCheats(Cheat::TimeDilation);
        EnableCheats(Cheat::GodMode);
        EnableCheats(Cheat::Stamina);
        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(GameFixes::UltraWide); }
    if (fix == GameFixes::Camera)               { g_Camera_fix_enabled = enabled; CameraDistanceFixEnabled(); }
    if (fix == GameFixes::HUD)                  { g_HUD_fix_enabled = enabled; HUDUpdate(true); }
    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::TimeDilation)         { g_TimeDilation_fix_enabled = enabled; EnableCheats(Cheat::TimeDilation); }
    if (fix == GameFixes::GodMode)              { g_GodMode_fix_enabled = enabled; EnableCheats(Cheat::GodMode); }
    if (fix == GameFixes::Stamina)              { g_Stamina_fix_enabled = enabled; EnableCheats(Cheat::Stamina); }
    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_CameraMultiplier = value;
    if (setting == GameSetting::WorldTimeDilation) g_WorldTimeDilationValue = value;
    if (setting == GameSetting::AITimeDilation) g_AITimeDilationValue = value;
    if (setting == GameSetting::HUD) {
        g_HUDOffsets = ((int)value * screenWidth) / 100;
        HUDUpdate(false);
    }
}
// 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 = (float)g_Health;
	infos->Stamina = (float)g_Stamina;
    infos->screenWidth = screenWidth;
	infos->screenHeight = screenHeight;
	infos->aspectRatio = g_AspectRatio;
}

// Code injection functions
static float g_UIScale = 1.f;
static UWidget* playerState = nullptr;
static UWidget* tipPanel = nullptr;
static UWidget* activeSlots = nullptr;
static UWidget* showItems = nullptr;
static UWidget* playerGuideLines = nullptr;
static UWidget* newHelpblock = nullptr;
static UWidget* coins = nullptr;
static UWidget* pose = nullptr;
static UWidget* gameSaving = nullptr;
static ULONGLONG lastScanResolutionTick = 0; // Last time Actors were scanned for enemies time dilation
static void ProcessEvent() {
    if (!PEHook && ProcessEventaddress) {
        PEHook = safetyhook::create_mid(ProcessEventaddress + 0xc,
            [](SafetyHookContext& ctx) {
                UObject* object = (UObject*)ctx.rcx;
                UFunction* func = (UFunction*)ctx.rdx;

                ULONGLONG now = GetTickCount64();
                if (now - lastScanResolutionTick >= DEFAULT_DELAY_BETWEEN_TICK) { // Delay between each tick to avoid ProcessEvent stuttering
                    lastScanResolutionTick = now;
                    GetResolution(screenWidth, screenHeight, g_AspectRatio);
                }
                if (object && func) {
                    std::string funcName = func->GetName();
                    if (funcName == "Construct") TrackWidgetConstruct((UUserWidget*)object);
                    if (funcName == "Destruct") TrackWidgetDestruct((UUserWidget*)object);
                    if (funcName == "Construct" || funcName == "Destruct") {
                        // Right HUD elements
                        auto HandleWidget = [&](auto* typedWidget, auto*& gWidgetPtr) { // Lambda to initialize pointers on Construct and nullify them on Destruct
                            if (funcName == "Construct") {
                                gWidgetPtr = typedWidget;
                                HUDUpdate(false);
                            }
                            else if (funcName == "Destruct") gWidgetPtr = nullptr;
                        };
                        if (object->IsA(UWB_NewPlayerGuidelines_C::StaticClass())) // valid
                            HandleWidget(static_cast<UWidget*>(object), playerGuideLines);
                        else if (object->IsA(UWB_GameSaving_C::StaticClass())) // valid
                            HandleWidget(static_cast<UWidget*>(object), gameSaving);
                        else if (object->IsA(UWB_ShowAddItemMain_New_C::StaticClass())) // valid
                            HandleWidget(static_cast<UWidget*>(object), showItems);
                        else if (object->IsA(UWB_NewHelpBlock_C::StaticClass())) // valid
                            HandleWidget(static_cast<UWidget*>(object), newHelpblock);
                        else if (object->IsA(UWB_PoseMain_C::StaticClass())) // valid
                            HandleWidget(static_cast<UWidget*>(object), pose);
                        else if (object->IsA(UWB_PlayerCoins_C::StaticClass())) // valid
                            HandleWidget(static_cast<UWidget*>(object), coins);
                        // Left HUD elements (player state, key tips, active slots)
                        else if (object->IsA(UWB_MainUI_C::StaticClass())) {
						    auto* HUDWidget = static_cast<UWB_MainUI_C*>(object);
                            if (funcName == "Construct") {
                                g_UIScale = SDK::UWidgetLayoutLibrary::GetViewportScale(HUDWidget);
							    //SetLogger(logger);
                                if (HUDWidget->WB_PlayerState) playerState = reinterpret_cast<UWidget*>(HUDWidget->WB_PlayerState);
                                if (HUDWidget->WB_KeyTipPanel_47) tipPanel = reinterpret_cast<UWidget*>(HUDWidget->WB_KeyTipPanel_47);
                                if (HUDWidget->WB_ActiveSlots) activeSlots = reinterpret_cast<UWidget*>(HUDWidget->WB_ActiveSlots);
                                HUDUpdate(false);
                            }
                            if (funcName == "Destruct") {
							    playerState = nullptr;
							    tipPanel = nullptr;
							    activeSlots = nullptr;
                            }
					    }
					}
                }
            });
    }
}

static void HUDUpdate(bool writeLog) {
    if (writeLog) logger->info("HUD & UI scaling fix {}", g_fix_enabled && g_HUD_fix_enabled ? "enabled" : "disabled");
    float HUDoffset = (g_fix_enabled && g_HUD_fix_enabled) ? g_HUDOffsets : 0.f;

    ApplyTransformOffset(playerState, HUDoffset / g_UIScale);
    ApplyTransformOffset(tipPanel, HUDoffset / g_UIScale);
    ApplyTransformOffset(pose, HUDoffset / g_UIScale);
    ApplyTransformOffset(activeSlots, -(HUDoffset / g_UIScale));
    ApplyTransformOffset(showItems, -(HUDoffset / g_UIScale));
    ApplyTransformOffset(playerGuideLines, -(HUDoffset / g_UIScale));
    ApplyTransformOffset(newHelpblock, -(HUDoffset / g_UIScale));
    ApplyTransformOffset(gameSaving, -(HUDoffset / g_UIScale));
    ApplyTransformOffset(coins, -(HUDoffset / g_UIScale));
}


static bool bHastoCompensateFOV = false;
static float currentAspect = 0.f;
static void FOVFixEnabled() {
    if (CameraComponentaddress) {
        if (!FOVHook) { // Hook only once
            FOVHook = safetyhook::create_mid(CameraComponentaddress + 0xa,
                [](SafetyHookContext& ctx) {
                    g_FOV_In = ctx.xmm0.f32[0];
                    if (g_fix_enabled && bHastoCompensateFOV)
                        ctx.xmm0.f32[0] = Maths::CompensateHorizontalFOV(g_FOV_In, currentAspect, g_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];
                });
        }
    }
    logger->info("FOV fix {}", g_fix_enabled && g_fov_fix_enabled ? "enabled" : "disabled");
}

static void UltraWideFixEnabled(GameFixes fix) {
    if (g_fix_enabled && g_ultrawide_fix_enabled && CameraComponentaddress/* && ARatioAxisConstraintaddress*/) {
	    Memory::PatchBytes(CameraComponentaddress + 0x1b, "\x31\xC9\x90\x90\x90\x90\x90", 7); // xor ecx,ecx bConstrainAspectRatio = false (for cutscenes)
        if (!UWHook) { // Hook only once
            UWHook = safetyhook::create_mid(CameraComponentaddress + 0x12,
                [](SafetyHookContext& ctx) {
					if (!ctx.rbx) return;

					currentAspect = *(float*)(ctx.rbx + 0x2f0);
                    if (currentAspect > baseAspect) bHastoCompensateFOV = true;
                    else bHastoCompensateFOV = false;
                });
        }
        else UWHook.enable();
        logger->info("Ultrawide fix enabled");
    }
    if (!(g_fix_enabled && g_ultrawide_fix_enabled) && CameraComponentaddress) {
        Memory::RestoreBytes(CameraComponentaddress + 0x1b);
        if (UWHook) UWHook.disable();
        bHastoCompensateFOV = false;
        logger->info("Ultrawide fix disabled");
    }
}

static void CameraDistanceFixEnabled() {
    if (g_fix_enabled && g_Camera_fix_enabled && CameraDistanceaddress) {
        if (!CameraHook) { // Hook only once
            CameraHook = safetyhook::create_mid(CameraDistanceaddress,
                [](SafetyHookContext& ctx) {
                    g_CameraIn = ctx.xmm0.f32[0];
                    ctx.xmm0.f32[0] *= g_CameraMultiplier;
                    g_CameraOut = ctx.xmm0.f32[0];
                });
        }
        else CameraHook.enable();
        logger->info("Camera distance fix enabled");
    }
    if (!(g_fix_enabled && g_Camera_fix_enabled) && CameraDistanceaddress) {
        if (CameraHook) CameraHook.disable();
        logger->info("Camera distance fix disabled");
    }
}

// -- Cheats --
static ULONGLONG lastScanActorTick = 0; // Last time Actors were scanned for applying 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;
            });
    }
    // Enemies time dilation
    if (Timedilationaddress && !TimeDilationHook) {
        TimeDilationHook = safetyhook::create_mid(Timedilationaddress,
            [](SafetyHookContext& ctx) {
                if (!ctx.rbx) return;
                UObject* object = (UObject*)ctx.rbx;
                if (!object || !object->Class) return;

                ULONGLONG now = GetTickCount64();
                if (now - lastScanActorTick >= DEFAULT_DELAY_BETWEEN_ACTOR_TICK) { // Delay between each tick to avoid ProcessEvent stuttering
                    lastScanActorTick = now;
                    if (object->IsA(ACharacter::StaticClass())) {
                        auto* character = static_cast<ACharacter*>(object);
                        if (!character) return;

                        if (!character->IsPlayerControlled()) { // NPC or enemy
                            auto* controller = character->GetController();
                            if (controller && controller->IsA(AImpl_BaseAIController_C::StaticClass())) { // enemy spotted
                                auto* aiController = static_cast<AImpl_BaseAIController_C*>(controller);
                                if (aiController && aiController->Perception) {
									if (!aiController->IsBoss) { // Won't apply to bosses to avoid breaking the fight
                                        if (g_Stealth_fix_enabled) aiController->EnableAlertPerception(false);
                                        else {
                                            aiController->EnableAlertPerception(true);
                                            aiController->FlushEnableSense();
                                            aiController->UpdateSenseTarget();
                                        }
                                    }
                                }
                                character->CustomTimeDilation = g_TimeDilation_fix_enabled ? g_AITimeDilationValue : 1.f;
                            }
                        }
                    }
                }
                
                if (object->IsA(ABP_CombatCharacter_Player_Final_C::StaticClass())) {
                    auto* character = static_cast<ABP_CombatCharacter_Player_Final_C*>(object);
                    if (character) {
                        if (character->IsPlayerControlled()) {
							double maxHP = 0.0;
                            double maxStamina = 0.0;
                            character->GetHp(&g_Health);
                            character->GetMaxHp(&maxHP);

                            UExtendedStatComponent_Stamina_C* staminaPtr = nullptr;
                            character->GetMyStamina(&staminaPtr);
                            if (staminaPtr) {
								staminaPtr->GetCurrentValue(&g_Stamina);
                                staminaPtr->GetMaxValue(&maxStamina);
                                if (g_Stamina_fix_enabled) staminaPtr->SetCurrentValue(maxStamina);
                            }
                            if (g_GodMode_fix_enabled) character->SetHp(maxHP);
                        }
                    }
                }
            });
    }

    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::Stamina) logger->info("Stamina cheat {}", g_Stamina_fix_enabled ? "enabled" : "disabled");
    if (cheat == Cheat::Stealth) logger->info("Stealth cheat {}", g_Stealth_fix_enabled ? "enabled" : "disabled");
}

// Memory patch fixes
static void DOFFixEnabled() {
    if (g_fix_enabled && g_DOF_fix_enabled && DOFaddress) {
        Memory::PatchBytes(DOFaddress + 0x3, "\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 + 0x3);
        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 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");
    }
}

static void FogFixEnabled() {
    if (g_fix_enabled && g_Fog_fix_enabled && Fogaddress) {
        Memory::PatchBytes(Fogaddress, "\xEB", 1); // jmp -> r.Fog 0
        logger->info("Fog fix enabled");
    }
    if (!(g_fix_enabled && g_Fog_fix_enabled) && Fogaddress) {
        Memory::RestoreBytes(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 --------------");
    ReactivateDevConsole(logger);
}

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