feat: added AstoIR build target

Author: Zffu

compiler/compiler_main/src/cli.rs

@@ -1,3 +1,5 @@
+use std::path::{Path, PathBuf};
+
 use clap::{Parser, Subcommand, ValueEnum};
 
 #[derive(Parser)]
@@ -34,7 +36,7 @@ pub enum CLICommand {
     #[command(visible_alias = "b", about = "Builds the given file(s)")]
     Build {
         #[arg(short = 'o')]
-        out: String,
+        out: PathBuf,
 
         #[arg(long, value_enum, default_value = "llvm")]
         platform: Platform,
@@ -46,7 +48,7 @@ pub enum CLICommand {
         linker: String,
 
         #[arg(required = true)]
-        input: Vec<String>,
+        input: Vec<PathBuf>,
     },
 
     #[command(visible_alias = "ver", about = "Displays the version")]

compiler/compiler_main/src/cmds/build.rs

@@ -0,0 +1,20 @@
+use std::{fs, path::PathBuf};
+
+use ast_parser::parse_ast_ctx;
+use astoir::run_astoir_mir;
+use lexer::lexer::lexer_parse_file;
+
+use crate::quietlyquit_if_errors;
+
+pub fn build_mir(path: String, out: PathBuf) {
+    let lexer = lexer_parse_file(&path);
+    quietlyquit_if_errors!();
+
+    let ast = parse_ast_ctx(&lexer.unwrap());
+    quietlyquit_if_errors!();
+
+    let mir = run_astoir_mir(ast.unwrap());
+    quietlyquit_if_errors!();
+
+    fs::write(out, format!("{}", mir.unwrap())).unwrap();
+}

compiler/compiler_main/src/cmds/mod.rs

@@ -1,3 +1,4 @@
+pub mod build;
 pub mod check;
 
 #[macro_export]
@@ -9,3 +10,11 @@ macro_rules! quietlyquit_if_errors {
         }
     };
 }
+
+#[macro_export]
+macro_rules! soft_panic {
+    ($lit:literal) => {
+        println!($lit);
+        std::process::exit(445)
+    };
+}

compiler/compiler_main/src/main.rs

@@ -1,10 +1,15 @@
-use std::time::Instant;
+use std::{
+    fs,
+    path::{Path, PathBuf},
+    process::Output,
+    time::Instant,
+};
 
 use clap::Parser;
 
 use crate::{
-    cli::{CLICommand, Cli},
-    cmds::check::run_check,
+    cli::{CLICommand, Cli, OutputFormat, Platform},
+    cmds::{build::build_mir, check::run_check},
     version::{GIT_HASH, VERSION},
 };
 
@@ -35,6 +40,45 @@ fn main() {
             )
         }
 
+        CLICommand::Build {
+            out,
+            platform,
+            format,
+            linker,
+            input,
+        } => {
+            let needs_dir = input.len() > 1 && format != OutputFormat::Executable;
+
+            if input.len() > 1 && out.extension().is_some() && format != OutputFormat::Executable {
+                soft_panic!(
+                    "Output must be a directory if theres more than one input and that the target isn't an executable"
+                );
+            }
+
+            if needs_dir && !out.exists() {
+                fs::create_dir_all(out.clone()).unwrap();
+            }
+
+            if platform == Platform::AstoIR && format != OutputFormat::IR {
+                soft_panic!("Only IR target is supported by AstoIR platform!");
+            }
+
+            match platform {
+                Platform::AstoIR => {
+                    for i in input {
+                        let mut outfile = PathBuf::from(i.file_name().unwrap());
+                        outfile.add_extension("air");
+
+                        let output_path = out.join(outfile);
+
+                        build_mir(i.to_str().unwrap().to_string(), output_path);
+                    }
+                }
+
+                _ => todo!(),
+            }
+        }
+
         _ => todo!(),
     }
 }