/*
* Copyright (C) 2023 Carlos Maniero
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/
#define MUNIT_ENABLE_ASSERT_ALIASES
#include "munit.h"
#include "lexer.h"
#include "parser.h"
#include "ast.h"

void
make_lexer_from_static_src(lexer_t *lexer, char *src)
{
  lexer->filepath = "test.pipa";
  lexer->srclen   = 0;
  lexer->cur      = 0;
  lexer->row      = 0;
  lexer->bol      = 0;
  lexer->src      = src;
  lexer->srclen   = strlen(src);
}

void
assert_parser_error(char* src, char* error_msg) {
  parser_t parser;
  lexer_t lexer;

  make_lexer_from_static_src(&lexer, src);
  parser_init(&parser, &lexer);

  ast_node_t *ast_function = ast_node_new();

  bool parsed = parser_parse_function_declaration(&parser, ast_function);
  assert_false(parsed);
  assert_int(1, ==, parser.errors_len);
  assert_string_equal(error_msg, parser.errors[0].message);
}

static MunitResult
test_parse_function(const MunitParameter params[],
                    void *user_data_or_fixture)
{
  parser_t parser;
  lexer_t lexer;

  make_lexer_from_static_src(&lexer, "main(): i32 { return 42; }");
  parser_init(&parser, &lexer);
  ast_node_t *ast_function = ast_node_new();

  bool parsed = parser_parse_function_declaration(&parser, ast_function);
  assert_true(parsed);

  char actual[5];

  string_view_to_str(&ast_function->data.function.name, actual);
  assert_string_equal("main", actual);
  assert_int(AST_FUNCTION_DECLARATION, ==, ast_function->kind);

  ast_node_t *ast_return = ast_function->data.function.body;

  assert_int(AST_RETURN_STMT, ==, ast_return->kind);
  assert_int(42, ==, ast_return->data.return_stmt.number);

  ast_node_destroy(ast_function);

  return MUNIT_OK;
}

static MunitResult
test_parse_basic_syntax_errors(const MunitParameter params[],
                               void *user_data_or_fixture)
{
  assert_parser_error("(): i32 { return 42; }"      , "expected 'TOKEN_NAME' but got '('");
  assert_parser_error("main): i32 { return 42; }"   , "expected '(' but got ')'");
  assert_parser_error("main(: i32 { return 42; }"   , "expected ')' but got ':'");
  assert_parser_error("main() i32 { return 42; }"   , "expected ':' but got 'TOKEN_NAME'");
  assert_parser_error("main(): { return 42; }"      , "expected 'TOKEN_NAME' but got '{'");
  assert_parser_error("main(): i32 return 42; }"    , "expected '{' but got 'TOKEN_NAME'");
  assert_parser_error("main(): i32 { 42; }"         , "expected 'TOKEN_NAME' but got 'TOKEN_NUMBER'");
  assert_parser_error("main(): i32 { return; }"     , "expected 'TOKEN_NUMBER' but got ';'");
  assert_parser_error("main(): i32 { return 42;"    , "expected '}' but got end of file");
  assert_parser_error("main(): beff { return 42; }" , "type 'beff' is not defined");
  assert_parser_error("main(): i32 { oxi 42; }"     , "expected 'return' keyword but got 'oxi'");

  return MUNIT_OK;
}

static MunitTest tests[] = {
  { "/test_parse_function", test_parse_function, NULL, NULL, MUNIT_TEST_OPTION_NONE, NULL },
  { "/test_parse_basic_syntax_errors", test_parse_basic_syntax_errors, NULL, NULL, MUNIT_TEST_OPTION_NONE, NULL },
  { NULL, NULL, NULL, NULL, MUNIT_TEST_OPTION_NONE, NULL }
};

static const MunitSuite suite = {
  "/lexer_test", tests, NULL, 1, MUNIT_SUITE_OPTION_NONE
};

int
main(int argc, char *argv[])
{
  return munit_suite_main(&suite, NULL, argc, argv);
  return EXIT_SUCCESS;
}