feat: allow multiple lines of hashbangs (google#715)

rockofox · Jan 21, 2024 · bd8b89c · bd8b89c
1 parent 906786a
commit bd8b89c
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/cli.ts b/src/cli.ts
@@ -179,7 +179,20 @@ async function importPath(filepath: string, origin = filepath) {
   const __filename = resolve(origin)
   const __dirname = dirname(__filename)
   const require = createRequire(origin)
+  const shebangRegex = /^#!(.*)*\n/
+  const file = await fs.readFile(filepath)
+
   Object.assign(global, { __filename, __dirname, require })
+
+  if (shebangRegex.test(file.toString())) {
+    return writeAndImport(
+      await file.toString().replace(shebangRegex, ''),
+      join(dirname(filepath), basename(filepath) + '.mjs'),
+      origin
+    )
+  }
+
+
   await import(url.pathToFileURL(filepath).toString())
 }