You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
2.3 KiB
114 lines
2.3 KiB
2 years ago
|
#include <csignal> // std::signal
|
||
|
|
#include <cstdlib> // std::exit
|
||
|
|
#include <iostream> // std::cin
|
||
|
|
#include <string> // std::getline
|
||
|
|
#include <string_view>
|
||
|
|
|
||
|
|
#include "environment.h"
|
||
|
|
#include "eval.h"
|
||
|
|
#include "ruc/argparser.h"
|
||
|
|
#include "ruc/format/color.h"
|
||
|
|
|
||
|
|
#include "ast.h"
|
||
|
|
#include "error.h"
|
||
|
|
#include "lexer.h"
|
||
|
|
#include "printer.h"
|
||
|
|
#include "reader.h"
|
||
|
|
#include "settings.h"
|
||
|
|
|
||
|
|
#if 1
|
||
|
|
auto read(std::string_view input) -> blaze::ASTNode*
|
||
|
|
{
|
||
|
|
blaze::Lexer lexer(input);
|
||
|
|
lexer.tokenize();
|
||
|
|
if (blaze::Settings::the().get("dump-lexer") == "1") {
|
||
|
|
lexer.dump();
|
||
|
|
}
|
||
|
|
|
||
|
|
blaze::Reader reader(std::move(lexer.tokens()));
|
||
|
|
reader.read();
|
||
|
|
if (blaze::Settings::the().get("dump-reader") == "1") {
|
||
|
|
reader.dump();
|
||
|
|
}
|
||
|
|
|
||
|
|
return reader.node();
|
||
|
|
}
|
||
|
|
|
||
|
|
auto eval(blaze::ASTNode* ast) -> blaze::ASTNode*
|
||
|
|
{
|
||
|
|
blaze::GlobalEnvironment env;
|
||
|
|
blaze::Eval eval(ast, &env);
|
||
|
|
eval.eval();
|
||
|
|
return eval.ast();
|
||
|
|
}
|
||
|
|
|
||
|
|
auto print(blaze::ASTNode* exp) -> void
|
||
|
|
{
|
||
|
|
blaze::Printer printer(exp);
|
||
|
|
printer.dump();
|
||
|
|
}
|
||
|
|
|
||
|
|
auto rep(std::string_view input) -> void
|
||
|
|
{
|
||
|
|
blaze::Error::the().clearErrors();
|
||
|
|
blaze::Error::the().setInput(input);
|
||
|
|
|
||
|
|
print(eval(read(input)));
|
||
|
|
}
|
||
|
|
|
||
|
|
static auto cleanup(int signal) -> void
|
||
|
|
{
|
||
|
|
print("\033[0m");
|
||
|
|
std::exit(signal);
|
||
|
|
}
|
||
|
|
|
||
|
|
auto main(int argc, char* argv[]) -> int
|
||
|
|
{
|
||
|
|
bool dump_lexer = false;
|
||
|
|
bool dump_reader = false;
|
||
|
|
bool pretty_print = false;
|
||
|
|
|
||
|
|
// CLI arguments
|
||
|
|
ruc::ArgParser arg_parser;
|
||
|
|
arg_parser.addOption(dump_lexer, 'l', "dump-lexer", nullptr, nullptr);
|
||
|
|
arg_parser.addOption(dump_reader, 'r', "dump-reader", nullptr, nullptr);
|
||
|
|
arg_parser.addOption(pretty_print, 'c', "color", nullptr, nullptr);
|
||
|
|
arg_parser.parse(argc, argv);
|
||
|
|
|
||
|
|
// Set settings
|
||
|
|
blaze::Settings::the().set("dump-lexer", dump_lexer ? "1" : "0");
|
||
|
|
blaze::Settings::the().set("dump-reader", dump_reader ? "1" : "0");
|
||
|
|
blaze::Settings::the().set("pretty-print", pretty_print ? "1" : "0");
|
||
|
|
|
||
|
|
// Signal callbacks
|
||
|
|
std::signal(SIGINT, cleanup);
|
||
|
|
std::signal(SIGTERM, cleanup);
|
||
|
|
|
||
|
|
while (true) {
|
||
|
|
if (!pretty_print) {
|
||
|
|
print("user> ");
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
print(fg(ruc::format::TerminalColor::Blue), "user>");
|
||
|
|
print(" \033[1m");
|
||
|
|
}
|
||
|
|
std::string line;
|
||
|
|
std::getline(std::cin, line);
|
||
|
|
if (pretty_print) {
|
||
|
|
print("\033[0m");
|
||
|
|
}
|
||
|
|
|
||
|
|
// Exit with Ctrl-D
|
||
|
|
if (std::cin.eof() || std::cin.fail()) {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
rep(line);
|
||
|
|
}
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
// - Add AST node printing support to ruc::format
|