#include <iostream>
#include "Lexer.h"
#include "Parser.h"

int main(int argc, char *argv[]) {
    if (argc != 2) {
        std::cout << "Usage: " << argv[0] << " filename" << std::endl;
        exit(1);
    }

    std::string filename = argv[1];
    std::ifstream fin(filename, std::ios::in);
    // test lexer
    if (!fin.is_open()) {
        std::cout << "Error: File " << filename << " not found" << std::endl;
        exit(2);
    }

    std::string input;
    while (!fin.eof()) {
        std::string line;
        std::getline(fin, line);
        line += "\n";
        input += line;
        line.clear();
        if (fin.eof())
        {
            break;
        }
    }
	input += '\0';
    Lexer *lexer = new Lexer(input);
    std::vector<Token> tokens = lexer->LexicalAnalyisis();
    for (auto token : tokens) {
        std::cout << token.GetStringValue();
        if (token._type != TOKEN_NEWLINE) {
            std::cout << std::endl;
        }

    }

    Parser *parser = new Parser(tokens);
    parser->parse();
    parser->printTree(nullptr);

    fin.close();

    delete lexer;
    delete parser;

    return 0;
}