#include "ast/expressions/identifier.hpp"
#include "ast/expressions/integer.hpp"
#include "ast/expressions/prefix.hpp"
#include "utils.hpp"

#include <doctest.h>
#include <sstream>

void test_integer_literal(ast::expression* expr, int value) {
    ast::integer_literal* int_lit = cast<ast::integer_literal>(expr);

    REQUIRE(int_lit->value == value);
    std::ostringstream oss;
    oss << value;
    REQUIRE(int_lit->token_literal() == oss.str());
}

TEST_SUITE("Parser: expression") {
    TEST_CASE_FIXTURE(
        ParserFixture,
        "Simple expression statement with identifier"
    ) {
        setup("foobar;");
        REQUIRE(program->statements.size() == 1);
        ast::expression_stmt* expression_stmt =
            cast<ast::expression_stmt>(program->statements[0]);

        ast::identifier* ident =
            cast<ast::identifier>(expression_stmt->expression);

        REQUIRE(ident->value == "foobar");
        REQUIRE(ident->token_literal() == "foobar");
    };

    TEST_CASE_FIXTURE(
        ParserFixture,
        "Simple expression statement with integer"
    ) {
        setup("5;");
        REQUIRE(program->statements.size() == 1);

        ast::expression_stmt* expression_stmt =
            cast<ast::expression_stmt>(program->statements[0]);

        test_integer_literal(expression_stmt->expression, 5);
    };

    TEST_CASE_FIXTURE(
        ParserFixture,
        "Simple expression statement with prefix before integer"
    ) {
        SUBCASE("Prefix: '!'") {
            setup("!5;");

            REQUIRE(program->statements.size() == 1);
            ast::expression_stmt* expression_stmt =
                cast<ast::expression_stmt>(program->statements[0]);

            ast::prefix_expr* prefix_expr =
                cast<ast::prefix_expr>(expression_stmt->expression);

            REQUIRE(prefix_expr->op == "!");
            test_integer_literal(prefix_expr->right, 5);
        }

        SUBCASE("Prefix: '-'") {
            setup("-15;");

            REQUIRE(program->statements.size() == 1);
            ast::expression_stmt* expression_stmt =
                cast<ast::expression_stmt>(program->statements[0]);

            ast::prefix_expr* prefix_expr =
                cast<ast::prefix_expr>(expression_stmt->expression);

            REQUIRE(prefix_expr->op == "-");
            test_integer_literal(prefix_expr->right, 15);
        }
    }
}