@csszyx/unplugin 0.10.11 → 0.11.0

package/dist/shared/{unplugin.CiSDltfp.mjs → unplugin.BS60onPk.mjs}

@@ -1,3 +1,4 @@
+import { createHash } from 'node:crypto';
 import * as fs from 'node:fs';
 import { mkdirSync, writeFileSync } from 'node:fs';
 import { createRequire } from 'node:module';
@@ -5,7 +6,7 @@ import * as path from 'node:path';
 import { dirname } from 'node:path';
 import { performance } from 'node:perf_hooks';
 import { fileURLToPath } from 'node:url';
-import { scanGlobalVarUsages, sortStrings as sortStrings$1, ensureRustTransformAvailable, transformSourceCode, transformRust, transformOxc, transformRustBatch, transform, isRustTransformAvailable } from '@csszyx/compiler';
+import { scanGlobalVarUsages, sortStrings as sortStrings$1, ensureRustTransformAvailable, transformSourceCode, transformRust, transformOxc, transformRustBatch, ASTBudgetExceededError, transform, isRustTransformAvailable } from '@csszyx/compiler';
 import { encode, compute_mangle_checksum } from '@csszyx/core';
 import { getNativePackageName } from '@csszyx/core/native';
 import { preprocess as preprocess$1 } from '@csszyx/svelte-adapter';
@@ -14,8 +15,8 @@ import { preprocess } from '@csszyx/vue-adapter';
 import { createUnplugin } from 'unplugin';
 import { mangleCSSSync } from '../css-mangler.mjs';
 import { s as sortStrings, e as escapeHtmlAttribute } from './unplugin.B1mblcm-.mjs';
-import { createHash } from 'node:crypto';
-import { r as resolveTransformCacheDir, c as createTransformCacheKey, a as readTransformCache, w as writeTransformCache, e as evictOldTransformCacheEntries, b as evictMemoryCacheToBudget } from './unplugin.CGqFVGlB.mjs';
+import { i as importsRuntimeHelper, f as findRuntimeImportClause } from './unplugin.B3RHYokB.mjs';
+import { r as resolveTransformCacheDir, c as createTransformCacheKey, a as readTransformCache, w as writeTransformCache, e as evictOldTransformCacheEntries, b as evictMemoryCacheToBudget } from './unplugin.ByzV6iZE.mjs';
 import postcss from 'postcss';
 import valueParser from 'postcss-value-parser';
 
@@ -802,19 +803,118 @@ function readRuntimeImportSymbols(source, clause) {
 }
 function findLocalImportSources(code) {
   const out = [];
-  const staticImportRe = /import\s+(?!type\b)(?:\S(?:.*\S)?\s+from\s+)?['"]([^'"]+)['"]/g;
-  const exportFromRe = /export\s+(?!type\b)\S(?:.*\S)?\s+from\s+['"]([^'"]+)['"]/g;
-  const dynamicImportRe = /import\s*\(\s*['"]([^'"]+)['"]\s*\)/g;
-  for (const re of [staticImportRe, exportFromRe, dynamicImportRe]) {
-    for (const match of code.matchAll(re)) {
-      const source = match[1];
-      if (source.startsWith(".") || source.startsWith("/")) {
-        out.push(source);
-      }
+  for (const line of code.split("\n")) {
+    for (const spec of staticImportSpecifiers(line)) {
+      pushIfLocal(out, spec);
     }
   }
+  for (const line of code.split("\n")) {
+    for (const spec of exportFromSpecifiers(line)) {
+      pushIfLocal(out, spec);
+    }
+  }
+  for (const match of code.matchAll(/import\s*\(\s*['"]([^'"]+)['"]\s*\)/g)) {
+    pushIfLocal(out, match[1]);
+  }
   return out;
 }
+function pushIfLocal(out, spec) {
+  if (spec !== null && (spec.startsWith(".") || spec.startsWith("/"))) {
+    out.push(spec);
+  }
+}
+function staticImportSpecifiers(line) {
+  const specs = [];
+  for (const kw of allOccurrences(line, "import")) {
+    const afterKw = skipSpaces(line, kw + "import".length);
+    if (afterKw === kw + "import".length || startsWithWord(line, afterKw, "type")) {
+      continue;
+    }
+    if (line[afterKw] === '"' || line[afterKw] === "'") {
+      const spec2 = readQuoted(line, afterKw);
+      if (spec2 !== null) {
+        specs.push(spec2);
+      }
+      continue;
+    }
+    const spec = specifierAfterFrom(line, afterKw);
+    if (spec !== null) {
+      specs.push(spec);
+    }
+  }
+  return specs;
+}
+function exportFromSpecifiers(line) {
+  const specs = [];
+  for (const kw of allOccurrences(line, "export")) {
+    const afterKw = skipSpaces(line, kw + "export".length);
+    if (afterKw === kw + "export".length || startsWithWord(line, afterKw, "type")) {
+      continue;
+    }
+    const spec = specifierAfterFrom(line, afterKw);
+    if (spec !== null) {
+      specs.push(spec);
+    }
+  }
+  return specs;
+}
+function specifierAfterFrom(line, clauseStart) {
+  const fromAt = findKeywordFrom(line, clauseStart);
+  if (fromAt === -1) {
+    return null;
+  }
+  if (skipSpaces(line, clauseStart) >= fromAt) {
+    return null;
+  }
+  const afterFrom = skipSpaces(line, fromAt + "from".length);
+  return readQuoted(line, afterFrom);
+}
+function findKeywordFrom(line, at) {
+  let i = line.indexOf("from", at);
+  while (i !== -1) {
+    const before = line[i - 1];
+    const after = line[i + "from".length];
+    if (before !== void 0 && /\s/.test(before) && after !== void 0 && /\s/.test(after)) {
+      return i;
+    }
+    i = line.indexOf("from", i + 1);
+  }
+  return -1;
+}
+function readQuoted(line, at) {
+  const quote = line[at];
+  if (quote !== '"' && quote !== "'") {
+    return null;
+  }
+  const close = line.indexOf(quote, at + 1);
+  if (close <= at + 1) {
+    return null;
+  }
+  return line.slice(at + 1, close);
+}
+function allOccurrences(line, word) {
+  const positions = [];
+  let i = line.indexOf(word);
+  while (i !== -1) {
+    positions.push(i);
+    i = line.indexOf(word, i + 1);
+  }
+  return positions;
+}
+function startsWithWord(line, at, word) {
+  if (!line.startsWith(word, at)) {
+    return false;
+  }
+  const after = line[at + word.length];
+  return after === void 0 || !/\w/.test(after);
+}
+function skipSpaces(line, at) {
+  let i = at;
+  while (i < line.length && /\s/.test(line[i])) {
+    i++;
+  }
+  return i;
+}
 function normalizeModuleId(id) {
   const clean = id.split("?")[0] ?? id;
   const cached = normalizedModuleIdCache.get(clean);
@@ -1113,10 +1213,9 @@ function parseThemeBlocks(cssContent) {
   };
   const stripped = stripLayerWrappers(cssContent);
   const blocks = extractThemeBlocks(stripped);
-  const propPattern = /--([a-z][a-z0-9-]*)(?:\s*:[^;]+)?;/g;
   for (const block of blocks) {
-    for (const match of block.matchAll(propPattern)) {
-      const categorized = categorizeProperty(match[1]);
+    for (const name of scanCustomPropertyNames(block)) {
+      const categorized = categorizeProperty(name);
       if (categorized) {
         result[categorized.category].add(categorized.token);
       }
@@ -1168,6 +1267,48 @@ function mergeThemes(themes) {
 function hasTokens(theme) {
   return Object.values(theme).some((arr) => arr.length > 0);
 }
+function scanCustomPropertyNames(block) {
+  const names = [];
+  let i = 0;
+  while (i < block.length) {
+    const dashes = block.indexOf("--", i);
+    if (dashes === -1) {
+      break;
+    }
+    let end = dashes + 2;
+    if (end >= block.length || !/[a-z]/.test(block[end])) {
+      i = dashes + 1;
+      continue;
+    }
+    end++;
+    while (end < block.length && /[a-z0-9-]/.test(block[end])) {
+      end++;
+    }
+    const name = block.slice(dashes + 2, end);
+    let matchEnd = -1;
+    let cursor = end;
+    while (cursor < block.length && /\s/.test(block[cursor])) {
+      cursor++;
+    }
+    if (block[cursor] === ":") {
+      const valueStart = cursor + 1;
+      const semi = block.indexOf(";", valueStart);
+      if (semi > valueStart) {
+        matchEnd = semi + 1;
+      }
+    }
+    if (matchEnd === -1 && block[end] === ";") {
+      matchEnd = end + 1;
+    }
+    if (matchEnd === -1) {
+      i = dashes + 1;
+      continue;
+    }
+    names.push(name);
+    i = matchEnd;
+  }
+  return names;
+}
 
 const GLOB_MAGIC_RE = /[*?[\]{}]/;
 const DEFAULT_IGNORED_DIRS = /* @__PURE__ */ new Set(["node_modules", ".git", ".next", ".turbo", "dist", "build"]);
@@ -1289,16 +1430,29 @@ function safeJsonForScriptTag(value, prettyPrint = false) {
   const json = prettyPrint ? JSON.stringify(value, null, 2) : JSON.stringify(value);
   return json.replace(/</g, "\\u003C").replace(/>/g, "\\u003E").replace(/&/g, "\\u0026").replace(/\u2028/g, "\\u2028").replace(/\u2029/g, "\\u2029");
 }
+function injectHtmlOpeningAttr(html, attr) {
+  const lower = html.toLowerCase();
+  let from = 0;
+  for (; ; ) {
+    const start = lower.indexOf("<html", from);
+    if (start === -1) {
+      return html;
+    }
+    const after = html[start + 5];
+    if (after === void 0 || after === ">" || after === "/" || /\s/.test(after)) {
+      const close = html.indexOf(">", start + 5);
+      if (close === -1) {
+        return html;
+      }
+      const nameEnd = start + 5;
+      return `${html.slice(0, nameEnd)}${attr}${html.slice(nameEnd)}`;
+    }
+    from = start + 5;
+  }
+}
 function injectChecksum(html, checksum, minify = false) {
   const attrName = minify ? "data-sz-cs" : "data-sz-checksum";
-  const htmlTagPattern = /<html([^>]*)>/i;
-  const match = html.match(htmlTagPattern);
-  if (!match) {
-    return html;
-  }
-  const existingAttrs = match[1];
-  const checksumAttr = ` ${attrName}="${checksum}"`;
-  return html.replace(htmlTagPattern, `<html${checksumAttr}${existingAttrs}>`);
+  return injectHtmlOpeningAttr(html, ` ${attrName}="${checksum}"`);
 }
 function injectMangleMapScript(html, mangleMap, options = {}) {
   const {
@@ -1327,14 +1481,7 @@ ${debugScript}`;
 function injectMangleMapAttribute(html, mangleMap, minify = false, varMangleMap = {}) {
   const attrName = minify ? "data-sz-m" : "data-sz-map";
   const jsonContent = JSON.stringify(createHydrationMangleMap(mangleMap, varMangleMap));
-  const htmlTagPattern = /<html([^>]*)>/i;
-  const match = html.match(htmlTagPattern);
-  if (!match) {
-    return html;
-  }
-  const existingAttrs = match[1];
-  const mapAttr = ` ${attrName}='${jsonContent}'`;
-  return html.replace(htmlTagPattern, `<html${mapAttr}${existingAttrs}>`);
+  return injectHtmlOpeningAttr(html, ` ${attrName}='${jsonContent}'`);
 }
 function injectHydrationData(html, mangleMap, checksum, options = {}) {
   const { mode = "script", minify = false } = options;
@@ -1622,11 +1769,26 @@ const MAX_SAFELIST_CLASSES = 1e5;
 const DEFAULT_VAR_MANGLE_MAP_MAX_BYTES = 100 * 1024;
 const GLOBAL_VAR_ALIAS_MAP_OWNER = "\0csszyx:global-var-aliases";
 const DIRECTIVE_PROLOGUE_PREFIX_RE = /^((?:\s|\/\/[^\n]*\n|\/\*(?:[^*]|\*(?!\/))*\*\/)*)(['"]use (?:client|server)['"];?\s*)/;
-const RUNTIME_HELPER_IMPORT_RE = {
-  _sz: /(?:import|export)\s+\{[^{}]*\b_sz\b[^{}]*\}\s*from\s*['"]@csszyx\/runtime['"]/,
-  _szMerge: /(?:import|export)\s+\{[^{}]*\b_szMerge\b[^{}]*\}\s*from\s*['"]@csszyx\/runtime['"]/,
-  __szColorVar: /(?:import|export)\s+\{[^{}]*\b__szColorVar\b[^{}]*\}\s*from\s*['"]@csszyx\/runtime['"]/
-};
+function findOpeningTag(source, tag) {
+  const lower = source.toLowerCase();
+  const marker = `<${tag}`;
+  let from = 0;
+  for (; ; ) {
+    const start = lower.indexOf(marker, from);
+    if (start === -1) {
+      return null;
+    }
+    const after = source[start + marker.length];
+    if (after === void 0 || after === ">" || after === "/" || /\s/.test(after)) {
+      const close = source.indexOf(">", start + marker.length);
+      if (close !== -1) {
+        return { start, close };
+      }
+      return null;
+    }
+    from = start + marker.length;
+  }
+}
 let _hasWarnedTsConfig = false;
 let _hasWarnedTransformCacheVersion = false;
 let _hasWarnedNativeFallback = false;
@@ -1898,7 +2060,7 @@ function shouldTrackGlobalVarSources(config) {
 }
 function recordGlobalVarSourceFile(state, filename, code) {
   const normalizedFilename = normalizeSourceFilename(filename);
-  if (!/\.[tj]sx?(?:\?.*)?$/.test(normalizedFilename)) {
+  if (!matchesScriptExtension(normalizedFilename, SCRIPT_ID_EXTENSIONS)) {
     return;
   }
   if (code === null) {
@@ -2245,6 +2407,11 @@ function findPackageVersionFromFile(file, fallback) {
 function normalizeSourceFilename(filename) {
   return filename.replace(/\\/g, "/");
 }
+const SCRIPT_ID_EXTENSIONS = [".ts", ".tsx", ".js", ".jsx"];
+const SOURCE_MODULE_EXTENSIONS = [...SCRIPT_ID_EXTENSIONS, ".cts", ".mts", ".cjs", ".mjs"];
+function matchesScriptExtension(id, extensions) {
+  return extensions.some((ext) => id.endsWith(ext) || id.includes(`${ext}?`));
+}
 function insertRuntimeImport(code, importStmt) {
   const directiveMatch = code.match(DIRECTIVE_PROLOGUE_PREFIX_RE);
   if (!directiveMatch) {
@@ -2453,18 +2620,49 @@ function mangleCodeClassesSync(code, mangleMap) {
     }
     return `${sep}${ws}"${mangled.join(" ")}"`;
   });
-  result = result.replace(/\bszs:\s*\{([^{}]*)\}/g, (whole, body) => {
-    const mangledBody = body.replace(
-      /"((?:[^"\\]|\\.)*)"/g,
-      (_m, inner) => `"${mangleClassString(inner)}"`
-    ).replace(
-      /'((?:[^'\\]|\\.)*)'/g,
-      (_m, inner) => `'${mangleClassString(inner)}'`
-    );
+  result = result.replace(/\bszsc:\s*\{([^{}]*)\}/g, (whole, body) => {
+    const mangledBody = mangleQuotedStringLiterals(body, mangleClassString);
     return whole.replace(body, mangledBody);
   });
   return result;
 }
+function mangleQuotedStringLiterals(body, mangle) {
+  let out = "";
+  let i = 0;
+  while (i < body.length) {
+    const quote = body[i];
+    if (quote !== '"' && quote !== "'") {
+      out += quote;
+      i++;
+      continue;
+    }
+    let inner = "";
+    let j = i + 1;
+    let closed = false;
+    while (j < body.length) {
+      const ch = body[j];
+      if (ch === "\\") {
+        inner += ch + (body[j + 1] ?? "");
+        j += 2;
+        continue;
+      }
+      if (ch === quote) {
+        closed = true;
+        break;
+      }
+      inner += ch;
+      j++;
+    }
+    if (!closed) {
+      out += quote;
+      i++;
+      continue;
+    }
+    out += `${quote}${mangle(inner)}${quote}`;
+    i = j + 1;
+  }
+  return out;
+}
 function assertGlobalVarMangleConfig(options) {
   const config = options.production?.mangleGlobalVars;
   const errors = validateGlobalVarMangleConfig(config);
@@ -2492,6 +2690,7 @@ function createCsszyxPlugins(options = {}) {
   let manglingEnabled = options.production?.mangle !== false;
   const mangleReserved = new Set(options.production?.mangleExclude ?? []);
   const astBudgetOverride = options.build?.astBudgetLimit;
+  const prescanAstBudget = astBudgetOverride ?? 5e5;
   const cacheRequested = (options.build?.cache ?? DEFAULT_BUILD_CONFIG.cache) !== false;
   const cacheVersionsKnown = PLUGIN_VERSION !== UNKNOWN_PACKAGE_VERSION && COMPILER_VERSION !== UNKNOWN_PACKAGE_VERSION;
   const cacheEnabled = cacheRequested && cacheVersionsKnown;
@@ -2539,6 +2738,7 @@ function createCsszyxPlugins(options = {}) {
   let evictedCacheRoot = null;
   const transformMemoryCache = /* @__PURE__ */ new Map();
   let transformMemoryCacheCodeChars = 0;
+  const prescanResultHandoff = /* @__PURE__ */ new Map();
   const state = {
     classes: /* @__PURE__ */ new Set(),
     parsedTheme: null,
@@ -2646,13 +2846,13 @@ function createCsszyxPlugins(options = {}) {
     return isHardIgnoredPath(id, compileSourceDirs);
   }
   function shouldProcessSource(id) {
-    return !isHardIgnored(id) && !isUserExcluded(id) && isUserIncluded(id) && (/\.([cm]?[tj]s|[tj]sx)(\?.*)?$/.test(id) || id.endsWith(".vue") || id.endsWith(".svelte"));
+    return !isHardIgnored(id) && !isUserExcluded(id) && isUserIncluded(id) && (matchesScriptExtension(id, SOURCE_MODULE_EXTENSIONS) || id.endsWith(".vue") || id.endsWith(".svelte"));
   }
   function shouldProcessCss(id) {
-    return !isHardIgnored(id) && !isUserExcluded(id) && /\.css(\?.*)?$/.test(id);
+    return !isHardIgnored(id) && !isUserExcluded(id) && matchesScriptExtension(id, [".css"]);
   }
-  function transformConfiguredSource(source, filename) {
-    const compilerOptions = createCompilerOptions();
+  function transformConfiguredSource(source, filename, astBudget) {
+    const compilerOptions = createCompilerOptions(astBudget);
     const effectiveFilename = normalizeSourceFilename(filename);
     const cacheRoot = resolveTransformCacheDir(state.rootDir, options.build?.cacheDir);
     if (cacheEnabled) {
@@ -2679,6 +2879,15 @@ function createCsszyxPlugins(options = {}) {
         rememberTransformCacheEntry(cacheKey.key, cached);
         return cached;
       }
+      if (astBudget === void 0) {
+        const handoff = prescanResultHandoff.get(effectiveFilename);
+        if (handoff) {
+          prescanResultHandoff.delete(effectiveFilename);
+          if (handoff.inputSha256 === cacheKey.inputSha256) {
+            return handoff.result;
+          }
+        }
+      }
     }
     let result;
     if (parserMode === "babel") {
@@ -2709,9 +2918,9 @@ function createCsszyxPlugins(options = {}) {
     }
     return result;
   }
-  function createCompilerOptions() {
+  function createCompilerOptions(astBudget = astBudgetOverride) {
     return {
-      astBudget: astBudgetOverride,
+      astBudget,
       mangleVars: options.production?.mangleVars === true,
       mangleVarHoistMaxDepth: options.production?.mangleVarHoistMaxDepth,
       globalVarAliases: earlyGlobalVarAliasEntries.length > 0 ? earlyGlobalVarAliasEntries : void 0,
@@ -2726,7 +2935,10 @@ function createCsszyxPlugins(options = {}) {
       nativeIdentity: parserMode === "rust" ? resolveNativeCacheIdentity() : void 0,
       parserMode,
       producer: parserMode,
-      astBudget: astBudgetOverride,
+      // The EFFECTIVE budget, not the raw override: prescan-lane results
+      // (larger budget) must not be served to the transform hook, whose
+      // smaller budget could not have produced them (and vice versa).
+      astBudget: compilerOptions.astBudget,
       mangleVars: compilerOptions.mangleVars,
       mangleVarHoistMaxDepth: compilerOptions.mangleVarHoistMaxDepth,
       globalVarAliases: normalizeGlobalVarAliasesForCache(compilerOptions.globalVarAliases),
@@ -2738,7 +2950,7 @@ function createCsszyxPlugins(options = {}) {
     if (parserMode !== "rust" || files.length <= 1) {
       return transformPrescanSourcesIndividually(files);
     }
-    const compilerOptions = createCompilerOptions();
+    const compilerOptions = createCompilerOptions(prescanAstBudget);
     const cacheRoot = resolveTransformCacheDir(state.rootDir, options.build?.cacheDir);
     const results = /* @__PURE__ */ new Map();
     const misses = [];
@@ -2808,7 +3020,11 @@ function createCsszyxPlugins(options = {}) {
         try {
           results.set(
             miss.filePath,
-            transformConfiguredSource(miss.content, miss.effectiveFilename)
+            transformConfiguredSource(
+              miss.content,
+              miss.effectiveFilename,
+              prescanAstBudget
+            )
           );
         } catch {
         }
@@ -2819,15 +3035,28 @@ function createCsszyxPlugins(options = {}) {
       return result ? { filePath: file.filePath, result } : null;
     }).filter((entry) => entry !== null);
   }
+  function warnPrescanBudgetSkip(filePath) {
+    console.warn(
+      `[csszyx] prescan skipped ${filePath}: the file exceeds the AST node budget, so NONE of its classes reached the safelist and their CSS will not be generated. Raise \`build.astBudgetLimit\` in the csszyx plugin options, or split the file.`
+    );
+  }
   function transformPrescanSourcesIndividually(files) {
     const results = [];
     for (const file of files) {
       try {
         results.push({
           filePath: file.filePath,
-          result: transformConfiguredSource(file.content, file.filePath)
+          result: transformConfiguredSource(
+            file.content,
+            file.filePath,
+            prescanAstBudget
+          )
         });
       } catch (err) {
+        if (err instanceof ASTBudgetExceededError) {
+          warnPrescanBudgetSkip(file.filePath);
+          continue;
+        }
         console.warn(
           `[csszyx] prescan skipped ${file.filePath}: transform failed, so none of its classes reached the safelist. ${err instanceof Error ? err.message : String(err)}`
         );
@@ -2955,7 +3184,23 @@ function createCsszyxPlugins(options = {}) {
       }
       scanDir(sourceDir);
     }
+    const prescanContentByPath = new Map(
+      prescanSources.map((file) => [file.filePath, file.content])
+    );
     for (const { filePath, result } of transformPrescanSources(prescanSources)) {
+      if (cacheEnabled && !result.diagnostics.some((d) => d.includes("AST budget exceeded"))) {
+        const content = prescanContentByPath.get(filePath);
+        if (content !== void 0) {
+          prescanResultHandoff.set(normalizeSourceFilename(filePath), {
+            inputSha256: createHash("sha256").update(content).digest("hex"),
+            result
+          });
+        }
+      }
+      if (result.diagnostics.some((d) => d.includes("AST budget exceeded"))) {
+        warnPrescanBudgetSkip(filePath);
+        continue;
+      }
       if (result.classes.size === 0 && result.rawClassNames.size === 0 && result.diagnostics.some((d) => d.includes("[csszyx] parse error in "))) {
         console.warn(
           `[csszyx] prescan skipped ${filePath}: the file failed to parse, so none of its classes reached the safelist. Fix the syntax error (or check the file extension matches its contents).`
@@ -3211,10 +3456,10 @@ function createCsszyxPlugins(options = {}) {
         if (shouldProcessSource(id)) {
           trackGlobalVarSourceFile(id, code);
         }
-        if (/\.[tj]sx?(\?.*)?$/.test(id)) {
+        if (matchesScriptExtension(id, SCRIPT_ID_EXTENSIONS)) {
           assertNoRSCBoundaryViolation(code, id);
         }
-        if (/\.css(\?.*)?$/.test(id)) {
+        if (matchesScriptExtension(id, [".css"])) {
           state.sawAnyCss = true;
           if (cssImportsTailwind(code)) {
             state.sawTailwindEntry = true;
@@ -3238,6 +3483,8 @@ function createCsszyxPlugins(options = {}) {
         let transformedCode = code;
         let usesRuntime = false;
         let usesMerge = false;
+        let usesSzcn = false;
+        let usesSzPart = false;
         let usesColorVar = false;
         let transformed = false;
         let szClasses;
@@ -3266,6 +3513,8 @@ function createCsszyxPlugins(options = {}) {
             transformedCode = result.code;
             usesRuntime = result.usesRuntime;
             usesMerge = result.usesMerge;
+            usesSzcn = result.usesSzcn;
+            usesSzPart = result.usesSzPart;
             usesColorVar = result.usesColorVar;
             transformed = result.transformed;
             szClasses = result.classes;
@@ -3274,12 +3523,17 @@ function createCsszyxPlugins(options = {}) {
             for (const msg of result.diagnostics) {
               if (msg.includes("unresolvable sz spread")) {
                 state.spreadWarnings.add(`${id}
+  ${msg}`);
+              } else if (msg.includes("AST budget exceeded")) {
+                console.warn(`[csszyx] ${id}
   ${msg}`);
               }
             }
             if (!quiet && result.diagnostics.length > 0 && process.env.NODE_ENV !== "production") {
               for (const msg of result.diagnostics) {
-                if (msg.includes("unresolvable sz spread")) continue;
+                if (msg.includes("unresolvable sz spread") || msg.includes("AST budget exceeded")) {
+                  continue;
+                }
                 this.warn(`[csszyx] ${id}
   ${msg}`);
               }
@@ -3294,16 +3548,14 @@ function createCsszyxPlugins(options = {}) {
         }
         if (transformedCode.includes("<html") && /(?:layout|Root|Document|app)\.tsx?$/i.test(id)) {
           const attrName = options.production?.minify ? "data-sz-cs" : "data-sz-checksum";
-          transformedCode = transformedCode.replace(
-            /<html([^>]*)>/i,
-            `<html$1 ${attrName}="${CHECKSUM_PLACEHOLDER}">`
-          );
+          const htmlTag = findOpeningTag(transformedCode, "html");
+          if (htmlTag) {
+            transformedCode = `${transformedCode.slice(0, htmlTag.close)} ${attrName}="${CHECKSUM_PLACEHOLDER}"${transformedCode.slice(htmlTag.close)}`;
+          }
           const debugScript = `<script dangerouslySetInnerHTML={{__html: \`(function(){var m=${MANGLE_MAP_PLACEHOLDER};var vm=${VAR_MANGLE_MAP_PLACEHOLDER};var gp=decodeURIComponent(${escapeJsonForInlineScript(JSON.stringify(encodedGlobalVarAliasPrefix))});var r={};var vr={};for(var k in m)r[m[k]]=k;for(var vk in vm){var vv=vm[vk];var vs=Array.isArray(vv)?vv:[vv];for(var vi=0;vi<vs.length;vi++)(vr[vs[vi]]||(vr[vs[vi]]=[])).push(vk)}window.__csszyx={mangleMap:m,varMangleMap:vm,checksum:"${CHECKSUM_PLACEHOLDER}",decode:function(c){return r[c]},encode:function(c){return m[c]},decodeVar:function(v){return vr[v]||[]},encodeVar:function(v){return vm[v]},decodeGlobalVar:function(v){var a=vr[v]||[];return v.indexOf(gp)===0?a[0]:void 0},decodeAll:function(el){return(el.className||"").split(" ").map(function(c){return r[c]||c})}}})()\`}} />`;
-          if (transformedCode.includes("<body")) {
-            transformedCode = transformedCode.replace(
-              /(<body[^>]*>)/i,
-              `$1${debugScript}`
-            );
+          const bodyTag = findOpeningTag(transformedCode, "body");
+          if (bodyTag) {
+            transformedCode = `${transformedCode.slice(0, bodyTag.close + 1)}${debugScript}${transformedCode.slice(bodyTag.close + 1)}`;
           }
           transformed = true;
         }
@@ -3315,21 +3567,23 @@ function createCsszyxPlugins(options = {}) {
           if (usesMerge) {
             imports.push("_szMerge");
           }
+          if (usesSzcn) {
+            imports.push("_szcn");
+          }
+          if (usesSzPart) {
+            imports.push("_szPart");
+          }
           if (usesColorVar) {
             imports.push("__szColorVar");
           }
           const hasRuntimeImport = imports.length > 0 && transformedCode.includes("@csszyx/runtime");
-          const needed = hasRuntimeImport ? imports.filter(
-            (name) => !RUNTIME_HELPER_IMPORT_RE[name]?.test(transformedCode)
-          ) : imports;
+          const needed = hasRuntimeImport ? imports.filter((name) => !importsRuntimeHelper(transformedCode, name)) : imports;
           if (needed.length > 0) {
-            const existingImport = transformedCode.match(
-              /^(import\s*\{[^}]*)\}\s*from\s*'@csszyx\/runtime'/m
-            );
+            const existingImport = findRuntimeImportClause(transformedCode);
             if (existingImport) {
               transformedCode = transformedCode.replace(
-                existingImport[0],
-                `${existingImport[1]}, ${needed.join(", ")} } from '@csszyx/runtime'`
+                existingImport.statement,
+                `${existingImport.prefixWithBody}, ${needed.join(", ")} } from '@csszyx/runtime'`
               );
             } else {
               const importStmt = `import { ${needed.join(", ")} } from '@csszyx/runtime';
@@ -3339,12 +3593,12 @@ function createCsszyxPlugins(options = {}) {
             transformed = true;
           }
         }
-        if (/\bszcn\s*\(/.test(code) && !transformedCode.includes(THEME_GROUPS_VIRTUAL_ID) && shouldProcessSource(id)) {
+        if ((usesSzcn || /\bszcn\s*\(/.test(code)) && !transformedCode.includes(THEME_GROUPS_VIRTUAL_ID) && shouldProcessSource(id)) {
           transformedCode = `import '${THEME_GROUPS_VIRTUAL_ID}';
 ${transformedCode}`;
           transformed = true;
         }
-        if (/\.[tj]sx?(\?.*)?$/.test(id)) {
+        if (matchesScriptExtension(id, SCRIPT_ID_EXTENSIONS)) {
           assertNoRSCBoundaryViolation(transformedCode, id);
           const record = createRSCModuleRecord(transformedCode, id);
           state.rscModules.set(record.id, record);
@@ -3402,6 +3656,7 @@ ${transformedCode}`;
           emitWarning(`[csszyx] ${warning}`);
         }
         state.spreadWarnings.clear();
+        prescanResultHandoff.clear();
         if (manglingEnabled && Object.keys(state.mangleMap).length > 0) {
           globalThis.__csszyx_ssr_mangle_map = state.mangleMap;
         }
@@ -3461,6 +3716,7 @@ ${transformedCode}`;
          * @param ctx - HMR context containing the changed file
          */
         handleHotUpdate(ctx) {
+          prescanResultHandoff.clear();
           const scanCss = options.build?.scanCss;
           if (scanCss) {
             const root = ctx.server.config.root || process.cwd();
@@ -3895,4 +4151,4 @@ const esbuildPlugin = (options = {}) => {
   };
 };
 
-export { isRSCServerModule as A, isTailwindReservedGlobalVar as B, mangleCodeClassesSync as C, mangleHybridHazardMessage as D, mergeThemes as E, missingTailwindEntryMessage as F, normalizeGlobalVarAliasesForCache as G, parseThemeBlocks as H, planGlobalVarAliases as I, readGlobalVarScanCache as J, recordGlobalVarSourceFile as K, resolveCompileSourceDirs as L, resolveGlobalVarScanCacheDir as M, resolveNativeCacheIdentity as N, rewriteGlobalVarCssAliases as O, rollupPlugin as P, scanGlobalVarCss as Q, shouldEmitWarning as R, shouldTrackGlobalVarSources as S, shouldWarnMissingTailwindEntry as T, shouldWarnUnscopedMonorepo as U, skippedSzFilesMessage as V, unscopedMonorepoMessage as W, validateGlobalVarAliasInputs as X, vitePlugin as Y, webpackPlugin as Z, writeGlobalVarScanCache as _, appendTailwindSourceDirective as a, assertNoRSCBoundaryViolation as b, assertNoRSCGraphViolation as c, collectMangleHybridHazards as d, computeSafelistRelPath as e, createGlobalVarAliasValidationOptions as f, createGlobalVarMapAssetSource as g, createGlobalVarScanCacheKey as h, createRSCModuleRecord as i, cssHasContentScope as j, cssImportsTailwind as k, deleteRSCModuleRecord as l, esbuildPlugin as m, extractGlobalVarAliasesForManifest as n, fileMayContainSafelistableSz as o, findRSCBoundaryViolation as p, findRSCGraphViolation as q, hasInjectableTailwindCandidate as r, hasTokens as s, hasUseClientDirective as t, unplugin as u, hasUseServerDirective as v, isCompileSourceOptedIn as w, isHardIgnoredPath as x, isMonorepoPackage as y, isPackagesSkippedSource as z };
+export { webpackPlugin as $, isPackagesSkippedSource as A, isRSCServerModule as B, isTailwindReservedGlobalVar as C, mangleCodeClassesSync as D, mangleHybridHazardMessage as E, mergeThemes as F, missingTailwindEntryMessage as G, normalizeGlobalVarAliasesForCache as H, parseThemeBlocks as I, planGlobalVarAliases as J, readGlobalVarScanCache as K, recordGlobalVarSourceFile as L, resolveCompileSourceDirs as M, resolveGlobalVarScanCacheDir as N, resolveNativeCacheIdentity as O, rewriteGlobalVarCssAliases as P, rollupPlugin as Q, scanCustomPropertyNames as R, scanGlobalVarCss as S, shouldEmitWarning as T, shouldTrackGlobalVarSources as U, shouldWarnMissingTailwindEntry as V, shouldWarnUnscopedMonorepo as W, skippedSzFilesMessage as X, unscopedMonorepoMessage as Y, validateGlobalVarAliasInputs as Z, vitePlugin as _, appendTailwindSourceDirective as a, writeGlobalVarScanCache as a0, assertNoRSCBoundaryViolation as b, assertNoRSCGraphViolation as c, collectMangleHybridHazards as d, computeSafelistRelPath as e, createGlobalVarAliasValidationOptions as f, createGlobalVarMapAssetSource as g, createGlobalVarScanCacheKey as h, createRSCModuleRecord as i, cssHasContentScope as j, cssImportsTailwind as k, deleteRSCModuleRecord as l, esbuildPlugin as m, extractGlobalVarAliasesForManifest as n, fileMayContainSafelistableSz as o, findLocalImportSources as p, findRSCBoundaryViolation as q, findRSCGraphViolation as r, hasInjectableTailwindCandidate as s, hasTokens as t, unplugin as u, hasUseClientDirective as v, hasUseServerDirective as w, isCompileSourceOptedIn as x, isHardIgnoredPath as y, isMonorepoPackage as z };