#include "config.hpp"
#include "panel_clock.hpp"
#include <cassert>
#include <cstdlib>
#include <filesystem>
#include <fstream>
#include <stdexcept>
#include <string>
#include <unistd.h>
using namespace frameyap;
namespace {
void put(const std::filesystem::path& p, const std::string& s) { std::ofstream out(p); out << s; assert(out); }
std::string get(const std::filesystem::path& p) { std::ifstream in(p); return {std::istreambuf_iterator<char>(in), {}}; }
template<class Fn> void fails(Fn fn) { bool raised = false; try { fn(); } catch (const std::exception&) { raised = true; } assert(raised); }
}
int main(int argc, char** argv) {
    assert(argc == 2);
    ::setenv("TZ", "UTC", 1); tzset();
    std::tm instant{};
    instant.tm_year = 124; instant.tm_mon = 0; instant.tm_mday = 2;
    instant.tm_hour = 15; instant.tm_min = 4;
    auto at = std::mktime(&instant);
    assert(panel_clock(at, false, DateFormat::MonthDayYear).time == "03:04 PM");
    assert(panel_clock(at, false, DateFormat::MonthDayYear).date == "01/02/2024");
    assert(panel_clock(at, true, DateFormat::DayMonthYear).time == "15:04");
    assert(panel_clock(at, true, DateFormat::DayMonthYear).date == "02/01/2024");
    assert(panel_clock(at, true, DateFormat::Iso).date == "2024-01-02");
    assert(panel_clock(at, true, DateFormat::Off).date.empty());
    const auto dir = std::filesystem::temp_directory_path() / ("frameyap-config-test-" + std::to_string(::getpid()));
    std::filesystem::create_directories(dir);
    ::setenv("XDG_CONFIG_HOME", dir.c_str(), 1);
    ::setenv("XDG_CACHE_HOME", dir.c_str(), 1);
    auto path = default_config_path();
    assert(path == dir / "frameyap/config.json");
    ::setenv("XDG_CONFIG_HOME", "relative", 1);
    ::setenv("HOME", dir.c_str(), 1);
    assert(default_config_path() == dir / ".config/frameyap/config.json");
    ::unsetenv("HOME");
    assert(default_config_path().empty());
    ::setenv("XDG_CONFIG_HOME", dir.c_str(), 1);
    assert(load_config(path).buttons.empty());
    assert((load_config(path).quick_inputs == std::vector<std::string>{"/new", "/questions", "/help"}));
    assert(load_config(path).backend == "redux");
    assert(!load_config(path).experimental_input_priority);
    assert(!load_config(path).advanced_debug);
    assert(!load_config(path).auto_insert);
    assert(!load_config(path).close_mic_when_idle);
    assert(!load_config(path).lock_layout);
    assert(!load_config(path).clock_24h);
    assert(load_config(path).date_format == DateFormat::MonthDayYear);
    assert(load_config(path).wrist.width == .30f);
    assert(load_config(path).gradient.enabled);
    assert(load_config(path).gradient.period_seconds == 30.f);
    assert(load_config(path).gradient.strength == .12f);
    auto example = load_config(std::filesystem::path(argv[1]) / "config.example.json");
    assert(example.buttons.at("ptt") == "/user/hand/right/input/x");
    assert(example.buttons.at("quick_chat") == "/user/hand/right/input/y");
    assert(example.buttons.at("enter").empty());
    assert(example.quick_inputs.size() == 3);
    assert(example.font.empty());
    assert(!example.experimental_input_priority);
    assert(!example.advanced_debug);
    assert(!example.auto_insert);
    assert(!example.close_mic_when_idle);
    assert(!example.lock_layout);
    assert(!example.clock_24h && example.date_format == DateFormat::MonthDayYear);
    assert(example.wrist.y == .18f && example.wrist.z == .089f);
    assert(example.gradient.enabled && example.gradient.period_seconds == 30.f && example.gradient.strength == .12f);
    std::filesystem::create_directories(path.parent_path());
    assert(save_backend(path, "fixture_cpu"));
    assert(load_config(path).backend == "fixture_cpu");
    assert(!save_backend(path, "../escaped") && !save_backend(path, "UPPER"));
    assert(save_advanced_debug(path, true));
    assert(load_config(path).advanced_debug);
    assert(get(path).find("\"advanced_debug\":true") != std::string::npos);
    assert(save_advanced_debug(path, false));
    assert(!load_config(path).advanced_debug);
    assert(save_auto_insert(path, true));
    assert(load_config(path).auto_insert);
    assert(save_auto_insert(path, false));
    assert(!load_config(path).auto_insert);
    assert(save_close_mic_when_idle(path, true));
    assert(load_config(path).close_mic_when_idle);
    assert(save_close_mic_when_idle(path, false));
    assert(!load_config(path).close_mic_when_idle);
    assert(save_lock_layout(path, true));
    assert(load_config(path).lock_layout);
    assert(save_lock_layout(path, false));
    assert(!load_config(path).lock_layout);
    assert(save_clock_24h(path, true) && load_config(path).clock_24h);
    assert(save_date_format(path, DateFormat::Iso) && load_config(path).date_format == DateFormat::Iso);
    assert(save_date_format(path, DateFormat::Off) && load_config(path).date_format == DateFormat::Off);
    assert(save_date_format(path, DateFormat::DayMonthYear) && load_config(path).date_format == DateFormat::DayMonthYear);
    assert(save_date_format(path, DateFormat::MonthDayYear));
    assert(save_clock_24h(path, false));
    for (const auto* invalid : {R"({"advanced_debug":"true"})", R"({"advanced_debug":0})",
                               R"({"advanced_debug":null})", R"({"advanced_debug":[]})"}) {
        put(path, invalid);
        fails([&] { load_config(path); });
        assert(!save_advanced_debug(path, true));
        assert(get(path) == invalid);
    }
    put(path, R"({"font":"escaped \u0061 and \"quotes\"","input_priority":"experimental","wrist":{"y":0.21},"theme":{"ink":"#123ABC"},"buttons":{"enter":""}})");
    const auto customized = get(path);
    assert(save_advanced_debug(path, true));
    assert(load_config(path).advanced_debug);
    assert(get(path).find(customized.substr(1, customized.size() - 2)) != std::string::npos);
    assert(save_advanced_debug(path, false));
    assert(!load_config(path).advanced_debug);
    auto before = get(path);
    assert(save_advanced_debug(path, false) && get(path) == before);
    for (const auto* invalid : {R"({"backend":"../escape"})", R"({"backend":""})",
                               R"({"backend":2})"}) {
        put(path, invalid); fails([&] { load_config(path); });
        assert(!save_backend(path, "redux"));
    }
    put(path, R"({"advanced_debug":false, "font":"kept"})");
    assert(save_advanced_debug(path, true));
    assert(get(path) == R"({"advanced_debug":true, "font":"kept"})");
    before = get(path);
    put(path, R"({"font":"bad","font":"duplicate"})");
    assert(!save_advanced_debug(path, true));
    assert(get(path) == R"({"font":"bad","font":"duplicate"})");
    put(path, before);
    auto link = dir / "linked-config";
    std::filesystem::create_symlink(path, link);
    assert(!save_advanced_debug(link, false));
    assert(!save_auto_insert(link, true));
    assert(!save_close_mic_when_idle(link, true));
    assert(!save_lock_layout(link, true));
    assert(get(path) == before);
    std::filesystem::remove(link);
    put(path, R"({"font":")" + std::string(4090, 'x') + R"("})");
    before = get(path);
    assert(!save_advanced_debug(path, true) && get(path) == before);
    put(path, R"({"input_priority":"experimental","advanced_debug":true})");
    auto experimental = load_config(path);
    assert(experimental.experimental_input_priority && experimental.advanced_debug);
    // A priority request must preserve the user's existing action manifest/bindings.
    assert(action_manifest(argv[1], experimental) == std::filesystem::absolute(std::filesystem::path(argv[1]) / "actions.json"));
    for (const auto* invalid : {R"({"clock_24h":"true"})", R"({"date_format":true})",
                               R"({"date_format":"us"})"}) {
        put(path, invalid); fails([&] { load_config(path); });
        assert(!save_clock_24h(path, true));
        assert(!save_date_format(path, DateFormat::Iso));
        assert(get(path) == invalid);
    }
    put(path, R"({"date_format":"off","font":"kept"})");
    assert(save_date_format(path, DateFormat::Iso));
    assert(get(path) == R"({"date_format":"iso","font":"kept"})");
    for (const auto* invalid : {R"({"close_mic_when_idle":"true"})", R"({"close_mic_when_idle":1})",
                               R"({"close_mic_when_idle":null})", R"({"close_mic_when_idle":[]})"}) {
        put(path, invalid);
        fails([&] { load_config(path); });
        assert(!save_close_mic_when_idle(path, true));
        assert(get(path) == invalid);
    }
    put(path, R"({"close_mic_when_idle":false,"font":"kept"})");
    assert(save_close_mic_when_idle(path, true));
    assert(load_config(path).close_mic_when_idle);
    assert(get(path) == R"({"close_mic_when_idle":true,"font":"kept"})");
    before = get(path);
    assert(save_close_mic_when_idle(path, true) && get(path) == before);
    assert(save_close_mic_when_idle(path, false));
    assert(get(path) == R"({"close_mic_when_idle":false,"font":"kept"})");
    put(path, R"({"auto_insert":"on"})");
    fails([&] { load_config(path); });
    assert(!save_auto_insert(path, true));
    put(path, R"({"auto_insert":false,"font":"kept"})");
    assert(save_auto_insert(path, true));
    assert(load_config(path).auto_insert);
    assert(get(path) == R"({"auto_insert":true,"font":"kept"})");
    for (const auto* invalid : {R"({"lock_layout":"true"})", R"({"lock_layout":1})",
                               R"({"lock_layout":null})", R"({"lock_layout":[]})",
                               R"({"lock_layout":{}})"}) {
        put(path, invalid);
        fails([&] { load_config(path); });
        assert(!save_lock_layout(path, true));
        assert(get(path) == invalid);
    }
    put(path, R"({"lock_layout":false,"font":"kept","date_format":"iso"})");
    assert(save_lock_layout(path, true));
    assert(load_config(path).lock_layout);
    assert(get(path) == R"({"lock_layout":true,"font":"kept","date_format":"iso"})");
    before = get(path);
    assert(save_lock_layout(path, true) && get(path) == before);
    assert(save_lock_layout(path, false));
    assert(get(path) == R"({"lock_layout":false,"font":"kept","date_format":"iso"})");
    put(path, R"({"font":"escaped \u0061","theme":{"ink":"#123ABC"},"auto_insert":true})");
    assert(save_lock_layout(path, true));
    assert(get(path) == R"({"font":"escaped \u0061","theme":{"ink":"#123ABC"},"auto_insert":true,"lock_layout":true})");
    assert(load_config(path).lock_layout && load_config(path).auto_insert);
    put(path, R"({"input_priority":"normal"})");
    assert(!load_config(path).experimental_input_priority);
    for (const auto* invalid : {R"({"input_priority":true})", R"({"input_priority":16777216})",
                               R"({"input_priority":"highest"})", R"({"input_priority":null})"}) {
        put(path, invalid);
        fails([&] { load_config(path); });
    }
    put(path, R"({"wrist":{"x":-0.02,"y":0.15,"z":0.1,"width":0.35,"roll_degrees":-35}})");
    auto tuned = load_config(path).wrist;
    assert(tuned.x == -.02f && tuned.y == .15f && tuned.z == .1f);
    assert(tuned.width == .35f && tuned.roll_degrees == -35.f);
    for (const auto* invalid : {R"({"wrist":{"width":0}})", R"({"wrist":{"x":true}})",
                               R"({"wrist":{"z":"0.1"}})", R"({"wrist":{"roll_degrees":181}})",
                               R"({"wrist":{"extra":1}})", R"({"wrist":null})",
                               R"({"wrist":{"x":1e999}})"}) {
        put(path, invalid); fails([&] { load_config(path); });
    }
    put(path, R"({"gradient":{"enabled":false,"period_seconds":45.5,"strength":0.2}})");
    auto gradient = load_config(path).gradient;
    assert(!gradient.enabled && gradient.period_seconds == 45.5f && gradient.strength == .2f);
    assert(save_auto_insert(path, true)); // unrelated settings retain the effect
    assert(load_config(path).gradient.period_seconds == 45.5f);
    put(path, R"({"gradient":{"strength":0}})");
    assert(load_config(path).gradient.enabled && load_config(path).gradient.period_seconds == 30.f);
    assert(load_config(path).gradient.strength == 0.f);
    put(path, R"({"gradient":{"period_seconds":5,"strength":0.3}})");
    assert(load_config(path).gradient.period_seconds == 5.f && load_config(path).gradient.strength == .3f);
    put(path, R"({"gradient":{"period_seconds":300}})");
    assert(load_config(path).gradient.period_seconds == 300.f);
    for (auto invalid : {R"({"gradient":true})", R"({"gradient":{"enabled":1}})",
                         R"({"gradient":{"enabled":"false"}})", R"({"gradient":{"period_seconds":0}})",
                         R"({"gradient":{"period_seconds":301}})", R"({"gradient":{"period_seconds":1e99}})",
                         R"({"gradient":{"period_seconds":true}})", R"({"gradient":{"strength":-0.1}})",
                         R"({"gradient":{"strength":0.31}})", R"({"gradient":{"strength":1e99}})",
                         R"({"gradient":{"strength":"0.1"}})", R"({"gradient":{"unknown":1}})"}) {
        put(path, invalid); fails([&] { load_config(path); });
    }
    put(path, R"({"theme":{"background":"#123ABC","accent":"#abcdef","frame_end":"#010203"},"font":"/nonexistent/face.ttf","buttons":{"ptt":"/user/hand/left/input/y","cancel":"/user/hand/right/input/b","right_grip":""}})");
    auto config = load_config(path);
    assert((config.theme.background == Rgba{0x12, 0x3a, 0xbc, 255}));
    assert((config.theme.accent == Rgba{0xab, 0xcd, 0xef, 255}));
    assert(config.buttons.at("right_grip").empty());
    assert(resolve_font(argv[1], config.font) == (std::filesystem::path(argv[1]) / "fonts/Inconsolata-Regular.ttf").string());
    auto manifest = action_manifest(argv[1], config);
    assert(std::filesystem::is_regular_file(manifest));
    assert(get(manifest).find("bindings_frame_controller.json") != std::string::npos);
    auto generated = get(manifest.parent_path() / "bindings_frame_controller.json");
    assert(generated.find("/user/hand/left/input/y") != std::string::npos);
    assert(generated.find("/actions/frameyap/in/cancel") != std::string::npos);
    assert(generated.find("/actions/frameyap/in/right_grip") == std::string::npos);
    assert(generated.find("/user/hand/right/input/a") != std::string::npos);
    assert(generated.find("/user/hand/right/input/y") != std::string::npos);
    assert(generated.find("/actions/frameyap/in/quick_chat") != std::string::npos);
    config.buttons["cancel"] = ""; config.buttons["insert"] = ""; config.buttons["enter"] = "";
    auto disabled = action_manifest(argv[1], config);
    generated = get(disabled.parent_path() / "bindings_frame_controller.json");
    assert(generated.find("/actions/frameyap/in/cancel") == std::string::npos);
    assert(generated.find("/actions/frameyap/in/insert") == std::string::npos);
    assert(generated.find("/actions/frameyap/in/enter") == std::string::npos);
    assert(action_manifest(argv[1], {}) == std::filesystem::absolute(std::filesystem::path(argv[1]) / "actions.json"));
    put(path, R"({"buttons":{"enter":"/user/hand/right/input/y"}})");
    generated = get(action_manifest(argv[1], load_config(path)).parent_path() / "bindings_frame_controller.json");
    assert(generated.find("/actions/frameyap/in/quick_chat") != std::string::npos);
    assert(generated.find("/actions/frameyap/in/enter") == std::string::npos);
    put(path, R"({"quick_inputs":["/new","custom phrase"]})");
    assert((load_config(path).quick_inputs == std::vector<std::string>{"/new", "custom phrase"}));
    for (const auto* invalid : {R"({"quick_inputs":[]})", R"({"quick_inputs":[42]})",
                               R"({"quick_inputs":["line\nfeed"]})", R"({"quick_inputs":["\u001b"]})",
                               R"({"quick_inputs":["", "/new"]})", R"({"quick_inputs":"/new"})"}) {
        put(path, invalid); fails([&] { load_config(path); });
    }
    put(path, R"({"buttons":{"ptt":"/user/hand/left/input/grip"}})");
    fails([&] { action_manifest(argv[1], load_config(path)); }); // overlapping bindings are never silently chosen
    put(path, R"({"buttons":{"ptt":"/user/hand/left/input/x\"}hack"}})");
    fails([&] { load_config(path); });
    put(path, R"({"theme":{"ink":"#FG0000"}})");
    fails([&] { load_config(path); });
    put(path, R"({"font":42})");
    fails([&] { load_config(path); });
    put(path, R"({"font":"a","font":"b"})");
    fails([&] { load_config(path); });
    put(path, "{\"font\":\"broken\"");
    fails([&] { load_config(path); });
    std::filesystem::remove_all(dir);
}