change build system to rollup

src/bundle/rewriters/js.ts

@@ -1,49 +1,78 @@
-import { parseModule } from "meriyah";
-import { generate } from "astring";
-import { encodeUrl } from "./url";
-import { replace } from "estraverse";
-
-// i am a cat. i like to be petted. i like to be fed. i like to be
-
-// js rewiter is NOT finished
-
-// location
-// window
-// self
-// globalThis
-// this
-// top
-// parent
-
-
-export function rewriteJs(js: string, origin?: URL) {
-    try {
-        const ast = parseModule(js);
-    
-        // const identifierList = [
-        //     "window",
-        //     "self",
-        //     "globalThis",
-        //     "parent",
-        //     "top",
-        //     "location",
-        //     ""
-        // ]
-    
-        replace(ast, {
-            enter: (node, parent) => {
-                if (["ImportDeclaration", "ImportExpression", "ExportAllDeclaration", "ExportNamedDeclaration"].includes(node.type) && node.source) {
-                    node.source.value = encodeUrl(node.source.value, origin);
-                }
-
-                return node;
-            },
-
-            fallback: "iteration"
-        })
-    
-        return generate(ast);
-    } catch (err) {
-        throw new Error(err);
-    }
+import { parseModule } from "meriyah";
+import { generate } from "astring";
+import { makeTraveler } from "astravel";
+import { encodeUrl } from "./url";
+import * as ESTree from "estree";
+
+// i am a cat. i like to be petted. i like to be fed. i like to be
+
+// js rewiter is NOT finished
+
+// location
+// window
+// self
+// globalThis
+// this
+// top
+// parent
+
+
+export function rewriteJs(js: string, origin?: URL) {
+    try {
+        const ast = parseModule(js, {
+            module: true,
+            webcompat: true
+        });
+    
+        const identifierList = [
+            "window",
+            "self",
+            "globalThis",
+            "this",
+            "parent",
+            "top",
+            "location"
+        ]
+    
+        const customTraveler = makeTraveler({
+            ImportDeclaration: (node: ESTree.ImportDeclaration) => {
+                node.source.value = encodeUrl(node.source.value as string, origin);
+            },
+            
+            ImportExpression: (node: ESTree.ImportExpression) => {
+                if (node.source.type === "Literal") {
+                    node.source.value = encodeUrl(node.source.value as string, origin);
+                } else if (node.source.type === "Identifier") {
+                    // this is for things that import something like
+                    // const moduleName = "name";
+                    // await import(moduleName);
+                    node.source.name = `__wrapImport(${node.source.name})`;
+                }
+            },
+    
+            ExportAllDeclaration: (node: ESTree.ExportAllDeclaration) => {
+                node.source.value = encodeUrl(node.source.value as string, origin);
+            },
+    
+            ExportNamedDeclaration: (node: ESTree.ExportNamedDeclaration) => {
+                // strings are Literals in ESTree syntax but these will always be strings
+                if (node.source) node.source.value = encodeUrl(node.source.value as string, origin);
+            },
+
+            // js rweriting notrdone
+            MemberExpression: (node: ESTree.MemberExpression) => {
+                if (node.object.type === "Identifier" && identifierList.includes(node.object.name)) {
+                    node.object.name = "__" + node.object.name;
+                }
+            }
+        });
+    
+        customTraveler.go(ast);
+    
+        return generate(ast);
+    } catch {
+        console.log(js);
+
+        return js;
+    }
 }