Merge pull request #11 from bertdeblock/support-inline-imports

Support inline imports
NullVoxPopuli · Mar 19, 2024 · a2cd103 · a2cd103
2 parents b6c7cd9 + 8a69f53
commit a2cd103
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/src/fixes/glint.js b/src/fixes/glint.js
@@ -32,6 +32,13 @@ export function fixGTSExtensions(contents) {
       path.node.source.value = path.node.source.value.replace(/\.gts$/, '');
     });
 
+  root
+    .find(j.TSImportType)
+    .filter((path) => path.node.argument.value.includes('.gts'))
+    .forEach((path) => {
+      path.node.argument.value = path.node.argument.value.replace(/\.gts$/, '');
+    });
+
   return root.toSource();
 }
 

diff --git a/src/fixes/glint.test.ts b/src/fixes/glint.test.ts
@@ -35,6 +35,18 @@ describe('fixGTSExtensions', () => {
 
     expect(result).toBe(`export { x } from "./foo";`);
   });
+
+  test('works on inline imports', () => {
+    let code = stripIndent`
+      import("@ember/component/template-only").TOC<import("./foo.gts").FooSignature>;
+    `;
+
+    let result = fixGTSExtensions(code);
+
+    expect(result).toBe(
+      `import("@ember/component/template-only").TOC<import("./foo").FooSignature>;`
+    );
+  });
 });
 
 describe('fixOwnReferences', () => {