main.cpp

/**
 * main.cpp
 * author: Bao Le
 */

#include <iostream>
#include <fstream>
#include <sstream>
#include <string>
#include <stdexcept>
#include "ast.hpp"
#include "lexer.hpp"
#include "parser.hpp"
#include "semantic.hpp"
#include "codegen.hpp"

std::string read_file(const std::string& path) {
    std::ifstream file(path);
    if (!file.is_open()) {
        throw std::runtime_error("Failed to open file: " + path);
    }
    // read the file into a string
    std::stringstream ss;
    ss << file.rdbuf();
    return ss.str();
}

int main(int argc, char* argv[]) {
    std::string source;
    if (argc > 1) {
        source = read_file(argv[1]);
    }
    else {
        std::ostringstream ss;
        ss << std::cin.rdbuf();
        source = ss.str();
    }
    // Lexer
    Lexer lexer(source);
    auto tokens = lexer.tokenize();

    // Parser
    Parser parser(std::move(tokens));
    auto ast = parser.parse_program();

    // Semantic Analysis
    SemanticAnalyzer analyzer(ast.get());
    analyzer.analyze();

    // Code generation
    CodeGenerator codegen(ast.get());
    codegen.generate("output.s");   // writes assembly to a file
}