#include <chrono>
#include <thread>

#include "../../shared.hpp"
#include "../../hyprctlCompat.hpp"
#include "../shared.hpp"
#include "tests.hpp"

static bool isActiveWindow(const std::string& class_, char fullscreen = '0', bool log = true) {
    std::string activeWin     = getFromSocket("/activewindow");
    auto        winClass      = Tests::getAttribute(activeWin, "class");
    auto        winFullscreen = Tests::getAttribute(activeWin, "fullscreen").back();
    if (winClass == class_ && winFullscreen == fullscreen)
        return true;
    else {
        if (log)
            NLog::log("{}Wrong active window: expected class {} fullscreen '{}', found class {}, fullscreen '{}'", Colors::RED, class_, fullscreen, winClass, winFullscreen);
        return false;
    }
}

static bool waitForActiveWindow(const std::string& class_, char fullscreen = '0', bool logLastCheck = true, int maxTries = 50) {
    int cnt = 0;
    while (!isActiveWindow(class_, fullscreen, false)) {
        ++cnt;
        std::this_thread::sleep_for(std::chrono::milliseconds(100));
        if (cnt > maxTries) {
            return isActiveWindow(class_, fullscreen, logLastCheck);
        }
    }
    return true;
}

// TODO: split this into multiple test cases
TEST_CASE(followMouseShrink) {
    NLog::log("{}Testing follow_mouse_shrink", Colors::GREEN);

    getFromSocket("/dispatch hl.dsp.focus({ workspace = 'name:follow_mouse_shrink' })");

    // follow_mouse 2 so cursor position determines focus (mode 1's delta threshold
    // is unreliable with movecursor/simulateMouseMovement). float_switch_override_focus 2
    // enables focus switching between floating windows.
    OK(getFromSocket("/eval hl.config({ input = { follow_mouse = 2 } })"));
    OK(getFromSocket("/eval hl.config({ input = { float_switch_override_focus = 2 } })"));

    // Spawn two floating windows with a 20px gap
    // fms_a: position (100,100), size 400x400 -> hitbox [100,499] x [100,499]
    if (!Tests::spawnKitty("fms_a"))
        FAIL_TEST("Couldn't spawn kitty");
    OK(getFromSocket("/dispatch hl.dsp.window.float({ action = 'set' })"));
    OK(getFromSocket("/dispatch hl.dsp.window.resize({ x = 400, y = 400 })"));
    OK(getFromSocket("/dispatch hl.dsp.window.move({ x = 100, y = 100 })"));

    // fms_b: position (520,100), size 400x400 -> hitbox [520,919] x [100,499]
    if (!Tests::spawnKitty("fms_b"))
        FAIL_TEST("Couldn't spawn kitty");
    OK(getFromSocket("/dispatch hl.dsp.window.float({ action = 'set' })"));
    OK(getFromSocket("/dispatch hl.dsp.window.resize({ x = 400, y = 400 })"));
    OK(getFromSocket("/dispatch hl.dsp.window.move({ x = 520, y = 100 })"));

    // --- Test 1: Baseline shrink=0, edge focus works ---
    NLog::log("{}Test 1: shrink=0, cursor at B's left edge focuses B", Colors::GREEN);
    OK(getFromSocket("/eval hl.config({ input = { follow_mouse_shrink = 0 } })"));
    // Focus A explicitly, then move cursor inside A so follow_mouse tracks it
    OK(getFromSocket("/dispatch hl.dsp.focus({ window = 'class:fms_a' })"));
    ASSERT(waitForActiveWindow("fms_a"), true);
    OK(getFromSocket("/dispatch hl.dsp.cursor.move({ x = 300, y = 300 })"));
    std::this_thread::sleep_for(std::chrono::milliseconds(200));
    // Move to just inside B's left edge
    OK(getFromSocket("/dispatch hl.dsp.cursor.move({ x = 521, y = 300 })"));
    ASSERT(waitForActiveWindow("fms_b"), true);

    // --- Test 2: Shrink=20, cursor in dead zone does NOT change focus ---
    NLog::log("{}Test 2: shrink=20, cursor in B's dead zone stays on A", Colors::GREEN);
    OK(getFromSocket("/eval hl.config({ input = { follow_mouse_shrink = 20 } })"));
    // Focus A explicitly
    OK(getFromSocket("/dispatch hl.dsp.focus({ window = 'class:fms_a' })"));
    ASSERT(waitForActiveWindow("fms_a"), true);
    OK(getFromSocket("/dispatch hl.dsp.cursor.move({ x = 300, y = 300 })"));
    std::this_thread::sleep_for(std::chrono::milliseconds(200));
    // Move to 530,300 -- 10px inside B, within 20px shrink zone (B's shrunk hitbox starts at 540)
    OK(getFromSocket("/dispatch hl.dsp.cursor.move({ x = 530, y = 300 })"));
    std::this_thread::sleep_for(std::chrono::milliseconds(500));
    ASSERT(isActiveWindow("fms_a"), true);

    // --- Test 3: Shrink=20, cursor well inside inactive window DOES focus it ---
    NLog::log("{}Test 3: shrink=20, cursor at B's center focuses B", Colors::GREEN);
    // Still focused on A from test 2
    OK(getFromSocket("/dispatch hl.dsp.cursor.move({ x = 720, y = 300 })"));
    ASSERT(waitForActiveWindow("fms_b"), true);

    // --- Test 4: Focused window's hitbox is NOT shrunk ---
    NLog::log("{}Test 4a: focused window hitbox is not shrunk", Colors::GREEN);
    // Focus A explicitly, move cursor inside A, then move near A's right edge
    OK(getFromSocket("/dispatch hl.dsp.focus({ window = 'class:fms_a' })"));
    ASSERT(waitForActiveWindow("fms_a"), true);
    OK(getFromSocket("/dispatch hl.dsp.cursor.move({ x = 300, y = 300 })"));
    std::this_thread::sleep_for(std::chrono::milliseconds(200));
    OK(getFromSocket("/dispatch hl.dsp.cursor.move({ x = 490, y = 300 })"));
    std::this_thread::sleep_for(std::chrono::milliseconds(500));
    ASSERT(isActiveWindow("fms_a"), true);

    NLog::log("{}Test 4b: inactive window hitbox IS shrunk at same position", Colors::GREEN);
    // Focus B explicitly, then move to (490,300). Now A is inactive with shrunk box ending at 479, so 490 is outside A.
    OK(getFromSocket("/dispatch hl.dsp.focus({ window = 'class:fms_b' })"));
    ASSERT(waitForActiveWindow("fms_b"), true);
    OK(getFromSocket("/dispatch hl.dsp.cursor.move({ x = 720, y = 300 })"));
    std::this_thread::sleep_for(std::chrono::milliseconds(200));
    OK(getFromSocket("/dispatch hl.dsp.cursor.move({ x = 490, y = 300 })"));
    std::this_thread::sleep_for(std::chrono::milliseconds(500));
    ASSERT(isActiveWindow("fms_b"), true);
}
