Function transform

packages/extension/mcp/tools/code/variables.ts:35–70

(
  styles: Map<string, Record<string, string>>,
  { pluginCode, config }: TransformOptions
)

}

export function transform(
  styles: Map<string, Record<string, string>>,
  { pluginCode, config }: TransformOptions
): Promise<Set<string>> {
  const { references, buckets } = collectRefs(styles)
  if (!references.length) return Promise.resolve(new Set())

  return runTransformVariableBatch(
    references.map(({ code, name, value }) => ({ code, name, value })),
    workerUnitOptions(config),
    pluginCode
  ).then((transformResults) => {
    const replacements = transformResults.map((result) => {
      const noFallback = stripFallback(result)
      const canonicalName = canonicalizeVarName(noFallback)
      return canonicalName ? toVarExpr(canonicalName) : noFallback
    })
    const usedNames = new Set<string>()
    // Keep original reference names to handle transforms that inline literal values.
    references.forEach(({ name }) => usedNames.add(`--${name}`))
    replacements.forEach((repl) => extractVarNames(repl).forEach((n) => usedNames.add(n)))

    for (const bucket of buckets.values()) {
      const style = styles.get(bucket.nodeId)
      if (!style) continue

      let occurrence = 0
      style[bucket.property] = replaceVarFunctions(bucket.value, ({ full }) => {
        const refIndex = bucket.matchIndices[occurrence++]!
        return replacements[refIndex] ?? full
      })
    }

    return usedNames
  })
}

export function collectRefs(styles: Map<string, Record<string, string>>) {
  const references: VariableReferenceInternal[] = []