#include <string>
#include <spdlog/spdlog.h>
#include <spdlog/sinks/rotating_file_sink.h>
#include <filesystem>
#include <safetyhook.hpp>
#include "GameInformations.h"
#include "ObfuscateString.h"
#include "Memory.hpp";
#include "Maths.hpp";
#include "UEngine.hpp";
#include "SDK/Basic.hpp"
#include "SDK/Engine_classes.hpp"

using namespace SDK;

// Constants
const std::string PLUGIN_NAME = "BloodOfMehran";
const std::string PLUGIN_LOG = PLUGIN_NAME + ".log";
const std::string gameExecutable = "BloodOfMehranClient-Win64-Shipping.exe";

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

// Plugin states
static bool AOBScanDone = false;
static bool g_fix_enabled = false;
static bool g_fov_fix_enabled = false;
static bool g_ultrawide_fix_enabled = false;
static bool g_Camera_fix_enabled = false;
static bool g_DOF_fix_enabled = false;
static bool g_CA_fix_enabled = false;
static int g_AdditionalFOVValue = 0;
static float g_CameraDistance = 1.f;

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

// AOB Unreal Engine offsets addresses
static uint8_t* GObjectsaddress = nullptr;
static uint8_t* GNamesaddress = 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* CameraComponentaddress = nullptr;
static uint8_t* Cameraaddress = nullptr;
static uint8_t* ConstrainAspectRatioaddress = nullptr;
static uint8_t* AspectRatioAxisConstraintaddress = nullptr;

// Hooking
static SafetyHookMid FOVHook{};
static SafetyHookMid CameraHook{};

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

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

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

        if (!AspectRatioAxisConstraintaddress) {
            constexpr auto AspectRatioAxisConstraintStringObfuscated = make_obfuscated<0x4A>("41 0F ?? ?? ?? ?? ?? ?? 48 8D ?? ?? ?? ?? ?? 4C ?? ?? 4D ?? ?? E8");
            AspectRatioAxisConstraintaddress = Memory::AOBScan(gameExecutable, AspectRatioAxisConstraintStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

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

        if (!DOFaddress) {
            constexpr auto DOFStringObfuscated = make_obfuscated<0x4A>("8B ?? ?? 48 ?? ?? E8 ?? ?? ?? ?? 0F ?? ?? 48 6B ?? ?? 48 8D");
            DOFaddress = Memory::AOBScan(gameExecutable, DOFStringObfuscated.decrypt(), PAGE_EXECUTE_READ);
            //"BloodOfMehranClient-Win64-Shipping.exe" + 2680657 - 74 03          - je "BloodOfMehranClient-Win64-Shipping.exe" + 268065C
            //"BloodOfMehranClient-Win64-Shipping.exe" + 2680659 - 48 8B C3       - mov rax, rbx
            //"BloodOfMehranClient-Win64-Shipping.exe" + 268065C - 8B 34 30       - mov esi, [rax + rsi]
            //"BloodOfMehranClient-Win64-Shipping.exe" + 268065F - 48 8B CF       - mov rcx, rdi
            //"BloodOfMehranClient-Win64-Shipping.exe" + 2680662 - E8 E9 AF 6B 01 - call "BloodOfMehranClient-Win64-Shipping.exe" + 3D3B650

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

        if (!CAaddress) {
            constexpr auto CAStringObfuscated = make_obfuscated<0x4A>("7F ?? 44 89 ?? ?? ?? ?? ?? 43 ?? ?? ?? 39 05");
            CAaddress = Memory::AOBScan(gameExecutable, 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 (!Cameraaddress) { // Low level function involved in displaying Widgets in real time
            constexpr auto CameraStringObfuscated = make_obfuscated<0x4A>("50 0F 11 ?? ?? ?? ?? ?? 48 8D ?? ?? 0F 11 ?? ?? ?? ?? ?? F2");
            Cameraaddress = Memory::AOBScan(gameExecutable, CameraStringObfuscated.decrypt(), PAGE_EXECUTE_READ);
            //"BloodOfMehranClient-Win64-Shipping.exe" + 40E494B - F2 0F 11 9B B8 02 00 00 - movsd[rbx + 000002B8], xmm3
            //"BloodOfMehranClient-Win64-Shipping.exe" + 40E4953 - E8 98 EB 0A FD          - call "BloodOfMehranClient-Win64-Shipping.exe" + 11934F0
            //"BloodOfMehranClient-Win64-Shipping.exe" + 40E4958 - F3 0F 10 83 30 02 00 00 - movss xmm0, [rbx + 00000230]
            //"BloodOfMehranClient-Win64-Shipping.exe" + 40E4960 - F2 0F 10 64 24 78       - movsd xmm4, [rsp + 78]
            //"BloodOfMehranClient-Win64-Shipping.exe" + 40E4966 - F2 0F 10 5C 24 70       - movsd xmm3, [rsp + 70]

            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));
                Cameraaddress += 0x30;
            }
        }

        if (CameraComponentaddress && Cameraaddress && FOVaddress && DOFaddress && CAaddress
            && ConstrainAspectRatioaddress && AspectRatioAxisConstraintaddress) {
            logger->info("All AOB signatures found. Ready to patch...");
            AOBScanDone = true;
        }

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

            constexpr auto GObjetcsStringObfuscated = make_obfuscated<0x4A>("48 8B ?? ?? ?? ?? ?? 48 8B ?? ?? 48 8D ?? ?? EB ?? 33");
            GObjectsaddress = Memory::AOBScan(gameExecutable, GObjetcsStringObfuscated.decrypt(), PAGE_EXECUTE_READ);
            constexpr auto GNamesStringObfuscated = make_obfuscated<0x4A>("48 8D ?? ?? ?? ?? ?? EB ?? 48 8D ?? ?? ?? ?? ?? E8 ?? ?? ?? ?? C6 05 ?? ?? ?? ?? ?? 0F ?? ?? 4C");
            GNamesaddress = Memory::AOBScan(gameExecutable, GNamesStringObfuscated.decrypt(), PAGE_EXECUTE_READ);
            constexpr auto AppendStringStringObfuscated = make_obfuscated<0x4A>("48 89 ?? ?? ?? 48 89 ?? ?? ?? 57 48 83 ?? ?? 80 3D ?? ?? ?? ?? ?? 48 8B F2 8B ?? 48");
            AppendStringaddress = Memory::AOBScan(gameExecutable, 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 33 ?? 48 89 ?? ?? ?? ?? ?? 4D ?? ?? 48");
            ProcessEventaddress = Memory::AOBScan(gameExecutable, ProcessEventStringObfuscated.decrypt(), PAGE_EXECUTE_READ);

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

            if (!GNamesaddress)
                logger->warn("GNames signature not found. Maybe your game has been updated and is no more compatible with this plugin.");
            else {
                uint32_t gNamesOffset = static_cast<uint32_t>(Memory::GetOffsetFromOpcode(GNamesaddress + 0x3) -
                    reinterpret_cast<uint8_t*>(GetModuleHandleA(gameExecutable.c_str())));
                logger->info("GNames offset is: 0x{:X}.", gNamesOffset);
                Offsets::GNames = static_cast<UC::uint32>(gNamesOffset); // Update GNames 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(gameExecutable, 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(gameExecutable, ProcessEventaddress);
                uint32_t gProcessEventOffset = *gProcessEventOffsetOpt;
                logger->info("Process Event offset is: 0x{:X}.", gProcessEventOffset);
                Offsets::ProcessEvent = static_cast<UC::uint32>(gProcessEventOffset);// Update ProcessEvent offset
            }
        }

        logger->info("-------------- Fixes initialisation -------------");
    }

    if (FOVaddress) FOVFixEnabled();
    if (Cameraaddress) CameraFixEnabled();
    if (DOFaddress) DOFFixEnabled();
    if (CAaddress) CAFixEnabled();
    if (AspectRatioAxisConstraintaddress && ConstrainAspectRatioaddress) UltraWideFixEnabled();
    if (!g_Console_Enabled && GObjectsaddress && GNamesaddress && AppendStringaddress && ProcessEventaddress)
        EnableConsole();
}

// Setters for Reshade addon call
extern "C" __declspec(dllexport) void SetFOVFixEnabled(bool enabled, bool init)
{
    g_fov_fix_enabled = enabled;
    if (!init) FOVFixEnabled();
}

extern "C" __declspec(dllexport) void SetUltraWideFixEnabled(bool enabled, bool init)
{
    g_ultrawide_fix_enabled = enabled;
    if (!init) UltraWideFixEnabled();
}

extern "C" __declspec(dllexport) void SetDOFFixEnabled(bool enabled, bool init)
{
    g_DOF_fix_enabled = enabled;
    if (!init) DOFFixEnabled();
}

extern "C" __declspec(dllexport) void SetCAFixEnabled(bool enabled, bool init)
{
    g_CA_fix_enabled = enabled;
    if (!init) CAFixEnabled();
}

extern "C" __declspec(dllexport) void SetCameraFixEnabled(bool enabled, bool init)
{
    g_Camera_fix_enabled = enabled;
    if (!init) CameraFixEnabled();
}

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

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

// 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->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,
                [](SafetyHookContext& ctx) {
                    g_FOV_In = ctx.xmm0.f32[0];
                    ctx.xmm0.f32[0] += (g_fix_enabled && g_fov_fix_enabled ? g_AdditionalFOVValue : g_FOV_In);
                    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 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");
    }
}

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

static void DOFFixEnabled() {
    if (g_fix_enabled && g_DOF_fix_enabled && DOFaddress) {
        Memory::PatchBytes(DOFaddress, "\x31\xF6\x90", 3); // xor esi,esi 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 r.SceneColorFringeQuality
        logger->info("Chromatics aberrations fix enabled");
    }
    if (!(g_fix_enabled && g_CA_fix_enabled) && CAaddress) {
        Memory::RestoreBytes(CAaddress);
        logger->info("Chromatics aberrations fix disabled");
    }
}

// UE Console creation
static void EnableConsole()
{
    logger->info("-------------- Console re-enabling --------------");
    if (!GObjectsaddress || !AppendStringaddress || !ProcessEventaddress) {
        logger->warn("Could not re-enable console");
        logger->info("------------------ User inputs ------------------");
        return;
    }

    std::thread([&]() {
        auto start = std::chrono::high_resolution_clock::now(); // Measure the time to renable console
        UEngine* Engine = nullptr;

        for (int i = 0; i < 100; ++i) { // gives 10 seconds to find UE Engine
            std::this_thread::sleep_for(std::chrono::milliseconds(100));
            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);
            auto end = std::chrono::high_resolution_clock::now();
            std::chrono::duration<double> elapsed = end - start;

            logger->info("Console fully reactivated in {:.3f}s and bound to key Tilde", elapsed.count());
            logger->info("------------------ User inputs ------------------");

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

static void InitializeLogger()
{
    try
    {
        std::filesystem::path log_path = std::filesystem::absolute(PLUGIN_LOG);
        if (std::filesystem::exists(log_path))
            std::filesystem::remove(log_path);
        logger = std::make_shared<spdlog::logger>("Blood of Mehran", 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;
}