72 lines
1.5 KiB
C++
72 lines
1.5 KiB
C++
#include <iostream>
|
|
#include <fstream>
|
|
#include <string>
|
|
#include <optional>
|
|
|
|
#include "lexer.hpp"
|
|
#include "token.hpp"
|
|
#include "ast.hpp"
|
|
#include "parser.hpp"
|
|
|
|
namespace dsp {
|
|
|
|
bool had_error = false;
|
|
|
|
}
|
|
|
|
static void run(std::string source)
|
|
{
|
|
using namespace dsp;
|
|
|
|
std::string_view buf(source);
|
|
lexer_t lexer(buf);
|
|
std::vector<token_t> tokens = lexer.lex_tokens();
|
|
parser_t parser(tokens);
|
|
std::optional<stmt_t *> stmt_o = parser.instruction();
|
|
if (stmt_o) {
|
|
dsp::ast_printer_t p(std::cout);
|
|
(*stmt_o)->accept(&p);
|
|
}
|
|
std::cout << std::endl << std::flush;
|
|
}
|
|
|
|
static void run_prompt()
|
|
{
|
|
constexpr auto prompt = "> ";
|
|
std::string line;
|
|
std::cout << prompt << std::flush;
|
|
while (std::getline(std::cin, line)) {
|
|
run(line);
|
|
std::cout << prompt << std::flush;
|
|
}
|
|
}
|
|
|
|
static int run_file(char const * const filename)
|
|
{
|
|
std::ifstream is {filename, std::ios::binary | std::ios::ate};
|
|
if (!is.is_open()) {
|
|
std::cerr << "failed to open " << filename << std::endl;
|
|
return -1;
|
|
}
|
|
const std::streampos size = is.tellg();
|
|
std::string buf(size, '\0');
|
|
is.seekg(0);
|
|
if (!is.read(&buf[0], size)) {
|
|
std::cerr << "read failed" << std::endl;
|
|
return -1;
|
|
}
|
|
run(buf);
|
|
return dsp::had_error;
|
|
}
|
|
|
|
int main(const int argc, char const * const argv[])
|
|
{
|
|
switch (argc) {
|
|
case 1: run_prompt(); return dsp::had_error;
|
|
case 2: return run_file(argv[1]);
|
|
default:
|
|
std::cerr << "Usage: " << argv[0] << " [filename]" << std::endl;
|
|
return -1;
|
|
}
|
|
}
|