/* Copyright (C) 2024 Aryadev Chavali

 * You may distribute and modify this code under the terms of the
 * GPLv2 license.  You should have received a copy of the GPLv2
 * license with this file.  If not, please write to:
 * aryadev@aryadevchavali.com.

 * Created: 2024-04-14
 * Author: Aryadev Chavali
 * Description: Entrypoint for assembly program
 */

#include <algorithm>
#include <cstdio>
#include <iostream>
#include <optional>
#include <string>
#include <tuple>
#include <vector>

extern "C"
{
#include <lib/inst.h>
}

#include "./base.hpp"
#include "./lexer.hpp"
#include "./preprocesser.hpp"

using std::cout, std::cerr, std::endl;
using std::pair, std::string, std::string_view, std::vector;

void usage(const char *program_name, FILE *fp)
{
  fprintf(fp,
          "Usage: %s FILE OUT-FILE\n"
          "\tFILE: Source code to compile\n"
          "\tOUT-FILE: Name of file to store bytecode\n",
          program_name);
}

int main(int argc, const char *argv[])
{
  if (argc == 1 || argc > 3)
  {
    usage(argv[0], stderr);
    return -1;
  }
  int ret                 = 0;
  const char *source_name = argv[1];
  const char *out_name    = argv[2];
  (void)out_name;

#if VERBOSE >= 1
  printf("[%sASSEMBLER%s]: Assembling `%s` to `%s`\n", TERM_YELLOW, TERM_RESET,
         source_name, out_name);
#endif

  auto file_source = read_file(source_name);

#if VERBOSE >= 1
  printf("[%sASSEMBLER%s]: `%s` -> %lu bytes\n", TERM_YELLOW, TERM_RESET,
         source_name, file_source.has_value() ? file_source.value().size() : 0);
#endif

  string source_str;
  string_view original;
  string_view src;
  vector<token_t *> tokens, preprocessed_tokens;
  lerr_t lerr;
  pp_err_t pp_err;

  // Highest scoped variable cut off point

  if (file_source.has_value())
    source_str = file_source.value();
  else
  {
    cerr << "ERROR: file `" << source_name << "` does not exist!" << endl;
    ret = -1;
    goto end;
  }
  original = string_view{source_str};
  src      = string_view{source_str};
  lerr     = tokenise_buffer(src, tokens);

  if (lerr.type != lerr_type_t::OK)
  {
    cerr << source_name << ":" << lerr << endl;
    ret = 255 - static_cast<int>(lerr.type);
    goto end;
  }
  else
  {

#if VERBOSE >= 1
    printf("[%sLEXER%s]: %lu bytes -> %lu tokens\n", TERM_GREEN, TERM_RESET,
           source_str.size(), tokens.size());
#endif

#if VERBOSE == 2
    printf("[%sLEXER%s]: Tokens "
           "parsed:\n----------------------------------------------------------"
           "----------------------\n",
           TERM_GREEN, TERM_RESET);
    for (auto token : tokens)
      cout << "\t" << *token << endl;
    printf("-------------------------------------------------------------"
           "-------------------\n");
#endif
  }

  // preprocessing
  pp_err = preprocesser(tokens, preprocessed_tokens);
  if (pp_err.type != pp_err_type_t::OK)
  {
    cerr << source_name << ":" << pp_err.reference->line << ":"
         << pp_err.reference->column << ":" << pp_err << endl;
    ret = 255 - static_cast<int>(pp_err.type);
    goto end;
  }
  else
  {

#if VERBOSE >= 1
    printf("[%sPREPROCESSOR%s]: %lu tokens -> %lu tokens\n", TERM_GREEN,
           TERM_RESET, tokens.size(), preprocessed_tokens.size());
#endif
#if VERBOSE == 2
    printf("[%sPREPROCESSOR%s]: Processed tokens: "
           "\n-----------------------------------------------------------------"
           "---------------\n",
           TERM_GREEN, TERM_RESET);
    for (auto token : preprocessed_tokens)
      cout << "\t" << *token << endl;
    printf("-------------------------------------------------------------"
           "-------------------\n");
#endif
  }

end:
  for (auto token : tokens)
    delete token;
  for (auto token : preprocessed_tokens)
    delete token;

  return ret;
}