#include "ast.hpp"

#include <filesystem>
#include <iostream>
#include <string>

using std::string;
using std::operator""s;

struct Config
{
    string exe_name; // compiler exe name
    std::filesystem::path input_file;
    std::filesystem::path output_file;

    bool emitast{false};

    Config(int argc, char **argv) : argc(argc), argv(argv)
    {
        parse_cmd_line();
        check();
    }

private:
    int argc{-1};
    char **argv{nullptr};

    void parse_cmd_line();
    void check();
    // print helper infomation and exit
    void print_help() const;
    void print_err(const string &msg) const;
};

int main(int argc, char **argv)
{
    Config config(argc, argv);
    auto syntax_tree = parse(config.input_file.c_str());
    auto ast = AST(syntax_tree);
    ASTPrinter printer;
    ast.run_visitor(printer);

    return 0;
}

void Config::parse_cmd_line()
{
    exe_name = argv[0];
    for (int i = 1; i < argc; ++i)
    {
        if (argv[i] == "-h"s || argv[i] == "--help"s)
        {
            print_help();
        }
        else if (argv[i] == "-o"s)
        {
            if (output_file.empty() && i + 1 < argc)
            {
                output_file = argv[i + 1];
                i += 1;
            }
            else
            {
                print_err("bad output file");
            }
        }
        else if (argv[i] == "-emit-ast"s)
        {
            emitast = true;
        }
        else
        {
            if (input_file.empty())
            {
                input_file = argv[i];
            }
            else
            {
                string err =
                    "unrecognized command-line option \'"s + argv[i] + "\'"s;
                print_err(err);
            }
        }
    }
}

void Config::check()
{
    if (input_file.empty())
    {
        print_err("no input file");
    }
    if (input_file.extension() != ".cminus")
    {
        print_err("file format not recognized");
    }
    if (output_file.empty())
    {
        output_file = input_file.stem();
    }
}

void Config::print_help() const
{
    std::cout << "Usage: " << exe_name
              << " [-h|--help] [-o <target-file>] [-mem2reg] [-emit-llvm] [-S] "
                 "<input-file>"
              << std::endl;
    exit(0);
}

void Config::print_err(const string &msg) const
{
    std::cout << exe_name << ": " << msg << std::endl;
    exit(-1);
}