#define _GNU_SOURCE
#include <dirent.h>
#include <fcntl.h>
#include <inttypes.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <unistd.h>

#include <unity/unity.h>

#include "config.h"
#include "persist.h"
#include "xce/cluster_def.h"
#include "xlist.h"

/* ------------------------------------------------------------------ */
/* Test helpers                                                         */
/* ------------------------------------------------------------------ */

static char g_tmpdir[256];

static persist_t* make_hook_test_persist(void)
{
    strcpy(g_tmpdir, "/tmp/xce_hook_test_XXXXXX");
    if (!mkdtemp(g_tmpdir))
        return NULL;

    persist_t* p = calloc(1, sizeof(persist_t));
    if (!p)
        return NULL;

    persist_config_t cfg = persist_config_defaults();
    cfg.wal_sync_ms = 1; /* 1 ms — fsync thread wakes up immediately on stop */

    persist_set_workdir(p, g_tmpdir, "hook-test");

    if (persist_init(p, &cfg, NULL) != 0) {
        free(p);
        return NULL;
    }

    if (persist_start(p) != 0) {
        persist_fini(p);
        free(p);
        return NULL;
    }

    return p;
}

static void cleanup_hook_test_persist(persist_t* p)
{
    persist_stop(p);
    persist_fini(p);
    free(p);
}

/* Zero-initialized stubs — capnp encoders handle all-NULL fields safely */
static worker_t g_stub_worker;
static job_t    g_stub_job;
static queue_t  g_stub_queue;

/* Variadic wrappers so tests can pass typed args through a real va_list */
static int call_worker_hook(int type, int sit, int detail, persist_t* p, ...)
{
    va_list ap;
    va_start(ap, p);
    int rc = persist_worker_hook_fun(type, sit, detail, p, ap);
    va_end(ap);
    return rc;
}

static int call_job_hook(int type, int sit, int detail, persist_t* p, ...)
{
    va_list ap;
    va_start(ap, p);
    int rc = persist_job_hook_fun(type, sit, detail, p, ap);
    va_end(ap);
    return rc;
}

static int call_queue_hook(int type, int sit, int detail, persist_t* p, ...)
{
    va_list ap;
    va_start(ap, p);
    int rc = persist_queue_hook_fun(type, sit, detail, p, ap);
    va_end(ap);
    return rc;
}

/* One-element xlist backed by a stack pointer array — no heap allocation needed */
#define STUB_JOB_XLIST(name) \
    void* name##_ptr_ = &g_stub_job; \
    xlist_t name = { .count = 1, .caps = 1, .delta = 1, .data = &name##_ptr_ }

/* ------------------------------------------------------------------ */
/* Tests — RED phase: functions not yet implemented, must fail to link  */
/* ------------------------------------------------------------------ */

static void test_job_hook_new_writes_wal_entry(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    /* JOB_NEW args: (cluster_t*, xlist_t* jlist) — list must have ≥1 job */
    STUB_JOB_XLIST(jlist);
    int rc = call_job_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                           HOOK_DETAIL_JOB_NEW, p,
                           (cluster_t*)NULL, &jlist);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_GREATER_THAN(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_job_hook_new_writes_wal_entry: PASS\n");
}

static void test_job_hook_start_writes_wal_entry(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    /* JOB_START args: (cluster_t*, job_pidinfo_t*, job_t*) */
    int rc = call_job_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                           HOOK_DETAIL_JOB_START, p,
                           (cluster_t*)NULL, (void*)NULL, &g_stub_job);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_GREATER_THAN(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_job_hook_start_writes_wal_entry: PASS\n");
}

static void test_job_hook_exit_writes_wal_entry(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    /* JOB_EXIT args: (cluster_t*, xce_jobfinish_t*, job_t*) */
    int rc = call_job_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                           HOOK_DETAIL_JOB_EXIT, p,
                           (cluster_t*)NULL, (void*)NULL, &g_stub_job);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_GREATER_THAN(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_job_hook_exit_writes_wal_entry: PASS\n");
}

static void test_job_hook_fail_writes_wal_entry(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    /* JOB_FAIL args: (cluster_t*, job_failinfo_t*, job_t*) */
    int rc = call_job_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                           HOOK_DETAIL_JOB_FAIL, p,
                           (cluster_t*)NULL, (void*)NULL, &g_stub_job);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_GREATER_THAN(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_job_hook_fail_writes_wal_entry: PASS\n");
}

static void test_job_hook_stop_writes_wal_entry(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    /* STOP/RESUME/UPDATE/CKPT/PREASON/RUSAGE default args: (cluster_t*, job_t*) */
    int rc = call_job_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                           HOOK_DETAIL_JOB_STOP, p,
                           (cluster_t*)NULL, &g_stub_job);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_GREATER_THAN(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_job_hook_stop_writes_wal_entry: PASS\n");
}

static void test_job_hook_resume_writes_wal_entry(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    int rc = call_job_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                           HOOK_DETAIL_JOB_RESUME, p,
                           (cluster_t*)NULL, &g_stub_job);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_GREATER_THAN(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_job_hook_resume_writes_wal_entry: PASS\n");
}

static void test_job_hook_update_writes_wal_entry(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    int rc = call_job_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                           HOOK_DETAIL_JOB_UPDATE, p,
                           (cluster_t*)NULL, &g_stub_job);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_GREATER_THAN(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_job_hook_update_writes_wal_entry: PASS\n");
}

static void test_job_hook_ckpt_writes_wal_entry(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    int rc = call_job_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                           HOOK_DETAIL_JOB_CKPT, p,
                           (cluster_t*)NULL, &g_stub_job);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_GREATER_THAN(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_job_hook_ckpt_writes_wal_entry: PASS\n");
}

static void test_job_hook_rusage_writes_wal_entry(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    int rc = call_job_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                           HOOK_DETAIL_JOB_RUSAGE_UPDATE, p,
                           (cluster_t*)NULL, &g_stub_job);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_GREATER_THAN(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_job_hook_rusage_writes_wal_entry: PASS\n");
}

static void test_job_hook_setpreason_writes_wal_entry(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    int rc = call_job_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                           HOOK_DETAIL_SETPREASON, p,
                           (cluster_t*)NULL, &g_stub_job);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_GREATER_THAN(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_job_hook_setpreason_writes_wal_entry: PASS\n");
}

static void test_job_hook_clrpreason_writes_wal_entry(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    int rc = call_job_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                           HOOK_DETAIL_CLRPREASON, p,
                           (cluster_t*)NULL, &g_stub_job);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_GREATER_THAN(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_job_hook_clrpreason_writes_wal_entry: PASS\n");
}

static void test_queue_hook_update_writes_wal_entry(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    /* QUEUE_UPDATE args: (cluster_t*, queue_t*) */
    int rc = call_queue_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                             HOOK_DETAIL_QUEUE_UPDATE, p,
                             (cluster_t*)NULL, &g_stub_queue);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_GREATER_THAN(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_queue_hook_update_writes_wal_entry: PASS\n");
}

static void test_worker_hook_join_writes_wal_entry(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    /* Unified JOIN signature: (cluster_t*, char* name, worker_t*) */
    int rc = call_worker_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                              HOOK_DETAIL_WORKER_JOIN, p,
                              (cluster_t*)NULL, "stub-host", &g_stub_worker);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_GREATER_THAN(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_worker_hook_join_writes_wal_entry: PASS\n");
}

static void test_worker_hook_update_writes_wal_entry(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    /* UPDATE signature: (cluster_t*, worker_t*) */
    int rc = call_worker_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                              HOOK_DETAIL_WORKER_UPDATE, p,
                              (cluster_t*)NULL, &g_stub_worker);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_GREATER_THAN(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_worker_hook_update_writes_wal_entry: PASS\n");
}

static void test_worker_hook_dead_writes_wal_entry(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    /* DEAD signature: (cluster_t*, worker_t*) */
    int rc = call_worker_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                              HOOK_DETAIL_WORKER_DEAD, p,
                              (cluster_t*)NULL, &g_stub_worker);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_GREATER_THAN(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_worker_hook_dead_writes_wal_entry: PASS\n");
}

static void test_job_hook_skips_unknown_detail(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    /* HOOK_DETAIL_SCHEDULER_JOIN has no WAL mapping — must be silently skipped */
    int rc = persist_job_hook_fun(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                                  HOOK_DETAIL_SCHEDULER_JOIN, p, NULL);
    TEST_ASSERT_EQUAL_INT(0, rc);

    /* LSN must NOT have advanced — no WAL entry written */
    TEST_ASSERT_EQUAL_UINT64(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_job_hook_skips_unknown_detail: PASS\n");
}

static void test_worker_hook_skips_unknown_detail(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    int rc = persist_worker_hook_fun(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT,
                                     HOOK_DETAIL_SCHEDULER_DEAD, p, NULL);
    TEST_ASSERT_EQUAL_INT(0, rc);
    TEST_ASSERT_EQUAL_UINT64(lsn_before, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_worker_hook_skips_unknown_detail: PASS\n");
}

static void test_job_hook_new_writes_multiple_entries_sequentially(void)
{
    persist_t* p = make_hook_test_persist();
    TEST_ASSERT_NOT_NULL(p);

    uint64_t lsn_before = p->current_lsn;

    /* Fire three hook calls — each must advance LSN */
    STUB_JOB_XLIST(jlist);
    call_job_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT, HOOK_DETAIL_JOB_NEW, p,
                  (cluster_t*)NULL, &jlist);
    call_job_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT, HOOK_DETAIL_JOB_START, p,
                  (cluster_t*)NULL, (void*)NULL, &g_stub_job);
    call_job_hook(HOOK_AFTER_EVENT, HOOK_AFTER_EVENT, HOOK_DETAIL_JOB_EXIT, p,
                  (cluster_t*)NULL, (void*)NULL, &g_stub_job);

    TEST_ASSERT_EQUAL_UINT64(lsn_before + 3, p->current_lsn);

    cleanup_hook_test_persist(p);
    printf("test_job_hook_new_writes_multiple_entries_sequentially: PASS\n");
}

/* ------------------------------------------------------------------ */
/* Suite entry point                                                    */
/* ------------------------------------------------------------------ */

void test_persist_hooks(void)
{
    printf("\n>> persist hook function tests\n\n");

    RUN_TEST(test_job_hook_new_writes_wal_entry);
    RUN_TEST(test_job_hook_start_writes_wal_entry);
    RUN_TEST(test_job_hook_exit_writes_wal_entry);
    RUN_TEST(test_job_hook_fail_writes_wal_entry);
    RUN_TEST(test_job_hook_stop_writes_wal_entry);
    RUN_TEST(test_job_hook_resume_writes_wal_entry);
    RUN_TEST(test_job_hook_update_writes_wal_entry);
    RUN_TEST(test_job_hook_ckpt_writes_wal_entry);
    RUN_TEST(test_job_hook_rusage_writes_wal_entry);
    RUN_TEST(test_job_hook_setpreason_writes_wal_entry);
    RUN_TEST(test_job_hook_clrpreason_writes_wal_entry);
    RUN_TEST(test_queue_hook_update_writes_wal_entry);
    RUN_TEST(test_worker_hook_join_writes_wal_entry);
    RUN_TEST(test_worker_hook_update_writes_wal_entry);
    RUN_TEST(test_worker_hook_dead_writes_wal_entry);
    RUN_TEST(test_job_hook_skips_unknown_detail);
    RUN_TEST(test_worker_hook_skips_unknown_detail);
    RUN_TEST(test_job_hook_new_writes_multiple_entries_sequentially);
}