@yahoo/uds 3.171.1 → 3.172.0

package/dist/css/dist/purger/optimized/ast/oxc.js

@@ -0,0 +1,253 @@
+/*! © 2026 Yahoo, Inc. UDS v0.0.0-development */
+import { parseSync, rawTransferSupported, visitorKeys } from "oxc-parser";
+//#region ../css/dist/purger/optimized/ast/oxc.mjs
+/*! © 2026 Yahoo, Inc. UDS CSS v0.0.0-development */
+const RAW_TRANSFER_AVAILABLE = rawTransferSupported();
+const getLangForFile = (filePath) => {
+	const lowered = filePath.toLowerCase();
+	if (lowered.endsWith(".tsx")) return "tsx";
+	if (lowered.endsWith(".ts") || lowered.endsWith(".mts") || lowered.endsWith(".cts")) return "ts";
+	return "jsx";
+};
+const isNodeLike = (value) => typeof value === "object" && value !== null && typeof value.type === "string";
+/**
+* Visit `node` and all descendants in document order (self included).
+*/
+const visitNodeAndDescendants = (node, parent, visit) => {
+	visit(node, parent);
+	const keys = visitorKeys[node.type] ?? [];
+	for (const key of keys) {
+		const value = node[key];
+		if (Array.isArray(value)) {
+			for (const child of value) if (isNodeLike(child)) visitNodeAndDescendants(child, node, visit);
+		} else if (isNodeLike(value)) visitNodeAndDescendants(value, node, visit);
+	}
+};
+/**
+* Visit all descendants of `node` (self excluded), mirroring
+* ts-morph `getDescendants()` / `getDescendantsOfKind()` scoped scans.
+*/
+const forEachDescendant = (node, visit) => {
+	visitNodeAndDescendants(node, null, (descendant) => {
+		if (descendant !== node) visit(descendant);
+	});
+};
+const getPatternIdentifier = (pattern) => {
+	if (pattern.type === "Identifier") return pattern;
+	if (pattern.type === "AssignmentPattern" && pattern.left.type === "Identifier") return pattern.left;
+	if (pattern.type === "RestElement" && pattern.argument.type === "Identifier") return pattern.argument;
+	return null;
+};
+const getPatternDefault = (pattern) => pattern.type === "AssignmentPattern" ? pattern.right : null;
+const getPatternTypeNode = (pattern) => {
+	return (pattern.type === "AssignmentPattern" ? pattern.left : pattern).typeAnnotation?.typeAnnotation ?? null;
+};
+const addDefinition = (source, definition) => {
+	const existing = source.definitions.get(definition.name);
+	if (existing) existing.push(definition);
+	else source.definitions.set(definition.name, [definition]);
+};
+const collectParameterDefinitions = (source, fn) => {
+	fn.params.forEach((param, index) => {
+		const identifier = getPatternIdentifier(param);
+		if (!identifier) return;
+		addDefinition(source, {
+			kind: "parameter",
+			name: identifier.name,
+			start: param.start,
+			node: param,
+			init: getPatternDefault(param),
+			typeNode: getPatternTypeNode(param),
+			paramFunction: fn,
+			paramIndex: index
+		});
+	});
+};
+const collectBindingDefinition = (source, element, pattern) => {
+	if (element.type === "Property") {
+		const identifier = getPatternIdentifier(element.value);
+		if (!identifier) return;
+		const key = element.key;
+		const bindingPropertyName = key.type === "Identifier" ? key.name : source.code.slice(key.start, key.end);
+		addDefinition(source, {
+			kind: "binding",
+			name: identifier.name,
+			start: element.start,
+			node: element,
+			init: getPatternDefault(element.value),
+			typeNode: null,
+			bindingPropertyName,
+			bindingPattern: pattern
+		});
+		return;
+	}
+	const identifier = getPatternIdentifier(element);
+	if (!identifier) return;
+	addDefinition(source, {
+		kind: "binding",
+		name: identifier.name,
+		start: element.start,
+		node: element,
+		init: getPatternDefault(element),
+		typeNode: null,
+		bindingPropertyName: identifier.name,
+		bindingPattern: pattern
+	});
+};
+const isTopLevel = (parent, source) => {
+	if (!parent) return false;
+	if (parent.type === "Program") return true;
+	if (parent.type === "ExportNamedDeclaration" || parent.type === "ExportDefaultDeclaration") return source.parents.get(parent)?.type === "Program";
+	return false;
+};
+const indexNode = (source, node, parent) => {
+	if (parent) source.parents.set(node, parent);
+	switch (node.type) {
+		case "JSXOpeningElement":
+			source.jsxElements.push(node);
+			break;
+		case "CallExpression":
+			source.callExpressions.push(node);
+			break;
+		case "JSXAttribute":
+			source.jsxAttributes.push(node);
+			break;
+		case "Property":
+			if (parent?.type === "ObjectExpression") source.objectProperties.push(node);
+			else if (parent?.type === "ObjectPattern") collectBindingDefinition(source, node, parent);
+			break;
+		case "RestElement":
+			if (parent?.type === "ObjectPattern" || parent?.type === "ArrayPattern") collectBindingDefinition(source, node, parent);
+			break;
+		case "Identifier":
+		case "AssignmentPattern":
+			if (parent?.type === "ArrayPattern") collectBindingDefinition(source, node, parent);
+			break;
+		case "VariableDeclarator":
+			source.variableDeclarators.push(node);
+			if (node.id.type === "Identifier") {
+				const name = node.id.name;
+				addDefinition(source, {
+					kind: "variable",
+					name,
+					start: node.start,
+					node,
+					init: node.init ?? null,
+					typeNode: getPatternTypeNode(node.id)
+				});
+				const declaration = source.parents.get(node);
+				if (declaration && isTopLevel(source.parents.get(declaration) ?? null, source)) {
+					if (!source.topLevelVariables.has(name)) source.topLevelVariables.set(name, node);
+				}
+			}
+			break;
+		case "FunctionDeclaration": {
+			const fn = node;
+			collectParameterDefinitions(source, fn);
+			if (fn.id && isTopLevel(parent, source)) {
+				const name = fn.id.name;
+				addDefinition(source, {
+					kind: "function",
+					name,
+					start: parent && parent.type !== "Program" && "start" in parent ? parent.start : fn.start,
+					node: fn,
+					init: null,
+					typeNode: null
+				});
+				if (!source.topLevelFunctions.has(name)) source.topLevelFunctions.set(name, fn);
+			}
+			break;
+		}
+		case "FunctionExpression":
+		case "ArrowFunctionExpression":
+			collectParameterDefinitions(source, node);
+			break;
+		case "TSEnumDeclaration":
+			if (isTopLevel(parent, source) && !source.topLevelEnums.has(node.id.name)) source.topLevelEnums.set(node.id.name, node);
+			break;
+		default: break;
+	}
+};
+/**
+* Parse a source file and build the purger's node indexes in a single walk.
+*/
+const parseSource = (filePath, code) => {
+	const parserOptions = { lang: getLangForFile(filePath) };
+	if (RAW_TRANSFER_AVAILABLE) parserOptions.experimentalRawTransfer = true;
+	const result = parseSync(filePath, code, parserOptions);
+	const source = {
+		filePath,
+		code,
+		program: result.program,
+		staticImports: result.module.staticImports,
+		parents: /* @__PURE__ */ new WeakMap(),
+		jsxElements: [],
+		callExpressions: [],
+		jsxAttributes: [],
+		objectProperties: [],
+		variableDeclarators: [],
+		topLevelVariables: /* @__PURE__ */ new Map(),
+		topLevelFunctions: /* @__PURE__ */ new Map(),
+		topLevelEnums: /* @__PURE__ */ new Map(),
+		definitions: /* @__PURE__ */ new Map(),
+		lineStarts: null
+	};
+	visitNodeAndDescendants(source.program, null, (node, parent) => indexNode(source, node, parent));
+	return source;
+};
+/** ts-morph `node.getText()` analog */
+const getText = (source, node) => source.code.slice(node.start, node.end);
+const getParent = (source, node) => source.parents.get(node);
+/** ts-morph `node.getStartLineNumber()` analog (1-based) */
+const getLineNumber = (source, offset) => {
+	if (!source.lineStarts) {
+		const lineStarts = [0];
+		for (let i = 0; i < source.code.length; i++) if (source.code[i] === "\n") lineStarts.push(i + 1);
+		source.lineStarts = lineStarts;
+	}
+	const lineStarts = source.lineStarts;
+	let low = 0;
+	let high = lineStarts.length - 1;
+	while (low < high) {
+		const mid = Math.ceil((low + high) / 2);
+		if (lineStarts[mid] <= offset) low = mid;
+		else high = mid - 1;
+	}
+	return low + 1;
+};
+/**
+* Position-aware local definition lookup, mirroring ts-morph's
+* `getLocalDefinitionNodes`: latest definition at or before the identifier.
+*/
+const findLocalDefinitions = (source, identifier) => {
+	if (identifier.type !== "Identifier") return [];
+	const candidates = source.definitions.get(identifier.name);
+	if (!candidates) return [];
+	const matches = candidates.filter((definition) => definition.start <= identifier.start);
+	if (matches.length === 0) return [];
+	return [...matches].sort((left, right) => right.start - left.start).slice(0, 1);
+};
+/** True for a string `Literal` node */
+const isStringLiteral = (node) => node.type === "Literal" && typeof node.value === "string";
+const getStringLiteralValue = (node) => node.value;
+/** True for a `TemplateLiteral` with no substitutions (ts NoSubstitutionTemplateLiteral) */
+const isNoSubstitutionTemplate = (node) => node.type === "TemplateLiteral" && node.expressions.length === 0;
+/** ts-morph `element.getTagNameNode().getText()` analog */
+const getJsxTagName = (source, element) => {
+	const name = element.name;
+	return name.type === "JSXIdentifier" ? name.name : source.code.slice(name.start, name.end);
+};
+/** ts-morph `attr.getNameNode().getText()` analog */
+const getJsxAttributeName = (source, attribute) => {
+	const name = attribute.name;
+	return name.type === "JSXIdentifier" ? name.name : source.code.slice(name.start, name.end);
+};
+/** ts-morph `prop.getName()` analog for object-literal properties */
+const getPropertyName = (source, property) => {
+	const key = property.key;
+	if (key.type === "Identifier") return key.name;
+	if (isStringLiteral(key)) return getStringLiteralValue(key);
+	return source.code.slice(key.start, key.end);
+};
+//#endregion
+export { findLocalDefinitions, forEachDescendant, getJsxAttributeName, getJsxTagName, getLineNumber, getParent, getPropertyName, getStringLiteralValue, getText, isNoSubstitutionTemplate, isStringLiteral, parseSource };

package/dist/css/dist/purger/optimized/ast/props.cjs

@@ -1,7 +1,6 @@
 /*! © 2026 Yahoo, Inc. UDS v0.0.0-development */
-require("../../../../../_virtual/_rolldown/runtime.cjs");
+const require_oxc = require("./oxc.cjs");
 const require_expressions = require("./expressions.cjs");
-let ts_morph = require("ts-morph");
 //#region ../css/dist/purger/optimized/ast/props.mjs
 /*! © 2026 Yahoo, Inc. UDS CSS v0.0.0-development */
 /**
@@ -11,27 +10,34 @@ let ts_morph = require("ts-morph");
 * For spread props like `<Box {...props} />`, marks them as fromSpread for later resolution.
 *
 * @param element The JSX element to extract props from
+* @param source The parsed source containing the element
 * @returns Array of extracted props with their values
 */
-const extractPropsFromElement = (element) => element.getAttributes().flatMap((attr) => {
-	if (ts_morph.Node.isJsxAttribute(attr)) {
-		const propName = attr.getNameNode().getText();
-		const stringLiterals = attr.getDescendantsOfKind(ts_morph.SyntaxKind.StringLiteral);
+const extractPropsFromElement = (element, source) => element.attributes.flatMap((attr) => {
+	if (attr.type === "JSXAttribute") {
+		const propName = require_oxc.getJsxAttributeName(source, attr);
+		const stringLiterals = [];
+		if (attr.value) {
+			if (require_oxc.isStringLiteral(attr.value)) stringLiterals.push(attr.value);
+			require_oxc.forEachDescendant(attr.value, (descendant) => {
+				if (require_oxc.isStringLiteral(descendant)) stringLiterals.push(descendant);
+			});
+		}
 		return [{
 			name: propName,
-			values: stringLiterals.length > 0 ? stringLiterals.map((lit) => lit.getLiteralText()) : (() => {
-				const initializer = attr.getInitializer();
-				if (!initializer || !ts_morph.Node.isJsxExpression(initializer)) return [];
-				const expression = initializer.getExpression();
-				return expression ? require_expressions.extractStringLiterals(expression) : [];
+			values: stringLiterals.length > 0 ? stringLiterals.map((lit) => require_oxc.getStringLiteralValue(lit)) : (() => {
+				const initializer = attr.value;
+				if (!initializer || initializer.type !== "JSXExpressionContainer") return [];
+				const expression = initializer.expression;
+				return expression && expression.type !== "JSXEmptyExpression" ? require_expressions.extractStringLiterals(expression, source) : [];
 			})(),
 			fromSpread: false
 		}];
 	}
-	if (ts_morph.Node.isJsxSpreadAttribute(attr)) {
-		const expression = attr.getExpression();
+	if (attr.type === "JSXSpreadAttribute") {
+		const expression = attr.argument;
 		return [{
-			name: `__spread__:${ts_morph.Node.isIdentifier(expression) ? expression.getText() : expression.getChildren().pop()?.getText() ?? ""}`,
+			name: `__spread__:${expression.type === "Identifier" ? expression.name : expression.type === "MemberExpression" && !expression.computed && expression.property.type === "Identifier" ? expression.property.name : require_oxc.getText(source, expression).split(".").pop() ?? ""}`,
 			values: [],
 			fromSpread: true
 		}];
@@ -42,6 +48,6 @@ const extractPropsFromElement = (element) => element.getAttributes().flatMap((at
 * Extract only explicit (non-spread) props from a JSX element.
 * Used when we want to avoid including spread-derived values.
 */
-const extractExplicitProps = (element) => extractPropsFromElement(element).filter((p) => !p.fromSpread);
+const extractExplicitProps = (element, source) => extractPropsFromElement(element, source).filter((p) => !p.fromSpread);
 //#endregion
 exports.extractExplicitProps = extractExplicitProps;

package/dist/css/dist/purger/optimized/ast/props.js

@@ -1,6 +1,6 @@
 /*! © 2026 Yahoo, Inc. UDS v0.0.0-development */
+import { forEachDescendant, getJsxAttributeName, getStringLiteralValue, getText, isStringLiteral } from "./oxc.js";
 import { extractStringLiterals } from "./expressions.js";
-import { Node, SyntaxKind } from "ts-morph";
 //#region ../css/dist/purger/optimized/ast/props.mjs
 /*! © 2026 Yahoo, Inc. UDS CSS v0.0.0-development */
 /**
@@ -10,27 +10,34 @@ import { Node, SyntaxKind } from "ts-morph";
 * For spread props like `<Box {...props} />`, marks them as fromSpread for later resolution.
 *
 * @param element The JSX element to extract props from
+* @param source The parsed source containing the element
 * @returns Array of extracted props with their values
 */
-const extractPropsFromElement = (element) => element.getAttributes().flatMap((attr) => {
-	if (Node.isJsxAttribute(attr)) {
-		const propName = attr.getNameNode().getText();
-		const stringLiterals = attr.getDescendantsOfKind(SyntaxKind.StringLiteral);
+const extractPropsFromElement = (element, source) => element.attributes.flatMap((attr) => {
+	if (attr.type === "JSXAttribute") {
+		const propName = getJsxAttributeName(source, attr);
+		const stringLiterals = [];
+		if (attr.value) {
+			if (isStringLiteral(attr.value)) stringLiterals.push(attr.value);
+			forEachDescendant(attr.value, (descendant) => {
+				if (isStringLiteral(descendant)) stringLiterals.push(descendant);
+			});
+		}
 		return [{
 			name: propName,
-			values: stringLiterals.length > 0 ? stringLiterals.map((lit) => lit.getLiteralText()) : (() => {
-				const initializer = attr.getInitializer();
-				if (!initializer || !Node.isJsxExpression(initializer)) return [];
-				const expression = initializer.getExpression();
-				return expression ? extractStringLiterals(expression) : [];
+			values: stringLiterals.length > 0 ? stringLiterals.map((lit) => getStringLiteralValue(lit)) : (() => {
+				const initializer = attr.value;
+				if (!initializer || initializer.type !== "JSXExpressionContainer") return [];
+				const expression = initializer.expression;
+				return expression && expression.type !== "JSXEmptyExpression" ? extractStringLiterals(expression, source) : [];
 			})(),
 			fromSpread: false
 		}];
 	}
-	if (Node.isJsxSpreadAttribute(attr)) {
-		const expression = attr.getExpression();
+	if (attr.type === "JSXSpreadAttribute") {
+		const expression = attr.argument;
 		return [{
-			name: `__spread__:${Node.isIdentifier(expression) ? expression.getText() : expression.getChildren().pop()?.getText() ?? ""}`,
+			name: `__spread__:${expression.type === "Identifier" ? expression.name : expression.type === "MemberExpression" && !expression.computed && expression.property.type === "Identifier" ? expression.property.name : getText(source, expression).split(".").pop() ?? ""}`,
 			values: [],
 			fromSpread: true
 		}];
@@ -41,6 +48,6 @@ const extractPropsFromElement = (element) => element.getAttributes().flatMap((at
 * Extract only explicit (non-spread) props from a JSX element.
 * Used when we want to avoid including spread-derived values.
 */
-const extractExplicitProps = (element) => extractPropsFromElement(element).filter((p) => !p.fromSpread);
+const extractExplicitProps = (element, source) => extractPropsFromElement(element, source).filter((p) => !p.fromSpread);
 //#endregion
 export { extractExplicitProps };

package/dist/css/dist/purger/optimized/ast/spread.cjs

@@ -1,32 +1,31 @@
 /*! © 2026 Yahoo, Inc. UDS v0.0.0-development */
-require("../../../../../_virtual/_rolldown/runtime.cjs");
+const require_oxc = require("./oxc.cjs");
 const require_jsx = require("./jsx.cjs");
 const require_props = require("./props.cjs");
-let ts_morph = require("ts-morph");
 //#region ../css/dist/purger/optimized/ast/spread.mjs
 /*! © 2026 Yahoo, Inc. UDS CSS v0.0.0-development */
 /**
-* Find all usages of a component across the entire project.
+* Find all usages of a component across the analyzed sources.
 * This is the key function for spread resolution - when we encounter
 * {...props} on a UDS component inside a wrapper, we trace back to
 * find all places where the wrapper is used and extract actual values.
 *
 * @param componentName Name of the component to find usages of
-* @param project The ts-morph project
+* @param sources The parsed sources to search
 * @param cache Optional cache to avoid re-scanning
 * @returns Array of component usages with their props
 */
-const findComponentUsages = (componentName, project, cache) => {
+const findComponentUsages = (componentName, sources, cache) => {
 	if (cache?.has(componentName)) return cache.get(componentName);
 	const usages = [];
-	project.getSourceFiles().forEach((sourceFile) => {
-		if (!fileReferencesComponent(sourceFile, componentName)) return;
-		[...sourceFile.getDescendantsOfKind(ts_morph.SyntaxKind.JsxSelfClosingElement), ...sourceFile.getDescendantsOfKind(ts_morph.SyntaxKind.JsxOpeningElement)].filter((el) => el.getTagNameNode().getText() === componentName).forEach((element) => {
-			const props = require_props.extractExplicitProps(element);
+	sources.forEach((source) => {
+		if (!fileReferencesComponent(source, componentName)) return;
+		source.jsxElements.filter((el) => require_oxc.getJsxTagName(source, el) === componentName).forEach((element) => {
+			const props = require_props.extractExplicitProps(element, source);
 			usages.push({
 				componentName,
-				filePath: sourceFile.getFilePath(),
-				line: element.getStartLineNumber(),
+				filePath: source.filePath,
+				line: require_oxc.getLineNumber(source, element.start),
 				props
 			});
 		});
@@ -37,10 +36,10 @@ const findComponentUsages = (componentName, project, cache) => {
 /**
 * Check if a source file references a component (by import or definition)
 */
-const fileReferencesComponent = (sourceFile, componentName) => {
-	if (sourceFile.getImportDeclarations().some((imp) => imp.getNamedImports().some((named) => named.getName() === componentName))) return true;
-	if (sourceFile.getVariableDeclaration(componentName)) return true;
-	if (sourceFile.getFunction(componentName)) return true;
+const fileReferencesComponent = (source, componentName) => {
+	if (source.staticImports.some((staticImport) => staticImport.entries.some((entry) => entry.importName.kind === "Name" && entry.importName.name === componentName))) return true;
+	if (source.topLevelVariables.has(componentName)) return true;
+	if (source.topLevelFunctions.has(componentName)) return true;
 	return false;
 };
 /**
@@ -54,14 +53,15 @@ const fileReferencesComponent = (sourceFile, componentName) => {
 *
 * @param element The JSX element with the spread
 * @param spreadIdentifier The identifier being spread (e.g., 'rest' from {...rest})
+* @param elementSource The parsed source containing the element
 * @param context The purge context for caching
 * @returns Map of prop names to their resolved values, or undefined if couldn't trace
 */
-const resolveSpreadFromUsages = (element, spreadIdentifier, context) => {
-	const parentInfo = require_jsx.getParentComponentInfo(element);
+const resolveSpreadFromUsages = (element, spreadIdentifier, elementSource, context) => {
+	const parentInfo = require_jsx.getParentComponentInfo(element, elementSource);
 	if (!parentInfo) return;
 	if (!(parentInfo.spreadRestIdentifier === spreadIdentifier || parentInfo.paramIdentifier === spreadIdentifier)) return;
-	const usages = findComponentUsages(parentInfo.componentName, context.project, context.componentUsageCache);
+	const usages = findComponentUsages(parentInfo.componentName, context.sources, context.componentUsageCache);
 	if (usages.length === 0) return;
 	context.stats.spreadsTraced++;
 	const resolvedProps = /* @__PURE__ */ new Map();
@@ -75,7 +75,7 @@ const resolveSpreadFromUsages = (element, spreadIdentifier, context) => {
 			resolvedProps.set(prop.name, existing);
 		});
 	});
-	const cacheKey = `${element.getSourceFile().getFilePath()}:${element.getStartLineNumber()}:${spreadIdentifier}`;
+	const cacheKey = `${elementSource.filePath}:${require_oxc.getLineNumber(elementSource, element.start)}:${spreadIdentifier}`;
 	context.spreadCache.set(cacheKey, Array.from(resolvedProps).map(([name, values]) => ({
 		name,
 		values,

package/dist/css/dist/purger/optimized/ast/spread.js

@@ -1,31 +1,31 @@
 /*! © 2026 Yahoo, Inc. UDS v0.0.0-development */
+import { getJsxTagName, getLineNumber } from "./oxc.js";
 import { getParentComponentInfo } from "./jsx.js";
 import { extractExplicitProps } from "./props.js";
-import { SyntaxKind } from "ts-morph";
 //#region ../css/dist/purger/optimized/ast/spread.mjs
 /*! © 2026 Yahoo, Inc. UDS CSS v0.0.0-development */
 /**
-* Find all usages of a component across the entire project.
+* Find all usages of a component across the analyzed sources.
 * This is the key function for spread resolution - when we encounter
 * {...props} on a UDS component inside a wrapper, we trace back to
 * find all places where the wrapper is used and extract actual values.
 *
 * @param componentName Name of the component to find usages of
-* @param project The ts-morph project
+* @param sources The parsed sources to search
 * @param cache Optional cache to avoid re-scanning
 * @returns Array of component usages with their props
 */
-const findComponentUsages = (componentName, project, cache) => {
+const findComponentUsages = (componentName, sources, cache) => {
 	if (cache?.has(componentName)) return cache.get(componentName);
 	const usages = [];
-	project.getSourceFiles().forEach((sourceFile) => {
-		if (!fileReferencesComponent(sourceFile, componentName)) return;
-		[...sourceFile.getDescendantsOfKind(SyntaxKind.JsxSelfClosingElement), ...sourceFile.getDescendantsOfKind(SyntaxKind.JsxOpeningElement)].filter((el) => el.getTagNameNode().getText() === componentName).forEach((element) => {
-			const props = extractExplicitProps(element);
+	sources.forEach((source) => {
+		if (!fileReferencesComponent(source, componentName)) return;
+		source.jsxElements.filter((el) => getJsxTagName(source, el) === componentName).forEach((element) => {
+			const props = extractExplicitProps(element, source);
 			usages.push({
 				componentName,
-				filePath: sourceFile.getFilePath(),
-				line: element.getStartLineNumber(),
+				filePath: source.filePath,
+				line: getLineNumber(source, element.start),
 				props
 			});
 		});
@@ -36,10 +36,10 @@ const findComponentUsages = (componentName, project, cache) => {
 /**
 * Check if a source file references a component (by import or definition)
 */
-const fileReferencesComponent = (sourceFile, componentName) => {
-	if (sourceFile.getImportDeclarations().some((imp) => imp.getNamedImports().some((named) => named.getName() === componentName))) return true;
-	if (sourceFile.getVariableDeclaration(componentName)) return true;
-	if (sourceFile.getFunction(componentName)) return true;
+const fileReferencesComponent = (source, componentName) => {
+	if (source.staticImports.some((staticImport) => staticImport.entries.some((entry) => entry.importName.kind === "Name" && entry.importName.name === componentName))) return true;
+	if (source.topLevelVariables.has(componentName)) return true;
+	if (source.topLevelFunctions.has(componentName)) return true;
 	return false;
 };
 /**
@@ -53,14 +53,15 @@ const fileReferencesComponent = (sourceFile, componentName) => {
 *
 * @param element The JSX element with the spread
 * @param spreadIdentifier The identifier being spread (e.g., 'rest' from {...rest})
+* @param elementSource The parsed source containing the element
 * @param context The purge context for caching
 * @returns Map of prop names to their resolved values, or undefined if couldn't trace
 */
-const resolveSpreadFromUsages = (element, spreadIdentifier, context) => {
-	const parentInfo = getParentComponentInfo(element);
+const resolveSpreadFromUsages = (element, spreadIdentifier, elementSource, context) => {
+	const parentInfo = getParentComponentInfo(element, elementSource);
 	if (!parentInfo) return;
 	if (!(parentInfo.spreadRestIdentifier === spreadIdentifier || parentInfo.paramIdentifier === spreadIdentifier)) return;
-	const usages = findComponentUsages(parentInfo.componentName, context.project, context.componentUsageCache);
+	const usages = findComponentUsages(parentInfo.componentName, context.sources, context.componentUsageCache);
 	if (usages.length === 0) return;
 	context.stats.spreadsTraced++;
 	const resolvedProps = /* @__PURE__ */ new Map();
@@ -74,7 +75,7 @@ const resolveSpreadFromUsages = (element, spreadIdentifier, context) => {
 			resolvedProps.set(prop.name, existing);
 		});
 	});
-	const cacheKey = `${element.getSourceFile().getFilePath()}:${element.getStartLineNumber()}:${spreadIdentifier}`;
+	const cacheKey = `${elementSource.filePath}:${getLineNumber(elementSource, element.start)}:${spreadIdentifier}`;
 	context.spreadCache.set(cacheKey, Array.from(resolvedProps).map(([name, values]) => ({
 		name,
 		values,