applyCleanup() — ui Function Reference

Source Code

packages/shadcn/src/utils/transformers/transform-cleanup.ts lines 52–124

export function applyCleanup(sourceFile: SourceFile) {
  // Collect attributes to remove (can't remove while iterating).
  const attributesToRemove: ReturnType<
    typeof sourceFile.getDescendantsOfKind<typeof SyntaxKind.JsxAttribute>
  > = []

  // Process all JSX className attributes.
  for (const attr of sourceFile.getDescendantsOfKind(SyntaxKind.JsxAttribute)) {
    const attrName = attr.getNameNode().getText()
    if (attrName !== "className" && attrName !== "classNames") {
      continue
    }

    const initializer = attr.getInitializer()

    // className="..."
    if (initializer?.isKind(SyntaxKind.StringLiteral)) {
      const currentValue = initializer.getLiteralValue()
      if (CN_MARKER_REGEX.test(currentValue)) {
        const newValue = stripCnMarkers(currentValue)
        if (newValue === "") {
          // Remove the entire attribute if className becomes empty.
          attributesToRemove.push(attr)
        } else if (newValue !== currentValue) {
          initializer.setLiteralValue(newValue)
        }
      }
    }

    // className={...} or classNames={{...}}
    if (initializer?.isKind(SyntaxKind.JsxExpression)) {
      processStringLiterals(initializer)
    }
  }

  // Remove empty className attributes.
  for (const attr of attributesToRemove) {
    attr.remove()
  }

  // Process cva() calls.
  for (const call of sourceFile.getDescendantsOfKind(
    SyntaxKind.CallExpression
  )) {
    if (call.getExpression().getText() !== "cva") {
      continue
    }

    for (const arg of call.getArguments()) {
      if (arg.isKind(SyntaxKind.StringLiteral)) {
        processStringLiteral(arg)
        continue
      }
      if (arg.isKind(SyntaxKind.NoSubstitutionTemplateLiteral)) {
        processStringLiteral(arg)
        continue
      }
      // Handle object arguments (variants).
      processStringLiterals(arg)
    }
  }

  // Process mergeProps() calls.
  for (const call of sourceFile.getDescendantsOfKind(
    SyntaxKind.CallExpression
  )) {
    if (call.getExpression().getText() !== "mergeProps") {
      continue
    }

    processStringLiterals(call)
  }
}