fix: nested classNames in components like Sonner were not having pref…

…ix applied correctly
shadcn-ui · Apr 19, 2024 · 156b1a4 · 156b1a4
1 parent bebc284
commit 156b1a4
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/packages/cli/src/utils/transformers/transform-tw-prefix.ts b/packages/cli/src/utils/transformers/transform-tw-prefix.ts
@@ -165,6 +165,40 @@ export const transformTwPrefixes: Transformer = async ({
           })
       }
     }
+
+    // Handling cases where there are descendant classNames, e.g., in a Sonner component.
+    if (node.isKind(SyntaxKind.JsxAttribute)) {
+      const attributeInitializer = node.getInitializer()
+
+      if (attributeInitializer?.isKind(SyntaxKind.JsxExpression)) {
+        const expressionContent = attributeInitializer.getExpression()
+
+        if (
+          expressionContent &&
+          expressionContent.getKind() === SyntaxKind.ObjectLiteralExpression
+        ) {
+          const propertyAssignments = expressionContent.getDescendantsOfKind(
+            SyntaxKind.PropertyAssignment
+          )
+
+          propertyAssignments.forEach((property) => {
+            if (property.getInitializer()?.isKind(SyntaxKind.StringLiteral)) {
+              if (property.getName() !== "variant") {
+                const classValue = property.getInitializer()
+                if (classValue) {
+                  classValue.replaceWithText(
+                    `"${applyPrefix(
+                      classValue.getText()?.replace(/"/g, ""),
+                      config.tailwind.prefix
+                    )}"`
+                  )
+                }
+              }
+            }
+          })
+        }
+      }
+    }
   })
 
   return sourceFile