mirror of
https://github.com/open-goal/jak-project.git
synced 2024-10-20 00:57:44 -04:00
85bef15364
* remove str files from inputs * removed unused verbose flag * allow some kind of conditional game building for JP extras * make `PrintBankInfo` do nothing * clang * Update gltf_mesh_extract.cpp * fix `*jak1-full-game*` in extractor * use json library to build json file * screw it red label support as well
148 lines
4.9 KiB
C++
148 lines
4.9 KiB
C++
#include <cstdio>
|
|
#include <regex>
|
|
|
|
#include "common/goos/ReplUtils.h"
|
|
#include "common/log/log.h"
|
|
#include "common/nrepl/ReplServer.h"
|
|
#include "common/util/FileUtil.h"
|
|
#include "common/versions.h"
|
|
|
|
#include "goalc/compiler/Compiler.h"
|
|
|
|
#include "third-party/CLI11.hpp"
|
|
#include "third-party/fmt/color.h"
|
|
#include "third-party/fmt/core.h"
|
|
|
|
void setup_logging() {
|
|
lg::set_file(file_util::get_file_path({"log/compiler.txt"}));
|
|
lg::set_file_level(lg::level::info);
|
|
lg::set_stdout_level(lg::level::info);
|
|
lg::set_flush_level(lg::level::info);
|
|
lg::initialize();
|
|
}
|
|
|
|
int main(int argc, char** argv) {
|
|
bool auto_listen = false;
|
|
bool auto_debug = false;
|
|
bool auto_find_user = false;
|
|
std::string cmd = "";
|
|
std::string username = "#f";
|
|
int nrepl_port = 8181;
|
|
|
|
CLI::App app{"OpenGOAL Compiler / REPL"};
|
|
app.add_option("-c,--cmd", cmd, "Specify a command to run");
|
|
app.add_option("-u,--user", username,
|
|
"Specify the username to use for your user profile in 'goal_src/user/'");
|
|
app.add_option("-p,--port", nrepl_port, "Specify the nREPL port. Defaults to 8181");
|
|
app.add_flag("--auto-lt", auto_listen,
|
|
"Attempt to automatically connect to the listener on startup");
|
|
app.add_flag("--auto-dbg", auto_debug,
|
|
"Attempt to automatically connect to the debugger on startup");
|
|
app.add_flag("--user-auto", auto_find_user,
|
|
"Attempt to automatically deduce the user, overrides '-user'");
|
|
app.validate_positionals();
|
|
CLI11_PARSE(app, argc, argv);
|
|
|
|
if (!file_util::setup_project_path(std::nullopt)) {
|
|
return 1;
|
|
}
|
|
|
|
if (auto_find_user) {
|
|
username = "#f";
|
|
std::regex allowed_chars("[0-9a-zA-Z\\-\\.\\!\\?<>]");
|
|
try {
|
|
auto text = std::make_shared<goos::FileText>(
|
|
file_util::get_file_path({"goal_src", "user", "user.txt"}), "goal_src/user/user.txt");
|
|
goos::TextStream ts(text);
|
|
ts.seek_past_whitespace_and_comments();
|
|
std::string found_username;
|
|
while (ts.text_remains()) {
|
|
auto character = ts.read();
|
|
if (!std::regex_match(std::string(1, character), allowed_chars)) {
|
|
break;
|
|
}
|
|
found_username.push_back(character);
|
|
}
|
|
if (!found_username.empty()) {
|
|
username = found_username;
|
|
}
|
|
} catch (std::exception& e) {
|
|
printf("error opening user desc file: %s\n", e.what());
|
|
}
|
|
}
|
|
|
|
setup_logging();
|
|
|
|
lg::info("OpenGOAL Compiler {}.{}", versions::GOAL_VERSION_MAJOR, versions::GOAL_VERSION_MINOR);
|
|
|
|
// Init REPL
|
|
ReplStatus status = ReplStatus::WANT_RELOAD;
|
|
std::function<bool()> shutdown_callback = [&]() { return status == ReplStatus::WANT_EXIT; };
|
|
ReplServer repl_server(shutdown_callback, nrepl_port);
|
|
bool repl_server_ok = repl_server.init_server();
|
|
std::thread nrepl_thread;
|
|
// the compiler may throw an exception if it fails to load its standard library.
|
|
try {
|
|
std::unique_ptr<Compiler> compiler;
|
|
std::mutex compiler_mutex;
|
|
// if a command is provided on the command line, no REPL just run the compiler on it
|
|
if (!cmd.empty()) {
|
|
compiler = std::make_unique<Compiler>();
|
|
compiler->run_front_end_on_string(cmd);
|
|
return 0;
|
|
}
|
|
// Start nREPL Server
|
|
if (repl_server_ok) {
|
|
nrepl_thread = std::thread([&]() {
|
|
while (!shutdown_callback()) {
|
|
auto resp = repl_server.get_msg();
|
|
if (resp) {
|
|
std::lock_guard<std::mutex> lock(compiler_mutex);
|
|
status = compiler->handle_repl_string(resp.value());
|
|
// Print out the prompt, just for better UX
|
|
compiler->print_to_repl(compiler->get_prompt());
|
|
}
|
|
std::this_thread::sleep_for(std::chrono::microseconds(50000));
|
|
}
|
|
});
|
|
}
|
|
// Run automatic forms if applicable
|
|
if (auto_debug || auto_listen) {
|
|
std::lock_guard<std::mutex> lock(compiler_mutex);
|
|
status = compiler->handle_repl_string("(lt)");
|
|
}
|
|
if (auto_debug) {
|
|
std::lock_guard<std::mutex> lock(compiler_mutex);
|
|
status = compiler->handle_repl_string("(dbg) (:cont)");
|
|
}
|
|
// Poll Terminal
|
|
while (status != ReplStatus::WANT_EXIT) {
|
|
if (status == ReplStatus::WANT_RELOAD) {
|
|
fmt::print("Reloading compiler...\n");
|
|
std::lock_guard<std::mutex> lock(compiler_mutex);
|
|
if (compiler) {
|
|
compiler->save_repl_history();
|
|
}
|
|
compiler = std::make_unique<Compiler>(username, std::make_unique<ReplWrapper>());
|
|
status = ReplStatus::OK;
|
|
}
|
|
std::string input_from_stdin = compiler->get_repl_input();
|
|
if (!input_from_stdin.empty()) {
|
|
// lock, while we compile
|
|
std::lock_guard<std::mutex> lock(compiler_mutex);
|
|
status = compiler->handle_repl_string(input_from_stdin);
|
|
}
|
|
}
|
|
} catch (std::exception& e) {
|
|
fmt::print(stderr, "Compiler Fatal Error: {}\n", e.what());
|
|
status = ReplStatus::WANT_EXIT;
|
|
}
|
|
|
|
// Cleanup
|
|
if (repl_server_ok) {
|
|
repl_server.shutdown_server();
|
|
nrepl_thread.join();
|
|
}
|
|
return 0;
|
|
}
|