#include <regex>
#include <set>

#include <glob/glob.h>
#include <polyhook2/MemProtector.hpp>

#include <koalabox/config.hpp>
#include <koalabox/globals.hpp>
#include <koalabox/hook.hpp>
#include <koalabox/http_client.hpp>
#include <koalabox/lib.hpp>
#include <koalabox/lib_monitor.hpp>
#include <koalabox/logger.hpp>
#include <koalabox/path.hpp>
#include <koalabox/paths.hpp>
#include <koalabox/platform.hpp>
#include <koalabox/util.hpp>

// static
#include "smoke_api/config.hpp"
#include "smoke_api/steamclient/steamclient.hpp"

#include "smoke_api.hpp"
#include "steam_api/steam_interfaces.hpp"
#include "steam_api/virtuals/steam_api_virtuals.hpp"

#include "build_config.h"

#ifdef KB_WIN
#include "koalabox/win.hpp"
#elif defined(KB_LINUX) && defined(KB_32)
#include "generated/32/proxy_exports.hpp"
#elif defined(KB_LINUX) && defined(KB_64)
#include "generated/64/proxy_exports.hpp"
#endif

// Hooking steam_api has shown itself to be less desirable than steamclient
// for the reasons outlined below:
//
// Calling original in flat functions will actually call the hooked functions
// because the original function redirects the execution to a function taken
// from self pointer, which would have been hooked by SteamInternal_*Interface
// functions.
//
// Furthermore, turns out that many flat functions share the same body,
// which looks like the following snippet:
//
//   mov rax, qword ptr ds:[rcx]
//   jmp qword ptr ds:[rax+immediate]
//
// This means that we end up inadvertently hooking unintended functions.
// Given that hooking steam_api has no apparent benefits, but has inherent flaws,
// the support for it has been dropped from this project.

namespace {
    namespace kb = koalabox;

    void* original_steamapi_handle = nullptr;
    bool is_hook_mode;

    void check_for_updates() {
        const auto latest_release_url = std::format(
            "https://api.github.com/repos/acidicoala/{}/releases/latest",
            PROJECT_NAME
        );
        const auto res = kb::http_client::get_json(latest_release_url);
        const auto latest_tag = res["tag_name"].get<std::string>();
        const auto current_tag = std::format("v{}", PROJECT_VERSION);

        if(current_tag == latest_tag) {
            LOG_DEBUG("Running the latest version");
        } else {
            const auto release_page = std::format(
                "https://github.com/acidicoala/{}/releases/{}",
                PROJECT_NAME, latest_tag
            );

            LOG_WARN("New version {} available: {}", latest_tag, release_page);
        }
    }

    std::set<std::string> find_steamclient_versions(void* steamapi_handle) {
        if(!steamapi_handle) {
            kb::util::panic("Invalid state. steamapi_handle is null.");
        }

        std::set<std::string> versions;

        // On Linux the section name depends on individual lib file, so we can't use generic constants
        // ReSharper disable once CppDFAUnreachableCode
        const std::string str_section_name = kb::platform::is_windows ? ".rdata" : ".rodata";
        const auto str_section = kb::lib::get_section_or_throw(steamapi_handle, str_section_name);
        const auto str_section_str = str_section.to_string();

        const std::regex pattern(R"(SteamClient\d{3})");
        const auto matches_begin = std::sregex_iterator(str_section_str.begin(), str_section_str.end(), pattern);
        const auto matches_end = std::sregex_iterator();

        for(std::sregex_iterator i = matches_begin; i != matches_end; ++i) {
            versions.insert(i->str());
        }

        LOG_DEBUG("Found {} steamclient version(s) in read-only section: {}", versions.size(), versions);

        return versions;
    }

#ifdef KB_WIN
    void warn_if_late_injection(const std::string& steamclient_version) {
        if(kb::util::is_wine_env()) {
            return;
        }

        LOG_WARN(
            "'{}' was already initialized. SmokeAPI might not work as expected.", steamclient_version
        );
        LOG_WARN(
            "Probable cause: SmokeAPI was injected too late. If possible, try injecting it earlier."
        );
        LOG_WARN("NOTE: You can safely ignore this warning if running under Proton or native Linux");
    }
#endif

    // ReSharper disable once CppDFAConstantFunctionResult
    bool on_steamclient_loaded(void* steamclient_handle) {
        KB_HOOK_DETOUR_MODULE(CreateInterface, steamclient_handle);

        if(!is_hook_mode) {
            return true;
        }

        // Check for late hooking

        static const auto CreateInterface$ = KB_LIB_GET_FUNC(steamclient_handle, CreateInterface);

        if(original_steamapi_handle) {
            // SteamAPI might have been initialized.
            // Hence, we need to query SteamClient interfaces and hook them if needed.
            const auto steamclient_versions = find_steamclient_versions(original_steamapi_handle);
            for(const auto& steamclient_version : steamclient_versions) {
                if(CreateInterface$(steamclient_version.c_str(), nullptr)) {
#ifdef KB_WIN
                    warn_if_late_injection(steamclient_version);
#endif

                    steam_interfaces::hook_steamclient_interface(steamclient_handle, steamclient_version);
                } else {
                    LOG_INFO("'{}' has not been initialized. Waiting for initialization.", steamclient_version);
                }
            }
        } else {
            LOG_ERROR("{} -> original_steamapi_handle is null", __func__);
        }

        return true;
    }

    void init_lib_monitor() {
        kb::lib_monitor::init_listener({{STEAMCLIENT_DLL, on_steamclient_loaded}});
    }

    std::optional<AppId_t> get_app_id_from_env() noexcept {
        if(const auto app_id_str = kb::util::get_env("SteamAppId")) {
            try {
                const auto app_id = std::stoi(*app_id_str);

                LOG_DEBUG("Found AppID from environment: {}", app_id);
                return app_id;
            } catch(const std::exception& e) {
                LOG_ERROR("Failed to parse AppID '{}' from environment: {}", *app_id_str, e.what());
            }
        }

        return std::nullopt;
    }

    std::optional<AppId_t> get_app_id_from_steam_client() noexcept {
        try {
            const DECLARE_ARGS();

            const auto& version_map = steam_interfaces::get_interface_name_to_version_map();
            THIS = CreateInterface(version_map.at("ISteamClient").c_str(), nullptr);
            if(THIS) {
                if(const auto get_steam_utils = SMK_FIND_INTERFACE_FUNC(THIS, ISteamClient, GetISteamUtils)) {
                    constexpr auto steam_pipe = 1;
                    const auto& utils_version = version_map.at("ISteamUtils");
                    THIS = get_steam_utils(ARGS(steam_pipe, utils_version.c_str()));
                    if(THIS) {
                        if(const auto get_app_id = SMK_FIND_INTERFACE_FUNC(THIS, ISteamUtils, GetAppID)) {
                            if(const auto app_id = get_app_id(ARGS())) {
                                LOG_DEBUG("Found AppID from ISteamUtils: {}", app_id);
                                return app_id;
                            }
                            LOG_ERROR("ISteamUtils::GetAppID returned 0");
                        }
                    }
                }
            } else {
                LOG_ERROR("Failed to create interface '{}'", version_map.at("ISteamClient"))
            }
        } catch(const std::exception& e) {
            LOG_ERROR("Failed to get app id. Unhandled exception: {}", e.what());
        }

        return std::nullopt;
    }

    void init_hook_mode([[maybe_unused]] void* self_module_handle) {
        is_hook_mode = true;
#ifdef KB_LINUX
        // Because we got injected via LD_PRELOAD,
        // Linux loader will resolve all SteamAPI exports in unlocker instead of original library.
        // Hence, we need to patch the stubs even in hook mode.

        const std::string lib_name = STEAM_API_MODULE ".so";
        for(const auto& lib_path : glob::rglob({"./" + lib_name, "**/" + lib_name})) {
            if(const auto lib_bitness = kb::lib::get_bitness(lib_path)) {
                if(static_cast<uint8_t>(*lib_bitness) == kb::platform::bitness) {
                    if(const auto lib_handle = kb::lib::load(lib_path)) {
                        LOG_INFO("Found original library: {}", kb::path::to_str(lib_path));

                        original_steamapi_handle = *lib_handle;
                        proxy_exports::init(self_module_handle, original_steamapi_handle);
                        return;
                    }
                }
            }
        }

        if(!original_steamapi_handle) {
            kb::util::panic("Failed to find original " + lib_name);
        }
#endif
    }

    void init_proxy_mode([[maybe_unused]] void* self_module_handle) {
        is_hook_mode = true;

        original_steamapi_handle = kb::lib::load_original_library(kb::paths::get_self_dir(), STEAM_API_MODULE);
#ifdef KB_LINUX
        proxy_exports::init(self_module_handle, original_steamapi_handle);
#endif
    }
}

namespace smoke_api {
    void init(void* self_module_handle) {
        try {
            static std::mutex section;
            const std::lock_guard lock(section);

            static bool init_complete = false;
            if(init_complete) {
                LOG_ERROR("{} is already initialized", PROJECT_NAME);
            }

            kb::globals::init_globals(self_module_handle, PROJECT_NAME);

            config::get() = kb::config::parse<config::Config>();

            if(config::get().logging) {
                kb::logger::init_file_logger(kb::paths::get_log_path());
            } else {
                kb::logger::init_null_logger();
            }

            LOG_INFO("{} v{}{} | Built at '{}'", PROJECT_NAME, PROJECT_VERSION, VERSION_SUFFIX, __TIMESTAMP__);
            LOG_DEBUG("Parsed config:\n{}", nlohmann::ordered_json(config::get()).dump(2));

            const auto exe_path = kb::lib::get_fs_path(nullptr);
            const auto exe_name = kb::path::to_str(exe_path.filename());

            LOG_DEBUG("Process name: '{}' [{}-bit]", exe_name, kb::platform::bitness);
            LOG_DEBUG("Self name: '{}'", kb::path::to_str(kb::lib::get_fs_path(self_module_handle).filename()));

#ifdef KB_WIN
            kb::win::check_self_duplicates();
#endif

#ifdef KB_DEBUG
            // TODO: Add config option to toggle this and show native OS notification
            // The real reason behind this is for automatic testing of HTTPs dependencies
            std::thread(check_for_updates).detach();
#endif

            // We need to hook functions in either mode
            kb::hook::init(true);

            if(kb::hook::is_hook_mode(self_module_handle, STEAM_API_MODULE)) {
                LOG_INFO("Detected hook mode");
                init_hook_mode(self_module_handle);
            } else {
                LOG_INFO("Detected proxy mode");
                init_proxy_mode(self_module_handle);
            }

            init_lib_monitor();

            init_complete = true;
            LOG_INFO("Initialization complete");
        } catch(const std::exception& e) {
            kb::util::panic(std::format("Initialization error: {}", e.what()));
        }
    }

    void shutdown() {
        try {
            static bool shutdown_complete = false;
            if(shutdown_complete) {
                LOG_ERROR("{} is already shut down", PROJECT_NAME);
            }

            if(original_steamapi_handle != nullptr) {
                kb::lib::unload(original_steamapi_handle);
                original_steamapi_handle = nullptr;
            }

            if(kb::lib_monitor::is_initialized()) {
                kb::lib_monitor::shutdown_listener();
            }

            // TODO: Unhook everything

            shutdown_complete = true;
            LOG_INFO("Shutdown complete");
        } catch(const std::exception& e) {
            LOG_ERROR("Shutdown error: {}", e.what());
        }

        kb::logger::shutdown();
    }

    AppId_t get_app_id() {
        static AppId_t cached_app_id = 0;
        if(cached_app_id) {
            return cached_app_id;
        }

        LOG_DEBUG("No cached App ID found. Searching in environment variables.");

        if(const auto opt_app_id = get_app_id_from_env()) {
            return cached_app_id = *opt_app_id;
        }

        LOG_WARN("Failed to find App ID in environment variables. Falling back to ISteamUtils::GetAppID.");

        // IDEA: Try to read steam_appid.txt here. SteamAppId env var is not available when it's present.
        //       But what if the ID specified in steam_appid.txt is invalid?

        if(const auto opt_app_id = get_app_id_from_steam_client()) {
            return cached_app_id = *opt_app_id;
        }

        LOG_ERROR("Failed to find App ID");

        return 0;
    }
}