#include "unity.h"
#include "config.h"
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>

/* 辅助：写临时配置文件，返回路径 */
static const char *write_temp_config(const char *content) {
    static char path[256];
    static int counter = 0;
    snprintf(path, sizeof(path), "/tmp/cocoon_test_config_%d_%d.json",
             (int)getpid(), counter++);
    FILE *fp = fopen(path, "w");
    if (fp) {
        fwrite(content, 1, strlen(content), fp);
        fclose(fp);
    }
    return path;
}

static void cleanup(const char *path) {
    remove(path);
}

/* ===== config_load_from_file ===== */

void test_load_valid_config(void) {
    const char *p = write_temp_config(
        "{\n"
        "  \"root_dir\": \"/var/www\",\n"
        "  \"port\": 8080,\n"
        "  \"threaded\": true,\n"
        "  \"num_workers\": 8,\n"
        "  \"max_connections\": 5000,\n"
        "  \"timeout_ms\": 60000,\n"
        "  \"log_level\": \"debug\",\n"
        "  \"gzip_enabled\": false\n"
        "}\n"
    );
    cocoon_config_t cfg = {0};
    TEST_ASSERT_TRUE(config_load_from_file(p, &cfg));
    TEST_ASSERT_EQUAL_STRING("/var/www", cfg.root_dir);
    TEST_ASSERT_EQUAL(8080, cfg.port);
    TEST_ASSERT_TRUE(cfg.threaded);
    TEST_ASSERT_EQUAL(8, cfg.num_workers);
    TEST_ASSERT_EQUAL(5000, cfg.max_connections);
    TEST_ASSERT_EQUAL(60000, cfg.timeout_ms);
    TEST_ASSERT_EQUAL(LOG_LEVEL_DEBUG, cfg.log_level);
    TEST_ASSERT_FALSE(cfg.gzip_enabled);
    free((void *)cfg.root_dir);
    cleanup(p);
}

void test_load_valid_minimal(void) {
    const char *p = write_temp_config("{\"port\": 3000}");
    cocoon_config_t cfg = {0};
    TEST_ASSERT_TRUE(config_load_from_file(p, &cfg));
    TEST_ASSERT_EQUAL(3000, cfg.port);
    free((void *)cfg.root_dir);
    cleanup(p);
}

void test_load_valid_with_comments(void) {
    const char *p = write_temp_config(
        "{\n"
        "  // this is a comment\n"
        "  \"port\": 9090\n"
        "}\n"
    );
    cocoon_config_t cfg = {0};
    TEST_ASSERT_TRUE(config_load_from_file(p, &cfg));
    TEST_ASSERT_EQUAL(9090, cfg.port);
    free((void *)cfg.root_dir);
    cleanup(p);
}

void test_load_invalid_json(void) {
    const char *p = write_temp_config("not json");
    cocoon_config_t cfg = {0};
    TEST_ASSERT_FALSE(config_load_from_file(p, &cfg));
    cleanup(p);
}

void test_load_missing_file(void) {
    cocoon_config_t cfg = {0};
    TEST_ASSERT_FALSE(config_load_from_file(
        "/tmp/cocoon_nonexist_12345.json", &cfg));
}

void test_load_empty_file(void) {
    const char *p = write_temp_config("");
    cocoon_config_t cfg = {0};
    TEST_ASSERT_FALSE(config_load_from_file(p, &cfg));
    cleanup(p);
}

void test_load_null_args(void) {
    TEST_ASSERT_FALSE(config_load_from_file(NULL, NULL));
    TEST_ASSERT_FALSE(config_load_from_file("/tmp/x", NULL));
}

/* ===== config_merge ===== */

void test_merge_override_all(void) {
    cocoon_config_t base = {
        .root_dir = strdup("/old"),
        .port = 8080,
        .threaded = false,
        .num_workers = 2,
        .max_connections = 100,
        .timeout_ms = 30000,
        .log_level = LOG_LEVEL_INFO,
        .gzip_enabled = true,
        .brotli_enabled = true
    };
    cocoon_config_t cmdline = {
        .root_dir = strdup("/new"),
        .port = 9090,
        .threaded = true,
        .num_workers = 4,
        .max_connections = 200,
        .timeout_ms = 60000,
        .log_level = LOG_LEVEL_DEBUG,
        .gzip_enabled = false,
        .brotli_enabled = false
    };
    config_merge(&base, &cmdline,
                 true, true, true, true, true, true, true, true, true, true, true, true,
                 false, false, false, false);
    TEST_ASSERT_EQUAL_STRING("/new", base.root_dir);
    TEST_ASSERT_EQUAL(9090, base.port);
    TEST_ASSERT_TRUE(base.threaded);
    TEST_ASSERT_EQUAL(4, base.num_workers);
    TEST_ASSERT_EQUAL(200, base.max_connections);
    TEST_ASSERT_EQUAL(60000, base.timeout_ms);
    TEST_ASSERT_EQUAL(LOG_LEVEL_DEBUG, base.log_level);
    TEST_ASSERT_FALSE(base.gzip_enabled);
    TEST_ASSERT_FALSE(base.brotli_enabled);
    free((void *)base.root_dir);
    free((void *)cmdline.root_dir);
}

void test_merge_no_override(void) {
    cocoon_config_t base = {
        .root_dir = strdup("/old"),
        .port = 8080,
        .threaded = false,
        .num_workers = 2,
        .log_level = LOG_LEVEL_INFO
    };
    cocoon_config_t cmdline = {
        .root_dir = "/new",
        .port = 9090,
        .num_workers = 4,
        .log_level = LOG_LEVEL_DEBUG
    };
    config_merge(&base, &cmdline,
                 false, false, false, false, false, false, false, false, false, false, false, false,
                 false, false, false, false);
    TEST_ASSERT_EQUAL_STRING("/old", base.root_dir);
    TEST_ASSERT_EQUAL(8080, base.port);
    TEST_ASSERT_FALSE(base.threaded);
    TEST_ASSERT_EQUAL(2, base.num_workers);
    TEST_ASSERT_EQUAL(LOG_LEVEL_INFO, base.log_level);
    free((void *)base.root_dir);
}

void test_merge_partial_override(void) {
    cocoon_config_t base = {
        .root_dir = strdup("/old"),
        .port = 8080,
        .num_workers = 2,
        .log_level = LOG_LEVEL_INFO
    };
    cocoon_config_t cmdline = {
        .root_dir = "/new",
        .port = 9090,
        .num_workers = 4,
        .log_level = LOG_LEVEL_DEBUG
    };
    config_merge(&base, &cmdline,
                 true, false, false, true, false, false, false, false, false, false, false, false,
                 false, false, false, false);
    TEST_ASSERT_EQUAL_STRING("/new", base.root_dir);   /* overridden */
    TEST_ASSERT_EQUAL(8080, base.port);                /* not overridden */
    TEST_ASSERT_EQUAL(2, base.num_workers);            /* not overridden */
    TEST_ASSERT_EQUAL(LOG_LEVEL_INFO, base.log_level);   /* not overridden */
    free((void *)base.root_dir);
}

void test_load_invalid_field_type(void) {
    /* port 为字符串而非数字，应解析失败或忽略 */
    const char *p = write_temp_config("{\"port\": \"not_a_number\"}");
    cocoon_config_t cfg = {0};
    /* 极简解析器可能忽略非法类型，不 crash 即可 */
    config_load_from_file(p, &cfg);
    /* port 保持默认值 0 */
    TEST_ASSERT_EQUAL(0, cfg.port);
    free((void *)cfg.root_dir);
    cleanup(p);
}

void test_load_log_level_invalid(void) {
    const char *p = write_temp_config("{\"log_level\": \"verbose\"}");
    cocoon_config_t cfg = {0};
    config_load_from_file(p, &cfg);
    /* 非法 log_level 保持默认值 INFO */
    TEST_ASSERT_EQUAL(LOG_LEVEL_INFO, cfg.log_level);
    free((void *)cfg.root_dir);
    cleanup(p);
}

void test_load_nested_ignored(void) {
    /* 极简解析器遇到嵌套对象会失败，但至少不 crash */
    const char *p = write_temp_config(
        "{\n"
        "  \"port\": 7777,\n"
        "  \"nested\": {\"a\": 1}\n"
        "}\n"
    );
    cocoon_config_t cfg = {0};
    /* 当前极简解析器不支持嵌套对象，返回 false，但不应 segfault */
    config_load_from_file(p, &cfg);
    TEST_ASSERT_TRUE(1); /* 走到这里说明没 crash */
    free((void *)cfg.root_dir);
    cleanup(p);
}

void test_load_array_ignored(void) {
    /* 极简解析器遇到数组会失败，但至少不 crash */
    const char *p = write_temp_config(
        "{\n"
        "  \"port\": 7777,\n"
        "  \"arr\": [1, 2, 3]\n"
        "}\n"
    );
    cocoon_config_t cfg = {0};
    config_load_from_file(p, &cfg);
    TEST_ASSERT_TRUE(1); /* 走到这里说明没 crash */
    free((void *)cfg.root_dir);
    cleanup(p);
}

void test_merge_cmdline_null_root_dir(void) {
    /* cmdline root_dir 为 NULL，不应覆盖 base */
    cocoon_config_t base = {.root_dir = strdup("/old"), .port = 8080};
    cocoon_config_t cmdline = {.root_dir = NULL, .port = 9090};
    config_merge(&base, &cmdline, true, true, false, false, false, false, false, false, false, false, false, false,
                 false, false, false, false);
    TEST_ASSERT_EQUAL_STRING("/old", base.root_dir); /* NULL 不覆盖 */
    TEST_ASSERT_EQUAL(9090, base.port);               /* port 覆盖 */
    free((void *)base.root_dir);
}

void test_merge_null_safety(void) {
    /* 不应 crash */
    config_merge(NULL, NULL, true, true, true, true, true, true, true, true, true, true, true, true,
                 true, true, true, true);
    TEST_ASSERT_TRUE(1);
}

void test_load_gzip_enabled(void) {
    /* gzip_enabled: true */
    const char *p1 = write_temp_config("{\"gzip_enabled\": true}");
    cocoon_config_t cfg1 = {0};
    TEST_ASSERT_TRUE(config_load_from_file(p1, &cfg1));
    TEST_ASSERT_TRUE(cfg1.gzip_enabled);
    free((void *)cfg1.root_dir);
    cleanup(p1);

    /* gzip_enabled: false */
    const char *p2 = write_temp_config("{\"gzip_enabled\": false}");
    cocoon_config_t cfg2 = {0};
    TEST_ASSERT_TRUE(config_load_from_file(p2, &cfg2));
    TEST_ASSERT_FALSE(cfg2.gzip_enabled);
    free((void *)cfg2.root_dir);
    cleanup(p2);

    /* 默认值应为 true（在 main 中设置，但这里 0 初始化后为 false） */
    const char *p3 = write_temp_config("{\"port\": 3000}");
    cocoon_config_t cfg3 = {.gzip_enabled = true};
    TEST_ASSERT_TRUE(config_load_from_file(p3, &cfg3));
    TEST_ASSERT_TRUE(cfg3.gzip_enabled); /* 未指定时保持原值 */
    free((void *)cfg3.root_dir);
    cleanup(p3);
}

void test_load_brotli_enabled(void) {
    /* brotli_enabled: true */
    const char *p1 = write_temp_config("{\"brotli_enabled\": true}");
    cocoon_config_t cfg1 = {0};
    TEST_ASSERT_TRUE(config_load_from_file(p1, &cfg1));
    TEST_ASSERT_TRUE(cfg1.brotli_enabled);
    free((void *)cfg1.root_dir);
    cleanup(p1);

    /* brotli_enabled: false */
    const char *p2 = write_temp_config("{\"brotli_enabled\": false}");
    cocoon_config_t cfg2 = {0};
    TEST_ASSERT_TRUE(config_load_from_file(p2, &cfg2));
    TEST_ASSERT_FALSE(cfg2.brotli_enabled);
    free((void *)cfg2.root_dir);
    cleanup(p2);

    /* 默认值应为 true */
    const char *p3 = write_temp_config("{\"port\": 3000}");
    cocoon_config_t cfg3 = {.brotli_enabled = true};
    TEST_ASSERT_TRUE(config_load_from_file(p3, &cfg3));
    TEST_ASSERT_TRUE(cfg3.brotli_enabled);
    free((void *)cfg3.root_dir);
    cleanup(p3);
}

void setUp(void) {}
void tearDown(void) {}

int main(void) {
    UNITY_BEGIN();

    RUN_TEST(test_load_valid_config);
    RUN_TEST(test_load_valid_minimal);
    RUN_TEST(test_load_valid_with_comments);
    RUN_TEST(test_load_invalid_json);
    RUN_TEST(test_load_missing_file);
    RUN_TEST(test_load_empty_file);
    RUN_TEST(test_load_null_args);
    RUN_TEST(test_load_invalid_field_type);
    RUN_TEST(test_load_log_level_invalid);
    RUN_TEST(test_load_nested_ignored);
    RUN_TEST(test_load_array_ignored);

    RUN_TEST(test_merge_override_all);
    RUN_TEST(test_merge_no_override);
    RUN_TEST(test_merge_partial_override);
    RUN_TEST(test_merge_null_safety);
    RUN_TEST(test_merge_cmdline_null_root_dir);

    RUN_TEST(test_load_gzip_enabled);
    RUN_TEST(test_load_brotli_enabled);
    return UNITY_END();
}