#include <argparse/argparse.hpp>
#include <fstream>
#include <iostream>
#include "IR/IR.h"
#include "semantic/semantic.h"

int main(int argc, char **argv) {
  argparse::ArgumentParser program("zmxcc");

  program.add_argument("input").help("input file path").required();

  program.add_argument("-o", "--output").help("output file path").nargs(1).required();

  try {
    program.parse_args(argc, argv);
  } catch (const std::runtime_error &err) {
    std::cerr << err.what() << std::endl;
    std::cerr << program;
    return 1;
  }

  auto input_file = program.get<std::string>("input");
  auto output_file = program.get<std::string>("output");
  std::ifstream fin(input_file);
  std::ofstream fout(output_file);
  std::shared_ptr<Program_ASTNode> ast;
  try {
    SemanticCheck(fin, ast);
    auto IR = BuildIR(ast);
    IR->RecursivePrint(fout);
  } catch (const SemanticError &err) {
    std::cout << err.what() << std::endl;
    return err.GetErrorCode();
  } catch (const std::exception &err) {
    std::cout << "Unknown error: " << err.what() << std::endl;
    return 254;
  } catch (...) {
    std::cout << "Unknown error without message" << std::endl;
    return 255;
  }
  return 0;
}