Riyyi
2 years ago
4 changed files with 141 additions and 1 deletions
@ -0,0 +1,133 @@
|
||||
/*
|
||||
* Copyright (C) 2023 Riyyi |
||||
* |
||||
* SPDX-License-Identifier: MIT |
||||
*/ |
||||
|
||||
#include <csignal> // std::signal |
||||
#include <cstdlib> // std::exit |
||||
#include <string> |
||||
#include <string_view> |
||||
|
||||
#include "ruc/argparser.h" |
||||
#include "ruc/format/color.h" |
||||
|
||||
#include "ast.h" |
||||
#include "environment.h" |
||||
#include "error.h" |
||||
#include "eval.h" |
||||
#include "forward.h" |
||||
#include "lexer.h" |
||||
#include "printer.h" |
||||
#include "reader.h" |
||||
#include "readline.h" |
||||
#include "settings.h" |
||||
|
||||
#if 1 |
||||
namespace blaze { |
||||
|
||||
static EnvironmentPtr s_outer_env = Environment::create(); |
||||
|
||||
static auto cleanup(int signal) -> void |
||||
{ |
||||
print("\033[0m\n"); |
||||
std::exit(signal); |
||||
} |
||||
|
||||
auto read(std::string_view input) -> ASTNodePtr |
||||
{ |
||||
Lexer lexer(input); |
||||
lexer.tokenize(); |
||||
if (Settings::the().get("dump-lexer") == "1") { |
||||
lexer.dump(); |
||||
} |
||||
|
||||
Reader reader(std::move(lexer.tokens())); |
||||
reader.read(); |
||||
if (Settings::the().get("dump-reader") == "1") { |
||||
reader.dump(); |
||||
} |
||||
|
||||
return reader.node(); |
||||
} |
||||
|
||||
auto eval(ASTNodePtr ast, EnvironmentPtr env) -> ASTNodePtr |
||||
{ |
||||
Eval eval(ast, env); |
||||
eval.eval(); |
||||
|
||||
return eval.ast(); |
||||
} |
||||
|
||||
static auto print(ASTNodePtr exp) -> std::string |
||||
{ |
||||
Printer printer; |
||||
|
||||
return printer.print(exp, true); |
||||
} |
||||
|
||||
static auto rep(std::string_view input, EnvironmentPtr env) -> std::string |
||||
{ |
||||
Error::the().clearErrors(); |
||||
Error::the().setInput(input); |
||||
|
||||
return print(eval(read(input), env)); |
||||
} |
||||
|
||||
static std::string_view lambdaTable[] = { |
||||
"(def! not (fn* (cond) (if cond false true)))", |
||||
}; |
||||
|
||||
static auto installLambdas(EnvironmentPtr env) -> void |
||||
{ |
||||
for (auto function : lambdaTable) { |
||||
rep(function, env); |
||||
} |
||||
} |
||||
|
||||
} // namespace blaze
|
||||
|
||||
auto main(int argc, char* argv[]) -> int |
||||
{ |
||||
bool dump_lexer = false; |
||||
bool dump_reader = false; |
||||
bool pretty_print = false; |
||||
std::string_view history_path = "~/.blaze-history"; |
||||
|
||||
// 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.addOption(history_path, 'h', "history-path", nullptr, nullptr, nullptr, ruc::ArgParser::Required::Yes); |
||||
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, blaze::cleanup); |
||||
std::signal(SIGTERM, blaze::cleanup); |
||||
|
||||
installFunctions(blaze::s_outer_env); |
||||
installLambdas(blaze::s_outer_env); |
||||
|
||||
blaze::Readline readline(pretty_print, history_path); |
||||
|
||||
std::string input; |
||||
while (readline.get(input)) { |
||||
if (pretty_print) { |
||||
print("\033[0m"); |
||||
} |
||||
print("{}\n", rep(input, blaze::s_outer_env)); |
||||
} |
||||
|
||||
if (pretty_print) { |
||||
print("\033[0m"); |
||||
} |
||||
|
||||
return 0; |
||||
} |
||||
#endif |
Loading…
Reference in new issue