Restructure and start working on CLI

libjrpc/src/compile.rs

@@ -0,0 +1,111 @@
+use std::{collections::HashMap, path::PathBuf};
+
+use anyhow::{Context, Result};
+
+use crate::{
+    ir::{EnumDefinition, ServiceDefinition, TypeDefinition},
+    IR,
+};
+
+pub trait Compile {
+    fn new(options: &HashMap<String, String>) -> Result<Self>
+    where
+        Self: Sized;
+
+    fn name(&self) -> String;
+
+    fn start(&mut self, ctx: &mut CompileContext) -> Result<()>;
+
+    fn generate_type(
+        &mut self,
+        ctx: &mut CompileContext,
+        definition: &TypeDefinition,
+    ) -> Result<()>;
+    fn generate_enum(
+        &mut self,
+        ctx: &mut CompileContext,
+        definition: &EnumDefinition,
+    ) -> Result<()>;
+    fn generate_service(
+        &mut self,
+        ctx: &mut CompileContext,
+        definition: &ServiceDefinition,
+    ) -> Result<()>;
+
+    fn finalize(&mut self, ctx: &mut CompileContext, ir: &IR) -> Result<()>;
+}
+
+pub struct CompileContext {
+    output_folder: PathBuf,
+}
+impl CompileContext {
+    pub fn new(output_folder: &str) -> Self {
+        CompileContext {
+            output_folder: output_folder.into(),
+        }
+    }
+
+    pub fn write_file(&self, filename: &str, content: String) -> Result<()> {
+        let res_path = self.output_folder.clone().join(filename);
+        let res_dir = res_path.parent().context("Path has no parent!")?;
+        std::fs::create_dir_all(res_dir)?;
+        std::fs::write(res_path, content)?;
+        Ok(())
+    }
+}
+
+pub struct FileGenerator {
+    content: Vec<String>,
+}
+
+impl FileGenerator {
+    pub fn new() -> Self {
+        FileGenerator {
+            content: Vec::new(),
+        }
+    }
+
+    pub fn a<T: ToString>(&mut self, indent: usize, content: T) {
+        let line = "    ".repeat(indent) + &content.to_string();
+        self.content.push(line);
+    }
+
+    pub fn a0<T: ToString>(&mut self, content: T) {
+        self.a(0, content);
+    }
+
+    pub fn a1<T: ToString>(&mut self, content: T) {
+        self.a(1, content);
+    }
+
+    pub fn a2<T: ToString>(&mut self, content: T) {
+        self.a(2, content);
+    }
+
+    pub fn a3<T: ToString>(&mut self, content: T) {
+        self.a(3, content);
+    }
+
+    pub fn a4<T: ToString>(&mut self, content: T) {
+        self.a(4, content);
+    }
+
+    pub fn a5<T: ToString>(&mut self, content: T) {
+        self.a(5, content);
+    }
+    pub fn a6<T: ToString>(&mut self, content: T) {
+        self.a(6, content);
+    }
+
+    pub fn add_line(&mut self, line: &str) {
+        self.content.push(line.to_string());
+    }
+
+    pub fn get_content(&self) -> String {
+        self.content.join("\n")
+    }
+
+    pub fn into_content(self) -> String {
+        self.get_content()
+    }
+}