@yahoo/uds 3.114.0-beta.2 → 3.114.0

package/dist/tailwind/dist/purger/optimized/purgeFromCode.js

@@ -8,20 +8,20 @@ import { Node, Project, SyntaxKind, ts } from "ts-morph";
 /**
 * Get component info with all its dependencies (using pre-generated data)
 */
-const getComponentWithDeps = (componentName) => {
+function getComponentWithDeps(componentName) {
 	const visited = /* @__PURE__ */ new Set();
 	const result = [];
-	const collect = (name) => {
+	function collect(name) {
 		if (visited.has(name)) return;
 		visited.add(name);
-		const info = resolveComponentInfo(name);
+		const info = componentCache[name];
 		if (!info) return;
 		result.push(info);
-		info.internalComponents.forEach((depName) => collect(depName));
-	};
+		for (const depName of info.internalComponents) collect(depName);
+	}
 	collect(componentName);
 	return result;
-};
+}
 /**
 * Auto-variants lookup (loaded at module init)
 */
@@ -30,96 +30,10 @@ let autoVariantsCache = null;
 * Pre-generated component data (loaded at module init)
 */
 let componentCache = null;
-/**
-* Map of normalized component name -> actual component cache key
-* Useful when internal components are imported without namespace separators
-* (e.g., MenuItemBase vs Menu.ItemBase).
-*/
-let componentNameLookup = null;
-const buildComponentNameLookup = (data) => {
-	const lookup = {};
-	Object.keys(data).forEach((key) => {
-		const normalized = normalizeComponentName(key);
-		if (!lookup[normalized]) lookup[normalized] = key;
-	});
-	return lookup;
-};
-const resolveComponentInfo = (name) => {
-	if (componentCache && componentCache[name]) return componentCache[name];
-	const normalized = normalizeComponentName(name);
-	const mappedName = componentNameLookup?.[normalized];
-	if (mappedName && componentCache) return componentCache[mappedName];
-};
-const getModuleSpecifierValue = (importDecl) => {
-	return importDecl.getModuleSpecifierValue() ?? importDecl.getModuleSpecifier().getText().replace(/^['"]|['"]$/g, "");
-};
-const isUdsComponentModule = (moduleSpecifier) => {
-	const cleaned = moduleSpecifier.replace(/^['"]|['"]$/g, "");
-	return cleaned === "@yahoo/uds" || cleaned.startsWith("@yahoo/uds/");
-};
-/**
-* Normalize component name to match autoVariants keys (lower camel, no punctuation).
-*/
-const normalizeComponentName = (name) => {
-	const cleaned = name.replace(/\W/g, "");
-	return cleaned.charAt(0).toLowerCase() + cleaned.slice(1);
-};
-/**
-* Fallback: infer prop -> variant key mapping from autoVariants when the
-* generated component data misses it (e.g., helpers like buildAvatarRootClasses).
-*/
-const getAutoVariantKeysForProp = (componentName, propName) => {
-	if (!autoVariantsCache) return [];
-	const segments = componentName.match(/[A-Z][a-z0-9]*/g) ?? [];
-	const candidates = new Set([normalizeComponentName(componentName)]);
-	if (segments.length > 0) Array.from({ length: segments.length }, (_, index) => index + 1).forEach((i) => {
-		const partial = segments.slice(0, i).join("");
-		candidates.add(normalizeComponentName(partial));
-	});
-	const propLower = propName.toLowerCase();
-	return Object.keys(autoVariantsCache).filter((key) => {
-		const keyLower = key.toLowerCase();
-		return Array.from(candidates).some((candidate) => keyLower.startsWith(candidate) && keyLower.includes(propLower));
-	});
-};
-const getVariantKeysForProp = (componentInfo, propName) => {
-	const keys = new Set(componentInfo.propToVariantKeys?.[propName] ?? []);
-	getAutoVariantKeysForProp(componentInfo.name, propName).forEach((inferred) => keys.add(inferred));
-	return [...keys];
-};
-const extractClassNamePrimitives = (sourceFile) => {
-	const classes = [];
-	sourceFile.getDescendantsOfKind(SyntaxKind.JsxAttribute).forEach((attr) => {
-		if (attr.getNameNode().getText() !== "className") return;
-		const initializer = attr.getInitializer();
-		if (!initializer) return;
-		const values = [];
-		if (Node.isStringLiteral(initializer)) values.push(initializer.getLiteralText());
-		else if (Node.isJsxExpression(initializer)) {
-			const expr = initializer.getExpression();
-			if (expr) values.push(...extractStringLiterals(expr));
-		}
-		values.forEach((raw) => {
-			classes.push(...raw.split(/\s+/).filter(Boolean));
-		});
-	});
-	sourceFile.getDescendantsOfKind(SyntaxKind.PropertyAssignment).forEach((prop) => {
-		if (prop.getName() !== "className") return;
-		const initializer = prop.getInitializer();
-		if (!initializer) return;
-		extractStringLiterals(initializer).forEach((raw) => {
-			classes.push(...raw.split(/\s+/).filter(Boolean));
-		});
-	});
-	return classes;
-};
-const purgeFromCodeOptimized = async (code, options) => {
+async function purgeFromCodeOptimized(code, options) {
 	const { variants, autoVariants, componentData } = options;
 	if (!autoVariantsCache) autoVariantsCache = autoVariants;
-	if (!componentCache) {
-		componentCache = componentData;
-		componentNameLookup = buildComponentNameLookup(componentData);
-	}
+	if (!componentCache) componentCache = componentData;
 	const startTime = performance.now();
 	const sourceFile = new Project({ useInMemoryFileSystem: true }).createSourceFile("input.tsx", code);
 	const stats = {
@@ -130,120 +44,105 @@ const purgeFromCodeOptimized = async (code, options) => {
 		expressionsResolved: 0
 	};
 	const imports = [];
-	sourceFile.getImportDeclarations().forEach((importDecl) => {
-		if (isUdsComponentModule(getModuleSpecifierValue(importDecl))) importDecl.getNamedImports().forEach((namedImport) => imports.push(namedImport.getName()));
-	});
+	for (const importDecl of sourceFile.getImportDeclarations()) if (importDecl.getModuleSpecifier().getText().includes("@yahoo/uds")) for (const namedImport of importDecl.getNamedImports()) imports.push(namedImport.getName());
 	const componentProps = /* @__PURE__ */ new Map();
-	const referencedComponents = /* @__PURE__ */ new Set();
 	const getStylesProps = extractGetStylesCalls(sourceFile, stats, variants);
-	imports.forEach((imp) => {
-		findComponentReferences(sourceFile, imp).forEach((reference) => {
-			const componentName = reference.getTagNameNode().getText() || imp;
-			referencedComponents.add(componentName);
-			const propsMap = componentProps.get(componentName) ?? /* @__PURE__ */ new Map();
-			extractPropsFromReference(reference, stats, sourceFile).forEach(([propName, values]) => {
-				if (values.length > 0) {
-					const existing = propsMap.get(propName) ?? /* @__PURE__ */ new Set();
-					values.forEach((val) => existing.add(val));
-					propsMap.set(propName, existing);
-				}
-			});
-			componentProps.set(componentName, propsMap);
-		});
-		if (!componentProps.has(imp)) componentProps.set(imp, /* @__PURE__ */ new Map());
-	});
+	for (const imp of imports) {
+		const references = findComponentReferences(sourceFile, imp);
+		const propsMap = componentProps.get(imp) ?? /* @__PURE__ */ new Map();
+		for (const reference of references) {
+			const props = extractPropsFromReference(reference, stats, sourceFile);
+			for (const [propName, values] of props) if (values.length > 0) {
+				const existing = propsMap.get(propName) ?? /* @__PURE__ */ new Set();
+				for (const val of values) existing.add(val);
+				propsMap.set(propName, existing);
+			}
+		}
+		componentProps.set(imp, propsMap);
+	}
 	const allComponents = [];
 	const seenComponents = /* @__PURE__ */ new Set();
-	new Set([...imports, ...referencedComponents]).forEach((componentName) => {
-		getComponentWithDeps(componentName).forEach((comp) => {
-			if (!seenComponents.has(comp.name)) {
-				seenComponents.add(comp.name);
-				allComponents.push(comp);
-			}
-		});
-	});
+	for (const componentName of imports) {
+		const componentsWithDeps = getComponentWithDeps(componentName);
+		for (const comp of componentsWithDeps) if (!seenComponents.has(comp.name)) {
+			seenComponents.add(comp.name);
+			allComponents.push(comp);
+		}
+	}
 	const safelist = [];
-	[...getStylesProps.entries()].forEach(([propName, values]) => {
+	for (const [propName, values] of getStylesProps) {
 		const variantGroup = variants[propName];
-		if (variantGroup) values.forEach((value) => {
+		if (variantGroup) for (const value of values) {
 			const cls = variantGroup[value];
 			if (cls) safelist.push(cls.replaceAll("\\", ""));
-		});
-	});
-	allComponents.forEach((componentInfo) => {
+		}
+	}
+	for (const componentInfo of allComponents) {
 		const userProps = componentProps.get(componentInfo.name) ?? /* @__PURE__ */ new Map();
-		Object.entries(componentInfo.defaultProps).forEach(([propName, propValue]) => {
-			if (userProps.has(propName)) return;
+		for (const [propName, propValue] of Object.entries(componentInfo.defaultProps)) {
+			if (userProps.has(propName)) continue;
 			const variantGroup = variants[propName];
 			if (variantGroup) {
 				const cls = variantGroup[propValue];
 				if (cls) safelist.push(cls.replaceAll("\\", ""));
 			}
-			const variantKeys = getVariantKeysForProp(componentInfo, propName);
-			if (variantKeys.length > 0) variantKeys.forEach((variantKey) => {
-				const mappedVariantGroup = variants[variantKey] ?? autoVariantsCache[variantKey];
+			const variantKeys = componentInfo.propToVariantKeys?.[propName];
+			if (variantKeys) for (const variantKey of variantKeys) {
+				const mappedVariantGroup = variants[variantKey];
 				if (mappedVariantGroup) {
 					const cls = mappedVariantGroup[propValue];
 					if (cls) safelist.push(cls.replaceAll("\\", ""));
 				}
-			});
-		});
-		Object.entries(componentInfo.getStylesLiterals).forEach(([propName, propValue]) => {
+			}
+		}
+		for (const [propName, propValue] of Object.entries(componentInfo.getStylesLiterals)) {
 			const variantGroup = variants[propName.replace(/:\d+$/, "")];
 			if (variantGroup) {
 				const cls = variantGroup[propValue];
 				if (cls) safelist.push(cls.replaceAll("\\", ""));
 			}
-		});
-		if (componentInfo.cxLiterals) componentInfo.cxLiterals.forEach((cls) => safelist.push(cls));
+		}
+		if (componentInfo.cxLiterals) for (const cls of componentInfo.cxLiterals) safelist.push(cls);
 		if (componentInfo.propToVariantKeys) {
 			const componentDefaults = options.variantDefaults?.[componentInfo.name] ?? {};
-			Object.keys(componentInfo.propToVariantKeys).forEach((propName) => {
-				if (userProps.has(propName)) return;
+			for (const [propName, variantKeys] of Object.entries(componentInfo.propToVariantKeys)) {
+				if (userProps.has(propName)) continue;
 				const defaultValue = componentDefaults[propName];
-				getVariantKeysForProp(componentInfo, propName).forEach((variantKey) => {
+				for (const variantKey of variantKeys) {
 					const variantGroup = autoVariantsCache[variantKey];
 					if (variantGroup) {
 						if (variantGroup["default"]) safelist.push(variantGroup["default"]);
 						if (defaultValue && variantGroup[defaultValue]) safelist.push(variantGroup[defaultValue]);
 					}
-				});
-			});
+				}
+			}
 		}
-		Object.entries(componentInfo.internalComponentProps).forEach(([_internalCompName, propsRecord]) => {
-			Object.entries(propsRecord).forEach(([propName, propValues]) => {
-				const variantGroup = variants[propName];
-				if (variantGroup) propValues.forEach((propValue) => {
-					const cls = variantGroup[propValue];
-					if (cls) safelist.push(cls.replaceAll("\\", ""));
-				});
-			});
-		});
-		[...userProps.entries()].forEach(([propName, usedValues]) => {
-			if (propName === "className") {
-				[...usedValues].forEach((value) => {
-					value.split(/\s+/).filter(Boolean).forEach((cls) => safelist.push(cls));
-				});
-				return;
+		for (const [_internalCompName, propsRecord] of Object.entries(componentInfo.internalComponentProps)) for (const [propName, propValues] of Object.entries(propsRecord)) {
+			const variantGroup = variants[propName];
+			if (variantGroup) for (const propValue of propValues) {
+				const cls = variantGroup[propValue];
+				if (cls) safelist.push(cls.replaceAll("\\", ""));
 			}
+		}
+		for (const [propName, usedValues] of userProps) {
 			const variantGroup = variants[propName];
-			if (variantGroup) [...usedValues].forEach((value) => {
+			if (variantGroup) for (const value of usedValues) {
 				const cls = variantGroup[value];
 				if (cls) safelist.push(cls.replaceAll("\\", ""));
-			});
-			[componentInfo, ...getComponentWithDeps(componentInfo.name).slice(1)].forEach((comp) => {
-				const variantKeys = getVariantKeysForProp(comp, propName);
-				if (variantKeys.length > 0) variantKeys.forEach((variantKey) => {
+			}
+			const componentsToCheck = [componentInfo, ...getComponentWithDeps(componentInfo.name).slice(1)];
+			for (const comp of componentsToCheck) {
+				const variantKeys = comp.propToVariantKeys?.[propName];
+				if (variantKeys) for (const variantKey of variantKeys) {
 					const mappedVariantGroup = variants[variantKey];
-					if (mappedVariantGroup) [...usedValues].forEach((value) => {
+					if (mappedVariantGroup) for (const value of usedValues) {
 						const cls = mappedVariantGroup[value];
 						if (cls) safelist.push(cls.replaceAll("\\", ""));
-					});
-				});
-			});
-		});
-	});
-	if (options.includeAllClassNamePrimitives) safelist.push(...extractClassNamePrimitives(sourceFile));
+					}
+				}
+			}
+		}
+	}
 	const finalSafelist = [...new Set(safelist)];
 	stats.timeMs = Math.round(performance.now() - startTime);
 	stats.classesGenerated = finalSafelist.length;
@@ -253,29 +152,28 @@ const purgeFromCodeOptimized = async (code, options) => {
 		components: [...seenComponents],
 		stats
 	};
-};
+}
 /**
 * Find JSX references for a component in the source file
 */
-const findComponentReferences = (sourceFile, componentName) => {
+function findComponentReferences(sourceFile, componentName) {
 	const references = [];
-	sourceFile.getImportDeclarations().forEach((importDecl) => {
-		if (!isUdsComponentModule(getModuleSpecifierValue(importDecl))) return;
-		importDecl.getNamedImports().forEach((namedImport) => {
-			if (namedImport.getName() === componentName) {
-				const identifier = namedImport.getFirstDescendantByKindOrThrow(ts.SyntaxKind.Identifier);
-				references.push(...findJsxReferences(identifier));
-			}
-		});
-	});
+	for (const importDecl of sourceFile.getImportDeclarations()) {
+		if (!importDecl.getModuleSpecifier().getText().includes("@yahoo/uds")) continue;
+		for (const namedImport of importDecl.getNamedImports()) if (namedImport.getName() === componentName) {
+			const identifier = namedImport.getFirstDescendantByKindOrThrow(ts.SyntaxKind.Identifier);
+			references.push(...findJsxReferences(identifier));
+		}
+	}
 	return references;
-};
+}
 /**
 * Extract props from a JSX reference.
 */
-const extractPropsFromReference = (reference, stats, sourceFile) => {
+function extractPropsFromReference(reference, stats, sourceFile) {
 	const props = [];
-	reference.getAttributes().forEach((attr) => {
+	const attributes = reference.getAttributes();
+	for (const attr of attributes) {
 		if (attr.asKind(SyntaxKind.JsxAttribute)) {
 			const jsxAttr = attr.asKindOrThrow(SyntaxKind.JsxAttribute);
 			const propName = jsxAttr.getNameNode().getText();
@@ -296,17 +194,16 @@ const extractPropsFromReference = (reference, stats, sourceFile) => {
 		}
 		if (attr.asKind(SyntaxKind.JsxSpreadAttribute)) {
 			stats.spreadsTraced++;
-			traceSpreadInFile(attr.asKindOrThrow(SyntaxKind.JsxSpreadAttribute).getExpression().getText(), sourceFile).forEach(([propName, values]) => {
-				if (values.length > 0) props.push([propName, values]);
-			});
+			const resolvedSpreadProps = traceSpreadInFile(attr.asKindOrThrow(SyntaxKind.JsxSpreadAttribute).getExpression().getText(), sourceFile);
+			for (const [propName, values] of resolvedSpreadProps) if (values.length > 0) props.push([propName, values]);
 		}
-	});
+	}
 	return props;
-};
+}
 /**
 * Trace spread props to their source within the same file.
 */
-const traceSpreadInFile = (spreadName, sourceFile) => {
+function traceSpreadInFile(spreadName, sourceFile) {
 	const props = [];
 	const varDecl = sourceFile.getVariableDeclaration(spreadName);
 	if (!varDecl) return props;
@@ -314,64 +211,62 @@ const traceSpreadInFile = (spreadName, sourceFile) => {
 	if (!initializer) return props;
 	const objLiteral = initializer.asKind(SyntaxKind.ObjectLiteralExpression) ? initializer.asKindOrThrow(SyntaxKind.ObjectLiteralExpression) : initializer.asKind(SyntaxKind.AsExpression) ? initializer.asKindOrThrow(SyntaxKind.AsExpression).getExpression().asKind(SyntaxKind.ObjectLiteralExpression) ? initializer.asKindOrThrow(SyntaxKind.AsExpression).getExpression().asKindOrThrow(SyntaxKind.ObjectLiteralExpression) : null : null;
 	if (!objLiteral) return props;
-	objLiteral.getProperties().forEach((prop) => {
-		if (prop.asKind(SyntaxKind.PropertyAssignment)) {
-			const propAssign = prop.asKindOrThrow(SyntaxKind.PropertyAssignment);
-			const propName = propAssign.getName();
-			const propInit = propAssign.getInitializer();
-			if (propInit) {
-				const values = extractStringLiterals(propInit);
-				if (values.length > 0) props.push([propName, values]);
-			}
+	for (const prop of objLiteral.getProperties()) if (prop.asKind(SyntaxKind.PropertyAssignment)) {
+		const propAssign = prop.asKindOrThrow(SyntaxKind.PropertyAssignment);
+		const propName = propAssign.getName();
+		const propInit = propAssign.getInitializer();
+		if (propInit) {
+			const values = extractStringLiterals(propInit);
+			if (values.length > 0) props.push([propName, values]);
 		}
-	});
+	}
 	return props;
-};
+}
 /**
 * Extract props from getStyles() calls in user code.
 * This handles cases like:
 *   const styles = getStyles({ backgroundColor: 'brand', spacing: '4' });
 */
-const extractGetStylesCalls = (sourceFile, stats, variants) => {
+function extractGetStylesCalls(sourceFile, stats, variants) {
 	const props = /* @__PURE__ */ new Map();
 	const VALID_VARIANTS = new Set(Object.keys(variants));
-	sourceFile.getDescendantsOfKind(SyntaxKind.CallExpression).forEach((call) => {
-		if (call.getExpression().getText() !== "getStyles") return;
+	const callExpressions = sourceFile.getDescendantsOfKind(SyntaxKind.CallExpression);
+	for (const call of callExpressions) {
+		if (call.getExpression().getText() !== "getStyles") continue;
 		const args = call.getArguments();
-		if (args.length === 0) return;
+		if (args.length === 0) continue;
 		const firstArg = args[0];
-		if (!firstArg.asKind(SyntaxKind.ObjectLiteralExpression)) return;
-		firstArg.asKindOrThrow(SyntaxKind.ObjectLiteralExpression).getProperties().forEach((prop) => {
-			if (prop.asKind(SyntaxKind.PropertyAssignment)) {
-				const propAssign = prop.asKindOrThrow(SyntaxKind.PropertyAssignment);
-				const propName = propAssign.getName();
-				const propInit = propAssign.getInitializer();
-				if (propInit && VALID_VARIANTS.has(propName)) {
-					const values = extractStringLiterals(propInit);
-					if (values.length > 0) {
-						const existing = props.get(propName) ?? /* @__PURE__ */ new Set();
-						values.forEach((v) => existing.add(v));
-						props.set(propName, existing);
-						stats.expressionsResolved++;
-					}
+		if (!firstArg.asKind(SyntaxKind.ObjectLiteralExpression)) continue;
+		const objLiteral = firstArg.asKindOrThrow(SyntaxKind.ObjectLiteralExpression);
+		for (const prop of objLiteral.getProperties()) if (prop.asKind(SyntaxKind.PropertyAssignment)) {
+			const propAssign = prop.asKindOrThrow(SyntaxKind.PropertyAssignment);
+			const propName = propAssign.getName();
+			const propInit = propAssign.getInitializer();
+			if (propInit && VALID_VARIANTS.has(propName)) {
+				const values = extractStringLiterals(propInit);
+				if (values.length > 0) {
+					const existing = props.get(propName) ?? /* @__PURE__ */ new Set();
+					for (const v of values) existing.add(v);
+					props.set(propName, existing);
+					stats.expressionsResolved++;
 				}
-			} else if (prop.asKind(SyntaxKind.ShorthandPropertyAssignment)) {
-				const shorthand = prop.asKindOrThrow(SyntaxKind.ShorthandPropertyAssignment);
-				const propName = shorthand.getName();
-				if (VALID_VARIANTS.has(propName)) {
-					const values = extractStringLiterals(shorthand.getNameNode());
-					if (values.length > 0) {
-						const existing = props.get(propName) ?? /* @__PURE__ */ new Set();
-						values.forEach((v) => existing.add(v));
-						props.set(propName, existing);
-						stats.expressionsResolved++;
-					}
+			}
+		} else if (prop.asKind(SyntaxKind.ShorthandPropertyAssignment)) {
+			const shorthand = prop.asKindOrThrow(SyntaxKind.ShorthandPropertyAssignment);
+			const propName = shorthand.getName();
+			if (VALID_VARIANTS.has(propName)) {
+				const values = extractStringLiterals(shorthand.getNameNode());
+				if (values.length > 0) {
+					const existing = props.get(propName) ?? /* @__PURE__ */ new Set();
+					for (const v of values) existing.add(v);
+					props.set(propName, existing);
+					stats.expressionsResolved++;
 				}
 			}
-		});
-	});
+		}
+	}
 	return props;
-};
+}
 
 //#endregion
 export { purgeFromCodeOptimized };