@gustcss/vite 0.10.0 → 0.10.1

package/README.md

@@ -92,15 +92,30 @@ The manifest defaults to `<output>.classes.json`; set `mangleMap` to use another
 
 Treat the manifest as generated output and ignore it in Git unless your workflow intentionally pins class mappings.
 
-The transformer supports HTML, JavaScript, TypeScript, JSX, and TSX. It rewrites static `class` and `className` attributes, static template segments, direct string arguments to `classList`, and `setAttribute('class', ...)`. Code examples inside HTML `<code>` and `<pre>` bodies are neither rewritten nor inspected; classes on the elements themselves are still rewritten. An unclosed `<code>` or `<pre>` fails the build instead of masking the remainder of the document.
+The transformer supports HTML, JavaScript, TypeScript, JSX, TSX, and Astro. It rewrites static `class` and `className` attributes, static template segments, direct string arguments to `classList`, and `setAttribute('class', ...)`. Code examples inside HTML `<code>` and `<pre>` bodies are neither rewritten nor inspected; classes on the elements themselves are still rewritten. An unclosed `<code>` or `<pre>` fails the build instead of masking the remainder of the document.
 
-JavaScript, TypeScript, JSX, and TSX transforms emit Source Map v3 mappings back to the original source. Comments and ordinary strings containing `class="..."` are not treated as JSX attributes. An ordinary string that still references a mapped class fails closed as ambiguous.
+**Astro templates** (`.astro`) are rewritten before the Astro compiler sees them. Astro's own Vite plugins run ahead of user plugins, so the rewrite happens in the plugin's `load` hook on the raw source, guided by the AST that `parse()` from `@astrojs/compiler` returns (the package ships with astro, so nothing extra has to be installed). Attribute kinds, the frontmatter, comments, `<style>` and `<script>` blocks are taken from that AST.
+
+**Rewritten:**
+- Static `class="…"` attributes and `class={"…"}` expressions that consist of a single string literal
+- `class:list` array elements (including the right operand of `cond && "flex"`) and quoted object keys, nested arrays and objects included
+- `classList.add("…")` / `setAttribute("class", "…")` calls in the frontmatter and in `<script>` blocks
+
+**Fail-closed** (the build stops when a mapped class appears in one of these places):
+- Any `class` expression other than a single string literal, such as `class={cond ? "flex" : "x"}` or `` class={`flex ${x}`} ``
+- Inside `class:list`: unquoted keys (`{ flex: on }`), shorthand properties (`{ hidden }`), template literals, comparison operands (`variant === "flex"`), subscripts (`map["flex"]`), function calls, method chains, or an expression that is not an array/object literal
+- Component props other than `class` / `class:list` (`<Card klass="flex" />`), `set:html` / `set:text` values, strings inside `{}` expressions (`{JSON.stringify("flex")}`), and ambiguous strings in the frontmatter or `<script>`. A `class` or `class:list` passed to a component is rewritten as a static value, like JSX `className`; forward `Astro.props.class` unchanged on the receiving side
+- `<style>` selectors (`.flex`, `@scope (.flex)`, `[class~="flex"]`); add hand-written selectors such as `.sr-only { … }` to `mangleExclude`
+
+HTML comments, `<code>` / `<pre>` bodies, ordinary non-class attributes (`<input type="flex">`) and `<style>` declarations (`border: 1px solid`) are neither rewritten nor inspected.
+
+JavaScript, TypeScript, JSX, TSX, and Astro frontmatter transforms emit Source Map v3 mappings back to the original source. Comments and ordinary strings containing `class="..."` are not treated as JSX attributes. An ordinary string that still references a mapped class fails closed as ambiguous.
 
 Conditional expressions, nested calls, selectors, and other ambiguous references fail the build. The plugin restores readable CSS after a failed transform.
 
 Add dynamically or externally referenced class names, including production E2E selectors, to `mangleExclude`.
 
-Vue, Astro, and Svelte templates are not supported by this option yet. A mapped class in one of these files fails the build instead of emitting mismatched output.
+Vue and Svelte templates are not supported by this option. A mapped class in one of these files fails the build instead of emitting mismatched output.
 
 Files under `node_modules` are neither rewritten nor inspected.
 

package/dist/index.cjs

@@ -31,6 +31,8 @@ path = __toESM(path, 1);
 let fs = require("fs");
 fs = __toESM(fs, 1);
 let crypto = require("crypto");
+let node_module = require("node:module");
+let node_url = require("node:url");
 //#endregion
 //#region src/mangle.js
 var import_runner = /* @__PURE__ */ __toESM((/* @__PURE__ */ __commonJSMin(((exports, module) => {
@@ -169,7 +171,7 @@ var import_runner = /* @__PURE__ */ __toESM((/* @__PURE__ */ __commonJSMin(((exp
 })))(), 1);
 const SUPPORTED_SOURCE = /\.(?:[cm]?[jt]sx?|html)(?:$|\?)/;
 const HTML_SOURCE = /\.html(?:$|\?)/;
-const UNSUPPORTED_FRAMEWORK_SOURCE = /\.(?:vue|astro|svelte)(?:$|\?)/;
+const UNSUPPORTED_FRAMEWORK_SOURCE = /\.(?:vue|svelte)(?:$|\?)/;
 const DEPENDENCY_SOURCE = /(?:^|[\\/])node_modules[\\/]/;
 function hasOwn(classes, token) {
 	return Object.hasOwn(classes, token);
@@ -215,6 +217,7 @@ function findRawTextClosing(code, tagName, start) {
 function analyzeHtml(code, id) {
 	const markup = new Array(code.length).fill(" ");
 	const scriptRanges = [];
+	const commentRanges = [];
 	const exampleStack = [];
 	let index = 0;
 	while (index < code.length) {
@@ -222,7 +225,12 @@ function analyzeHtml(code, id) {
 		if (open < 0) break;
 		if (code.startsWith("<!--", open)) {
 			const commentEnd = code.indexOf("-->", open + 4);
-			index = commentEnd < 0 ? code.length : commentEnd + 3;
+			const end = commentEnd < 0 ? code.length : commentEnd + 3;
+			commentRanges.push({
+				start: open,
+				end
+			});
+			index = end;
 			continue;
 		}
 		const tagEnd = findHtmlTagEnd(code, open);
@@ -258,7 +266,8 @@ function analyzeHtml(code, id) {
 	if (exampleStack.length > 0) throw new Error(`[gustcss] unclosed <${exampleStack.at(-1)}> in ${id}`);
 	return {
 		markup: markup.join(""),
-		scriptRanges
+		scriptRanges,
+		commentRanges
 	};
 }
 function canStartRegex(code, index, previous) {
@@ -486,7 +495,7 @@ function rejectAmbiguousClassExpressions(code, structure, classes, id) {
 	while ((match = pattern.exec(structure)) !== null) {
 		const open = code.indexOf("{", match.index);
 		const end = findBalancedEnd(code, open, "{", "}");
-		if (end < 0) break;
+		if (end < 0) throw new Error(`[gustcss] incomplete class/className expression in ${id}. Ensure the expression is properly closed.`);
 		const ambiguous = findMappedToken(code.slice(open + 1, end - 1), classes);
 		if (ambiguous) throw new Error(`[gustcss] mapped class "${ambiguous}" remains in an ambiguous class expression in ${id}. Use a static class attribute/template segment, or add it to mangleExclude.`);
 		pattern.lastIndex = end;
@@ -580,30 +589,53 @@ function sourcesForFile(code, id) {
 	const inspection = new Array(code.length).fill(" ");
 	for (const range of html.scriptRanges) {
 		const script = code.slice(range.start, range.end);
-		const js = createJsMasks(script);
+		const jsScript = createJsMasks(script);
 		for (let offset = 0; offset < script.length; offset += 1) {
-			structure[range.start + offset] = js.structure[offset];
-			inspection[range.start + offset] = js.inspection[offset];
+			structure[range.start + offset] = jsScript.structure[offset];
+			inspection[range.start + offset] = jsScript.inspection[offset];
 		}
 	}
+	for (const range of html.commentRanges) for (let offset = range.start; offset < range.end; offset += 1) inspection[offset] = " ";
 	return {
 		attributes: html.markup,
 		structure: structure.join(""),
 		inspection: inspection.join("")
 	};
 }
-function collectRecognizedContextEdits(code, classes, id) {
+function collectEdits(code, classes, id) {
 	const edits = [];
 	const sources = sourcesForFile(code, id);
+	const commentRanges = [];
+	let commentIndex = 0;
+	while ((commentIndex = code.indexOf("<!--", commentIndex)) >= 0) {
+		const commentEnd = code.indexOf("-->", commentIndex + 4);
+		if (commentEnd < 0) {
+			commentRanges.push({
+				start: commentIndex,
+				end: code.length
+			});
+			break;
+		}
+		commentRanges.push({
+			start: commentIndex,
+			end: commentEnd + 3
+		});
+		commentIndex = commentEnd + 3;
+	}
+	function isInComment(pos) {
+		return commentRanges.some((range) => pos >= range.start && pos < range.end);
+	}
 	const attributePattern = /((?<![:\w-])class(?:Name)?\s*=\s*)(?:(["'])([\s\S]*?)(\2)|(\{\s*)(["'])([\s\S]*?)(\6)(\s*\}))/g;
 	let match;
 	while ((match = attributePattern.exec(sources.attributes)) !== null) {
+		if (isInComment(match.index)) continue;
 		if (!HTML_SOURCE.test(id) && !isJsxAttributeAt(sources.structure, match.index)) continue;
 		if (match[2] !== void 0) collectClassListEdits(match[3], classes, match.index + match[1].length + 1, edits);
 		else collectClassListEdits(match[7], classes, match.index + match[1].length + match[5].length + 1, edits);
 	}
 	const templatePattern = /((?<![:\w-])class(?:Name)?\s*=\s*\{\s*`)([\s\S]*?)(`\s*\})/g;
 	while ((match = templatePattern.exec(sources.attributes)) !== null) {
+		if (isInComment(match.index)) continue;
 		if (!HTML_SOURCE.test(id) && !isJsxAttributeAt(sources.structure, match.index)) continue;
 		collectTemplateBodyEdits(match[2], classes, match.index + match[1].length, edits);
 	}
@@ -613,7 +645,27 @@ function collectRecognizedContextEdits(code, classes, id) {
 		if (sources.structure[match.index] === " ") continue;
 		collectClassListEdits(match[3], classes, match.index + match[1].length + 1, edits);
 	}
-	return applyEdits(code, edits);
+	return edits;
+}
+function collectRecognizedContextEdits(code, classes, id) {
+	return applyEdits(code, collectEdits(code, classes, id));
+}
+/**
+* Collect the rewrites for a source fragment (TypeScript frontmatter, an inline
+* `<script>` body, …) and run the same fail-closed checks as `transformSource`,
+* but return the edits instead of the rewritten code so the caller can place
+* them inside a larger document.
+*/
+function collectFragmentEdits(code, classes, id) {
+	const edits = collectEdits(code, classes, id);
+	const rewritten = applyEdits(code, edits).code;
+	const sources = sourcesForFile(rewritten, id);
+	rejectDynamicSetAttributeCalls(rewritten, sources.structure, classes, id);
+	rejectDynamicClassNameAssignments(rewritten, sources.structure, classes, id);
+	rejectAmbiguousClassExpressions(rewritten, sources.structure, classes, id);
+	const ambiguous = findMappedInQuotedSegments(sources.inspection, classes);
+	if (ambiguous) throw new Error(`[gustcss] mapped class "${ambiguous}" remains in an ambiguous string in ${id}. Move it to a static class/className or classList context, or add it to mangleExclude.`);
+	return edits;
 }
 const BASE64 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
 function encodeVLQ(value) {
@@ -701,7 +753,7 @@ function createClassNameTransformer(classes) {
 	const transformWithSourceMap = (code, id) => {
 		if (UNSUPPORTED_FRAMEWORK_SOURCE.test(id)) {
 			const referenced = findMappedToken(code, classes);
-			if (referenced) throw new Error(`[gustcss] class-name mangling does not support Vue, Astro, or Svelte templates yet (${id}). Add "${referenced}" to mangleExclude or disable mangleClassNames.`);
+			if (referenced) throw new Error(`[gustcss] class-name mangling does not support Vue or Svelte templates yet (${id}). Add "${referenced}" to mangleExclude or disable mangleClassNames.`);
 			return {
 				code,
 				map: null
@@ -721,9 +773,392 @@ function createClassNameTransformer(classes) {
 	};
 	const transform = (code, id) => transformWithSourceMap(code, id).code;
 	transform.withSourceMap = transformWithSourceMap;
+	transform.classes = classes;
 	return transform;
 }
 //#endregion
+//#region src/astro.js
+/**
+* Class-name mangling for `.astro` sources, driven by the AST that
+* `@astrojs/compiler`'s `parse()` returns.
+*
+* Only unambiguous class contexts are rewritten; any other place where a
+* mapped class name shows up fails the build (fail-closed), because a class
+* that reaches the DOM unrenamed would no longer match the mangled CSS.
+*/
+const CLASS_VALUED_BEFORE = /* @__PURE__ */ new Set([
+	"[",
+	",",
+	"&&",
+	"||"
+]);
+const CLASS_VALUED_AFTER = /* @__PURE__ */ new Set([
+	",",
+	"]",
+	"&&",
+	"||"
+]);
+const OPERATORS = [
+	"===",
+	"!==",
+	"&&",
+	"||",
+	"??",
+	"=>",
+	"==",
+	"!=",
+	"?.",
+	"<=",
+	">="
+];
+const PUNCTUATION = "[]{}(),:?.!+-*/%<>=&|~^;";
+/** Map UTF-8 byte offsets (what the compiler reports) to string indexes. */
+function byteToIndexMap(code) {
+	const map = [];
+	let byte = 0;
+	for (let index = 0; index < code.length; index += 1) {
+		const point = code.codePointAt(index);
+		const width = point < 128 ? 1 : point < 2048 ? 2 : point < 65536 ? 3 : 4;
+		for (let i = 0; i < width; i += 1) map[byte + i] = index;
+		byte += width;
+		if (point > 65535) index += 1;
+	}
+	map[byte] = code.length;
+	return map;
+}
+/**
+* Tokenize a JavaScript expression just far enough to know, for every string
+* literal and identifier, what surrounds it. Template literals are reported
+* as strings with quote "`" and are never rewritten. Returns null when the
+* expression cannot be tokenized (unterminated string, unknown character).
+*/
+function tokenizeExpression(source) {
+	const tokens = [];
+	let index = 0;
+	while (index < source.length) {
+		const char = source[index];
+		if (/\s/.test(char)) {
+			index += 1;
+			continue;
+		}
+		if (source.startsWith("//", index)) {
+			const newline = source.indexOf("\n", index);
+			index = newline < 0 ? source.length : newline + 1;
+			continue;
+		}
+		if (source.startsWith("/*", index)) {
+			const close = source.indexOf("*/", index + 2);
+			if (close < 0) return null;
+			index = close + 2;
+			continue;
+		}
+		if (char === "\"" || char === "'" || char === "`") {
+			const start = index;
+			index += 1;
+			while (index < source.length && source[index] !== char) {
+				if (source[index] === "\\") index += 1;
+				index += 1;
+			}
+			if (index >= source.length) return null;
+			tokens.push({
+				type: "string",
+				quote: char,
+				start,
+				end: index + 1,
+				value: source.slice(start + 1, index)
+			});
+			index += 1;
+			continue;
+		}
+		if (/[A-Za-z_$]/.test(char)) {
+			const start = index;
+			while (index < source.length && /[\w$]/.test(source[index])) index += 1;
+			tokens.push({
+				type: "identifier",
+				start,
+				end: index,
+				value: source.slice(start, index)
+			});
+			continue;
+		}
+		if (/[0-9]/.test(char)) {
+			const start = index;
+			while (index < source.length && /[\w.]/.test(source[index])) index += 1;
+			tokens.push({
+				type: "number",
+				start,
+				end: index,
+				value: source.slice(start, index)
+			});
+			continue;
+		}
+		const operator = OPERATORS.find((candidate) => source.startsWith(candidate, index));
+		if (operator) {
+			tokens.push({
+				type: "operator",
+				start: index,
+				end: index + operator.length,
+				value: operator
+			});
+			index += operator.length;
+			continue;
+		}
+		if (PUNCTUATION.includes(char)) {
+			tokens.push({
+				type: "operator",
+				start: index,
+				end: index + 1,
+				value: char
+			});
+			index += 1;
+			continue;
+		}
+		return null;
+	}
+	return tokens;
+}
+/**
+* Rewrite the class-valued string literals of a `class:list` expression.
+* Accepted shapes are array literals, object literals and their nesting.
+* Inside them only array elements (also as the right operand of `&&` / `||`)
+* and quoted object keys are class-valued; every other occurrence of a mapped
+* class name fails closed.
+*/
+function collectClassListExpressionEdits(source, classes, baseOffset, edits, id) {
+	const fail = (token, reason) => {
+		throw new Error(`[gustcss] mapped class "${token}" ${reason} in class:list of ${id}. Use an array/object literal with quoted entries, or add it to mangleExclude.`);
+	};
+	const tokens = tokenizeExpression(source);
+	const first = tokens && tokens[0];
+	const last = tokens && tokens[tokens.length - 1];
+	if (!(tokens && tokens.length >= 2 && (first.value === "[" && last.value === "]" || first.value === "{" && last.value === "}"))) {
+		const mapped = findMappedToken(source, classes);
+		if (mapped) fail(mapped, "remains in an expression that is not an array or object literal");
+		return;
+	}
+	const stack = [];
+	for (let i = 0; i < tokens.length; i += 1) {
+		const token = tokens[i];
+		const previous = tokens[i - 1];
+		const next = tokens[i + 1];
+		const context = stack[stack.length - 1];
+		if (token.type === "operator") {
+			if (token.value === "[") {
+				const subscript = previous && (previous.type === "identifier" || previous.type === "string" || previous.value === ")" || previous.value === "]");
+				stack.push(subscript ? "subscript" : "array");
+			} else if (token.value === "{") stack.push("object");
+			else if (token.value === "(") stack.push("call");
+			else if (token.value === "]" || token.value === "}" || token.value === ")") {
+				if (stack.length === 0) fail(findMappedToken(source, classes) || "?", "appears in an unbalanced");
+				stack.pop();
+			}
+			continue;
+		}
+		if (token.type === "identifier") {
+			if (!hasOwn(classes, token.value)) continue;
+			if (context === "object" && previous && (previous.value === "{" || previous.value === ",") && next && (next.value === ":" || next.value === "," || next.value === "}")) fail(token.value, `is an unquoted object key (write it as "${token.value}")`);
+			fail(token.value, "is referenced as an identifier");
+		}
+		if (token.type !== "string") continue;
+		const mapped = findMappedToken(token.value, classes);
+		if (token.quote === "`") {
+			if (mapped) fail(mapped, "remains in a template literal");
+			continue;
+		}
+		const arrayElement = context === "array" && previous && CLASS_VALUED_BEFORE.has(previous.value) && next && CLASS_VALUED_AFTER.has(next.value);
+		const objectKey = context === "object" && previous && (previous.value === "{" || previous.value === ",") && next && next.value === ":";
+		if (arrayElement || objectKey) collectClassListEdits(token.value, classes, baseOffset + token.start + 1, edits);
+		else if (mapped) fail(mapped, "remains in a position that is not an array element or object key");
+	}
+	if (stack.length !== 0) fail(findMappedToken(source, classes) || "?", "appears in an unbalanced");
+}
+/**
+* Find a mapped class referenced by a stylesheet selector, including at-rule
+* preludes (`@scope (.flex)`) and attribute selectors (`[class~="flex"]`).
+* Declarations and comments are ignored.
+*/
+function findMappedClassInStylesheet(css, classes) {
+	const source = css.replace(/\/\*[\s\S]*?\*\//g, " ");
+	let prelude = "";
+	let quote = null;
+	for (let i = 0; i < source.length; i += 1) {
+		const char = source[i];
+		if (quote) {
+			prelude += char;
+			if (char === "\\") {
+				prelude += source[i + 1] || "";
+				i += 1;
+			} else if (char === quote) quote = null;
+			continue;
+		}
+		if (char === "\"" || char === "'") {
+			quote = char;
+			prelude += char;
+			continue;
+		}
+		if (char === "{") {
+			const found = findMappedClassInSelector(prelude.trim(), classes);
+			if (found) return found;
+			prelude = "";
+		} else if (char === "}" || char === ";") prelude = "";
+		else prelude += char;
+	}
+	return null;
+}
+function findMappedClassInSelector(selector, classes) {
+	const classPattern = /\.((?:\\.|[A-Za-z0-9_-])+)/g;
+	let match;
+	while ((match = classPattern.exec(selector)) !== null) {
+		const name = match[1].replace(/\\(.)/g, "$1");
+		if (hasOwn(classes, name)) return name;
+	}
+	const attributePattern = /\[\s*class\s*[~|^$*]?=\s*(["']?)([^\]"']+)\1\s*[is]?\s*\]/g;
+	while ((match = attributePattern.exec(selector)) !== null) {
+		const mapped = findMappedToken(match[2], classes);
+		if (mapped) return mapped;
+	}
+	return null;
+}
+/**
+* Locate the value of an attribute from its start offset (the attribute
+* name). Returns the index range of the value and its delimiter.
+*/
+function locateAttributeValue(code, attribute, start) {
+	let index = start + attribute.name.length;
+	while (index < code.length && /\s/.test(code[index])) index += 1;
+	if (code[index] !== "=") return null;
+	index += 1;
+	while (index < code.length && /\s/.test(code[index])) index += 1;
+	const delimiter = code[index];
+	if (delimiter === "\"" || delimiter === "'" || delimiter === "`") {
+		const end = code.indexOf(delimiter, index + 1);
+		return end < 0 ? null : {
+			start: index + 1,
+			end,
+			delimiter
+		};
+	}
+	if (delimiter === "{") {
+		const end = findBalancedEnd(code, index, "{", "}");
+		return end < 0 ? null : {
+			start: index + 1,
+			end: end - 1,
+			delimiter
+		};
+	}
+	return null;
+}
+const STRING_LITERAL = /^\s*(["'])((?:\\.|(?!\1)[^\\])*)\1\s*$/;
+/**
+* Create the `.astro` transformer. `parse` is `@astrojs/compiler`'s parse
+* function, injected so the plugin and the tests can supply their own copy.
+*/
+function createAstroTransformer({ parse, classes }) {
+	return async function transformAstro(code, id) {
+		const { ast } = await parse(code, { position: true });
+		const toIndex = byteToIndexMap(code);
+		const edits = [];
+		const offsetOf = (node) => toIndex[node.position.start.offset];
+		const stop = (mapped, where) => {
+			throw new Error(`[gustcss] mapped class "${mapped}" remains in ${where} in ${id}. Move it to a static class attribute or class:list literal, or add it to mangleExclude.`);
+		};
+		const fragmentEdits = (text, start, kind) => {
+			for (const edit of collectFragmentEdits(text, classes, `${id}#${kind}.ts`)) edits.push({
+				start: start + edit.start,
+				end: start + edit.end,
+				replacement: edit.replacement
+			});
+		};
+		const visitAttribute = (attribute, node) => {
+			const start = offsetOf(attribute);
+			const isComponent = node.type === "component";
+			const value = attribute.value || "";
+			if (attribute.name === "class") {
+				const location = locateAttributeValue(code, attribute, start);
+				if (location && attribute.kind === "quoted") {
+					collectClassListEdits(code.slice(location.start, location.end), classes, location.start, edits);
+					return;
+				}
+				if (location && attribute.kind === "expression") {
+					const expression = code.slice(location.start, location.end);
+					const literal = STRING_LITERAL.exec(expression);
+					if (literal) {
+						const inner = expression.indexOf(literal[1]) + 1;
+						collectClassListEdits(literal[2], classes, location.start + inner, edits);
+						return;
+					}
+				}
+				const mapped = findMappedToken(value, classes);
+				if (mapped) stop(mapped, `a dynamic class attribute (${attribute.kind})`);
+				return;
+			}
+			if (attribute.name === "class:list") {
+				const location = locateAttributeValue(code, attribute, start);
+				if (!location || attribute.kind !== "expression") {
+					const mapped = findMappedToken(value, classes);
+					if (mapped) stop(mapped, `a class:list attribute of kind ${attribute.kind}`);
+					return;
+				}
+				collectClassListExpressionEdits(code.slice(location.start, location.end), classes, location.start, edits, id);
+				return;
+			}
+			if (attribute.kind === "spread") {
+				const mapped = findMappedToken(`${attribute.name} ${value}`, classes);
+				if (mapped) stop(mapped, `a spread attribute on <${node.name}>`);
+				return;
+			}
+			if (attribute.name === "set:html" || attribute.name === "set:text" || isComponent) {
+				const mapped = findMappedToken(value, classes);
+				if (mapped) stop(mapped, isComponent ? `the "${attribute.name}" prop of <${node.name}>` : `a ${attribute.name} value`);
+			}
+		};
+		const visit = (node) => {
+			switch (node.type) {
+				case "frontmatter": {
+					const start = code.indexOf(node.value, offsetOf(node));
+					if (start >= 0) fragmentEdits(node.value, start, "frontmatter");
+					return;
+				}
+				case "comment":
+				case "text":
+				case "doctype": return;
+				case "expression":
+					for (const child of node.children) if (child.type === "text") {
+						const mapped = findMappedToken(child.value, classes);
+						if (mapped && /["'`]/.test(child.value)) stop(mapped, "a template expression");
+					} else visit(child);
+					return;
+				case "element":
+				case "component":
+				case "custom-element":
+				case "fragment":
+					for (const attribute of node.attributes) visitAttribute(attribute, node);
+					if (node.type === "element" && node.name === "style") {
+						for (const child of node.children) {
+							if (child.type !== "text") continue;
+							const mapped = findMappedClassInStylesheet(child.value, classes);
+							if (mapped) stop(mapped, "a <style> selector");
+						}
+						return;
+					}
+					if (node.type === "element" && node.name === "script") {
+						for (const child of node.children) if (child.type === "text") fragmentEdits(child.value, offsetOf(child), "script");
+						return;
+					}
+					for (const child of node.children) visit(child);
+					return;
+				default: for (const child of node.children || []) visit(child);
+			}
+		};
+		visit(ast);
+		const applied = applyEdits(code, edits);
+		return {
+			code: applied.code,
+			map: createSourceMap(code, applied.code, applied.origins, id)
+		};
+	};
+}
+//#endregion
 //#region src/index.js
 /**
 * @gustcss/vite - Vite plugin for CSS Utility Generator
@@ -738,6 +1173,24 @@ function createClassNameTransformer(classes) {
 * are provided by the developer in their build configuration. This plugin is
 * designed for build-time use only and should not process untrusted input.
 */
+const ASTRO_SOURCE = /\.astro(?:$|\?)/;
+const ASTRO_MODULE = /^[^\0?]*\.astro$/;
+/**
+* Resolve @astrojs/compiler parse function from the project.
+*/
+async function resolveAstroCompiler(root) {
+	const projectRequire = (0, node_module.createRequire)(path.default.join(root, "package.json"));
+	const attempts = [
+		() => projectRequire.resolve("@astrojs/compiler"),
+		() => (0, node_module.createRequire)(projectRequire.resolve("astro/package.json")).resolve("@astrojs/compiler"),
+		() => (0, node_module.createRequire)(require("url").pathToFileURL(__filename).href).resolve("@astrojs/compiler")
+	];
+	for (const attempt of attempts) try {
+		const mod = await import((0, node_url.pathToFileURL)(attempt()).href);
+		if (typeof mod.parse === "function") return mod.parse;
+	} catch {}
+	return null;
+}
 function cssUtility(opts = {}) {
 	const output = opts.output || "src/styles/utility.css";
 	const content = opts.content || ["./src/**/*.{js,ts,jsx,tsx,astro,vue}"];
@@ -750,6 +1203,8 @@ function cssUtility(opts = {}) {
 	let rollbackContext = null;
 	let manifestSnapshot = null;
 	let watchProcess = null;
+	const loadedAstroModules = /* @__PURE__ */ new Set();
+	let astroTransformer = null;
 	function findBinary(cwd) {
 		return import_runner.default.resolveBinary({
 			cwd,
@@ -821,15 +1276,19 @@ function cssUtility(opts = {}) {
 			fs: fs.default
 		}, (tempConfigPath) => run(tempConfigPath));
 	}
+	let viteConfig = null;
 	return {
 		name: "gustcss",
 		...mangleClassNames ? { enforce: "pre" } : {},
 		configResolved(config) {
+			viteConfig = config;
 			const cwd = config.root || process.cwd();
 			const binaryPath = findBinary(cwd);
 			try {
 				const enableMangle = mangleClassNames && config.command === "build";
 				buildCSS(cwd, binaryPath, enableMangle);
+				loadedAstroModules.clear();
+				astroTransformer = null;
 				rollbackContext = enableMangle ? {
 					cwd,
 					binaryPath
@@ -839,8 +1298,30 @@ function cssUtility(opts = {}) {
 				throw error;
 			}
 		},
+		async load(id) {
+			if (!classNameTransformer || !ASTRO_MODULE.test(id)) return null;
+			const filePath = id.startsWith("/@fs/") ? id.slice(4) : id;
+			if (!fs.default.existsSync(filePath)) throw new Error(`[gustcss] cannot read ${id} to rewrite its class names for mangling. Disable mangleClassNames or exclude its classes with mangleExclude.`);
+			const code = fs.default.readFileSync(filePath, "utf-8");
+			loadedAstroModules.add(id);
+			if (!astroTransformer) {
+				const parse = await resolveAstroCompiler(viteConfig?.root || process.cwd());
+				if (!parse) throw new Error(`[gustcss] @astrojs/compiler could not be resolved from the project, so ${id} cannot be mangled. Install astro, or disable mangleClassNames.`);
+				astroTransformer = createAstroTransformer({
+					parse,
+					classes: classNameTransformer.classes
+				});
+			}
+			const result = await astroTransformer(code, filePath);
+			return result.code === code ? null : result;
+		},
 		transform(code, id) {
 			if (!classNameTransformer) return null;
+			if (ASTRO_MODULE.test(id)) {
+				if (!loadedAstroModules.has(id)) throw new Error(`[gustcss] ${id} was compiled without passing through the gustcss load hook, so its class names cannot be mangled safely. Disable mangleClassNames or move the plugin so it loads .astro sources first.`);
+				return null;
+			}
+			if (ASTRO_SOURCE.test(id)) return null;
 			const transformed = classNameTransformer.withSourceMap(code, id);
 			return transformed.code === code ? null : transformed;
 		},
@@ -907,9 +1388,15 @@ function cssUtility(opts = {}) {
 				}
 				if (tempConfigPath && fs.default.existsSync(tempConfigPath)) fs.default.unlinkSync(tempConfigPath);
 			};
+			const close = server.close.bind(server);
+			server.close = async () => {
+				cleanup();
+				return close();
+			};
 			server.httpServer?.on("close", cleanup);
 			process.once("SIGINT", cleanup);
 			process.once("SIGTERM", cleanup);
+			process.once("exit", cleanup);
 		},
 		writeBundle() {
 			if (!manifestSnapshot || fs.default.existsSync(manifestSnapshot.path)) return;

package/dist/index.mjs

@@ -3,6 +3,7 @@ import { spawn, spawnSync } from "child_process";
 import path from "path";
 import fs from "fs";
 import { randomBytes } from "crypto";
+import { pathToFileURL } from "node:url";
 //#region \0rolldown/runtime.js
 var __create = Object.create;
 var __defProp = Object.defineProperty;
@@ -164,7 +165,7 @@ var import_runner = /* @__PURE__ */ __toESM((/* @__PURE__ */ __commonJSMin(((exp
 })))(), 1);
 const SUPPORTED_SOURCE = /\.(?:[cm]?[jt]sx?|html)(?:$|\?)/;
 const HTML_SOURCE = /\.html(?:$|\?)/;
-const UNSUPPORTED_FRAMEWORK_SOURCE = /\.(?:vue|astro|svelte)(?:$|\?)/;
+const UNSUPPORTED_FRAMEWORK_SOURCE = /\.(?:vue|svelte)(?:$|\?)/;
 const DEPENDENCY_SOURCE = /(?:^|[\\/])node_modules[\\/]/;
 function hasOwn(classes, token) {
 	return Object.hasOwn(classes, token);
@@ -210,6 +211,7 @@ function findRawTextClosing(code, tagName, start) {
 function analyzeHtml(code, id) {
 	const markup = new Array(code.length).fill(" ");
 	const scriptRanges = [];
+	const commentRanges = [];
 	const exampleStack = [];
 	let index = 0;
 	while (index < code.length) {
@@ -217,7 +219,12 @@ function analyzeHtml(code, id) {
 		if (open < 0) break;
 		if (code.startsWith("<!--", open)) {
 			const commentEnd = code.indexOf("-->", open + 4);
-			index = commentEnd < 0 ? code.length : commentEnd + 3;
+			const end = commentEnd < 0 ? code.length : commentEnd + 3;
+			commentRanges.push({
+				start: open,
+				end
+			});
+			index = end;
 			continue;
 		}
 		const tagEnd = findHtmlTagEnd(code, open);
@@ -253,7 +260,8 @@ function analyzeHtml(code, id) {
 	if (exampleStack.length > 0) throw new Error(`[gustcss] unclosed <${exampleStack.at(-1)}> in ${id}`);
 	return {
 		markup: markup.join(""),
-		scriptRanges
+		scriptRanges,
+		commentRanges
 	};
 }
 function canStartRegex(code, index, previous) {
@@ -481,7 +489,7 @@ function rejectAmbiguousClassExpressions(code, structure, classes, id) {
 	while ((match = pattern.exec(structure)) !== null) {
 		const open = code.indexOf("{", match.index);
 		const end = findBalancedEnd(code, open, "{", "}");
-		if (end < 0) break;
+		if (end < 0) throw new Error(`[gustcss] incomplete class/className expression in ${id}. Ensure the expression is properly closed.`);
 		const ambiguous = findMappedToken(code.slice(open + 1, end - 1), classes);
 		if (ambiguous) throw new Error(`[gustcss] mapped class "${ambiguous}" remains in an ambiguous class expression in ${id}. Use a static class attribute/template segment, or add it to mangleExclude.`);
 		pattern.lastIndex = end;
@@ -575,30 +583,53 @@ function sourcesForFile(code, id) {
 	const inspection = new Array(code.length).fill(" ");
 	for (const range of html.scriptRanges) {
 		const script = code.slice(range.start, range.end);
-		const js = createJsMasks(script);
+		const jsScript = createJsMasks(script);
 		for (let offset = 0; offset < script.length; offset += 1) {
-			structure[range.start + offset] = js.structure[offset];
-			inspection[range.start + offset] = js.inspection[offset];
+			structure[range.start + offset] = jsScript.structure[offset];
+			inspection[range.start + offset] = jsScript.inspection[offset];
 		}
 	}
+	for (const range of html.commentRanges) for (let offset = range.start; offset < range.end; offset += 1) inspection[offset] = " ";
 	return {
 		attributes: html.markup,
 		structure: structure.join(""),
 		inspection: inspection.join("")
 	};
 }
-function collectRecognizedContextEdits(code, classes, id) {
+function collectEdits(code, classes, id) {
 	const edits = [];
 	const sources = sourcesForFile(code, id);
+	const commentRanges = [];
+	let commentIndex = 0;
+	while ((commentIndex = code.indexOf("<!--", commentIndex)) >= 0) {
+		const commentEnd = code.indexOf("-->", commentIndex + 4);
+		if (commentEnd < 0) {
+			commentRanges.push({
+				start: commentIndex,
+				end: code.length
+			});
+			break;
+		}
+		commentRanges.push({
+			start: commentIndex,
+			end: commentEnd + 3
+		});
+		commentIndex = commentEnd + 3;
+	}
+	function isInComment(pos) {
+		return commentRanges.some((range) => pos >= range.start && pos < range.end);
+	}
 	const attributePattern = /((?<![:\w-])class(?:Name)?\s*=\s*)(?:(["'])([\s\S]*?)(\2)|(\{\s*)(["'])([\s\S]*?)(\6)(\s*\}))/g;
 	let match;
 	while ((match = attributePattern.exec(sources.attributes)) !== null) {
+		if (isInComment(match.index)) continue;
 		if (!HTML_SOURCE.test(id) && !isJsxAttributeAt(sources.structure, match.index)) continue;
 		if (match[2] !== void 0) collectClassListEdits(match[3], classes, match.index + match[1].length + 1, edits);
 		else collectClassListEdits(match[7], classes, match.index + match[1].length + match[5].length + 1, edits);
 	}
 	const templatePattern = /((?<![:\w-])class(?:Name)?\s*=\s*\{\s*`)([\s\S]*?)(`\s*\})/g;
 	while ((match = templatePattern.exec(sources.attributes)) !== null) {
+		if (isInComment(match.index)) continue;
 		if (!HTML_SOURCE.test(id) && !isJsxAttributeAt(sources.structure, match.index)) continue;
 		collectTemplateBodyEdits(match[2], classes, match.index + match[1].length, edits);
 	}
@@ -608,7 +639,27 @@ function collectRecognizedContextEdits(code, classes, id) {
 		if (sources.structure[match.index] === " ") continue;
 		collectClassListEdits(match[3], classes, match.index + match[1].length + 1, edits);
 	}
-	return applyEdits(code, edits);
+	return edits;
+}
+function collectRecognizedContextEdits(code, classes, id) {
+	return applyEdits(code, collectEdits(code, classes, id));
+}
+/**
+* Collect the rewrites for a source fragment (TypeScript frontmatter, an inline
+* `<script>` body, …) and run the same fail-closed checks as `transformSource`,
+* but return the edits instead of the rewritten code so the caller can place
+* them inside a larger document.
+*/
+function collectFragmentEdits(code, classes, id) {
+	const edits = collectEdits(code, classes, id);
+	const rewritten = applyEdits(code, edits).code;
+	const sources = sourcesForFile(rewritten, id);
+	rejectDynamicSetAttributeCalls(rewritten, sources.structure, classes, id);
+	rejectDynamicClassNameAssignments(rewritten, sources.structure, classes, id);
+	rejectAmbiguousClassExpressions(rewritten, sources.structure, classes, id);
+	const ambiguous = findMappedInQuotedSegments(sources.inspection, classes);
+	if (ambiguous) throw new Error(`[gustcss] mapped class "${ambiguous}" remains in an ambiguous string in ${id}. Move it to a static class/className or classList context, or add it to mangleExclude.`);
+	return edits;
 }
 const BASE64 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/";
 function encodeVLQ(value) {
@@ -696,7 +747,7 @@ function createClassNameTransformer(classes) {
 	const transformWithSourceMap = (code, id) => {
 		if (UNSUPPORTED_FRAMEWORK_SOURCE.test(id)) {
 			const referenced = findMappedToken(code, classes);
-			if (referenced) throw new Error(`[gustcss] class-name mangling does not support Vue, Astro, or Svelte templates yet (${id}). Add "${referenced}" to mangleExclude or disable mangleClassNames.`);
+			if (referenced) throw new Error(`[gustcss] class-name mangling does not support Vue or Svelte templates yet (${id}). Add "${referenced}" to mangleExclude or disable mangleClassNames.`);
 			return {
 				code,
 				map: null
@@ -716,9 +767,392 @@ function createClassNameTransformer(classes) {
 	};
 	const transform = (code, id) => transformWithSourceMap(code, id).code;
 	transform.withSourceMap = transformWithSourceMap;
+	transform.classes = classes;
 	return transform;
 }
 //#endregion
+//#region src/astro.js
+/**
+* Class-name mangling for `.astro` sources, driven by the AST that
+* `@astrojs/compiler`'s `parse()` returns.
+*
+* Only unambiguous class contexts are rewritten; any other place where a
+* mapped class name shows up fails the build (fail-closed), because a class
+* that reaches the DOM unrenamed would no longer match the mangled CSS.
+*/
+const CLASS_VALUED_BEFORE = /* @__PURE__ */ new Set([
+	"[",
+	",",
+	"&&",
+	"||"
+]);
+const CLASS_VALUED_AFTER = /* @__PURE__ */ new Set([
+	",",
+	"]",
+	"&&",
+	"||"
+]);
+const OPERATORS = [
+	"===",
+	"!==",
+	"&&",
+	"||",
+	"??",
+	"=>",
+	"==",
+	"!=",
+	"?.",
+	"<=",
+	">="
+];
+const PUNCTUATION = "[]{}(),:?.!+-*/%<>=&|~^;";
+/** Map UTF-8 byte offsets (what the compiler reports) to string indexes. */
+function byteToIndexMap(code) {
+	const map = [];
+	let byte = 0;
+	for (let index = 0; index < code.length; index += 1) {
+		const point = code.codePointAt(index);
+		const width = point < 128 ? 1 : point < 2048 ? 2 : point < 65536 ? 3 : 4;
+		for (let i = 0; i < width; i += 1) map[byte + i] = index;
+		byte += width;
+		if (point > 65535) index += 1;
+	}
+	map[byte] = code.length;
+	return map;
+}
+/**
+* Tokenize a JavaScript expression just far enough to know, for every string
+* literal and identifier, what surrounds it. Template literals are reported
+* as strings with quote "`" and are never rewritten. Returns null when the
+* expression cannot be tokenized (unterminated string, unknown character).
+*/
+function tokenizeExpression(source) {
+	const tokens = [];
+	let index = 0;
+	while (index < source.length) {
+		const char = source[index];
+		if (/\s/.test(char)) {
+			index += 1;
+			continue;
+		}
+		if (source.startsWith("//", index)) {
+			const newline = source.indexOf("\n", index);
+			index = newline < 0 ? source.length : newline + 1;
+			continue;
+		}
+		if (source.startsWith("/*", index)) {
+			const close = source.indexOf("*/", index + 2);
+			if (close < 0) return null;
+			index = close + 2;
+			continue;
+		}
+		if (char === "\"" || char === "'" || char === "`") {
+			const start = index;
+			index += 1;
+			while (index < source.length && source[index] !== char) {
+				if (source[index] === "\\") index += 1;
+				index += 1;
+			}
+			if (index >= source.length) return null;
+			tokens.push({
+				type: "string",
+				quote: char,
+				start,
+				end: index + 1,
+				value: source.slice(start + 1, index)
+			});
+			index += 1;
+			continue;
+		}
+		if (/[A-Za-z_$]/.test(char)) {
+			const start = index;
+			while (index < source.length && /[\w$]/.test(source[index])) index += 1;
+			tokens.push({
+				type: "identifier",
+				start,
+				end: index,
+				value: source.slice(start, index)
+			});
+			continue;
+		}
+		if (/[0-9]/.test(char)) {
+			const start = index;
+			while (index < source.length && /[\w.]/.test(source[index])) index += 1;
+			tokens.push({
+				type: "number",
+				start,
+				end: index,
+				value: source.slice(start, index)
+			});
+			continue;
+		}
+		const operator = OPERATORS.find((candidate) => source.startsWith(candidate, index));
+		if (operator) {
+			tokens.push({
+				type: "operator",
+				start: index,
+				end: index + operator.length,
+				value: operator
+			});
+			index += operator.length;
+			continue;
+		}
+		if (PUNCTUATION.includes(char)) {
+			tokens.push({
+				type: "operator",
+				start: index,
+				end: index + 1,
+				value: char
+			});
+			index += 1;
+			continue;
+		}
+		return null;
+	}
+	return tokens;
+}
+/**
+* Rewrite the class-valued string literals of a `class:list` expression.
+* Accepted shapes are array literals, object literals and their nesting.
+* Inside them only array elements (also as the right operand of `&&` / `||`)
+* and quoted object keys are class-valued; every other occurrence of a mapped
+* class name fails closed.
+*/
+function collectClassListExpressionEdits(source, classes, baseOffset, edits, id) {
+	const fail = (token, reason) => {
+		throw new Error(`[gustcss] mapped class "${token}" ${reason} in class:list of ${id}. Use an array/object literal with quoted entries, or add it to mangleExclude.`);
+	};
+	const tokens = tokenizeExpression(source);
+	const first = tokens && tokens[0];
+	const last = tokens && tokens[tokens.length - 1];
+	if (!(tokens && tokens.length >= 2 && (first.value === "[" && last.value === "]" || first.value === "{" && last.value === "}"))) {
+		const mapped = findMappedToken(source, classes);
+		if (mapped) fail(mapped, "remains in an expression that is not an array or object literal");
+		return;
+	}
+	const stack = [];
+	for (let i = 0; i < tokens.length; i += 1) {
+		const token = tokens[i];
+		const previous = tokens[i - 1];
+		const next = tokens[i + 1];
+		const context = stack[stack.length - 1];
+		if (token.type === "operator") {
+			if (token.value === "[") {
+				const subscript = previous && (previous.type === "identifier" || previous.type === "string" || previous.value === ")" || previous.value === "]");
+				stack.push(subscript ? "subscript" : "array");
+			} else if (token.value === "{") stack.push("object");
+			else if (token.value === "(") stack.push("call");
+			else if (token.value === "]" || token.value === "}" || token.value === ")") {
+				if (stack.length === 0) fail(findMappedToken(source, classes) || "?", "appears in an unbalanced");
+				stack.pop();
+			}
+			continue;
+		}
+		if (token.type === "identifier") {
+			if (!hasOwn(classes, token.value)) continue;
+			if (context === "object" && previous && (previous.value === "{" || previous.value === ",") && next && (next.value === ":" || next.value === "," || next.value === "}")) fail(token.value, `is an unquoted object key (write it as "${token.value}")`);
+			fail(token.value, "is referenced as an identifier");
+		}
+		if (token.type !== "string") continue;
+		const mapped = findMappedToken(token.value, classes);
+		if (token.quote === "`") {
+			if (mapped) fail(mapped, "remains in a template literal");
+			continue;
+		}
+		const arrayElement = context === "array" && previous && CLASS_VALUED_BEFORE.has(previous.value) && next && CLASS_VALUED_AFTER.has(next.value);
+		const objectKey = context === "object" && previous && (previous.value === "{" || previous.value === ",") && next && next.value === ":";
+		if (arrayElement || objectKey) collectClassListEdits(token.value, classes, baseOffset + token.start + 1, edits);
+		else if (mapped) fail(mapped, "remains in a position that is not an array element or object key");
+	}
+	if (stack.length !== 0) fail(findMappedToken(source, classes) || "?", "appears in an unbalanced");
+}
+/**
+* Find a mapped class referenced by a stylesheet selector, including at-rule
+* preludes (`@scope (.flex)`) and attribute selectors (`[class~="flex"]`).
+* Declarations and comments are ignored.
+*/
+function findMappedClassInStylesheet(css, classes) {
+	const source = css.replace(/\/\*[\s\S]*?\*\//g, " ");
+	let prelude = "";
+	let quote = null;
+	for (let i = 0; i < source.length; i += 1) {
+		const char = source[i];
+		if (quote) {
+			prelude += char;
+			if (char === "\\") {
+				prelude += source[i + 1] || "";
+				i += 1;
+			} else if (char === quote) quote = null;
+			continue;
+		}
+		if (char === "\"" || char === "'") {
+			quote = char;
+			prelude += char;
+			continue;
+		}
+		if (char === "{") {
+			const found = findMappedClassInSelector(prelude.trim(), classes);
+			if (found) return found;
+			prelude = "";
+		} else if (char === "}" || char === ";") prelude = "";
+		else prelude += char;
+	}
+	return null;
+}
+function findMappedClassInSelector(selector, classes) {
+	const classPattern = /\.((?:\\.|[A-Za-z0-9_-])+)/g;
+	let match;
+	while ((match = classPattern.exec(selector)) !== null) {
+		const name = match[1].replace(/\\(.)/g, "$1");
+		if (hasOwn(classes, name)) return name;
+	}
+	const attributePattern = /\[\s*class\s*[~|^$*]?=\s*(["']?)([^\]"']+)\1\s*[is]?\s*\]/g;
+	while ((match = attributePattern.exec(selector)) !== null) {
+		const mapped = findMappedToken(match[2], classes);
+		if (mapped) return mapped;
+	}
+	return null;
+}
+/**
+* Locate the value of an attribute from its start offset (the attribute
+* name). Returns the index range of the value and its delimiter.
+*/
+function locateAttributeValue(code, attribute, start) {
+	let index = start + attribute.name.length;
+	while (index < code.length && /\s/.test(code[index])) index += 1;
+	if (code[index] !== "=") return null;
+	index += 1;
+	while (index < code.length && /\s/.test(code[index])) index += 1;
+	const delimiter = code[index];
+	if (delimiter === "\"" || delimiter === "'" || delimiter === "`") {
+		const end = code.indexOf(delimiter, index + 1);
+		return end < 0 ? null : {
+			start: index + 1,
+			end,
+			delimiter
+		};
+	}
+	if (delimiter === "{") {
+		const end = findBalancedEnd(code, index, "{", "}");
+		return end < 0 ? null : {
+			start: index + 1,
+			end: end - 1,
+			delimiter
+		};
+	}
+	return null;
+}
+const STRING_LITERAL = /^\s*(["'])((?:\\.|(?!\1)[^\\])*)\1\s*$/;
+/**
+* Create the `.astro` transformer. `parse` is `@astrojs/compiler`'s parse
+* function, injected so the plugin and the tests can supply their own copy.
+*/
+function createAstroTransformer({ parse, classes }) {
+	return async function transformAstro(code, id) {
+		const { ast } = await parse(code, { position: true });
+		const toIndex = byteToIndexMap(code);
+		const edits = [];
+		const offsetOf = (node) => toIndex[node.position.start.offset];
+		const stop = (mapped, where) => {
+			throw new Error(`[gustcss] mapped class "${mapped}" remains in ${where} in ${id}. Move it to a static class attribute or class:list literal, or add it to mangleExclude.`);
+		};
+		const fragmentEdits = (text, start, kind) => {
+			for (const edit of collectFragmentEdits(text, classes, `${id}#${kind}.ts`)) edits.push({
+				start: start + edit.start,
+				end: start + edit.end,
+				replacement: edit.replacement
+			});
+		};
+		const visitAttribute = (attribute, node) => {
+			const start = offsetOf(attribute);
+			const isComponent = node.type === "component";
+			const value = attribute.value || "";
+			if (attribute.name === "class") {
+				const location = locateAttributeValue(code, attribute, start);
+				if (location && attribute.kind === "quoted") {
+					collectClassListEdits(code.slice(location.start, location.end), classes, location.start, edits);
+					return;
+				}
+				if (location && attribute.kind === "expression") {
+					const expression = code.slice(location.start, location.end);
+					const literal = STRING_LITERAL.exec(expression);
+					if (literal) {
+						const inner = expression.indexOf(literal[1]) + 1;
+						collectClassListEdits(literal[2], classes, location.start + inner, edits);
+						return;
+					}
+				}
+				const mapped = findMappedToken(value, classes);
+				if (mapped) stop(mapped, `a dynamic class attribute (${attribute.kind})`);
+				return;
+			}
+			if (attribute.name === "class:list") {
+				const location = locateAttributeValue(code, attribute, start);
+				if (!location || attribute.kind !== "expression") {
+					const mapped = findMappedToken(value, classes);
+					if (mapped) stop(mapped, `a class:list attribute of kind ${attribute.kind}`);
+					return;
+				}
+				collectClassListExpressionEdits(code.slice(location.start, location.end), classes, location.start, edits, id);
+				return;
+			}
+			if (attribute.kind === "spread") {
+				const mapped = findMappedToken(`${attribute.name} ${value}`, classes);
+				if (mapped) stop(mapped, `a spread attribute on <${node.name}>`);
+				return;
+			}
+			if (attribute.name === "set:html" || attribute.name === "set:text" || isComponent) {
+				const mapped = findMappedToken(value, classes);
+				if (mapped) stop(mapped, isComponent ? `the "${attribute.name}" prop of <${node.name}>` : `a ${attribute.name} value`);
+			}
+		};
+		const visit = (node) => {
+			switch (node.type) {
+				case "frontmatter": {
+					const start = code.indexOf(node.value, offsetOf(node));
+					if (start >= 0) fragmentEdits(node.value, start, "frontmatter");
+					return;
+				}
+				case "comment":
+				case "text":
+				case "doctype": return;
+				case "expression":
+					for (const child of node.children) if (child.type === "text") {
+						const mapped = findMappedToken(child.value, classes);
+						if (mapped && /["'`]/.test(child.value)) stop(mapped, "a template expression");
+					} else visit(child);
+					return;
+				case "element":
+				case "component":
+				case "custom-element":
+				case "fragment":
+					for (const attribute of node.attributes) visitAttribute(attribute, node);
+					if (node.type === "element" && node.name === "style") {
+						for (const child of node.children) {
+							if (child.type !== "text") continue;
+							const mapped = findMappedClassInStylesheet(child.value, classes);
+							if (mapped) stop(mapped, "a <style> selector");
+						}
+						return;
+					}
+					if (node.type === "element" && node.name === "script") {
+						for (const child of node.children) if (child.type === "text") fragmentEdits(child.value, offsetOf(child), "script");
+						return;
+					}
+					for (const child of node.children) visit(child);
+					return;
+				default: for (const child of node.children || []) visit(child);
+			}
+		};
+		visit(ast);
+		const applied = applyEdits(code, edits);
+		return {
+			code: applied.code,
+			map: createSourceMap(code, applied.code, applied.origins, id)
+		};
+	};
+}
+//#endregion
 //#region src/index.js
 /**
 * @gustcss/vite - Vite plugin for CSS Utility Generator
@@ -733,6 +1167,24 @@ function createClassNameTransformer(classes) {
 * are provided by the developer in their build configuration. This plugin is
 * designed for build-time use only and should not process untrusted input.
 */
+const ASTRO_SOURCE = /\.astro(?:$|\?)/;
+const ASTRO_MODULE = /^[^\0?]*\.astro$/;
+/**
+* Resolve @astrojs/compiler parse function from the project.
+*/
+async function resolveAstroCompiler(root) {
+	const projectRequire = createRequire(path.join(root, "package.json"));
+	const attempts = [
+		() => projectRequire.resolve("@astrojs/compiler"),
+		() => createRequire(projectRequire.resolve("astro/package.json")).resolve("@astrojs/compiler"),
+		() => createRequire(import.meta.url).resolve("@astrojs/compiler")
+	];
+	for (const attempt of attempts) try {
+		const mod = await import(pathToFileURL(attempt()).href);
+		if (typeof mod.parse === "function") return mod.parse;
+	} catch {}
+	return null;
+}
 function cssUtility(opts = {}) {
 	const output = opts.output || "src/styles/utility.css";
 	const content = opts.content || ["./src/**/*.{js,ts,jsx,tsx,astro,vue}"];
@@ -745,6 +1197,8 @@ function cssUtility(opts = {}) {
 	let rollbackContext = null;
 	let manifestSnapshot = null;
 	let watchProcess = null;
+	const loadedAstroModules = /* @__PURE__ */ new Set();
+	let astroTransformer = null;
 	function findBinary(cwd) {
 		return import_runner.default.resolveBinary({
 			cwd,
@@ -816,15 +1270,19 @@ function cssUtility(opts = {}) {
 			fs
 		}, (tempConfigPath) => run(tempConfigPath));
 	}
+	let viteConfig = null;
 	return {
 		name: "gustcss",
 		...mangleClassNames ? { enforce: "pre" } : {},
 		configResolved(config) {
+			viteConfig = config;
 			const cwd = config.root || process.cwd();
 			const binaryPath = findBinary(cwd);
 			try {
 				const enableMangle = mangleClassNames && config.command === "build";
 				buildCSS(cwd, binaryPath, enableMangle);
+				loadedAstroModules.clear();
+				astroTransformer = null;
 				rollbackContext = enableMangle ? {
 					cwd,
 					binaryPath
@@ -834,8 +1292,30 @@ function cssUtility(opts = {}) {
 				throw error;
 			}
 		},
+		async load(id) {
+			if (!classNameTransformer || !ASTRO_MODULE.test(id)) return null;
+			const filePath = id.startsWith("/@fs/") ? id.slice(4) : id;
+			if (!fs.existsSync(filePath)) throw new Error(`[gustcss] cannot read ${id} to rewrite its class names for mangling. Disable mangleClassNames or exclude its classes with mangleExclude.`);
+			const code = fs.readFileSync(filePath, "utf-8");
+			loadedAstroModules.add(id);
+			if (!astroTransformer) {
+				const parse = await resolveAstroCompiler(viteConfig?.root || process.cwd());
+				if (!parse) throw new Error(`[gustcss] @astrojs/compiler could not be resolved from the project, so ${id} cannot be mangled. Install astro, or disable mangleClassNames.`);
+				astroTransformer = createAstroTransformer({
+					parse,
+					classes: classNameTransformer.classes
+				});
+			}
+			const result = await astroTransformer(code, filePath);
+			return result.code === code ? null : result;
+		},
 		transform(code, id) {
 			if (!classNameTransformer) return null;
+			if (ASTRO_MODULE.test(id)) {
+				if (!loadedAstroModules.has(id)) throw new Error(`[gustcss] ${id} was compiled without passing through the gustcss load hook, so its class names cannot be mangled safely. Disable mangleClassNames or move the plugin so it loads .astro sources first.`);
+				return null;
+			}
+			if (ASTRO_SOURCE.test(id)) return null;
 			const transformed = classNameTransformer.withSourceMap(code, id);
 			return transformed.code === code ? null : transformed;
 		},
@@ -902,9 +1382,15 @@ function cssUtility(opts = {}) {
 				}
 				if (tempConfigPath && fs.existsSync(tempConfigPath)) fs.unlinkSync(tempConfigPath);
 			};
+			const close = server.close.bind(server);
+			server.close = async () => {
+				cleanup();
+				return close();
+			};
 			server.httpServer?.on("close", cleanup);
 			process.once("SIGINT", cleanup);
 			process.once("SIGTERM", cleanup);
+			process.once("exit", cleanup);
 		},
 		writeBundle() {
 			if (!manifestSnapshot || fs.existsSync(manifestSnapshot.path)) return;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gustcss/vite",
-  "version": "0.10.0",
+  "version": "0.10.1",
   "description": "Vite plugin for GustCSS",
   "main": "dist/index.mjs",
   "module": "dist/index.mjs",
@@ -37,9 +37,10 @@
     "access": "public"
   },
   "dependencies": {
-    "gustcss": "^0.10.0"
+    "gustcss": "^0.10.1"
   },
   "devDependencies": {
+    "@astrojs/compiler": "2.13.0",
     "rolldown": "^1.1.0",
     "vitest": "^4.1.8"
   },