#include "gtest/gtest.h"

#include "bolt/CST.hpp"
#include "bolt/Diagnostics.hpp"
#include "bolt/DiagnosticEngine.hpp"
#include "bolt/Scanner.hpp"
#include "bolt/Parser.hpp"
#include "bolt/Checker.hpp"

using namespace bolt;

auto checkSourceFile(std::string Input) {
  DiagnosticStore DS;
  TextFile T { "#<anonymous>", Input };
  VectorStream<std::string, Char> Chars { Input, EOF };
  Scanner S(T, Chars);
  Punctuator PT(S);
  Parser P(T, PT, DS);
  LanguageConfig Config;
  auto SF = P.parseSourceFile();
  SF->setParents();
  Checker C(Config, DS);
  C.check(SF);
  return std::make_tuple(SF, C, DS);
}

auto checkExpression(std::string Input) {
  auto [SF, C, DS] = checkSourceFile(Input);
  return std::make_tuple(
    static_cast<ExpressionStatement*>(SF->Elements[0])->Expression,
    C,
    DS
  );
}

TEST(CheckerTest, InfersIntFromIntegerLiteral) {
  auto [Expr, Checker, DS] = checkExpression("1");
  ASSERT_EQ(DS.countDiagnostics(), 0);
  ASSERT_EQ(Checker.getType(Expr), Checker.getIntType());
}

TEST(CheckerTest, TestIllegalTypingVariable) {
  auto [SF, C, DS] = checkSourceFile("let a: Int = \"foo\"");
  ASSERT_EQ(DS.countDiagnostics(), 1);
  auto D1 = DS.Diagnostics[0];
  ASSERT_EQ(D1->getKind(), DiagnosticKind::UnificationError);
  auto Diag = static_cast<UnificationErrorDiagnostic*>(D1);
  // TODO these types have to be sorted first
  ASSERT_EQ(Diag->getLeft(), C.getIntType());
  ASSERT_EQ(Diag->getRight(), C.getStringType());
}