#define IMGUI_DISABLE_INCLUDE_IMCONFIG_H
#define IMGUI_HAS_DOCK 1

#include "GameInformations.h"
#include <imgui.h>
#include <reshade.hpp>
#include <fstream>
#include <string>
#include <Windows.h>

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

// Core game dll functions declarations
typedef void (*SetBoolFn)(bool, bool);
typedef void (*SetIntFn)(int);
typedef void (*SetFloatFn)(float);
typedef float (*GetFloatFn)();
typedef bool (*GetBoolFn)();

static HMODULE fixLib = nullptr;
static LONG g_coreInitialized = 0;
static SetBoolFn SetFixEnabled = nullptr;
static SetBoolFn SetFOVFixEnabled = nullptr;
static SetBoolFn SetDOFFixEnabled = nullptr;
static SetBoolFn SetCAFixEnabled = nullptr;
static SetBoolFn SetVignettingFixEnabled = nullptr;
static SetBoolFn SetCameraFixEnabled = nullptr;
static SetBoolFn SetVolumetricFogFixEnabled = nullptr;
static SetBoolFn SetFogDensityFixEnabled = nullptr;
static SetBoolFn SetFogMaxOpacityFixEnabled = nullptr;
static SetIntFn SetFOV = nullptr;
static SetFloatFn SetCameraDistance = nullptr;
static SetFloatFn SetFogDensity = nullptr;
static SetFloatFn SetFogMaxOpacity = nullptr;
static GetGameInfosStruct GetGameInfos = nullptr;

// Plugin variables for checkboxes and sliders
static bool fov_fix_enabled = false;
static bool DOF_fix_enabled = false;
static bool CA_fix_enabled = false;
static bool Vignetting_fix_enabled = false;
static bool camera_fix_enabled = false;
static bool volumetric_fog_fix_enabled = false;
static bool Fog_density_fix_enabled = false;
static bool Fog_max_opacity_fix_enabled = false;
static bool fix_enabled = false;
static int worldFOVvalue = 0;
static float cameraDistanceValue = 1.f;
static float fogDensityValue = 0.4f;
static float fogMaxOpacityValue = 0.4f;

static bool popup_Informations = false;

// Plugin settings
const char* SETTINGS_FILE = "PluginSettings.ini";
const char* GENERAL_FIX_SETTING = "GeneralFIX=";
const char* WORLD_FOV_FIX_SETTING = "WorldFOVFIX=";
const char* CAMERA_FIX_SETTING = "CameraFIX=";
const char* DOF_FIX_SETTING = "DOFFIX=";
const char* CA_FIX_SETTING = "CAFIX=";
const char* VIGNETTING_FIX_SETTING = "VignettingFIX=";
const char* VOLUMETRIC_FOG_FIX_SETTING = "VolumetricFogFIX=";
const char* FOG_DENSITY_FIX_SETTING = "FogDensityFIX=";
const char* FOG_MAX_OPACITY_FIX_SETTING = "FogMaxOpacityFIX=";
const char* WORLD_FOV_SETTING = "WorldFOV=";
const char* CAMERA_DISTANCE_SETTING = "CameraDistance=";
const char* FOG_DENSITY_SETTING = "FogDensity=";
const char* FOG_MAX_OPACITY_SETTING = "FogMaxOpacity=";
const char* FIX_VERSION = "1.0.3";
const char* FIX_INFORMATIONS = "This fix allows to:\n - Control FOV in game.\n - Control camera distance.\n - Disable depth of field.\n - Disable chromatic aberrations.\n - Disable vignetting.\n - Control fog.\n - Re enable console.\n\nDisabling Fog will not entirely remove it.";
const char* DONATION_URL = "https://buymeacoffee.com/k4sh44";

// Scaling factor based on screen resolution
float scale = 1.f;

// Load and unload game core dll functions /!\ necessary
static void LoadFixDLL()
{
    if (InterlockedCompareExchange(&g_coreInitialized, 1, 0) != 0)
        return;

    if (GetModuleHandleA("SilentHillfCore.dll") == nullptr) {
        fixLib = LoadLibraryA("SilentHillfCore.dll");

        if (!fixLib) {
            MessageBoxA(nullptr, "Impossible to load game core dll", "Erreur", MB_OK);
            return;
        }

        SetFixEnabled = (SetBoolFn)GetProcAddress(fixLib, "SetFixEnabled");
        SetFOVFixEnabled = (SetBoolFn)GetProcAddress(fixLib, "SetFOVFixEnabled");
        SetDOFFixEnabled = (SetBoolFn)GetProcAddress(fixLib, "SetDOFFixEnabled");
        SetCAFixEnabled = (SetBoolFn)GetProcAddress(fixLib, "SetCAFixEnabled");
        SetVignettingFixEnabled = (SetBoolFn)GetProcAddress(fixLib, "SetVignettingFixEnabled");
        SetVolumetricFogFixEnabled = (SetBoolFn)GetProcAddress(fixLib, "SetVolumetricFogFixEnabled");
        SetFogDensityFixEnabled = (SetBoolFn)GetProcAddress(fixLib,"SetFogDensityFixEnabled");
        SetFogMaxOpacityFixEnabled = (SetBoolFn)GetProcAddress(fixLib, "SetFogMaxOpacityFixEnabled");
        SetCameraFixEnabled = (SetBoolFn)GetProcAddress(fixLib, "SetCameraFixEnabled");
        SetFOV = (SetIntFn)GetProcAddress(fixLib, "SetFOV");
        SetCameraDistance = (SetFloatFn)GetProcAddress(fixLib, "SetCameraDistance");
        SetFogDensity = (SetFloatFn)GetProcAddress(fixLib,"SetFogDensity");
        SetFogMaxOpacity = (SetFloatFn)GetProcAddress(fixLib, "SetFogMaxOpacity");
        GetGameInfos = (GetGameInfosStruct)GetProcAddress(fixLib, "GetGameInfos");

        // Apply initial values loaded from settings
        if (SetFOV) SetFOV(worldFOVvalue);
        if (SetCameraDistance) SetCameraDistance(cameraDistanceValue);
        if (SetFogDensity) SetFogDensity(fogDensityValue);
        if (SetFogMaxOpacity) SetFogMaxOpacity(fogMaxOpacityValue);
        if (SetFOVFixEnabled) SetFOVFixEnabled(fov_fix_enabled, true);
        if (SetCameraFixEnabled) SetCameraFixEnabled(camera_fix_enabled, true);
        if (SetDOFFixEnabled) SetDOFFixEnabled(DOF_fix_enabled, true);
        if (SetCAFixEnabled) SetCAFixEnabled(CA_fix_enabled, true);
        if (SetVignettingFixEnabled) SetVignettingFixEnabled(Vignetting_fix_enabled, true);
        if (SetVolumetricFogFixEnabled) SetVolumetricFogFixEnabled(volumetric_fog_fix_enabled, true);
        if (SetFogDensityFixEnabled) SetFogDensityFixEnabled(Fog_density_fix_enabled, true);
        if (SetFogMaxOpacityFixEnabled) SetFogMaxOpacityFixEnabled(Fog_max_opacity_fix_enabled, true);
        if (SetFixEnabled) SetFixEnabled(fix_enabled, true);
    }
}

// Settings functions
static void SaveSettings()
{
    std::ofstream file(SETTINGS_FILE);
    if (file.is_open())
    {
        file << GENERAL_FIX_SETTING << (fix_enabled ? "1" : "0") << "\n";
        file << WORLD_FOV_FIX_SETTING << (fov_fix_enabled ? "1" : "0") << "\n";
        file << CAMERA_FIX_SETTING << (camera_fix_enabled ? "1" : "0") << "\n";
        file << DOF_FIX_SETTING << (DOF_fix_enabled ? "1" : "0") << "\n";
        file << CA_FIX_SETTING << (CA_fix_enabled ? "1" : "0") << "\n";
        file << VIGNETTING_FIX_SETTING << (Vignetting_fix_enabled ? "1" : "0") << "\n";
        file << VOLUMETRIC_FOG_FIX_SETTING << (volumetric_fog_fix_enabled ? "1" : "0") << "\n";
        file << FOG_DENSITY_FIX_SETTING << (Fog_density_fix_enabled ? "1" : "0") << "\n";
        file << FOG_MAX_OPACITY_FIX_SETTING << (Fog_max_opacity_fix_enabled ? "1" : "0") << "\n";
        file << WORLD_FOV_SETTING << worldFOVvalue << "\n";
        file << CAMERA_DISTANCE_SETTING << cameraDistanceValue << "\n";
        file << FOG_DENSITY_SETTING << fogDensityValue << "\n";
        file << FOG_MAX_OPACITY_SETTING << fogMaxOpacityValue << "\n";
        file.close();
    }
}

static void LoadSettings()
{
    std::ifstream file(SETTINGS_FILE);
    if (file.is_open())
    {
        std::string line;
        while (std::getline(file, line))
        {
            if (line.find(GENERAL_FIX_SETTING) == 0)
            {
                std::string val = line.substr(strlen(GENERAL_FIX_SETTING));
                fix_enabled = (val == "1" || val == "true");
            }
            else if (line.find(WORLD_FOV_FIX_SETTING) == 0)
            {
                std::string val = line.substr(strlen(WORLD_FOV_FIX_SETTING));
                fov_fix_enabled = (val == "1" || val == "true");
            }
            else if (line.find(CAMERA_FIX_SETTING) == 0)
            {
                std::string val = line.substr(strlen(CAMERA_FIX_SETTING));
                camera_fix_enabled = (val == "1" || val == "true");
            }
            else if (line.find(DOF_FIX_SETTING) == 0)
            {
                std::string val = line.substr(strlen(DOF_FIX_SETTING));
                DOF_fix_enabled = (val == "1" || val == "true");
            }
            else if (line.find(CA_FIX_SETTING) == 0)
            {
                std::string val = line.substr(strlen(CA_FIX_SETTING));
                CA_fix_enabled = (val == "1" || val == "true");
            }
            else if (line.find(VIGNETTING_FIX_SETTING) == 0)
            {
                std::string val = line.substr(strlen(VIGNETTING_FIX_SETTING));
                Vignetting_fix_enabled = (val == "1" || val == "true");
            }
            else if (line.find(VOLUMETRIC_FOG_FIX_SETTING) == 0)
            {
                std::string val = line.substr(strlen(VOLUMETRIC_FOG_FIX_SETTING));
                volumetric_fog_fix_enabled = (val == "1" || val == "true");
            }
            else if (line.find(FOG_DENSITY_FIX_SETTING) == 0)
            {
                std::string val = line.substr(strlen(FOG_DENSITY_FIX_SETTING));
                Fog_density_fix_enabled = (val == "1" || val == "true");
            }
            else if (line.find(FOG_MAX_OPACITY_FIX_SETTING) == 0)
            {
                std::string val = line.substr(strlen(FOG_MAX_OPACITY_FIX_SETTING));
                Fog_max_opacity_fix_enabled = (val == "1" || val == "true");
            }
            else if (line.find(WORLD_FOV_SETTING) == 0)
                worldFOVvalue = std::stoi(line.substr(strlen(WORLD_FOV_SETTING)));
            else if (line.find(CAMERA_DISTANCE_SETTING) == 0)
                cameraDistanceValue = std::stof(line.substr(strlen(CAMERA_DISTANCE_SETTING)));
            else if (line.find(FOG_DENSITY_SETTING) == 0)
                fogDensityValue = std::stof(line.substr(strlen(FOG_DENSITY_SETTING)));
            else if (line.find(FOG_MAX_OPACITY_SETTING) == 0)
                fogMaxOpacityValue = std::stof(line.substr(strlen(FOG_MAX_OPACITY_SETTING)));
        }
        file.close();
    }
}

// Initialize ImGui widgets for Reshade
static void on_overlay_draw(reshade::api::effect_runtime* runtime)
{
    ImGuiStyle& style = ImGui::GetStyle();
    style.ItemSpacing = ImVec2(8 * scale, 8 * scale);     // Spacing between widgets
    style.FramePadding = ImVec2(3 * scale, 3 * scale);    // Widgets padding
    style.WindowPadding = ImVec2(10 * scale, 10 * scale); // Overlay padding
    style.CellPadding = ImVec2(10 * scale, 10 * scale);   // Table cells padding

    ImGui::SetNextWindowSize(ImVec2(350 * scale, 150 * scale), ImGuiCond_Once);

    if (ImGui::Button("Like my work ? Consider donation")) ShellExecuteA(NULL, "open", DONATION_URL, NULL, NULL, SW_SHOWNORMAL); // Donation
    ImGui::SameLine();
    if (ImGui::Button("Fix informations")) popup_Informations = true; // Fix information

    if (popup_Informations)
    {
        ImGui::Begin("Informations", &popup_Informations);
        ImGui::Text("Version : %s", FIX_VERSION);
        ImGui::Text(FIX_INFORMATIONS);
        ImGui::End();
    }

    if (ImGui::BeginTable("FixesTable", 2, ImGuiTableFlags_SizingStretchSame)) {
        ImGui::TableSetupColumn("LeftFix", ImGuiTableColumnFlags_WidthStretch, 0.3f);
        ImGui::TableSetupColumn("RightFix", ImGuiTableColumnFlags_WidthStretch, 0.7f);
        ImGui::TableNextRow();

        // Drawing a left column with slider and general fix
        ImGui::TableSetColumnIndex(0);
        if (ImGui::CollapsingHeader("Enable fixes", ImGuiTreeNodeFlags_DefaultOpen))
            if (ImGui::Checkbox("Fix enabled", &fix_enabled)) { 
                if (SetFixEnabled) SetFixEnabled(fix_enabled, false);
                SaveSettings();
            }

        // Individual fixes
        ImGui::TableSetColumnIndex(1);
        if (ImGui::CollapsingHeader("Individual fixes", ImGuiTreeNodeFlags_DefaultOpen)) {
            if (ImGui::BeginTable("IndividualFixesTable", 2, ImGuiTableFlags_SizingStretchSame)) {
                ImGui::TableSetupColumn("IndFix1", ImGuiTableColumnFlags_WidthStretch, 0.35f);
                ImGui::TableSetupColumn("IndFix2", ImGuiTableColumnFlags_WidthStretch, 0.65f);
                ImGui::TableNextRow();

                ImGui::TableSetColumnIndex(0);
                if (ImGui::Checkbox("FOV", &fov_fix_enabled)) {
                    if (SetFOVFixEnabled) SetFOVFixEnabled(fov_fix_enabled, false);
                    SaveSettings();
                }

                if (ImGui::Checkbox("Vignetting", &Vignetting_fix_enabled)) {
                    if (SetVignettingFixEnabled) SetVignettingFixEnabled(Vignetting_fix_enabled, false);
                    SaveSettings();
                }

                if (ImGui::Checkbox("Fog density", &Fog_density_fix_enabled)) {
                    if (SetFogDensityFixEnabled) SetFogDensityFixEnabled(Fog_density_fix_enabled, false);
                    SaveSettings();
                }

                if (ImGui::Checkbox("Fog opacity", &Fog_max_opacity_fix_enabled)) {
                    if (SetFogMaxOpacityFixEnabled) SetFogMaxOpacityFixEnabled(Fog_max_opacity_fix_enabled, false);
                    SaveSettings();
                }

                ImGui::TableSetColumnIndex(1);
                if (ImGui::Checkbox("Camera", &camera_fix_enabled)) {
                    if (SetCameraFixEnabled) SetCameraFixEnabled(camera_fix_enabled, false);
                    SaveSettings();
                }

                if (ImGui::Checkbox("Depth of field", &DOF_fix_enabled)) {
                    if (SetDOFFixEnabled) SetDOFFixEnabled(DOF_fix_enabled, false);
                    SaveSettings();
                }

                if (ImGui::Checkbox("Chromatic aberrations", &CA_fix_enabled)) {
                    if (SetCAFixEnabled) SetCAFixEnabled(CA_fix_enabled, false);
                    SaveSettings();
                }

                if (ImGui::Checkbox("Volumetric fog", &volumetric_fog_fix_enabled)) {
                    if (SetVolumetricFogFixEnabled) SetVolumetricFogFixEnabled(volumetric_fog_fix_enabled, false);
                    SaveSettings();
                }
                ImGui::EndTable();
            }
        }
        ImGui::EndTable();
    }

    ImGui::SetCursorPosY(ImGui::GetCursorPosY() - 2 * style.ItemSpacing.y);
    if (ImGui::BeginTable("FixesSliders", 2, ImGuiTableFlags_SizingStretchSame)) {
        ImGui::TableNextRow();
        // Sliders
        ImGui::TableSetColumnIndex(0);
        if (ImGui::CollapsingHeader("In game additional FOV", ImGuiTreeNodeFlags_DefaultOpen))
        {
            ImGui::SetNextItemWidth(180 * scale);
            if (ImGui::SliderInt("##FOVValue", &worldFOVvalue, -20, 50)) {
                if (SetFOV) SetFOV(worldFOVvalue);
                SaveSettings();
            }
        }
        if (ImGui::IsItemHovered()) {
            ImGui::BeginTooltip();
            ImGui::Text("This will affect in game FOV only.");
            ImGui::EndTooltip();
        }

        if (ImGui::CollapsingHeader("Camera distance (*)", ImGuiTreeNodeFlags_DefaultOpen))
        {
            ImGui::SetNextItemWidth(180 * scale);
            if (ImGui::SliderFloat("##CameraDistanceValue", &cameraDistanceValue, 0, 3)) {
                if (SetCameraDistance) SetCameraDistance(cameraDistanceValue);
                SaveSettings();
            }
        }
        if (ImGui::IsItemHovered()) {
            ImGui::BeginTooltip();
            ImGui::Text("Value is a multiplier.");
            ImGui::EndTooltip();
        }

        ImGui::TableSetColumnIndex(1);
        if (ImGui::CollapsingHeader("Fog density (*)", ImGuiTreeNodeFlags_DefaultOpen))
        {
            ImGui::SetNextItemWidth(180 * scale);
            if (ImGui::SliderFloat("##FogDensityValue", &fogDensityValue, 0, 5)) {
                if (SetFogDensity) SetFogDensity(fogDensityValue);
                SaveSettings();
            }
        }
        if (ImGui::IsItemHovered()) {
            ImGui::BeginTooltip();
            ImGui::Text("This will override engine dynamic value.");
            ImGui::EndTooltip();
        }

        if (ImGui::CollapsingHeader("Fog opacity (*)", ImGuiTreeNodeFlags_DefaultOpen))
        {
            ImGui::SetNextItemWidth(180 * scale);
            if (ImGui::SliderFloat("##FogOpacityValue", &fogMaxOpacityValue, 0, 1)) {
                if (SetFogMaxOpacity) SetFogMaxOpacity(fogMaxOpacityValue);
                SaveSettings();
            }
        }
        if (ImGui::IsItemHovered()) {
            ImGui::BeginTooltip();
            ImGui::Text("This will override engine dynamic value.");
            ImGui::EndTooltip();
        }
        ImGui::EndTable();
    }

    // Fix status
    if (ImGui::CollapsingHeader("Fix informations", ImGuiTreeNodeFlags_DefaultOpen)) {
        ImGui::Text("Screen width: %d, height: %d, aspect ratio: %.2f", screenWidth, screenHeight, aspectRatio);
        if (GetGameInfos) {
            GameInfos infos{};
            GetGameInfos(&infos);
            if (infos.consoleEnabled)
                ImGui::Text("Console enabled and bound to key Tilde");
            ImGui::TextColored(ImColor(48, 179, 25), "FOV In: %.2f, Out: %.2f", infos.FOVIn, infos.FOVOut);
            ImGui::TextColored(ImColor(48, 179, 25), "Camera distance In: %.2f, Out: %.2f", infos.cameraIn, infos.cameraOut);
            ImGui::TextColored(ImColor(48, 179, 25), "Default fog density: %.3f, opacity: %3.f", infos.defaultFogDensity, infos.fogMaxOpacity);
        }
    }
}

// Main dll intrance
BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD ul_reason_for_call, LPVOID)
{
    switch (ul_reason_for_call)
    {
        case DLL_PROCESS_ATTACH:
            if (!reshade::register_addon(hinstDLL))
                return FALSE;
            LoadSettings();

            reshade::register_overlay("Silent Hill f", &on_overlay_draw);
            reshade::register_event<reshade::addon_event::init_effect_runtime>(
                [](reshade::api::effect_runtime* runtime) {
                    LoadFixDLL();
                });
            break;
        case DLL_PROCESS_DETACH:
            reshade::unregister_addon(hinstDLL);
            break;
    }
    return TRUE;
}