fix(toExports): don't strip extension for package import (#291)

Co-authored-by: Anthony Fu <[email protected]>
unjs · Nov 9, 2023 · cf28aa3 · cf28aa3
1 parent a90f45c
commit cf28aa3
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/src/utils.ts b/src/utils.ts
@@ -118,7 +118,9 @@ export function toExports (imports: Import[], fileDir?: string, includeType = fa
   const map = toImportModuleMap(imports, includeType)
   return Object.entries(map)
     .flatMap(([name, imports]) => {
-      name = name.replace(/\.[a-z]+$/, '')
+      if (isFilePath(name)) {
+        name = name.replace(/\.[a-zA-Z]+$/, '')
+      }
       if (fileDir && isAbsolute(name)) {
         name = relative(fileDir, name)
         if (!name.match(/^[.\/]/)) {
@@ -311,3 +313,7 @@ export function resolveIdAbsolute (id: string, parentId?: string) {
     url: parentId
   })
 }
+
+function isFilePath (path: string) {
+  return path.startsWith('.') || isAbsolute(path) || path.includes('://')
+}
diff --git a/test/to-export.test.ts b/test/to-export.test.ts
@@ -66,13 +66,19 @@ describe('toExports', () => {
   it('strip extensions', () => {
     const imports: Import[] = [
       { from: 'test1.ts', name: 'foo', as: 'foo' },
-      { from: 'test2.mjs', name: 'foobar', as: 'foobar' }
+      { from: 'test2.mjs', name: 'foobar', as: 'foobar' },
+      { from: './test1.ts', name: 'foo', as: 'foo' },
+      { from: './test2.mjs', name: 'foobar', as: 'foobar' },
+      { from: 'test1.ts/test1.ts', name: 'foo', as: 'foo' }
     ]
 
     expect(toExports(imports))
       .toMatchInlineSnapshot(`
-        "export { foo } from 'test1';
-        export { foobar } from 'test2';"
+        "export { foo } from 'test1.ts';
+        export { foobar } from 'test2.mjs';
+        export { foo } from './test1';
+        export { foobar } from './test2';
+        export { foo } from 'test1.ts/test1.ts';"
       `)
   })