fix: WASM compile function

librasn · May 14, 2024 · 9dbec29 · 9dbec29
1 parent 7ebd524
commit 9dbec29
Showing 1 changed file with 22 additions and 16 deletions.
diff --git a/rasn-compiler/src/lib.rs b/rasn-compiler/src/lib.rs
@@ -62,22 +62,28 @@ pub struct Generated {
 
 #[cfg(target_family = "wasm")]
 #[wasm_bindgen]
-pub fn compile(asn1: &str) -> Result<Generated, JsValue> {
-    Compiler::new()
-        .add_asn_literal(asn1)
-        .compile_to_string()
-        .map(|result| Generated {
-            rust: result.generated,
-            warnings: result
-                .warnings
-                .into_iter()
-                .fold(String::new(), |mut acc, w| {
-                    acc += &w.to_string();
-                    acc += "\n";
-                    acc
-                }),
-        })
-        .map_err(|e| JsValue::from(e.to_string()))
+pub fn compile(asn1: &str, backend: &str) -> Result<Generated, JsValue> {
+    if backend == "typescript" {
+        Compiler::<crate::prelude::TypescriptBackend, _>::new()
+            .add_asn_literal(asn1)
+            .compile_to_string()
+    } else {
+        Compiler::<crate::prelude::RasnBackend, _>::new()
+            .add_asn_literal(asn1)
+            .compile_to_string()
+    }
+    .map(|result| Generated {
+        rust: result.generated,
+        warnings: result
+            .warnings
+            .into_iter()
+            .fold(String::new(), |mut acc, w| {
+                acc += &w.to_string();
+                acc += "\n";
+                acc
+            }),
+    })
+    .map_err(|e| JsValue::from(e.to_string()))
 }
 
 /// The rasn compiler