add new() to all transpilers

Author: KSDaemon

rust/cubetranspilers/src/check_dup_prop_transpiler.rs

@@ -17,7 +17,11 @@ pub struct CheckDupPropTransformVisitor {
 }
 
 impl CheckDupPropTransformVisitor {
-    fn emit_error(&self, span: Span, message: &str) {
+    pub fn new(source_map: Option<PluginSourceMapProxy>) -> Self {
+        CheckDupPropTransformVisitor { source_map }
+    }
+
+    fn emit_error(&mut self, span: Span, message: &str) {
         HANDLER.with(|handler| {
             handler
                 .struct_span_err(span, &self.format_msg(span, message))

rust/cubetranspilers/src/cube_prop_ctx_transpiler.rs

@@ -73,6 +73,20 @@ pub struct CubePropTransformVisitor {
 }
 
 impl CubePropTransformVisitor {
+    pub fn new(
+        cube_names: HashSet<String>,
+        cube_symbols: HashMap<String, HashMap<String, bool>>,
+        context_symbols: HashMap<String, String>,
+        source_map: Option<PluginSourceMapProxy>,
+    ) -> Self {
+        CubePropTransformVisitor {
+            source_map,
+            cube_names,
+            cube_symbols,
+            context_symbols,
+        }
+    }
+
     fn emit_error(&self, span: Span, message: &str) {
         HANDLER.with(|handler| {
             handler

rust/cubetranspilers/src/import_export_transpiler.rs

@@ -15,6 +15,10 @@ pub struct ImportExportTransformVisitor {
 }
 
 impl ImportExportTransformVisitor {
+    pub fn new(source_map: Option<PluginSourceMapProxy>) -> Self {
+        ImportExportTransformVisitor { source_map }
+    }
+
     fn emit_error(&self, span: Span, message: &str) {
         HANDLER.with(|handler| {
             handler

rust/cubetranspilers/src/lib.rs

@@ -30,7 +30,7 @@ pub enum Transpilers {
     ValidationTranspiler,
 }
 
-#[derive(Deserialize, Clone, Debug)]
+#[derive(Deserialize, Clone, Debug, Default)]
 #[serde(rename_all = "camelCase")]
 pub struct TransformConfig {
     pub file_name: String,
@@ -40,18 +40,6 @@ pub struct TransformConfig {
     pub context_symbols: HashMap<String, String>,
 }
 
-impl Default for TransformConfig {
-    fn default() -> Self {
-        Self {
-            file_name: String::new(),
-            transpilers: Vec::new(),
-            cube_names: HashSet::new(),
-            cube_symbols: HashMap::new(),
-            context_symbols: HashMap::new(),
-        }
-    }
-}
-
 #[derive(Serialize, Clone, Debug)]
 #[serde(rename_all = "camelCase")]
 pub struct TransformResult {
@@ -84,7 +72,9 @@ pub fn run_transpilers(
     };
 
     let sm_cell = OnceCell::new();
-    sm_cell.set(sf.clone()).map_err(|_err| anyhow!("Failed to init OnceCell with source file"))?;
+    sm_cell
+        .set(sf.clone())
+        .map_err(|_err| anyhow!("Failed to init OnceCell with source file"))?;
 
     let plugin_source_map = PluginSourceMapProxy {
         source_file: sm_cell,
@@ -95,30 +85,26 @@ pub fn run_transpilers(
         .into_iter()
         .for_each(|transpiler| match transpiler {
             Transpilers::CubeCheckDuplicatePropTranspiler => {
-                let mut visitor = CheckDupPropTransformVisitor {
-                    source_map: Some(plugin_source_map.clone()),
-                };
+                let mut visitor =
+                    CheckDupPropTransformVisitor::new(Some(plugin_source_map.clone()));
                 program.visit_mut_with(&mut visitor);
             }
             Transpilers::CubePropContextTranspiler => {
-                let mut visitor = CubePropTransformVisitor {
-                    cube_names: transform_config.cube_names.clone(),
-                    cube_symbols: transform_config.cube_symbols.clone(),
-                    context_symbols: transform_config.context_symbols.clone(),
-                    source_map: Some(plugin_source_map.clone()),
-                };
+                let mut visitor = CubePropTransformVisitor::new(
+                    transform_config.cube_names.clone(),
+                    transform_config.cube_symbols.clone(),
+                    transform_config.context_symbols.clone(),
+                    Some(plugin_source_map.clone()),
+                );
                 program.visit_mut_with(&mut visitor);
             }
             Transpilers::ImportExportTranspiler => {
-                let mut visitor = ImportExportTransformVisitor {
-                    source_map: Some(plugin_source_map.clone()),
-                };
+                let mut visitor =
+                    ImportExportTransformVisitor::new(Some(plugin_source_map.clone()));
                 program.visit_mut_with(&mut visitor);
             }
             Transpilers::ValidationTranspiler => {
-                let mut visitor = ValidationTransformVisitor {
-                    source_map: Some(plugin_source_map.clone()),
-                };
+                let mut visitor = ValidationTransformVisitor::new(Some(plugin_source_map.clone()));
                 program.visit_mut_with(&mut visitor);
             }
         });

rust/cubetranspilers/src/validation_transpiler.rs

@@ -16,6 +16,10 @@ pub struct ValidationTransformVisitor {
 }
 
 impl ValidationTransformVisitor {
+    pub fn new(source_map: Option<PluginSourceMapProxy>) -> Self {
+        ValidationTransformVisitor { source_map }
+    }
+
     fn emit_warn(&self, span: Span, message: &str) {
         HANDLER.with(|handler| {
             handler