feat: support scripts. (#3)

* test: commented.

eslint.config.js

@@ -11,4 +11,10 @@ export default tseslint.config(
       'no-console': 'error',
     },
   },
+  {
+    files: ['test/fixtures/**/*'],
+    rules: {
+      '@typescript-eslint/no-unused-vars': 'off',
+    },
+  },
 )

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@knighted/reparse",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "description": "Multiple swc parsings of the same file with correct spans.",
   "type": "module",
   "main": "dist/reparse.js",

src/child.js

@@ -5,7 +5,7 @@ import { parseSync, parseFileSync } from '@swc/core'
 const baseConfig = {
   comments: false,
   target: 'esnext',
-  isModule: true,
+  isModule: 'unknown',
 }
 const tsConfig = {
   syntax: 'typescript',

src/childSync.js

@@ -5,7 +5,7 @@ import { parseSync, parseFileSync } from '@swc/core'
 const baseConfig = {
   comments: false,
   target: 'esnext',
-  isModule: true,
+  isModule: 'unknown',
 }
 const tsConfig = {
   syntax: 'typescript',

src/reparse.ts

@@ -1,12 +1,12 @@
 import { fork, spawnSync } from 'node:child_process'
 import { join } from 'node:path'
 
-import type { Module } from '@swc/core'
+import type { Module, Script } from '@swc/core'
 
 type Lang = 'ts' | 'es'
 
-const isModule = (msg: unknown): msg is Module => {
-  if (msg && typeof msg === 'object' && 'type' in msg && msg.type === 'Module') {
+const isModuleOrScript = (msg: unknown): msg is Module | Script => {
+  if (msg && typeof msg === 'object' && 'type' in msg && (msg.type === 'Module' || msg.type === 'Script')) {
     return true
   }
 
@@ -17,9 +17,9 @@ const forkChild = (...args: string[]) => {
     serialization: 'advanced',
   })
 
-  return new Promise<Module>((resolve, reject) => {
+  return new Promise<Module | Script>((resolve, reject) => {
     child.on('message', msg => {
-      if (isModule(msg)) {
+      if (isModuleOrScript(msg)) {
         resolve(msg)
       }
 

test/fixtures/commented.js

@@ -0,0 +1,12 @@
+/**
+ * Some start comment.
+ */
+
+function foo() {
+  // Some inline comment.
+  return 'bar'
+}
+
+/**
+ * Some end comment.
+ */

test/reparse.ts

@@ -30,4 +30,20 @@ describe('@knighted/reparse', () => {
     assert.equal(ast3.span.start, ast4.span.start)
     assert.equal(ast3.span.end, ast4.span.end)
   })
+
+  it('works with comments', async () => {
+    const ast1 = await reparseFile(join(fixtures, 'commented.js'))
+    const ast2 = await reparseFile(join(fixtures, 'commented.js'))
+
+    assert.equal(ast1.span.start, ast2.span.start)
+    assert.equal(ast1.span.end, ast2.span.end)
+  })
+
+  it.skip('works with large files', async () => {
+    const ast1 = await reparseFile(join(fixtures, 'large.js'))
+    const ast2 = await reparseFile(join(fixtures, 'large.js'))
+
+    assert.equal(ast1.span.start, ast2.span.start)
+    assert.equal(ast1.span.end, ast2.span.end)
+  })
 })