feat: handle async functions

Valerioageno · Apr 25, 2024 · 31f12f7 · 31f12f7
1 parent 8595584
commit 31f12f7
Showing 1 changed file with 84 additions and 6 deletions.
diff --git a/src/ssr.rs b/src/ssr.rs
@@ -147,17 +147,37 @@ where
 
         // TODO: transform this into an iterator
         for key in self.fn_map.keys() {
-            let result = match self.fn_map[key].call(scope, undef, &[params]) {
+            let render_fn = self.fn_map[key];
+
+            let result = match render_fn.call(scope, undef, &[params]) {
                 Some(val) => val,
                 None => return Err("Failed to call function"),
             };
 
-            let result = match result.to_string(scope) {
-                Some(val) => val,
-                None => return Err("Failed to parse the result to string"),
-            };
+            if result.is_promise() {
+                dbg!("This is a promise");
+
+                let promise = v8::Local::<v8::Promise>::try_from(result).unwrap();
+
+                while promise.state() == v8::PromiseState::Pending {
+                    scope.perform_microtask_checkpoint();
+                }
+
+                let result = promise.result(scope);
+                let result = match result.to_string(scope) {
+                    Some(val) => val,
+                    None => return Err("Failed to parse the result to string"),
+                };
+
+                rendered = format!("{}{}", rendered, result.to_rust_string_lossy(scope));
+            } else {
+                let result = match result.to_string(scope) {
+                    Some(val) => val,
+                    None => return Err("Failed to parse the result to string"),
+                };
 
-            rendered = format!("{}{}", rendered, result.to_rust_string_lossy(scope));
+                rendered = format!("{}{}", rendered, result.to_rust_string_lossy(scope));
+            }
         }
 
         Ok(rendered)
@@ -282,4 +302,62 @@ mod tests {
 
         assert_eq!(js2.render_to_string(None).unwrap(), "I don't accept params");
     }
+
+    #[test]
+    fn entry_point_is_async() {
+        init_test();
+
+        let mut js = Ssr::from(
+            r##"var SSR = {x: async () => "<html></html>"};"##.to_string(),
+            "SSR",
+        )
+        .unwrap();
+
+        assert_eq!(js.render_to_string(None).unwrap(), "<html></html>");
+        assert_eq!(
+            js.render_to_string(Some(r#"{"Hello world"}"#)).unwrap(),
+            "<html></html>"
+        );
+    }
+
+    #[test]
+    fn entry_point_is_async_with_params() {
+        init_test();
+
+        let mut js = Ssr::from(
+            r##"var SSR = {x: async (params) => "These are our parameters: " + params};"##
+                .to_string(),
+            "SSR",
+        )
+        .unwrap();
+
+        assert_eq!(
+            js.render_to_string(Some(r#"{"Hello world"}"#)).unwrap(),
+            "These are our parameters: {\"Hello world\"}"
+        );
+    }
+
+    #[test]
+    fn entry_point_is_async_with_nested_async() {
+        init_test();
+
+        let mut js = Ssr::from(
+            r##"
+            const asyncFn = async () => {
+                return "Hello world"
+            }
+            var SSR = {x: async () => {
+                return await asyncFn()
+            }};
+            "##
+            .to_string(),
+            "SSR",
+        )
+        .unwrap();
+
+        assert_eq!(
+            js.render_to_string(Some(r#"{"Hello world"}"#)).unwrap(),
+            "Hello world"
+        );
+    }
 }