#include <assert.h>
#include <stdio.h>

#include "interp.h"
#include "parser.h"
#include "sexp.h"

ParseResult parse_str(Parser *parser, const char* str) {
    Parser_set_string(parser, str);    
    return parse_sexp(parser);
}

#define ATOM_TEST(_str, _type_enum, _field, _expect) \
{ \
    res = parse_str(&parser, (_str)); \
    assert(!ParseResult_is_err(res)); \
    sexp = *Interp_ref(&interp, res.val); \
    assert(sexp.type == (_type_enum)); \
    assert(sexp._field == (_expect)); \
}

#define STRING_TEST(_str, _type_enum, _expect) \
{ \
    res = parse_str(&parser, (_str)); \
    assert(!ParseResult_is_err(res)); \
    sexp = *Interp_ref(&interp, res.val); \
    assert(sexp.type == _type_enum); \
    assert(strcmp(sexp.str, (_expect)) == 0); \
}

#define ERROR_TEST(_str) \
{ \
    res = parse_str(&parser, (_str)); \
    assert(ParseResult_is_err(res)); \
}

int main() {
    printf("[TEST] parser\n");
    Interp interp;
    Parser parser;
    Interp_init(&interp);
    Parser_init(&parser);
    parser.ctx = &interp;

    ParseResult res;
    SExp sexp;

    ATOM_TEST("1.11", kRealSExp, real, 1.11);
    ATOM_TEST("-1.11", kRealSExp, real, -1.11);
    ATOM_TEST("1.11e10", kRealSExp, real, 1.11e10);
    ATOM_TEST("1.11 ", kRealSExp, real, 1.11);
    ATOM_TEST("1.11e10 ", kRealSExp, real, 1.11e10);
    ATOM_TEST(" 1.11 ", kRealSExp, real, 1.11);
    ATOM_TEST(" 1.11e10 ", kRealSExp, real, 1.11e10);
    ERROR_TEST("123.1x");

    ATOM_TEST("42", kIntegerSExp, integer, 42);
    ATOM_TEST("-42", kIntegerSExp, integer, -42);
    ERROR_TEST("123x");

    ATOM_TEST("#t", kBooleanSExp, boolean, true);
    ATOM_TEST("#f", kBooleanSExp, boolean, false);
    ERROR_TEST("#x");

    ATOM_TEST("#\\t", kCharSExp, character, 't');
    ATOM_TEST("#\\newline", kCharSExp, character, '\n');
    ERROR_TEST("#\\uwu");

    STRING_TEST("\"test\"", kStringSExp, "test");
    STRING_TEST("\"t\\nest\"", kStringSExp, "t\nest");
    STRING_TEST("!uwu", kSymbolSExp, "!uwu");
    STRING_TEST("-", kSymbolSExp, "-");
    STRING_TEST("+", kSymbolSExp, "+");
    ERROR_TEST("\"t\\xst\"");
    ERROR_TEST("-abc");
    ERROR_TEST("@1");
    ERROR_TEST("a|");

    Interp_free(&interp);
    Parser_free(&parser);
    printf("[PASS] parser\n");
}