trix-ui 0.2.8 → 0.2.9

package/dist/styles/transform-style-map.js

@@ -1,429 +1,429 @@
-import { Node, } from "ts-morph";
-/**
- * Classes that should never be removed during transformation.
- * These are typically used as CSS selectors or for other purposes
- * that require the class name to remain in the code.
- */
-const ALLOWLIST = new Set(["cn-menu-target"]);
-function isStringLiteralLike(node) {
-    return (Node.isStringLiteral(node) || Node.isNoSubstitutionTemplateLiteral(node));
-}
-export const transformStyleMap = async ({ sourceFile, styleMap, }) => {
-    const matchedClasses = new Set();
-    applyToCvaCalls(sourceFile, styleMap, matchedClasses);
-    applyToClassNameAttributes(sourceFile, styleMap, matchedClasses);
-    applyToMergePropsCalls(sourceFile, styleMap, matchedClasses);
-    return sourceFile;
-};
-function applyStyleToCvaString(stringNode, styleMap, matchedClasses) {
-    const stringValue = stringNode.getLiteralText();
-    const cnClasses = extractCnClasses(stringValue);
-    if (cnClasses.length === 0) {
-        return;
-    }
-    // Process all cn-* classes, not just the first one
-    const unmatchedClasses = cnClasses.filter((cnClass) => !matchedClasses.has(cnClass));
-    if (unmatchedClasses.length === 0) {
-        // All classes already matched, just clean up non-allowlisted ones
-        const updated = removeCnClasses(stringValue);
-        stringNode.setLiteralValue(updated);
-        return;
-    }
-    const tailwindClassesToApply = unmatchedClasses
-        .map((cnClass) => styleMap[cnClass])
-        .filter((classes) => Boolean(classes));
-    if (tailwindClassesToApply.length > 0) {
-        const mergedClasses = tailwindClassesToApply.join(" ");
-        const updated = removeCnClasses(mergeClasses(mergedClasses, stringValue));
-        stringNode.setLiteralValue(updated);
-        unmatchedClasses.forEach((cnClass) => matchedClasses.add(cnClass));
-    }
-    else {
-        // No styles to apply, but still need to clean up non-allowlisted classes
-        const updated = removeCnClasses(stringValue);
-        stringNode.setLiteralValue(updated);
-    }
-}
-function applyToCvaCalls(sourceFile, styleMap, matchedClasses) {
-    sourceFile.forEachDescendant((node) => {
-        if (!Node.isCallExpression(node)) {
-            return;
-        }
-        const expression = node.getExpression();
-        if (!Node.isIdentifier(expression) || expression.getText() !== "cva") {
-            return;
-        }
-        const baseArg = node.getArguments()[0];
-        if (Node.isStringLiteral(baseArg)) {
-            applyStyleToCvaString(baseArg, styleMap, matchedClasses);
-        }
-        const configArg = node.getArguments()[1];
-        if (!configArg || !Node.isObjectLiteralExpression(configArg)) {
-            return;
-        }
-        const variantsProp = configArg
-            .getProperties()
-            .find((prop) => Node.isPropertyAssignment(prop) &&
-            Node.isIdentifier(prop.getNameNode()) &&
-            prop.getNameNode().getText() === "variants");
-        if (!variantsProp || !Node.isPropertyAssignment(variantsProp)) {
-            return;
-        }
-        const variantsObj = variantsProp.getInitializer();
-        if (!variantsObj || !Node.isObjectLiteralExpression(variantsObj)) {
-            return;
-        }
-        variantsObj.getProperties().forEach((typeProp) => {
-            if (!Node.isPropertyAssignment(typeProp)) {
-                return;
-            }
-            const typeObj = typeProp.getInitializer();
-            if (!typeObj || !Node.isObjectLiteralExpression(typeObj)) {
-                return;
-            }
-            typeObj.getProperties().forEach((variantProp) => {
-                if (!Node.isPropertyAssignment(variantProp)) {
-                    return;
-                }
-                const variantValue = variantProp.getInitializer();
-                if (variantValue && Node.isStringLiteral(variantValue)) {
-                    applyStyleToCvaString(variantValue, styleMap, matchedClasses);
-                }
-            });
-        });
-    });
-}
-function applyToClassNameAttributes(sourceFile, styleMap, matchedClasses) {
-    sourceFile.forEachDescendant((node) => {
-        if (!Node.isJsxAttribute(node) ||
-            node.getNameNode().getText() !== "className") {
-            return;
-        }
-        const initializer = node.getInitializer();
-        if (!initializer) {
-            return;
-        }
-        const cnClasses = extractCnClassesFromAttribute(initializer);
-        if (cnClasses.length === 0) {
-            return;
-        }
-        const jsxElement = node.getParent()?.getParent();
-        if (!jsxElement ||
-            (!Node.isJsxOpeningElement(jsxElement) &&
-                !Node.isJsxSelfClosingElement(jsxElement))) {
-            return;
-        }
-        const unmatchedClasses = cnClasses.filter((cnClass) => !matchedClasses.has(cnClass));
-        if (unmatchedClasses.length === 0) {
-            // Even if all classes are already matched, we still need to clean them up
-            cleanCnClassesFromAttribute(initializer);
-            return;
-        }
-        const tailwindClassesToApply = unmatchedClasses
-            .map((cnClass) => styleMap[cnClass])
-            .filter((classes) => Boolean(classes));
-        if (tailwindClassesToApply.length > 0) {
-            const mergedClasses = tailwindClassesToApply.join(" ");
-            applyClassesToElement(jsxElement, mergedClasses);
-        }
-        else {
-            cleanCnClassesFromAttribute(initializer);
-        }
-    });
-}
-function extractCnClassesFromAttribute(initializer) {
-    const classes = [];
-    if (isStringLiteralLike(initializer)) {
-        return extractCnClasses(initializer.getLiteralText());
-    }
-    if (!Node.isJsxExpression(initializer)) {
-        return classes;
-    }
-    const expression = initializer.getExpression();
-    if (!expression) {
-        return classes;
-    }
-    if (isStringLiteralLike(expression)) {
-        return extractCnClasses(expression.getLiteralText());
-    }
-    if (Node.isCallExpression(expression) && isCnCall(expression)) {
-        for (const argument of expression.getArguments()) {
-            if (isStringLiteralLike(argument)) {
-                classes.push(...extractCnClasses(argument.getLiteralText()));
-            }
-        }
-    }
-    return classes;
-}
-function cleanCnClassesFromAttribute(initializer) {
-    if (isStringLiteralLike(initializer)) {
-        const cleaned = removeCnClasses(initializer.getLiteralText());
-        initializer.setLiteralValue(cleaned);
-        return;
-    }
-    if (!Node.isJsxExpression(initializer)) {
-        return;
-    }
-    const expression = initializer.getExpression();
-    if (!expression) {
-        return;
-    }
-    if (isStringLiteralLike(expression)) {
-        const cleaned = removeCnClasses(expression.getLiteralText());
-        expression.setLiteralValue(cleaned);
-        return;
-    }
-    if (Node.isCallExpression(expression) && isCnCall(expression)) {
-        for (const argument of expression.getArguments()) {
-            if (isStringLiteralLike(argument)) {
-                const cleaned = removeCnClasses(argument.getLiteralText());
-                argument.setLiteralValue(cleaned);
-            }
-        }
-        removeEmptyArgumentsFromCnCall(expression);
-    }
-}
-function extractCnClasses(str) {
-    const matches = str.matchAll(/\bcn-[\w-]+\b/g);
-    return Array.from(matches, (match) => match[0]);
-}
-function extractCnClass(str) {
-    const classes = extractCnClasses(str);
-    return classes[0] ?? null;
-}
-function removeCnClasses(str) {
-    return str
-        .replace(/\bcn-[\w-]+\b/g, (match) => {
-        // Preserve allowlisted classes
-        if (ALLOWLIST.has(match)) {
-            return match;
-        }
-        return "";
-    })
-        .replace(/\s+/g, " ")
-        .trim();
-}
-function removeEmptyArgumentsFromCnCall(callExpression) {
-    if (!isCnCall(callExpression)) {
-        return;
-    }
-    const args = callExpression.getArguments();
-    const nonEmptyArgs = args.filter((arg) => {
-        if (isStringLiteralLike(arg)) {
-            const text = arg.getLiteralText().trim();
-            return text !== "";
-        }
-        return true;
-    });
-    if (nonEmptyArgs.length !== args.length) {
-        const argTexts = nonEmptyArgs.map((arg) => arg.getText());
-        const parent = callExpression.getParent();
-        if (parent && Node.isJsxExpression(parent)) {
-            parent.replaceWithText(`{cn(${argTexts.join(", ")})}`);
-        }
-        else {
-            callExpression.replaceWithText(`cn(${argTexts.join(", ")})`);
-        }
-    }
-}
-function applyClassesToElement(element, tailwindClasses) {
-    if (!Node.isJsxOpeningElement(element) &&
-        !Node.isJsxSelfClosingElement(element)) {
-        return;
-    }
-    const attribute = element
-        .getAttributes()
-        .find((attr) => Node.isJsxAttribute(attr) &&
-        attr.getNameNode().getText() === "className");
-    if (!attribute || !Node.isJsxAttribute(attribute)) {
-        element.addAttribute({
-            name: "className",
-            initializer: `{cn(${JSON.stringify(tailwindClasses)})}`,
-        });
-        return;
-    }
-    const initializer = attribute.getInitializer();
-    if (!initializer) {
-        attribute.setInitializer(`{cn(${JSON.stringify(tailwindClasses)})}`);
-        return;
-    }
-    if (isStringLiteralLike(initializer)) {
-        const existing = initializer.getLiteralText();
-        const updated = removeCnClasses(mergeClasses(tailwindClasses, existing));
-        initializer.setLiteralValue(updated);
-        return;
-    }
-    if (!Node.isJsxExpression(initializer)) {
-        return;
-    }
-    const expression = initializer.getExpression();
-    if (!expression) {
-        attribute.setInitializer(`{cn(${JSON.stringify(tailwindClasses)})}`);
-        return;
-    }
-    if (isStringLiteralLike(expression)) {
-        const existing = expression.getLiteralText();
-        const updated = removeCnClasses(mergeClasses(tailwindClasses, existing));
-        expression.setLiteralValue(updated);
-        return;
-    }
-    if (Node.isCallExpression(expression) && isCnCall(expression)) {
-        const firstArg = expression.getArguments()[0];
-        if (isStringLiteralLike(firstArg)) {
-            const existing = firstArg.getLiteralText();
-            const updated = removeCnClasses(mergeClasses(tailwindClasses, existing));
-            firstArg.setLiteralValue(updated);
-            for (let i = 1; i < expression.getArguments().length; i++) {
-                const arg = expression.getArguments()[i];
-                if (isStringLiteralLike(arg)) {
-                    const argText = arg.getLiteralText();
-                    const cleaned = removeCnClasses(argText);
-                    if (cleaned !== argText) {
-                        arg.setLiteralValue(cleaned);
-                    }
-                }
-            }
-            removeEmptyArgumentsFromCnCall(expression);
-            return;
-        }
-        const argumentTexts = expression
-            .getArguments()
-            .map((argument) => {
-            if (isStringLiteralLike(argument)) {
-                const cleaned = removeCnClasses(argument.getLiteralText());
-                return cleaned ? JSON.stringify(cleaned) : null;
-            }
-            return argument.getText();
-        })
-            .filter((arg) => arg !== null);
-        const updatedArguments = [JSON.stringify(tailwindClasses), ...argumentTexts];
-        attribute.setInitializer(`{cn(${updatedArguments.join(", ")})}`);
-        return;
-    }
-    attribute.setInitializer(`{cn(${JSON.stringify(tailwindClasses)}, ${expression.getText()})}`);
-}
-function mergeClasses(newClasses, existing) {
-    const existingParts = existing.split(/\s+/).filter(Boolean);
-    const newParts = newClasses.split(/\s+/).filter(Boolean);
-    const combined = [...newParts, ...existingParts];
-    return combined.join(" ").trim();
-}
-function isCnCall(call) {
-    const expression = call.getExpression();
-    return Node.isIdentifier(expression) && expression.getText() === "cn";
-}
-function applyToMergePropsCalls(sourceFile, styleMap, matchedClasses) {
-    sourceFile.forEachDescendant((node) => {
-        if (!Node.isCallExpression(node)) {
-            return;
-        }
-        const expression = node.getExpression();
-        if (!Node.isIdentifier(expression) ||
-            expression.getText() !== "mergeProps") {
-            return;
-        }
-        // Look for object literals in mergeProps arguments
-        for (const arg of node.getArguments()) {
-            if (!Node.isObjectLiteralExpression(arg)) {
-                continue;
-            }
-            // Find className property in the object literal
-            const classNameProp = arg
-                .getProperties()
-                .find((prop) => Node.isPropertyAssignment(prop) &&
-                Node.isIdentifier(prop.getNameNode()) &&
-                prop.getNameNode().getText() === "className");
-            if (!classNameProp || !Node.isPropertyAssignment(classNameProp)) {
-                continue;
-            }
-            const classNameInitializer = classNameProp.getInitializer();
-            if (!classNameInitializer) {
-                continue;
-            }
-            // Handle cn() calls in className
-            if (Node.isCallExpression(classNameInitializer) &&
-                isCnCall(classNameInitializer)) {
-                const cnClasses = extractCnClassesFromCnCall(classNameInitializer);
-                if (cnClasses.length === 0) {
-                    continue;
-                }
-                const unmatchedClasses = cnClasses.filter((cnClass) => !matchedClasses.has(cnClass));
-                if (unmatchedClasses.length === 0) {
-                    // Clean up cn-* classes even if already matched
-                    cleanCnClassesFromCnCall(classNameInitializer);
-                    continue;
-                }
-                const tailwindClassesToApply = unmatchedClasses
-                    .map((cnClass) => styleMap[cnClass])
-                    .filter((classes) => Boolean(classes));
-                if (tailwindClassesToApply.length > 0) {
-                    const mergedClasses = tailwindClassesToApply.join(" ");
-                    applyClassesToCnCall(classNameInitializer, mergedClasses, matchedClasses, unmatchedClasses);
-                }
-                else {
-                    cleanCnClassesFromCnCall(classNameInitializer);
-                }
-            }
-        }
-    });
-}
-function extractCnClassesFromCnCall(cnCall) {
-    const classes = [];
-    for (const argument of cnCall.getArguments()) {
-        if (isStringLiteralLike(argument)) {
-            classes.push(...extractCnClasses(argument.getLiteralText()));
-        }
-    }
-    return classes;
-}
-function cleanCnClassesFromCnCall(cnCall) {
-    for (const argument of cnCall.getArguments()) {
-        if (isStringLiteralLike(argument)) {
-            const cleaned = removeCnClasses(argument.getLiteralText());
-            argument.setLiteralValue(cleaned);
-        }
-    }
-    removeEmptyArgumentsFromCnCall(cnCall);
-}
-function applyClassesToCnCall(cnCall, tailwindClasses, matchedClasses, unmatchedClasses) {
-    const firstArg = cnCall.getArguments()[0];
-    if (isStringLiteralLike(firstArg)) {
-        const existing = firstArg.getLiteralText();
-        const updated = removeCnClasses(mergeClasses(tailwindClasses, existing));
-        firstArg.setLiteralValue(updated);
-        // Mark classes as matched
-        unmatchedClasses.forEach((cnClass) => matchedClasses.add(cnClass));
-        // Clean up cn-* classes from remaining arguments
-        for (let i = 1; i < cnCall.getArguments().length; i++) {
-            const arg = cnCall.getArguments()[i];
-            if (isStringLiteralLike(arg)) {
-                const argText = arg.getLiteralText();
-                const cleaned = removeCnClasses(argText);
-                if (cleaned !== argText) {
-                    arg.setLiteralValue(cleaned);
-                }
-            }
-        }
-        removeEmptyArgumentsFromCnCall(cnCall);
-        return;
-    }
-    // If first arg is not a string literal, prepend tailwind classes
-    const argumentTexts = cnCall
-        .getArguments()
-        .map((argument) => {
-        if (isStringLiteralLike(argument)) {
-            const cleaned = removeCnClasses(argument.getLiteralText());
-            return cleaned ? JSON.stringify(cleaned) : null;
-        }
-        return argument.getText();
-    })
-        .filter((arg) => arg !== null);
-    const updatedArguments = [JSON.stringify(tailwindClasses), ...argumentTexts];
-    // Mark classes as matched
-    unmatchedClasses.forEach((cnClass) => matchedClasses.add(cnClass));
-    const parent = cnCall.getParent();
-    if (parent) {
-        cnCall.replaceWithText(`cn(${updatedArguments.join(", ")})`);
-    }
-}
+import { Node, } from "ts-morph";
+/**
+ * Classes that should never be removed during transformation.
+ * These are typically used as CSS selectors or for other purposes
+ * that require the class name to remain in the code.
+ */
+const ALLOWLIST = new Set(["cn-menu-target"]);
+function isStringLiteralLike(node) {
+    return (Node.isStringLiteral(node) || Node.isNoSubstitutionTemplateLiteral(node));
+}
+export const transformStyleMap = async ({ sourceFile, styleMap, }) => {
+    const matchedClasses = new Set();
+    applyToCvaCalls(sourceFile, styleMap, matchedClasses);
+    applyToClassNameAttributes(sourceFile, styleMap, matchedClasses);
+    applyToMergePropsCalls(sourceFile, styleMap, matchedClasses);
+    return sourceFile;
+};
+function applyStyleToCvaString(stringNode, styleMap, matchedClasses) {
+    const stringValue = stringNode.getLiteralText();
+    const cnClasses = extractCnClasses(stringValue);
+    if (cnClasses.length === 0) {
+        return;
+    }
+    // Process all cn-* classes, not just the first one
+    const unmatchedClasses = cnClasses.filter((cnClass) => !matchedClasses.has(cnClass));
+    if (unmatchedClasses.length === 0) {
+        // All classes already matched, just clean up non-allowlisted ones
+        const updated = removeCnClasses(stringValue);
+        stringNode.setLiteralValue(updated);
+        return;
+    }
+    const tailwindClassesToApply = unmatchedClasses
+        .map((cnClass) => styleMap[cnClass])
+        .filter((classes) => Boolean(classes));
+    if (tailwindClassesToApply.length > 0) {
+        const mergedClasses = tailwindClassesToApply.join(" ");
+        const updated = removeCnClasses(mergeClasses(mergedClasses, stringValue));
+        stringNode.setLiteralValue(updated);
+        unmatchedClasses.forEach((cnClass) => matchedClasses.add(cnClass));
+    }
+    else {
+        // No styles to apply, but still need to clean up non-allowlisted classes
+        const updated = removeCnClasses(stringValue);
+        stringNode.setLiteralValue(updated);
+    }
+}
+function applyToCvaCalls(sourceFile, styleMap, matchedClasses) {
+    sourceFile.forEachDescendant((node) => {
+        if (!Node.isCallExpression(node)) {
+            return;
+        }
+        const expression = node.getExpression();
+        if (!Node.isIdentifier(expression) || expression.getText() !== "cva") {
+            return;
+        }
+        const baseArg = node.getArguments()[0];
+        if (Node.isStringLiteral(baseArg)) {
+            applyStyleToCvaString(baseArg, styleMap, matchedClasses);
+        }
+        const configArg = node.getArguments()[1];
+        if (!configArg || !Node.isObjectLiteralExpression(configArg)) {
+            return;
+        }
+        const variantsProp = configArg
+            .getProperties()
+            .find((prop) => Node.isPropertyAssignment(prop) &&
+            Node.isIdentifier(prop.getNameNode()) &&
+            prop.getNameNode().getText() === "variants");
+        if (!variantsProp || !Node.isPropertyAssignment(variantsProp)) {
+            return;
+        }
+        const variantsObj = variantsProp.getInitializer();
+        if (!variantsObj || !Node.isObjectLiteralExpression(variantsObj)) {
+            return;
+        }
+        variantsObj.getProperties().forEach((typeProp) => {
+            if (!Node.isPropertyAssignment(typeProp)) {
+                return;
+            }
+            const typeObj = typeProp.getInitializer();
+            if (!typeObj || !Node.isObjectLiteralExpression(typeObj)) {
+                return;
+            }
+            typeObj.getProperties().forEach((variantProp) => {
+                if (!Node.isPropertyAssignment(variantProp)) {
+                    return;
+                }
+                const variantValue = variantProp.getInitializer();
+                if (variantValue && Node.isStringLiteral(variantValue)) {
+                    applyStyleToCvaString(variantValue, styleMap, matchedClasses);
+                }
+            });
+        });
+    });
+}
+function applyToClassNameAttributes(sourceFile, styleMap, matchedClasses) {
+    sourceFile.forEachDescendant((node) => {
+        if (!Node.isJsxAttribute(node) ||
+            node.getNameNode().getText() !== "className") {
+            return;
+        }
+        const initializer = node.getInitializer();
+        if (!initializer) {
+            return;
+        }
+        const cnClasses = extractCnClassesFromAttribute(initializer);
+        if (cnClasses.length === 0) {
+            return;
+        }
+        const jsxElement = node.getParent()?.getParent();
+        if (!jsxElement ||
+            (!Node.isJsxOpeningElement(jsxElement) &&
+                !Node.isJsxSelfClosingElement(jsxElement))) {
+            return;
+        }
+        const unmatchedClasses = cnClasses.filter((cnClass) => !matchedClasses.has(cnClass));
+        if (unmatchedClasses.length === 0) {
+            // Even if all classes are already matched, we still need to clean them up
+            cleanCnClassesFromAttribute(initializer);
+            return;
+        }
+        const tailwindClassesToApply = unmatchedClasses
+            .map((cnClass) => styleMap[cnClass])
+            .filter((classes) => Boolean(classes));
+        if (tailwindClassesToApply.length > 0) {
+            const mergedClasses = tailwindClassesToApply.join(" ");
+            applyClassesToElement(jsxElement, mergedClasses);
+        }
+        else {
+            cleanCnClassesFromAttribute(initializer);
+        }
+    });
+}
+function extractCnClassesFromAttribute(initializer) {
+    const classes = [];
+    if (isStringLiteralLike(initializer)) {
+        return extractCnClasses(initializer.getLiteralText());
+    }
+    if (!Node.isJsxExpression(initializer)) {
+        return classes;
+    }
+    const expression = initializer.getExpression();
+    if (!expression) {
+        return classes;
+    }
+    if (isStringLiteralLike(expression)) {
+        return extractCnClasses(expression.getLiteralText());
+    }
+    if (Node.isCallExpression(expression) && isCnCall(expression)) {
+        for (const argument of expression.getArguments()) {
+            if (isStringLiteralLike(argument)) {
+                classes.push(...extractCnClasses(argument.getLiteralText()));
+            }
+        }
+    }
+    return classes;
+}
+function cleanCnClassesFromAttribute(initializer) {
+    if (isStringLiteralLike(initializer)) {
+        const cleaned = removeCnClasses(initializer.getLiteralText());
+        initializer.setLiteralValue(cleaned);
+        return;
+    }
+    if (!Node.isJsxExpression(initializer)) {
+        return;
+    }
+    const expression = initializer.getExpression();
+    if (!expression) {
+        return;
+    }
+    if (isStringLiteralLike(expression)) {
+        const cleaned = removeCnClasses(expression.getLiteralText());
+        expression.setLiteralValue(cleaned);
+        return;
+    }
+    if (Node.isCallExpression(expression) && isCnCall(expression)) {
+        for (const argument of expression.getArguments()) {
+            if (isStringLiteralLike(argument)) {
+                const cleaned = removeCnClasses(argument.getLiteralText());
+                argument.setLiteralValue(cleaned);
+            }
+        }
+        removeEmptyArgumentsFromCnCall(expression);
+    }
+}
+function extractCnClasses(str) {
+    const matches = str.matchAll(/\bcn-[\w-]+\b/g);
+    return Array.from(matches, (match) => match[0]);
+}
+function extractCnClass(str) {
+    const classes = extractCnClasses(str);
+    return classes[0] ?? null;
+}
+function removeCnClasses(str) {
+    return str
+        .replace(/\bcn-[\w-]+\b/g, (match) => {
+        // Preserve allowlisted classes
+        if (ALLOWLIST.has(match)) {
+            return match;
+        }
+        return "";
+    })
+        .replace(/\s+/g, " ")
+        .trim();
+}
+function removeEmptyArgumentsFromCnCall(callExpression) {
+    if (!isCnCall(callExpression)) {
+        return;
+    }
+    const args = callExpression.getArguments();
+    const nonEmptyArgs = args.filter((arg) => {
+        if (isStringLiteralLike(arg)) {
+            const text = arg.getLiteralText().trim();
+            return text !== "";
+        }
+        return true;
+    });
+    if (nonEmptyArgs.length !== args.length) {
+        const argTexts = nonEmptyArgs.map((arg) => arg.getText());
+        const parent = callExpression.getParent();
+        if (parent && Node.isJsxExpression(parent)) {
+            parent.replaceWithText(`{cn(${argTexts.join(", ")})}`);
+        }
+        else {
+            callExpression.replaceWithText(`cn(${argTexts.join(", ")})`);
+        }
+    }
+}
+function applyClassesToElement(element, tailwindClasses) {
+    if (!Node.isJsxOpeningElement(element) &&
+        !Node.isJsxSelfClosingElement(element)) {
+        return;
+    }
+    const attribute = element
+        .getAttributes()
+        .find((attr) => Node.isJsxAttribute(attr) &&
+        attr.getNameNode().getText() === "className");
+    if (!attribute || !Node.isJsxAttribute(attribute)) {
+        element.addAttribute({
+            name: "className",
+            initializer: `{cn(${JSON.stringify(tailwindClasses)})}`,
+        });
+        return;
+    }
+    const initializer = attribute.getInitializer();
+    if (!initializer) {
+        attribute.setInitializer(`{cn(${JSON.stringify(tailwindClasses)})}`);
+        return;
+    }
+    if (isStringLiteralLike(initializer)) {
+        const existing = initializer.getLiteralText();
+        const updated = removeCnClasses(mergeClasses(tailwindClasses, existing));
+        initializer.setLiteralValue(updated);
+        return;
+    }
+    if (!Node.isJsxExpression(initializer)) {
+        return;
+    }
+    const expression = initializer.getExpression();
+    if (!expression) {
+        attribute.setInitializer(`{cn(${JSON.stringify(tailwindClasses)})}`);
+        return;
+    }
+    if (isStringLiteralLike(expression)) {
+        const existing = expression.getLiteralText();
+        const updated = removeCnClasses(mergeClasses(tailwindClasses, existing));
+        expression.setLiteralValue(updated);
+        return;
+    }
+    if (Node.isCallExpression(expression) && isCnCall(expression)) {
+        const firstArg = expression.getArguments()[0];
+        if (isStringLiteralLike(firstArg)) {
+            const existing = firstArg.getLiteralText();
+            const updated = removeCnClasses(mergeClasses(tailwindClasses, existing));
+            firstArg.setLiteralValue(updated);
+            for (let i = 1; i < expression.getArguments().length; i++) {
+                const arg = expression.getArguments()[i];
+                if (isStringLiteralLike(arg)) {
+                    const argText = arg.getLiteralText();
+                    const cleaned = removeCnClasses(argText);
+                    if (cleaned !== argText) {
+                        arg.setLiteralValue(cleaned);
+                    }
+                }
+            }
+            removeEmptyArgumentsFromCnCall(expression);
+            return;
+        }
+        const argumentTexts = expression
+            .getArguments()
+            .map((argument) => {
+            if (isStringLiteralLike(argument)) {
+                const cleaned = removeCnClasses(argument.getLiteralText());
+                return cleaned ? JSON.stringify(cleaned) : null;
+            }
+            return argument.getText();
+        })
+            .filter((arg) => arg !== null);
+        const updatedArguments = [JSON.stringify(tailwindClasses), ...argumentTexts];
+        attribute.setInitializer(`{cn(${updatedArguments.join(", ")})}`);
+        return;
+    }
+    attribute.setInitializer(`{cn(${JSON.stringify(tailwindClasses)}, ${expression.getText()})}`);
+}
+function mergeClasses(newClasses, existing) {
+    const existingParts = existing.split(/\s+/).filter(Boolean);
+    const newParts = newClasses.split(/\s+/).filter(Boolean);
+    const combined = [...newParts, ...existingParts];
+    return combined.join(" ").trim();
+}
+function isCnCall(call) {
+    const expression = call.getExpression();
+    return Node.isIdentifier(expression) && expression.getText() === "cn";
+}
+function applyToMergePropsCalls(sourceFile, styleMap, matchedClasses) {
+    sourceFile.forEachDescendant((node) => {
+        if (!Node.isCallExpression(node)) {
+            return;
+        }
+        const expression = node.getExpression();
+        if (!Node.isIdentifier(expression) ||
+            expression.getText() !== "mergeProps") {
+            return;
+        }
+        // Look for object literals in mergeProps arguments
+        for (const arg of node.getArguments()) {
+            if (!Node.isObjectLiteralExpression(arg)) {
+                continue;
+            }
+            // Find className property in the object literal
+            const classNameProp = arg
+                .getProperties()
+                .find((prop) => Node.isPropertyAssignment(prop) &&
+                Node.isIdentifier(prop.getNameNode()) &&
+                prop.getNameNode().getText() === "className");
+            if (!classNameProp || !Node.isPropertyAssignment(classNameProp)) {
+                continue;
+            }
+            const classNameInitializer = classNameProp.getInitializer();
+            if (!classNameInitializer) {
+                continue;
+            }
+            // Handle cn() calls in className
+            if (Node.isCallExpression(classNameInitializer) &&
+                isCnCall(classNameInitializer)) {
+                const cnClasses = extractCnClassesFromCnCall(classNameInitializer);
+                if (cnClasses.length === 0) {
+                    continue;
+                }
+                const unmatchedClasses = cnClasses.filter((cnClass) => !matchedClasses.has(cnClass));
+                if (unmatchedClasses.length === 0) {
+                    // Clean up cn-* classes even if already matched
+                    cleanCnClassesFromCnCall(classNameInitializer);
+                    continue;
+                }
+                const tailwindClassesToApply = unmatchedClasses
+                    .map((cnClass) => styleMap[cnClass])
+                    .filter((classes) => Boolean(classes));
+                if (tailwindClassesToApply.length > 0) {
+                    const mergedClasses = tailwindClassesToApply.join(" ");
+                    applyClassesToCnCall(classNameInitializer, mergedClasses, matchedClasses, unmatchedClasses);
+                }
+                else {
+                    cleanCnClassesFromCnCall(classNameInitializer);
+                }
+            }
+        }
+    });
+}
+function extractCnClassesFromCnCall(cnCall) {
+    const classes = [];
+    for (const argument of cnCall.getArguments()) {
+        if (isStringLiteralLike(argument)) {
+            classes.push(...extractCnClasses(argument.getLiteralText()));
+        }
+    }
+    return classes;
+}
+function cleanCnClassesFromCnCall(cnCall) {
+    for (const argument of cnCall.getArguments()) {
+        if (isStringLiteralLike(argument)) {
+            const cleaned = removeCnClasses(argument.getLiteralText());
+            argument.setLiteralValue(cleaned);
+        }
+    }
+    removeEmptyArgumentsFromCnCall(cnCall);
+}
+function applyClassesToCnCall(cnCall, tailwindClasses, matchedClasses, unmatchedClasses) {
+    const firstArg = cnCall.getArguments()[0];
+    if (isStringLiteralLike(firstArg)) {
+        const existing = firstArg.getLiteralText();
+        const updated = removeCnClasses(mergeClasses(tailwindClasses, existing));
+        firstArg.setLiteralValue(updated);
+        // Mark classes as matched
+        unmatchedClasses.forEach((cnClass) => matchedClasses.add(cnClass));
+        // Clean up cn-* classes from remaining arguments
+        for (let i = 1; i < cnCall.getArguments().length; i++) {
+            const arg = cnCall.getArguments()[i];
+            if (isStringLiteralLike(arg)) {
+                const argText = arg.getLiteralText();
+                const cleaned = removeCnClasses(argText);
+                if (cleaned !== argText) {
+                    arg.setLiteralValue(cleaned);
+                }
+            }
+        }
+        removeEmptyArgumentsFromCnCall(cnCall);
+        return;
+    }
+    // If first arg is not a string literal, prepend tailwind classes
+    const argumentTexts = cnCall
+        .getArguments()
+        .map((argument) => {
+        if (isStringLiteralLike(argument)) {
+            const cleaned = removeCnClasses(argument.getLiteralText());
+            return cleaned ? JSON.stringify(cleaned) : null;
+        }
+        return argument.getText();
+    })
+        .filter((arg) => arg !== null);
+    const updatedArguments = [JSON.stringify(tailwindClasses), ...argumentTexts];
+    // Mark classes as matched
+    unmatchedClasses.forEach((cnClass) => matchedClasses.add(cnClass));
+    const parent = cnCall.getParent();
+    if (parent) {
+        cnCall.replaceWithText(`cn(${updatedArguments.join(", ")})`);
+    }
+}
 //# sourceMappingURL=transform-style-map.js.map