Merge pull request #6 from NullVoxPopuli/fix-extensions-on-exports

Fix extensions on exports

.prettierignore

@@ -1,4 +1,5 @@
 *.yaml
 *.yml
+*.md
 
 declarations/

src/fixes/glint.js

@@ -10,7 +10,7 @@ const j = jscodeshift.withParser('ts');
 export function fixGTSExtensions(contents) {
   const root = j(contents);
 
-  const fixed = root
+  root
     .find(j.ImportDeclaration)
     // @ts-expect-error
     .filter((path) => path.node.source.value.includes('.gts'))
@@ -19,10 +19,20 @@ export function fixGTSExtensions(contents) {
       //       moduleResolution = "bundler"
       // @ts-expect-error
       path.node.source.value = path.node.source.value.replace(/\.gts$/, '');
-    })
-    .toSource();
+    });
 
-  return fixed;
+  root
+    .find(j.ExportNamedDeclaration)
+    // @ts-expect-error
+    .filter((path) => path.node.source?.value?.includes('.gts'))
+    .forEach((path) => {
+      // TODO: this may only be appropriate when
+      //       moduleResolution = "bundler"
+      // @ts-expect-error
+      path.node.source.value = path.node.source.value.replace(/\.gts$/, '');
+    });
+
+  return root.toSource();
 }
 
 /**

src/fixes/glint.test.ts

@@ -15,6 +15,26 @@ describe('fixGTSExtensions', () => {
 
     expect(result).toBe(`import x from "./foo";`);
   });
+
+  test('works on default exports', () => {
+    let code = stripIndent`
+      export x from './foo.gts';
+    `;
+
+    let result = fixGTSExtensions(code);
+
+    expect(result).toBe(`export x from "./foo";`);
+  });
+
+  test('works on named exports', () => {
+    let code = stripIndent`
+      export { x } from './foo.gts';
+    `;
+
+    let result = fixGTSExtensions(code);
+
+    expect(result).toBe(`export { x } from "./foo";`);
+  });
 });
 
 describe('fixOwnReferences', () => {

src/index.js

@@ -71,6 +71,11 @@ export async function fixBadDeclarationOutput(
   }
 
   for (let filePath of files) {
-    await fixFile(filePath, fixesToApply);
+    try {
+      await fixFile(filePath, fixesToApply);
+    } catch (e) {
+      console.error(`Errored while processing ${filePath}`);
+      throw e;
+    }
   }
 }