fix: support hashbang (closes #18)

unjs · Mar 3, 2021 · 6a1f816 · 6a1f816
1 parent 9014df5
commit 6a1f816
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/jiti.ts b/src/jiti.ts
@@ -133,12 +133,18 @@ export default function createJITI (_filename: string = process.cwd(), opts: JIT
   }
 
   function transform (filename: string, source: string, transformOptions: any) {
-    return getCache(filename, source, () => opts.transform!({
+    let transfortmed = getCache(filename, source, () => opts.transform!({
       source,
       filename,
       legacy: opts.legacy,
       ...transformOptions
     }))
+
+    if (transfortmed.startsWith('#!')) {
+      transfortmed = '// ' + transfortmed
+    }
+
+    return transfortmed
   }
 
   function jiti (id: string) {

diff --git a/test/fixtures/hashbang/index.ts b/test/fixtures/hashbang/index.ts
@@ -0,0 +1,4 @@
+#!/usr/bin/env node
+// in the Script Goal
+'use strict'
+console.log(1)