fixed inline constant tag name in JSX transformation

pionxzh · Apr 30, 2024 · d5440f5 · d5440f5
1 parent ba0d7bc
commit d5440f5
Showing 1 changed file with 19 additions and 3 deletions.
diff --git a/packages/unminify/src/transformations/un-jsx.ts b/packages/unminify/src/transformations/un-jsx.ts
@@ -102,17 +102,16 @@ export const transformAST: ASTTransformation<typeof Schema> = (context, params)
         // bottom-up transformation
         .reverse()
         .forEach((path) => {
-            const jsxElement = toJSX(j, path, pragmas, pragmaFrags)
+            const jsxElement = toJSX(j, path, pragmas, pragmaFrags, root)
             if (jsxElement) {
                 const parentWithComments = j.ExpressionStatement.check(path.parent.node) ? path.parent : path
                 removePureAnnotation(j, parentWithComments.node)
-
                 path.replace(jsxElement)
             }
         })
 }
 
-function toJSX(j: JSCodeshift, path: ASTPath<CallExpression>, pragmas: string[], pragmaFrags: string[]): JSXElement | JSXFragment | null {
+function toJSX(j: JSCodeshift, path: ASTPath<CallExpression>, pragmas: string[], pragmaFrags: string[], root: Collection): JSXElement | JSXFragment | null {
     const scope = path.scope
     assertScopeExists(scope)
 
@@ -131,6 +130,23 @@ function toJSX(j: JSCodeshift, path: ASTPath<CallExpression>, pragmas: string[],
     if (isCapitalizationInvalid(j, type)) return null
 
     let tag = toJsxTag(j, type)
+
+    // constant tag name will convert into JSX tag
+    root.find(j.VariableDeclarator, {
+        id: {
+            type: 'Identifier',
+            name: tag.name,
+        },
+    }).forEach((path) => {
+        if (!path.node) return
+        if ('id' in path.node) {
+            const { id: { loc: { identifierName } } } = path.node
+            if (tag.name === identifierName) {
+                tag = { ...tag, name: path.node?.init?.value }
+            }
+        }
+    })
+
     // If a tag cannot be converted to JSX tag, convert it to a variable
     if (!tag && !j.SpreadElement.check(type)) {
         const name = generateName('Component', scope)