#include <assert.h>
#include <pprocessor.h>
#include <string.h>
#include <utest/acutest.h>

static cbool process_input(const char *input, cstring_t *output) {
    smcc_pp_t pp;
    core_mem_stream_t mem_stream;
    core_stream_t *output_stream;

    // 初始化预处理器
    output_stream = pp_init(
        &pp, core_mem_stream_init(&mem_stream, input, strlen(input), false));

    // 获取输出结果
    int ch;
    *output = cstring_new();
    while (1) {
        ch = core_stream_next_char(output_stream);
        if (ch == core_stream_eof) {
            break;
        }
        cstring_push(output, (char)ch);
    }
    // 清理资源
    pp_drop(&pp);

    return true;
}

#define CHECK_PP_OUTPUT_EXACT(input, expect)                                   \
    do {                                                                       \
        cstring_t output;                                                      \
        process_input(input, &output);                                         \
        assert(output.data != NULL);                                           \
        TEST_CHECK(strcmp(output.data, expect) == 0);                          \
    } while (0)

#define CHECK_PP_OUTPUT_CONTAIN(input, expect)                                 \
    do {                                                                       \
        cstring_t output;                                                      \
        process_input(input, &output);                                         \
        assert(output.data != NULL);                                           \
        TEST_CHECK(strstr(output.data, expect) != NULL);                       \
    } while (0)

static void test_define_simple_object_macro(void) {
    TEST_CASE("simple object-like macro");
    CHECK_PP_OUTPUT_EXACT("#define MAX 100\nMAX\n", "100\n");
    CHECK_PP_OUTPUT_EXACT("#define NAME test\r\nNAME\n", "test\n");
}

static void test_define_complex_object_macro(void) {
    TEST_CASE("complex object-like macro");
    CHECK_PP_OUTPUT_EXACT("#define VALUE (100 + 50)\nVALUE\n", "(100 + 50)\n");
    CHECK_PP_OUTPUT_EXACT("#define PI 3.14159\nPI\n", "3.14159\n");
}

static void test_define_function_macro(void) {
    TEST_CASE("function-like macro");
    CHECK_PP_OUTPUT_EXACT("#define ADD(a,b) a + b\nADD(1, 2)\n", "1 + 2\n");
    CHECK_PP_OUTPUT_EXACT(
        "#define MAX(a,b) ((a) > (b) ? (a) : (b))\nMAX(10, 20)\n",
        "((10) > (20) ? (10) : (20))\n");
}

static void test_define_stringify_operator(void) {
    TEST_CASE("stringify operator (#)");
    CHECK_PP_OUTPUT_EXACT("#define STRINGIFY(x) #x\nSTRINGIFY(hello)\n",
                          "\"hello\"\n");
    CHECK_PP_OUTPUT_EXACT("#define STR(x) #x\nSTR(test value)\n",
                          "\"test value\"\n");
}

static void test_define_concat_operator(void) {
    TEST_CASE("concatenation operator (##)");
    CHECK_PP_OUTPUT_EXACT("#define CONCAT(a,b) a##b\nCONCAT(hello,world)\n",
                          "helloworld\n");
    CHECK_PP_OUTPUT_EXACT("#define JOIN(pre,suf) pre##suf\nJOIN(var,123)\n",
                          "var123\n");
}

static void test_define_nested_macros(void) {
    TEST_CASE("nested macros");
    CHECK_PP_OUTPUT_EXACT(
        "#define MAX 100\n#define TWICE_MAX (MAX * 2)\nTWICE_MAX\n",
        "(100 * 2)\n");
    CHECK_PP_OUTPUT_EXACT(
        "#define A 1\n#define B (A + 1)\n#define C (B + 1)\nC\n",
        "((1 + 1) + 1)\n");
}

TEST_LIST = {
    {"test_define_simple_object_macro", test_define_simple_object_macro},
    {"test_define_complex_object_macro", test_define_complex_object_macro},
    {"test_define_function_macro", test_define_function_macro},
    {"test_define_stringify_operator", test_define_stringify_operator},
    {"test_define_concat_operator", test_define_concat_operator},
    {"test_define_nested_macros", test_define_nested_macros},
    {NULL, NULL},
};