Merge pull request #713 from patricklx/patch-4

check if template is in heritage clause
typed-ember · Mar 21, 2024 · dee489b · dee489b
2 parents ee54032 + 1a1a150
commit dee489b
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/packages/core/__tests__/transform/rewrite.test.ts b/packages/core/__tests__/transform/rewrite.test.ts
@@ -410,6 +410,29 @@ describe('Transform: rewriteModule', () => {
   });
 
   describe('ember-template-imports', () => {
+    test('in class extends', () => {
+      let customEnv = GlintEnvironment.load(['ember-loose', 'ember-template-imports']);
+      let script = {
+        filename: 'test.gts',
+        contents: stripIndent`
+          import Component, { hbs } from 'special/component';
+          export default class MyComponent extends Component(<template></template>) {
+            
+          }
+        `,
+      };
+
+      let transformedModule = rewriteModule(ts, { script }, customEnv);
+
+      expect(transformedModule?.transformedContents).toMatchInlineSnapshot(`
+        "import Component, { hbs } from 'special/component';
+        export default class MyComponent extends Component(({} as typeof import(\\"@glint/environment-ember-template-imports/-private/dsl\\")).templateExpression(function(𝚪, χ: typeof import(\\"@glint/environment-ember-template-imports/-private/dsl\\")) {
+          𝚪; χ;
+        })) {
+          
+        }"
+      `);
+    });
     test('embedded gts templates', () => {
       let customEnv = GlintEnvironment.load(['ember-loose', 'ember-template-imports']);
       let script = {

diff --git a/packages/core/src/transform/template/inlining/index.ts b/packages/core/src/transform/template/inlining/index.ts
@@ -21,6 +21,9 @@ export function isEmbeddedInClass(ts: TSLib, node: ts.Node): boolean {
     // TODO: this should likely actually filter on whether the template appears in a
     // static block or property definition, but just "am I in a class body" is the
     // current status quo and has been ok so far.
+    if (ts.isHeritageClause(current)) {
+      return false;
+    }
     if (ts.isClassLike(current)) {
       return true;
     }