From bd1f2934c6ec98e4dcee629764295b94446141c9 Mon Sep 17 00:00:00 2001
From: patrick <pa.trickjlp@gmail.com>
Date: Tue, 25 Jun 2024 11:59:17 +0200
Subject: [PATCH] add template only name

---
 src/plugin.ts | 26 +++++++++++++++++++++++---
 1 file changed, 23 insertions(+), 3 deletions(-)

diff --git a/src/plugin.ts b/src/plugin.ts
index 50bd5a8..b08b42f 100644
--- a/src/plugin.ts
+++ b/src/plugin.ts
@@ -1,7 +1,7 @@
 import type { NodePath } from '@babel/traverse';
 import type * as Babel from '@babel/core';
 import type { types as t } from '@babel/core';
-import { ImportUtil, type Importer } from 'babel-import-util';
+import { ImportUtil, Importer } from 'babel-import-util';
 import { ExpressionParser } from './expression-parser';
 import { JSUtils, ExtendedPluginBuilder } from './js-utils';
 import type { EmberTemplateCompiler, PreprocessOptions } from './ember-template-compiler';
@@ -496,13 +496,23 @@ function insertCompiledTemplate<EnvSpecificOptions>(
 
     let expression = t.callExpression(templateFactoryIdentifier, [templateExpression]);
 
+    let assignment = target.parent;
+    let assignmentName: string | null = null;
+    if (assignment.type === 'AssignmentExpression' && assignment.left.type === 'Identifier') {
+      assignmentName = assignment.left.name;
+    }
+
+    if (assignment.type === 'VariableDeclarator' && assignment.id.type === 'Identifier') {
+      assignmentName = assignment.id.name;
+    }
+
     if (config.rfc931Support) {
       expression = t.callExpression(i.import('@ember/component', 'setComponentTemplate'), [
         expression,
         backingClass?.node ??
           t.callExpression(
             i.import('@ember/component/template-only', 'default', 'templateOnly'),
-            []
+            ['@ember/component/template-only', assignmentName]
           ),
       ]);
     }
@@ -606,6 +616,16 @@ function updateCallForm<EnvSpecificOptions>(
     convertStrictMode(babel, target);
     removeEvalAndScope(target);
     target.node.arguments = target.node.arguments.slice(0, 2);
+    let assignment = target.parent;
+    let assignmentName: string | null = null;
+    if (assignment.type === 'AssignmentExpression' && assignment.left.type === 'Identifier') {
+      assignmentName = assignment.left.name;
+    }
+
+    if (assignment.type === 'VariableDeclarator' && assignment.id.type === 'Identifier') {
+      assignmentName = assignment.id.name;
+    }
+
     state.recursionGuard.add(target.node);
     state.util.replaceWith(target, (i) =>
       babel.types.callExpression(i.import('@ember/component', 'setComponentTemplate'), [
@@ -613,7 +633,7 @@ function updateCallForm<EnvSpecificOptions>(
         backingClass?.node ??
           babel.types.callExpression(
             i.import('@ember/component/template-only', 'default', 'templateOnly'),
-            []
+            ['@ember/component/template-only', assignmentName]
           ),
       ])
     );