@csszyx/unplugin 0.10.11 → 0.11.0

package/dist/shared/{unplugin.q0k37crS.cjs → unplugin.CP2cz9bA.cjs}

@@ -1,5 +1,6 @@
 'use strict';
 
+const node_crypto = require('node:crypto');
 const fs = require('node:fs');
 const node_module = require('node:module');
 const path = require('node:path');
@@ -14,8 +15,8 @@ const vueAdapter = require('@csszyx/vue-adapter');
 const unplugin$1 = require('unplugin');
 const cssMangler = require('../css-mangler.cjs');
 const htmlEscape = require('./unplugin.BCwRIUs_.cjs');
-const node_crypto = require('node:crypto');
-const transformCache = require('./unplugin.D32Rb29j.cjs');
+const runtimeImportScan = require('./unplugin.Rov-j_Wm.cjs');
+const transformCache = require('./unplugin.D0N9bprz.cjs');
 const postcss = require('postcss');
 const valueParser = require('postcss-value-parser');
 
@@ -822,19 +823,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);
@@ -1133,10 +1233,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);
       }
@@ -1188,6 +1287,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"]);
@@ -1309,16 +1450,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 {
@@ -1347,14 +1501,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;
@@ -1642,19 +1789,34 @@ 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;
 const _loggedActiveParsers = /* @__PURE__ */ new Set();
 const _babelFallbackFiles = /* @__PURE__ */ new Set();
-const requireFromHere = node_module.createRequire((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('shared/unplugin.q0k37crS.cjs', document.baseURI).href)));
+const requireFromHere = node_module.createRequire((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('shared/unplugin.CP2cz9bA.cjs', document.baseURI).href)));
 const PLUGIN_VERSION = findPackageVersionFromFile(
-  node_url.fileURLToPath((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('shared/unplugin.q0k37crS.cjs', document.baseURI).href))),
+  node_url.fileURLToPath((typeof document === 'undefined' ? require('u' + 'rl').pathToFileURL(__filename).href : (_documentCurrentScript && _documentCurrentScript.tagName.toUpperCase() === 'SCRIPT' && _documentCurrentScript.src || new URL('shared/unplugin.CP2cz9bA.cjs', document.baseURI).href))),
   UNKNOWN_PACKAGE_VERSION
 );
 const COMPILER_VERSION = findPackageVersionFromModule("@csszyx/compiler", UNKNOWN_PACKAGE_VERSION);
@@ -1918,7 +2080,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) {
@@ -2265,6 +2427,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) {
@@ -2473,18 +2640,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 = types.validateGlobalVarMangleConfig(config);
@@ -2512,6 +2710,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 ?? types.DEFAULT_BUILD_CONFIG.cache) !== false;
   const cacheVersionsKnown = PLUGIN_VERSION !== UNKNOWN_PACKAGE_VERSION && COMPILER_VERSION !== UNKNOWN_PACKAGE_VERSION;
   const cacheEnabled = cacheRequested && cacheVersionsKnown;
@@ -2559,6 +2758,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,
@@ -2666,13 +2866,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 = transformCache.resolveTransformCacheDir(state.rootDir, options.build?.cacheDir);
     if (cacheEnabled) {
@@ -2699,6 +2899,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") {
@@ -2729,9 +2938,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,
@@ -2746,7 +2955,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),
@@ -2758,7 +2970,7 @@ function createCsszyxPlugins(options = {}) {
     if (parserMode !== "rust" || files.length <= 1) {
       return transformPrescanSourcesIndividually(files);
     }
-    const compilerOptions = createCompilerOptions();
+    const compilerOptions = createCompilerOptions(prescanAstBudget);
     const cacheRoot = transformCache.resolveTransformCacheDir(state.rootDir, options.build?.cacheDir);
     const results = /* @__PURE__ */ new Map();
     const misses = [];
@@ -2828,7 +3040,11 @@ function createCsszyxPlugins(options = {}) {
         try {
           results.set(
             miss.filePath,
-            transformConfiguredSource(miss.content, miss.effectiveFilename)
+            transformConfiguredSource(
+              miss.content,
+              miss.effectiveFilename,
+              prescanAstBudget
+            )
           );
         } catch {
         }
@@ -2839,15 +3055,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 compiler.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)}`
         );
@@ -2975,7 +3204,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: node_crypto.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).`
@@ -3231,10 +3476,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;
@@ -3258,6 +3503,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;
@@ -3286,6 +3533,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;
@@ -3294,12 +3543,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}`);
               }
@@ -3314,16 +3568,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;
         }
@@ -3335,21 +3587,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) => !runtimeImportScan.importsRuntimeHelper(transformedCode, name)) : imports;
           if (needed.length > 0) {
-            const existingImport = transformedCode.match(
-              /^(import\s*\{[^}]*)\}\s*from\s*'@csszyx\/runtime'/m
-            );
+            const existingImport = runtimeImportScan.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';
@@ -3359,12 +3613,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);
@@ -3422,6 +3676,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;
         }
@@ -3481,6 +3736,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();
@@ -3930,6 +4186,7 @@ exports.deleteRSCModuleRecord = deleteRSCModuleRecord;
 exports.esbuildPlugin = esbuildPlugin;
 exports.extractGlobalVarAliasesForManifest = extractGlobalVarAliasesForManifest;
 exports.fileMayContainSafelistableSz = fileMayContainSafelistableSz;
+exports.findLocalImportSources = findLocalImportSources;
 exports.findRSCBoundaryViolation = findRSCBoundaryViolation;
 exports.findRSCGraphViolation = findRSCGraphViolation;
 exports.hasInjectableTailwindCandidate = hasInjectableTailwindCandidate;
@@ -3956,6 +4213,7 @@ exports.resolveGlobalVarScanCacheDir = resolveGlobalVarScanCacheDir;
 exports.resolveNativeCacheIdentity = resolveNativeCacheIdentity;
 exports.rewriteGlobalVarCssAliases = rewriteGlobalVarCssAliases;
 exports.rollupPlugin = rollupPlugin;
+exports.scanCustomPropertyNames = scanCustomPropertyNames;
 exports.scanGlobalVarCss = scanGlobalVarCss;
 exports.shouldEmitWarning = shouldEmitWarning;
 exports.shouldTrackGlobalVarSources = shouldTrackGlobalVarSources;