#include "triton/lang/error.h"

#include "triton/lang/ast.h"
#include "triton/lang/token.h"

#include <cstdarg>
#include <cstdio>
#include <cstring>
#include <string>


#define ANSI_COLOR_RED     "\x1b[31m"
#define ANSI_COLOR_GREEN   "\x1b[32m"
#define ANSI_COLOR_YELLOW  "\x1b[33m"
#define ANSI_COLOR_BLUE    "\x1b[34m"
#define ANSI_COLOR_MAGENTA "\x1b[35m"
#define ANSI_COLOR_CYAN    "\x1b[36m"
#define ANSI_COLOR_RESET   "\x1b[0m"


void Error(const char* format, ...) {
  fprintf(stderr,
          ANSI_COLOR_RED "error: " ANSI_COLOR_RESET);

  va_list args;
  va_start(args, format);
  vfprintf(stderr, format, args);
  va_end(args);

  fprintf(stderr, "\n");

  exit(-1);
}


[[noreturn]]
static void VError(const SourceLocation& loc,
                   const char* format,
                   va_list args) {
  const char* filename = nullptr;
  if(loc.filename_)
    filename = loc.filename_->c_str();
  fprintf(stderr,
          "%s:%d:%d: " ANSI_COLOR_RED "error: " ANSI_COLOR_RESET,
          filename,
          loc.line_,
          loc.column_);
  vfprintf(stderr, format, args);
  fprintf(stderr, "\n    ");

  bool sawNoSpace = false;
  int nspaces = 0;
  for (auto p = loc.lineBegin_; *p != '\n' && *p != 0; p++) {
    if (!sawNoSpace && (*p == ' ' || *p == '\t')) {
      ++nspaces;
    } else {
      sawNoSpace = true;
      fputc(*p, stderr);
    }
  }

  fprintf(stderr, "\n    ");
  for (unsigned i = 1; i + nspaces < loc.column_; ++i)
    fputc(' ', stderr);
  fprintf(stderr, "^\n");
  exit(-1);
}


void Error(const SourceLocation& loc, const char* format, ...) {
  va_list args;
  va_start(args, format);
  VError(loc, format, args);
  va_end(args);
}


void Error(const Token* tok, const char* format, ...) {
  va_list args;
  va_start(args, format);
  VError(tok->loc_, format, args);
  va_end(args);
}


void Error(const Expr* expr, const char* format, ...) {
  va_list args;
  va_start(args, format);
  VError(expr->Tok()->loc_, format, args);
  va_end(args);
}