feat: add debug option to cli

Cargo.toml

@@ -9,6 +9,7 @@ version = "0.1.0"
 lalrpop = "0.19.7"
 
 [dependencies]
+clap = { version = "4.1.8", features = ["derive"] }
 im = "15.1.0"
 lalrpop-util = {version = "0.19.7", features = ["lexer"]}
 libc = "0.2.139"

src/cli.rs

@@ -0,0 +1,8 @@
+use clap::Parser;
+
+#[derive(Parser, Debug, Default)]
+#[command(author, version, about, long_about = None)]
+pub struct Options {
+    #[arg(short, long)]
+    pub debug: bool,
+}

src/codegen.rs

@@ -7,7 +7,10 @@ use llvm_sys::{
     LLVMDiagnosticSeverity::{LLVMDSError, LLVMDSNote, LLVMDSRemark, LLVMDSWarning},
 };
 
-use crate::macros::{cstr, llvm_wrapper};
+use crate::{
+    cli::Options,
+    macros::{cstr, llvm_wrapper},
+};
 
 pub use llvm_sys::{core::*, execution_engine::*, prelude::*};
 
@@ -30,6 +33,7 @@ pub struct Codegen {
     pub environment: compile::Environment,
     pub global_environment: *mut GlobalEnvironment,
     pub global_sym: LLVMValueRef,
+    pub options: Box<crate::cli::Options>,
 }
 
 impl Drop for Codegen {
@@ -60,14 +64,16 @@ impl Codegen {
                 global_environment,
                 current_fn: std::ptr::null_mut(),
                 global_sym: std::ptr::null_mut(),
+                options: Box::default(),
             }
         }
     }
 
-    pub fn install_error_handling(self) -> Self {
+    pub fn install_error_handling(mut self) -> Self {
         unsafe {
             // enable diagnostic messages
-            let diagnostic_context = LLVMContextGetDiagnosticContext(self.context);
+            let diagnostic_context = self.options.as_mut() as *mut _ as *mut c_void;
+            let handle_diagnostic = handle_diagnostic as extern "C" fn(_, *mut c_void);
             let handle_fn: Option<extern "C" fn(_, _)> = Some(handle_diagnostic);
             LLVMContextSetDiagnosticHandler(self.context, handle_fn, diagnostic_context);
 
@@ -112,8 +118,14 @@ impl Codegen {
 }
 
 #[allow(clippy::not_unsafe_ptr_arg_deref)]
-pub extern "C" fn handle_diagnostic(info: LLVMDiagnosticInfoRef, _context: *mut c_void) {
+pub extern "C" fn handle_diagnostic(info: LLVMDiagnosticInfoRef, context: *mut libc::c_void) {
+    let options = context as *mut Options;
+
     unsafe {
+        if !(*options).debug {
+            return;
+        }
+
         let kind = match LLVMGetDiagInfoSeverity(info) {
             LLVMDSError => "error",
             LLVMDSWarning => "warning",

src/lib.rs

@@ -1,5 +1,6 @@
 #![feature(box_syntax, box_patterns)]
 #![feature(mem_copy_fn)]
+pub mod cli;
 pub mod closure;
 pub mod codegen;
 pub mod macros;

src/main.rs

@@ -1,3 +1,8 @@
+use clap::Parser;
+use soft::cli::Options;
+
 fn main() {
-    soft::repl::run()
+    let options = Options::parse();
+
+    soft::repl::run(&options);
 }

src/repl.rs

@@ -6,26 +6,31 @@ use crate::{
     runtime::ValueRef,
 };
 
-pub fn run() {
+pub fn run(options: &crate::cli::Options) {
     Codegen::install_execution_targets();
 
-    let mut rl = DefaultEditor::new().expect("cannot create a repl");
+    let mut editor = DefaultEditor::new().expect("cannot create a repl");
 
     let global_environment = Box::leak(Box::default());
 
     loop {
-        let readline = rl.readline("> ");
+        let readline = editor.readline("> ");
 
         match readline {
             Ok(line) => {
-                rl.add_history_entry(line.as_str())
+                editor
+                    .add_history_entry(line.as_str())
                     .expect("cannot add to the history");
 
                 let mut codegen = Codegen::new(global_environment)
                     .install_error_handling()
                     .install_primitives()
                     .install_global_environment();
 
+                if options.debug {
+                    codegen.dump_module();
+                }
+
                 let engine = ExecutionEngine::try_new(codegen.module)
                     .unwrap()
                     .install_primitive_symbols(&codegen.environment)
@@ -79,7 +84,6 @@ fn eval_line(
         .compile_main(converted)
         .map_err(|err| format!("[error] Could not compile expression: {err:?}"))?;
 
-    codegen.dump_module();
     codegen.verify_module().unwrap_or_else(|error| {
         for line in error.split('\n') {
             println!("[error*] {}", line);