transform module imports

rescript-lang · Oct 10, 2023 · 40851c8 · 40851c8
1 parent b5d902b
commit 40851c8
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/jscomp/frontend/bs_builtin_ppx.ml b/jscomp/frontend/bs_builtin_ppx.ml
@@ -601,7 +601,10 @@ let rec structure_mapper ~await_context (self : mapper) (stru : Ast_structure.t)
                        {txt = safe_module_type_name; loc}
                        ~typ:(Mty.typeof_ ~loc me)))
                 :: aux expr)
-            | Pexp_fun (_, _, _, expr) -> aux expr
+            | Pexp_let (_, vbs, expr) -> aux expr @ spelunk_vbs acc vbs
+            | Pexp_ifthenelse (_, then_expr, Some else_expr) ->
+              aux then_expr @ aux else_expr
+            | Pexp_fun (_, _, _, expr) | Pexp_newtype (_, expr) -> aux expr
             | _ -> acc
           in
           aux pvb_expr @ spelunk_vbs acc tl

diff --git a/jscomp/test/Import.js b/jscomp/test/Import.js