// DragonX Wallet - ImGui Edition
// Copyright 2024-2026 The Hush Developers
// Released under the GPLv3

#include "settings.h"
#include "version.h"

#include <nlohmann/json.hpp>
#include <fstream>
#include <filesystem>

#include "../util/logger.h"

#ifdef _WIN32
#include <shlobj.h>
#else
#include <pwd.h>
#include <unistd.h>
#endif

namespace fs = std::filesystem;
using json = nlohmann::json;

namespace dragonx {
namespace config {

Settings::Settings() = default;
Settings::~Settings() = default;

std::string Settings::getDefaultPath()
{
#ifdef _WIN32
    char path[MAX_PATH];
    if (SUCCEEDED(SHGetFolderPathA(NULL, CSIDL_APPDATA, NULL, 0, path))) {
        std::string dir = std::string(path) + "\\ObsidianDragon";
        fs::create_directories(dir);
        return dir + "\\settings.json";
    }
    return "settings.json";
#elif defined(__APPLE__)
    const char* home = getenv("HOME");
    if (!home) {
        struct passwd* pw = getpwuid(getuid());
        home = pw->pw_dir;
    }
    std::string dir = std::string(home) + "/Library/Application Support/ObsidianDragon";
    fs::create_directories(dir);
    return dir + "/settings.json";
#else
    const char* home = getenv("HOME");
    if (!home) {
        struct passwd* pw = getpwuid(getuid());
        home = pw->pw_dir;
    }
    std::string dir = std::string(home) + "/.config/ObsidianDragon";
    fs::create_directories(dir);
    return dir + "/settings.json";
#endif
}

bool Settings::load()
{
    return load(getDefaultPath());
}

bool Settings::load(const std::string& path)
{
    settings_path_ = path;
    
    std::ifstream file(path);
    if (!file.is_open()) {
        return false;
    }
    
    try {
        json j;
        file >> j;
        
        if (j.contains("theme")) theme_ = j["theme"].get<std::string>();
        if (j.contains("save_ztxs")) save_ztxs_ = j["save_ztxs"].get<bool>();
        if (j.contains("auto_shield")) auto_shield_ = j["auto_shield"].get<bool>();
        if (j.contains("use_tor")) use_tor_ = j["use_tor"].get<bool>();
        if (j.contains("allow_custom_fees")) allow_custom_fees_ = j["allow_custom_fees"].get<bool>();
        if (j.contains("default_fee")) default_fee_ = j["default_fee"].get<double>();
        if (j.contains("fetch_prices")) fetch_prices_ = j["fetch_prices"].get<bool>();
        if (j.contains("tx_explorer_url")) tx_explorer_url_ = j["tx_explorer_url"].get<std::string>();
        if (j.contains("address_explorer_url")) address_explorer_url_ = j["address_explorer_url"].get<std::string>();
        if (j.contains("language")) language_ = j["language"].get<std::string>();
        if (j.contains("skin_id")) skin_id_ = j["skin_id"].get<std::string>();
        if (j.contains("acrylic_enabled")) acrylic_enabled_ = j["acrylic_enabled"].get<bool>();
        if (j.contains("acrylic_quality")) acrylic_quality_ = j["acrylic_quality"].get<int>();
        if (j.contains("blur_multiplier")) blur_multiplier_ = j["blur_multiplier"].get<float>();
        if (j.contains("noise_opacity")) noise_opacity_ = j["noise_opacity"].get<float>();
        if (j.contains("gradient_background")) gradient_background_ = j["gradient_background"].get<bool>();
        // Migrate legacy reduced_transparency bool -> ui_opacity float
        if (j.contains("ui_opacity")) {
            ui_opacity_ = j["ui_opacity"].get<float>();
        } else if (j.contains("reduced_transparency") && j["reduced_transparency"].get<bool>()) {
            ui_opacity_ = 1.0f; // legacy: reduced = fully opaque
        }
        if (j.contains("window_opacity")) window_opacity_ = j["window_opacity"].get<float>();
        if (j.contains("balance_layout")) {
            if (j["balance_layout"].is_string())
                balance_layout_ = j["balance_layout"].get<std::string>();
            else if (j["balance_layout"].is_number_integer()) {
                // Legacy migration: convert old int index to string ID
                static const char* legacyIds[] = {
                    "classic","donut","consolidated","dashboard",
                    "vertical-stack","shield","timeline","two-row","minimal"
                };
                int idx = j["balance_layout"].get<int>();
                if (idx >= 0 && idx < 9) balance_layout_ = legacyIds[idx];
            }
        }
        if (j.contains("scanline_enabled")) scanline_enabled_ = j["scanline_enabled"].get<bool>();
        if (j.contains("hidden_addresses") && j["hidden_addresses"].is_array()) {
            hidden_addresses_.clear();
            for (const auto& a : j["hidden_addresses"])
                if (a.is_string()) hidden_addresses_.insert(a.get<std::string>());
        }
        if (j.contains("favorite_addresses") && j["favorite_addresses"].is_array()) {
            favorite_addresses_.clear();
            for (const auto& a : j["favorite_addresses"])
                if (a.is_string()) favorite_addresses_.insert(a.get<std::string>());
        }
        if (j.contains("wizard_completed")) wizard_completed_ = j["wizard_completed"].get<bool>();
        if (j.contains("auto_lock_timeout")) auto_lock_timeout_ = j["auto_lock_timeout"].get<int>();
        if (j.contains("unlock_duration")) unlock_duration_ = j["unlock_duration"].get<int>();
        if (j.contains("pin_enabled")) pin_enabled_ = j["pin_enabled"].get<bool>();
        if (j.contains("keep_daemon_running")) keep_daemon_running_ = j["keep_daemon_running"].get<bool>();
        if (j.contains("stop_external_daemon")) stop_external_daemon_ = j["stop_external_daemon"].get<bool>();
        if (j.contains("debug_categories") && j["debug_categories"].is_array()) {
            debug_categories_.clear();
            for (const auto& c : j["debug_categories"])
                if (c.is_string()) debug_categories_.insert(c.get<std::string>());
        }
        if (j.contains("theme_effects_enabled")) theme_effects_enabled_ = j["theme_effects_enabled"].get<bool>();
        if (j.contains("low_spec_mode")) low_spec_mode_ = j["low_spec_mode"].get<bool>();
        if (j.contains("selected_exchange")) selected_exchange_ = j["selected_exchange"].get<std::string>();
        if (j.contains("selected_pair")) selected_pair_ = j["selected_pair"].get<std::string>();
        if (j.contains("pool_url"))        pool_url_       = j["pool_url"].get<std::string>();
        if (j.contains("pool_algo"))       pool_algo_      = j["pool_algo"].get<std::string>();
        if (j.contains("pool_worker"))     pool_worker_    = j["pool_worker"].get<std::string>();
        if (j.contains("pool_threads"))    pool_threads_   = j["pool_threads"].get<int>();
        if (j.contains("pool_tls"))        pool_tls_       = j["pool_tls"].get<bool>();
        if (j.contains("pool_hugepages"))  pool_hugepages_ = j["pool_hugepages"].get<bool>();
        if (j.contains("pool_mode"))       pool_mode_      = j["pool_mode"].get<bool>();
        if (j.contains("font_scale") && j["font_scale"].is_number())
            font_scale_ = std::max(1.0f, std::min(1.5f, j["font_scale"].get<float>()));
        if (j.contains("window_width") && j["window_width"].is_number_integer())
            window_width_ = j["window_width"].get<int>();
        if (j.contains("window_height") && j["window_height"].is_number_integer())
            window_height_ = j["window_height"].get<int>();
        
        return true;
    } catch (const std::exception& e) {
        DEBUG_LOGF("Failed to parse settings: %s\n", e.what());
        return false;
    }
}

bool Settings::save()
{
    if (settings_path_.empty()) {
        settings_path_ = getDefaultPath();
    }
    return save(settings_path_);
}

bool Settings::save(const std::string& path)
{
    json j;
    
    j["theme"] = theme_;
    j["save_ztxs"] = save_ztxs_;
    j["auto_shield"] = auto_shield_;
    j["use_tor"] = use_tor_;
    j["allow_custom_fees"] = allow_custom_fees_;
    j["default_fee"] = default_fee_;
    j["fetch_prices"] = fetch_prices_;
    j["tx_explorer_url"] = tx_explorer_url_;
    j["address_explorer_url"] = address_explorer_url_;
    j["language"] = language_;
    j["skin_id"] = skin_id_;
    j["acrylic_enabled"] = acrylic_enabled_;
    j["acrylic_quality"] = acrylic_quality_;
    j["blur_multiplier"] = blur_multiplier_;
    j["noise_opacity"] = noise_opacity_;
    j["gradient_background"] = gradient_background_;
    j["ui_opacity"] = ui_opacity_;
    j["window_opacity"] = window_opacity_;
    j["balance_layout"] = balance_layout_;  // saved as string ID
    j["scanline_enabled"] = scanline_enabled_;
    j["hidden_addresses"] = json::array();
    for (const auto& addr : hidden_addresses_)
        j["hidden_addresses"].push_back(addr);
    j["favorite_addresses"] = json::array();
    for (const auto& addr : favorite_addresses_)
        j["favorite_addresses"].push_back(addr);
    j["wizard_completed"] = wizard_completed_;
    j["auto_lock_timeout"] = auto_lock_timeout_;
    j["unlock_duration"] = unlock_duration_;
    j["pin_enabled"] = pin_enabled_;
    j["keep_daemon_running"] = keep_daemon_running_;
    j["stop_external_daemon"] = stop_external_daemon_;
    j["debug_categories"] = json::array();
    for (const auto& cat : debug_categories_)
        j["debug_categories"].push_back(cat);
    j["theme_effects_enabled"] = theme_effects_enabled_;
    j["low_spec_mode"] = low_spec_mode_;
    j["selected_exchange"] = selected_exchange_;
    j["selected_pair"] = selected_pair_;
    j["pool_url"]       = pool_url_;
    j["pool_algo"]      = pool_algo_;
    j["pool_worker"]    = pool_worker_;
    j["pool_threads"]   = pool_threads_;
    j["pool_tls"]       = pool_tls_;
    j["pool_hugepages"] = pool_hugepages_;
    j["pool_mode"]      = pool_mode_;
    j["font_scale"]     = font_scale_;
    if (window_width_ > 0 && window_height_ > 0) {
        j["window_width"]  = window_width_;
        j["window_height"] = window_height_;
    }
    
    try {
        // Ensure directory exists
        fs::path p(path);
        fs::create_directories(p.parent_path());
        
        std::ofstream file(path);
        if (!file.is_open()) {
            return false;
        }
        
        file << j.dump(4);
        return true;
    } catch (const std::exception& e) {
        DEBUG_LOGF("Failed to save settings: %s\n", e.what());
        return false;
    }
}

} // namespace config
} // namespace dragonx