transform-cleanup.ts — ui Source File

Source Code

import { Transformer } from "@/src/utils/transformers"
import {
  NoSubstitutionTemplateLiteral,
  Node,
  Project,
  ScriptKind,
  SourceFile,
  StringLiteral,
  SyntaxKind,
} from "ts-morph"

// Regex to match cn-* marker classes (e.g., cn-rtl-flip, cn-logical-sides).
const CN_MARKER_REGEX = /\bcn-[a-z-]+\b/
const CN_MARKER_REGEX_GLOBAL = /\bcn-[a-z-]+\b/g

// Helper to strip all cn-* marker classes from a className string.
export function stripCnMarkers(className: string) {
  return className
    .replace(CN_MARKER_REGEX_GLOBAL, "")
    .replace(/\s+/g, " ")
    .trim()
}

type StringLikeLiteral = StringLiteral | NoSubstitutionTemplateLiteral

// Processes a string-like literal and strips cn-* markers.
function processStringLiteral(node: StringLikeLiteral) {
  const currentValue = node.getLiteralValue()
  if (!CN_MARKER_REGEX.test(currentValue)) {
    return
  }

  const newValue = stripCnMarkers(currentValue)
  if (newValue !== currentValue) {
    node.setLiteralValue(newValue)
  }
}

function processStringLiterals(node: Node) {
  for (const stringLit of node.getDescendantsOfKind(SyntaxKind.StringLiteral)) {
    processStringLiteral(stringLit)
  }

  for (const templateLit of node.getDescendantsOfKind(
    SyntaxKind.NoSubstitutionTemplateLiteral
  )) {
    processStringLiteral(templateLit)
  }
}

// Apply cleanup to a SourceFile directly (used by transformDirection).
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()
// ... (87 more lines)