Add empty compiler

Author: AlexDenisov

CMakeLists.txt

@@ -2,5 +2,13 @@ cmake_minimum_required(VERSION 3.27)
 project(LightStorm)
 
 set(CMAKE_CXX_STANDARD 20)
+set(CMAKE_CXX_STANDARD_REQUIRED ON)
+set(CMAKE_CXX_EXTENSIONS OFF)
 
-add_executable(LightStorm main.cpp)
+include(cmake/mruby.cmake)
+
+find_package(LLVM CONFIG REQUIRED)
+message("Using LLVM_CONFIG: ${LLVM_CONFIG}")
+
+add_subdirectory(lib)
+add_subdirectory(tools)

cmake/mruby.cmake

@@ -0,0 +1,21 @@
+include(ExternalProject)
+set (MRUBY_DIR "${CMAKE_SOURCE_DIR}/third_party/mruby")
+set (MRUBY_BINARY "${MRUBY_DIR}/bin/mruby")
+set (MRUBY_STATIC "${MRUBY_DIR}/build/host/lib/libmruby.a")
+
+ExternalProject_Add(mruby
+  SOURCE_DIR ${MRUBY_DIR}
+  CONFIGURE_COMMAND ""
+  BUILD_COMMAND ${CMAKE_COMMAND} -E env rake all --verbose
+  BUILD_IN_SOURCE ON
+  INSTALL_COMMAND ""
+  BUILD_BYPRODUCTS ${MRUBY_STATIC} ${MRUBY_BINARY}
+)
+
+add_executable(mruby_binary IMPORTED GLOBAL)
+set_property(TARGET mruby_binary PROPERTY IMPORTED_LOCATION ${MRUBY_BINARY})
+add_dependencies(mruby_binary mruby)
+
+add_library(mruby_static STATIC IMPORTED GLOBAL)
+set_property(TARGET mruby_static PROPERTY IMPORTED_LOCATION ${MRUBY_STATIC})
+add_dependencies(mruby_static mruby)

include/lightstorm/compiler/compiler.h

@@ -0,0 +1,14 @@
+#pragma once
+
+#include <filesystem>
+
+namespace lightstorm {
+
+class Compiler {
+public:
+  void compileSourceFile(const std::filesystem::path &file_path);
+
+private:
+};
+
+} // namespace lightstorm

lib/CMakeLists.txt

@@ -0,0 +1 @@
+add_subdirectory(compiler)

lib/compiler/CMakeLists.txt

@@ -0,0 +1,10 @@
+add_library(lightstorm_compiler
+  compiler.cpp
+)
+target_include_directories(lightstorm_compiler
+  PRIVATE
+  ${CMAKE_SOURCE_DIR}/third_party/mruby/include
+  ${CMAKE_SOURCE_DIR}/include
+)
+target_link_libraries(lightstorm_compiler PRIVATE mruby_static)
+target_compile_options(lightstorm_compiler PRIVATE -fno-rtti -fno-exceptions)

lib/compiler/compiler.cpp

@@ -0,0 +1,55 @@
+#include "lightstorm/compiler/compiler.h"
+#include <cassert>
+#include <iostream>
+#include <mruby.h>
+#include <mruby/compile.h>
+
+using namespace lightstorm;
+
+extern "C" {
+void mrb_codedump_all(mrb_state *, struct RProc *);
+}
+
+void Compiler::compileSourceFile(const std::filesystem::path &file_path) {
+  assert(exists(file_path) && "Cannot compile file");
+  mrb_state *mrb = mrb_open();
+  assert(mrb && "Out of memory?");
+  mrbc_context *mrbc = mrbc_context_new(mrb);
+  assert(mrbc && "Out of memory?");
+  mrbc_filename(mrb, mrbc, file_path.c_str());
+  mrbc->capture_errors = TRUE;
+  mrbc->no_optimize = TRUE;
+
+  FILE *f = fopen(file_path.c_str(), "r");
+  assert(f && "No file?");
+
+  struct mrb_parser_state *p = mrb_parse_file(mrb, f, mrbc);
+
+  if (!p) {
+    /* only occur when memory ran out */
+    std::cerr << "Failed to create parser state (out of memory)\n";
+    fclose(f);
+    mrbc_context_free(mrb, mrbc);
+    mrb_close(mrb);
+    return;
+  }
+
+  if (0 < p->nerr) {
+    /* parse error */
+    std::cerr << file_path.c_str() << ":" << p->error_buffer[0].lineno << ": "
+              << p->error_buffer[0].message << "\n";
+    fclose(f);
+    mrb_parser_free(p);
+    mrbc_context_free(mrb, mrbc);
+    mrb_close(mrb);
+    return;
+  }
+
+  struct RProc *proc = mrb_generate_code(mrb, p);
+  mrb_codedump_all(mrb, proc);
+
+  fclose(f);
+  mrb_parser_free(p);
+  mrbc_context_free(mrb, mrbc);
+  mrb_close(mrb);
+}

main.cpp

@@ -0,0 +1 @@
+add_subdirectory(lightstorm)

tools/CMakeLists.txt

@@ -0,0 +1,8 @@
+add_executable(lightstorm lightstorm.cpp)
+target_link_libraries(lightstorm PRIVATE lightstorm_compiler LLVMOption)
+target_compile_options(lightstorm PRIVATE -fno-rtti -fno-exceptions)
+target_include_directories(lightstorm
+  PRIVATE
+  ${CMAKE_SOURCE_DIR}/include
+  ${LLVM_INCLUDE_DIRS}
+)

tools/lightstorm/CMakeLists.txt

@@ -0,0 +1,32 @@
+#include <filesystem>
+#include <llvm/Support/CommandLine.h>
+#include <llvm/Support/FileSystem.h>
+
+#include "lightstorm/compiler/compiler.h"
+
+llvm::cl::OptionCategory LightstormCategory("lightstorm");
+
+llvm::cl::opt<std::string> Input(llvm::cl::Positional, llvm::cl::Required,
+                                 llvm::cl::desc("Input file"), llvm::cl::cat(LightstormCategory));
+
+int main(int argc, char **argv) {
+  llvm::llvm_shutdown_obj shutdownGuard;
+
+  llvm::cl::HideUnrelatedOptions(LightstormCategory);
+  llvm::cl::ParseCommandLineOptions(argc, argv);
+
+  std::error_code ec;
+  if (!std::filesystem::exists(Input.getValue(), ec)) {
+    llvm::errs() << "Input file doesn't exist: ";
+    if (ec) {
+      llvm::errs() << ec.message();
+    }
+    llvm::errs() << "\n";
+    return 1;
+  }
+
+  lightstorm::Compiler compiler;
+  compiler.compileSourceFile(Input.getValue());
+
+  return 0;
+}