wesl 0.7.21 → 0.7.23

package/dist/index.js

@@ -334,8 +334,8 @@ function throwClickableError(params) {
 	}
 }
 /** Warn the user about an identifier and throw a clickable exception */
-function failIdent(ident$1, msg) {
-	const { refIdentElem, originalName } = ident$1;
+function failIdent(ident, msg) {
+	const { refIdentElem, originalName } = ident;
 	const baseMessage = msg ?? `'${originalName}'`;
 	if (refIdentElem) failIdentElem(refIdentElem, baseMessage);
 	else throw new Error(baseMessage);
@@ -504,11 +504,78 @@ function childIdent(child) {
 }
 
 //#endregion
-//#region src/LowerAndEmit.ts
-/** Valid WGSL standard attributes (from spec). Non-WGSL attributes are stripped.
-* See: https://www.w3.org/TR/WGSL/#attributes
-* Note: @builtin, @diagnostic, @interpolate are parsed as separate attribute types. */
-const wgslStandardAttributes$1 = new Set([
+//#region src/StandardTypes.ts
+const stdFns = `bitcast all any select arrayLength 
+  abs acos acosh asin asinh atan atanh atan2 ceil clamp cos cosh 
+  countLeadingZeros countOneBits countTrailingZeros cross 
+  degrees determinant distance dot dot4U8Packed dot4I8Packed 
+  exp exp2 extractBits faceForward firstLeadingBit firstTrailingBit 
+  floor fma fract frexp insertBits inverseSqrt ldexp length log log2
+  max min mix modf normalize pow quantizeToF16 radians reflect refract
+  reverseBits round saturate sign sin sinh smoothstep sqrt step tan tanh
+  transpose trunc
+  dpdx dpdxCoarse dpdxFine dpdy dpdyCoarse dpdyFine fwidth 
+  fwidthCoarse fwidthFine
+  textureDimensions textureGather textureGatherCompare textureLoad
+  textureNumLayers textureNumLevels textureNumSamples
+  textureSample textureSampleBias textureSampleCompare textureSampleCompareLevel
+  textureSampleGrad textureSampleLevel textureSampleBaseClampToEdge
+  textureStore
+  atomicLoad atomicStore atomicAdd atomicSub atomicMax atomicMin
+  atomicAnd atomicOr atomicXor atomicExchange atomicCompareExchangeWeak
+  pack4x8snorm pack4x8unorm pack4xI8 pack4xU8 pack4xI8Clamp pack4xU8Clamp
+  pack2x16snorm pack2x16unorm pack2x16float
+  unpack4x8snorm unpack4x8unorm unpack4xI8 unpack4xU8 
+  unpack2x16snorm unpack2x16unorm unpack2x16float
+  storageBarrier textureBarrier workgroupBarrier workgroupUniformLoad
+  subgroupAdd subgroupAll subgroupAnd subgroupAny subgroupBallot 
+  subgroupBroadcast subgroupBroadcastFirst subgroupElect 
+  subgroupExclusiveAdd subgroupExclusiveMul subgroupInclusiveAdd 
+  subgroupInclusiveMul subgroupMax subgroupMin subgroupMul subgroupOr 
+  subgroupShuffle subgroupShuffleUp subgroupShuffleXor subgroupXor
+  quadBroadcast quadSwapDiagonal quadSwapX quadSwapY`.split(/\s+/);
+const sampledTextureTypes = `
+  texture_1d texture_2d texture_2d_array texture_3d 
+  texture_cube texture_cube_array
+`;
+const multisampledTextureTypes = `
+  texture_multisampled_2d texture_depth_multisampled_2d
+`;
+const textureStorageTypes = `
+  texture_storage_1d texture_storage_2d texture_storage_2d_array 
+  texture_storage_3d
+`;
+const stdTypes = `array atomic bool f16 f32 i32 
+  mat2x2 mat2x3 mat2x4 mat3x2 mat3x3 mat3x4 mat4x2 mat4x3 mat4x4
+  mat2x2f mat2x3f mat2x4f mat3x2f mat3x3f mat3x4f
+  mat4x2f mat4x3f mat4x4f
+  mat2x2h mat2x3h mat2x4h mat3x2h mat3x3h mat3x4h
+  mat4x2h mat4x3h mat4x4h
+  u32 vec2 vec3 vec4 ptr
+  vec2i vec3i vec4i vec2u vec3u vec4u
+  vec2f vec3f vec4f vec2h vec3h vec4h
+  ${sampledTextureTypes}
+  ${multisampledTextureTypes}
+  texture_external
+  ${textureStorageTypes}
+  texture_depth_2d texture_depth_2d_array texture_depth_cube
+  texture_depth_cube_array
+  sampler sampler_comparison
+  rgba8unorm rgba8snorm rgba8uint rgba8sint
+  rgba16uint rgba16sint rgba16float 
+  r32uint r32sint r32float rg32uint rg32sint rg32float
+  rgba32uint rgba32sint rgba32float
+  bgra8unorm`.split(/\s+/);
+/** https://www.w3.org/TR/WGSL/#predeclared-enumerants  */
+const stdEnumerants = `read write read_write 
+  function private workgroup uniform storage
+  rgba8unorm rgba8snorm rgba8uint rgba8sint 
+  rgba16uint rgba16sint rgba16float 
+  r32uint r32sint r32float rg32uint rg32sint rg32float
+  rgba32uint rgba32sint rgba32float bgra8unorm`.split(/\s+/);
+/** WGSL standard attributes whose params need binding (e.g., @workgroup_size).
+* See: https://www.w3.org/TR/WGSL/#attributes */
+const wgslStandardAttributes = new Set([
 	"align",
 	"binding",
 	"blend_src",
@@ -524,6 +591,21 @@ const wgslStandardAttributes$1 = new Set([
 	"vertex",
 	"workgroup_size"
 ]);
+/** return true if the name is for a built in type (not a user struct) */
+function stdType(name) {
+	return stdTypes.includes(name);
+}
+/** return true if the name is for a built in fn (not a user function) */
+function stdFn(name) {
+	return stdFns.includes(name) || stdType(name);
+}
+/** return true if the name is for a built in enumerant */
+function stdEnumerant(name) {
+	return stdEnumerants.includes(name);
+}
+
+//#endregion
+//#region src/LowerAndEmit.ts
 /** Traverse the AST, starting from root elements, emitting WGSL for each. */
 function lowerAndEmit(params) {
 	const { srcBuilder, rootElems, conditions } = params;
@@ -717,13 +799,13 @@ function emitContentsWithTrimming(elem, ctx) {
 	const validElements = filterValidElements(elem.contents, ctx.conditions);
 	const firstEmit = validElements.findIndex((e) => !isConditionalAttr(e));
 	const lastEmit = validElements.findLastIndex((e) => !isConditionalAttr(e));
-	validElements.forEach((elem$1, i) => {
-		if (elem$1.kind === "text") {
-			let text = elem$1.srcModule.src.slice(elem$1.start, elem$1.end);
+	validElements.forEach((elem, i) => {
+		if (elem.kind === "text") {
+			let text = elem.srcModule.src.slice(elem.start, elem.end);
 			if (i === firstEmit) text = text.trimStart();
 			if (i === lastEmit) text = text.trimEnd();
-			if (text) ctx.srcBuilder.add(text, elem$1.start, elem$1.end);
-		} else lowerAndEmitElem(elem$1, ctx);
+			if (text) ctx.srcBuilder.add(text, elem.start, elem.end);
+		} else lowerAndEmitElem(elem, ctx);
 	});
 }
 function isConditionalAttr(e) {
@@ -803,7 +885,7 @@ function emitAttribute(e, ctx) {
 	const { kind } = e.attribute;
 	if (kind === "@if" || kind === "@elif" || kind === "@else") return false;
 	if (kind === "@attribute") {
-		if (!wgslStandardAttributes$1.has(e.attribute.name)) return false;
+		if (!wgslStandardAttributes.has(e.attribute.name)) return false;
 		emitStandardAttribute(e, ctx);
 		return true;
 	}
@@ -885,13 +967,13 @@ function displayName(declIdent) {
 	return declIdent.mangledName || declIdent.originalName;
 }
 /** Trace through refersTo links until we find the declaration. */
-function findDecl(ident$1) {
-	let i = ident$1;
+function findDecl(ident) {
+	let i = ident;
 	do {
 		if (i.kind === "decl") return i;
 		i = i.refersTo;
 	} while (i);
-	throw new Error(`unresolved identifer: ${ident$1.originalName}`);
+	throw new Error(`unresolved identifer: ${ident.originalName}`);
 }
 
 //#endregion
@@ -1122,7 +1204,7 @@ function scopeToString(scope, indent = 0, shortIdents = true) {
 	if (attrStrings) str.add(attrStrings + " ");
 	if (kind === "partial") str.add("-");
 	str.add("{ ");
-	const last$1 = contents.length - 1;
+	const last = contents.length - 1;
 	let lastWasScope = false;
 	let hasBlock = false;
 	contents.forEach((elem, i) => {
@@ -1135,10 +1217,10 @@ function scopeToString(scope, indent = 0, shortIdents = true) {
 		} else {
 			if (lastWasScope) str.add("  ");
 			lastWasScope = false;
-			const ident$1 = elem;
-			if (shortIdents) str.add(identShortString(ident$1));
-			else str.add(identToString(ident$1));
-			if (i < last$1) str.add(" ");
+			const ident = elem;
+			if (shortIdents) str.add(identShortString(ident));
+			else str.add(identToString(ident));
+			if (i < last) str.add(" ");
 		}
 	});
 	if (!hasBlock && str.oneLine) str.add(" }");
@@ -1154,17 +1236,17 @@ function scopeToStringLong(scope) {
 	return scopeToString(scope, 0, false);
 }
 /** name of an identifier, with decls prefixed with '%' */
-function identShortString(ident$1) {
-	const { kind, originalName } = ident$1;
+function identShortString(ident) {
+	const { kind, originalName } = ident;
 	return `${kind === "decl" ? "%" : ""}${originalName}`;
 }
-function identToString(ident$1) {
-	if (!ident$1) return JSON.stringify(ident$1);
-	const { kind, originalName } = ident$1;
-	const idStr = ident$1.id ? `#${ident$1.id}` : "";
-	if (kind === "ref") return `${originalName} ${idStr} -> ${identToString(ident$1.refersTo)}`;
+function identToString(ident) {
+	if (!ident) return JSON.stringify(ident);
+	const { kind, originalName } = ident;
+	const idStr = ident.id ? `#${ident.id}` : "";
+	if (kind === "ref") return `${originalName} ${idStr} -> ${identToString(ident.refersTo)}`;
 	else {
-		const { mangledName } = ident$1;
+		const { mangledName } = ident;
 		return `%${originalName}${mangledName ? `(${mangledName})` : ""} ${idStr} `;
 	}
 }
@@ -1655,11 +1737,11 @@ function parseIdent(ctx, conditionRef) {
 	const path = parseModulePath(ctx.stream);
 	if (!path) return null;
 	const { parts, start, end } = path;
-	const ident$1 = ctx.createRefIdent(parts.join("::"));
-	if (conditionRef) ident$1.conditionRef = true;
-	const refIdentElem = makeRefIdentElem(ctx, ident$1, start, end);
+	const ident = ctx.createRefIdent(parts.join("::"));
+	if (conditionRef) ident.conditionRef = true;
+	const refIdentElem = makeRefIdentElem(ctx, ident, start, end);
 	if (!conditionRef) {
-		ctx.saveIdent(ident$1);
+		ctx.saveIdent(ident);
 		ctx.addElem(refIdentElem);
 	}
 	return refIdentElem;
@@ -2202,8 +2284,8 @@ function parseElseChain(ctx) {
 	while (stream.matchText("else")) {
 		if (stream.matchText("if")) {
 			expectExpression(ctx, "Expected expression after 'else if'");
-			const body$1 = expectCompound(ctx, "Expected '{' after else if");
-			ctx.addElem(body$1);
+			const body = expectCompound(ctx, "Expected '{' after else if");
+			ctx.addElem(body);
 			continue;
 		}
 		const body = expectCompound(ctx, "Expected '{' after else");
@@ -3162,8 +3244,8 @@ var ParsingContext = class {
 			declElem: null
 		};
 	}
-	saveIdent(ident$1) {
-		this.state.context.scope.contents.push(ident$1);
+	saveIdent(ident) {
+		this.state.context.scope.contents.push(ident);
 	}
 };
 
@@ -3252,7 +3334,7 @@ function toRegexSource(nameExp) {
 	}
 }
 function verifyNonCapturing(name, exp) {
-	if ((/* @__PURE__ */ new RegExp("|" + exp.source)).exec("").length > 1) throw new Error(`match expression groups must be non-capturing: ${name}: /${exp.source}/. Use (?:...) instead.`);
+	if (new RegExp("|" + exp.source).exec("").length > 1) throw new Error(`match expression groups must be non-capturing: ${name}: /${exp.source}/. Use (?:...) instead.`);
 }
 const regexSpecials = /[$+*.?|(){}[\]\\/^]/g;
 function escapeRegex(s) {
@@ -3595,115 +3677,16 @@ function flatImports(ast, conditions) {
 	return flat;
 }
 
-//#endregion
-//#region src/StandardTypes.ts
-const stdFns = `bitcast all any select arrayLength 
-  abs acos acosh asin asinh atan atanh atan2 ceil clamp cos cosh 
-  countLeadingZeros countOneBits countTrailingZeros cross 
-  degrees determinant distance dot dot4U8Packed dot4I8Packed 
-  exp exp2 extractBits faceForward firstLeadingBit firstTrailingBit 
-  floor fma fract frexp insertBits inverseSqrt ldexp length log log2
-  max min mix modf normalize pow quantizeToF16 radians reflect refract
-  reverseBits round saturate sign sin sinh smoothstep sqrt step tan tanh
-  transpose trunc
-  dpdx dpdxCoarse dpdxFine dpdy dpdyCoarse dpdyFine fwidth 
-  fwidthCoarse fwidthFine
-  textureDimensions textureGather textureGatherCompare textureLoad
-  textureNumLayers textureNumLevels textureNumSamples
-  textureSample textureSampleBias textureSampleCompare textureSampleCompareLevel
-  textureSampleGrad textureSampleLevel textureSampleBaseClampToEdge
-  textureStore
-  atomicLoad atomicStore atomicAdd atomicSub atomicMax atomicMin
-  atomicAnd atomicOr atomicXor atomicExchange atomicCompareExchangeWeak
-  pack4x8snorm pack4x8unorm pack4xI8 pack4xU8 pack4xI8Clamp pack4xU8Clamp
-  pack2x16snorm pack2x16unorm pack2x16float
-  unpack4x8snorm unpack4x8unorm unpack4xI8 unpack4xU8 
-  unpack2x16snorm unpack2x16unorm unpack2x16float
-  storageBarrier textureBarrier workgroupBarrier workgroupUniformLoad
-  subgroupAdd subgroupAll subgroupAnd subgroupAny subgroupBallot 
-  subgroupBroadcast subgroupBroadcastFirst subgroupElect 
-  subgroupExclusiveAdd subgroupExclusiveMul subgroupInclusiveAdd 
-  subgroupInclusiveMul subgroupMax subgroupMin subgroupMul subgroupOr 
-  subgroupShuffle subgroupShuffleUp subgroupShuffleXor subgroupXor
-  quadBroadcast quadSwapDiagonal quadSwapX quadSwapY`.split(/\s+/);
-const sampledTextureTypes = `
-  texture_1d texture_2d texture_2d_array texture_3d 
-  texture_cube texture_cube_array
-`;
-const multisampledTextureTypes = `
-  texture_multisampled_2d texture_depth_multisampled_2d
-`;
-const textureStorageTypes = `
-  texture_storage_1d texture_storage_2d texture_storage_2d_array 
-  texture_storage_3d
-`;
-const stdTypes = `array atomic bool f16 f32 i32 
-  mat2x2 mat2x3 mat2x4 mat3x2 mat3x3 mat3x4 mat4x2 mat4x3 mat4x4
-  mat2x2f mat2x3f mat2x4f mat3x2f mat3x3f mat3x4f
-  mat4x2f mat4x3f mat4x4f
-  mat2x2h mat2x3h mat2x4h mat3x2h mat3x3h mat3x4h
-  mat4x2h mat4x3h mat4x4h
-  u32 vec2 vec3 vec4 ptr
-  vec2i vec3i vec4i vec2u vec3u vec4u
-  vec2f vec3f vec4f vec2h vec3h vec4h
-  ${sampledTextureTypes}
-  ${multisampledTextureTypes}
-  texture_external
-  ${textureStorageTypes}
-  texture_depth_2d texture_depth_2d_array texture_depth_cube
-  texture_depth_cube_array
-  sampler sampler_comparison
-  rgba8unorm rgba8snorm rgba8uint rgba8sint
-  rgba16uint rgba16sint rgba16float 
-  r32uint r32sint r32float rg32uint rg32sint rg32float
-  rgba32uint rgba32sint rgba32float
-  bgra8unorm`.split(/\s+/);
-/** https://www.w3.org/TR/WGSL/#predeclared-enumerants  */
-const stdEnumerants = `read write read_write 
-  function private workgroup uniform storage
-  rgba8unorm rgba8snorm rgba8uint rgba8sint 
-  rgba16uint rgba16sint rgba16float 
-  r32uint r32sint r32float rg32uint rg32sint rg32float
-  rgba32uint rgba32sint rgba32float bgra8unorm`.split(/\s+/);
-/** return true if the name is for a built in type (not a user struct) */
-function stdType(name) {
-	return stdTypes.includes(name);
-}
-/** return true if the name is for a built in fn (not a user function) */
-function stdFn(name) {
-	return stdFns.includes(name) || stdType(name);
-}
-/** return true if the name is for a built in enumerant */
-function stdEnumerant(name) {
-	return stdEnumerants.includes(name);
-}
-
 //#endregion
 //#region src/BindIdents.ts
-/** WGSL standard attributes whose params need binding (e.g., @workgroup_size). */
-const wgslStandardAttributes = new Set([
-	"align",
-	"binding",
-	"blend_src",
-	"compute",
-	"const",
-	"fragment",
-	"group",
-	"id",
-	"invariant",
-	"location",
-	"must_use",
-	"size",
-	"vertex",
-	"workgroup_size"
-]);
 /** Bind ref idents to declarations and mangle global declaration names. */
 function bindIdents(params) {
 	const { rootAst, resolver, virtuals, accumulateUnbound } = params;
 	const { conditions = {}, mangler = minimalMangle } = params;
+	const { discoveryMode } = params;
 	const packageName = rootAst.srcModule.modulePath.split("::")[0];
-	const validRootDecls = findValidRootDecls(rootAst.rootScope, conditions);
-	const { globalNames, knownDecls } = initRootDecls(validRootDecls);
+	const rootDecls = discoveryMode ? findAllRootDecls(rootAst.rootScope) : findValidRootDecls(rootAst.rootScope, conditions);
+	const { globalNames, knownDecls } = initRootDecls(rootDecls);
 	const bindContext = {
 		resolver,
 		conditions,
@@ -3714,13 +3697,14 @@ function bindIdents(params) {
 		foundScopes: /* @__PURE__ */ new Set(),
 		globalNames,
 		globalStatements: /* @__PURE__ */ new Map(),
-		unbound: accumulateUnbound ? [] : void 0
+		unbound: accumulateUnbound ? [] : void 0,
+		discoveryMode
 	};
 	const liveDecls = {
-		decls: new Map(validRootDecls.map((d) => [d.originalName, d])),
+		decls: new Map(rootDecls.map((d) => [d.originalName, d])),
 		parent: null
 	};
-	const fromRootDecls = validRootDecls.flatMap((decl) => processDependentScope(decl, bindContext));
+	const fromRootDecls = rootDecls.flatMap((decl) => processDependentScope(decl, bindContext));
 	const fromRefs = bindIdentsRecursive(rootAst.rootScope, bindContext, liveDecls);
 	const newStatements = [...bindContext.globalStatements.values()];
 	return {
@@ -3756,10 +3740,11 @@ function* validItems(scope, conditions) {
 }
 /** Find all conditionally valid declarations at the root level. */
 function findValidRootDecls(rootScope, conditions) {
-	const found = [];
-	for (const item of validItems(rootScope, conditions)) if (item.kind === "decl") found.push(item);
-	else if (item.kind === "partial") collectDecls(item, found);
-	return found;
+	return collectDecls(validItems(rootScope, conditions));
+}
+/** Find all declarations at the root level, ignoring conditions. */
+function findAllRootDecls(rootScope) {
+	return collectDecls(rootScope.contents);
 }
 /** Find a public declaration with the given original name. */
 function publicDecl(scope, name, conditions) {
@@ -3785,7 +3770,8 @@ function bindIdentsRecursive(scope, bindContext, liveDecls) {
 function processScope(scope, bindContext, liveDecls) {
 	const newGlobals = [];
 	const newFromChildren = [];
-	for (const child of validItems(scope, bindContext.conditions)) if (child.kind === "decl") liveDecls.decls.set(child.originalName, child);
+	const items = bindContext.discoveryMode ? scope.contents : validItems(scope, bindContext.conditions);
+	for (const child of items) if (child.kind === "decl") liveDecls.decls.set(child.originalName, child);
 	else if (child.kind === "ref") {
 		const newDecl = handleRef(child, liveDecls, bindContext);
 		if (newDecl) newGlobals.push(newDecl);
@@ -3811,25 +3797,26 @@ function processDependentScope(decl, ctx) {
 * Mutates to mangle declarations and mark std references.
 * @return found declaration, or undefined if already processed
 */
-function handleRef(ident$1, liveDecls, bindContext) {
-	if (ident$1.refersTo || ident$1.std) return;
-	if (ident$1.conditionRef) return;
-	if (ident$1.attrParam && !wgslStandardAttributes.has(ident$1.attrParam)) return;
-	const foundDecl = findDeclInModule(ident$1, liveDecls) ?? findQualifiedImport(ident$1, bindContext);
+function handleRef(ident, liveDecls, bindContext) {
+	if (ident.refersTo || ident.std) return;
+	if (ident.conditionRef) return;
+	if (ident.attrParam && !wgslStandardAttributes.has(ident.attrParam)) return;
+	const foundDecl = findDeclInModule(ident, liveDecls) ?? findQualifiedImport(ident, bindContext);
 	if (foundDecl) {
-		ident$1.refersTo = foundDecl.decl;
-		return handleNewDecl(ident$1, foundDecl, bindContext);
+		ident.refersTo = foundDecl.decl;
+		return handleNewDecl(ident, foundDecl, bindContext);
 	}
-	if (stdWgsl(ident$1.originalName)) {
-		ident$1.std = true;
+	if (stdWgsl(ident.originalName)) {
+		ident.std = true;
 		return;
 	}
-	if (!bindContext.unbound) failIdent(ident$1, `unresolved identifier '${ident$1.originalName}'`);
+	if (!bindContext.unbound) failIdent(ident, `unresolved identifier '${ident.originalName}'`);
 }
+/** Follow new global declarations into their dependent scopes. */
 function handleDecls(newGlobals, bindContext) {
 	return newGlobals.flatMap((decl) => processDependentScope(decl, bindContext));
 }
-/** If found declaration is new, mangle its name. Return if it's a global declaration. */
+/** If found declaration is new, mangle its name. @return the decl if it's global. */
 function handleNewDecl(refIdent, foundDecl, ctx) {
 	const { decl, moduleAst } = foundDecl;
 	const { knownDecls, globalNames, mangler, globalStatements } = ctx;
@@ -3845,18 +3832,19 @@ function handleNewDecl(refIdent, foundDecl, ctx) {
 	return decl;
 }
 /** Search current scope and parent scopes for a matching declaration. */
-function findDeclInModule(ident$1, liveDecls) {
-	const found = liveDecls.decls.get(ident$1.originalName);
+function findDeclInModule(ident, liveDecls) {
+	const found = liveDecls.decls.get(ident.originalName);
 	if (found) return {
 		decl: found,
-		moduleAst: ident$1.ast
+		moduleAst: ident.ast
 	};
-	if (liveDecls.parent) return findDeclInModule(ident$1, liveDecls.parent);
+	if (liveDecls.parent) return findDeclInModule(ident, liveDecls.parent);
 }
 /** Match a ref ident to a declaration in another module via import or qualified ident. */
 function findQualifiedImport(refIdent, ctx) {
-	const { conditions, unbound } = ctx;
-	const flatImps = flatImports(refIdent.ast, conditions);
+	const { conditions, unbound, discoveryMode } = ctx;
+	const conds = discoveryMode ? void 0 : conditions;
+	const flatImps = flatImports(refIdent.ast, conds);
 	const identParts = refIdent.originalName.split("::");
 	const pathParts = matchingImport(identParts, flatImps) ?? qualifiedIdent(identParts);
 	if (!pathParts) {
@@ -3910,7 +3898,7 @@ function virtualModule(moduleName, ctx) {
 /** Get cached valid root declarations, computing on first access. */
 function getValidRootDecls(rootScope, conditions) {
 	const lexScope = rootScope;
-	if (!lexScope._validRootDecls) lexScope._validRootDecls = findValidRootDecls(rootScope, conditions);
+	lexScope._validRootDecls ??= findValidRootDecls(rootScope, conditions);
 	return lexScope._validRootDecls;
 }
 /** Given a global declIdent, return the liveDecls for its root scope. */
@@ -3939,13 +3927,167 @@ function setMangledName(proposedName, decl, globalNames, srcModule, mangler) {
 function stdWgsl(name) {
 	return stdType(name) || stdFn(name) || stdEnumerant(name);
 }
+/** @return identParts if it's a qualified path (has ::). */
 function qualifiedIdent(identParts) {
 	if (identParts.length > 1) return identParts;
 }
-/** Collect all declarations in a scope (used when scope is already validated). */
-function collectDecls(scope, found) {
-	for (const item of scope.contents) if (item.kind === "decl") found.push(item);
-	else if (item.kind === "partial") collectDecls(item, found);
+/** Collect all declarations from scope items, recursing into partial scopes. */
+function collectDecls(items) {
+	return [...items].flatMap((item) => {
+		const { kind } = item;
+		if (kind === "decl") return [item];
+		if (kind === "partial") return collectDecls(item.contents);
+		return [];
+	});
+}
+
+//#endregion
+//#region src/PathUtil.ts
+/** simplistic path manipulation utilities */
+/** return path with ./ and foo/.. elements removed */
+function normalize(path) {
+	const noDots = path.split("/").filter((s) => s !== ".");
+	const noDbl = [];
+	noDots.forEach((s) => {
+		if (s !== "") if (s === ".." && noDbl.length && noDbl[noDbl.length - 1] !== "..") noDbl.pop();
+		else noDbl.push(s);
+	});
+	return noDbl.join("/");
+}
+/** return path w/o a suffix.
+* e.g. /foo/bar.wgsl => /foo/bar */
+function noSuffix(path) {
+	const lastSlash = path.lastIndexOf("/");
+	const lastStart = lastSlash === -1 ? 0 : lastSlash + 1;
+	const suffix = path.indexOf(".", lastStart);
+	const suffixStart = suffix === -1 ? path.length : suffix;
+	return path.slice(0, suffixStart);
+}
+
+//#endregion
+//#region src/ModuleResolver.ts
+const libRegex = /^lib\.w[eg]sl$/i;
+/** Module resolver for in-memory source records. Lazy by default. */
+var RecordResolver = class {
+	astCache = /* @__PURE__ */ new Map();
+	sources;
+	packageName;
+	debugWeslRoot;
+	constructor(sources, options = {}) {
+		const { packageName = "package", debugWeslRoot } = options;
+		this.sources = sources;
+		this.packageName = packageName;
+		this.debugWeslRoot = normalizeDebugRoot(debugWeslRoot);
+	}
+	resolveModule(modulePath) {
+		const cached = this.astCache.get(modulePath);
+		if (cached) return cached;
+		const source = this.findSource(modulePath);
+		if (source === void 0) return void 0;
+		const ast = parseSrcModule({
+			modulePath,
+			debugFilePath: this.modulePathToDebugPath(modulePath),
+			src: source
+		});
+		this.astCache.set(modulePath, ast);
+		return ast;
+	}
+	findSource(modulePath) {
+		if (this.sources[modulePath] !== void 0) return this.sources[modulePath];
+		const filePath = this.moduleToFilePath(modulePath);
+		if (filePath === void 0) return void 0;
+		return findInVariants(this.sources, filePath);
+	}
+	/** Convert module path to file path, or undefined if not local. */
+	moduleToFilePath(modulePath) {
+		return moduleToRelativePath(modulePath, this.packageName);
+	}
+	modulePathToDebugPath(modulePath) {
+		const filePath = this.moduleToFilePath(modulePath) ?? modulePath;
+		return this.debugWeslRoot + filePath + ".wesl";
+	}
+	/** Parse all modules and return entries. */
+	allModules() {
+		for (const filePath of Object.keys(this.sources)) {
+			const treatLibAsRoot = this.packageName !== "package";
+			const modulePath = fileToModulePath(filePath, this.packageName, treatLibAsRoot);
+			this.resolveModule(modulePath);
+		}
+		return this.astCache.entries();
+	}
+};
+/** Composite resolver that tries each resolver in order until one succeeds. */
+var CompositeResolver = class {
+	resolvers;
+	constructor(resolvers) {
+		this.resolvers = resolvers;
+	}
+	resolveModule(modulePath) {
+		for (const resolver of this.resolvers) {
+			const ast = resolver.resolveModule(modulePath);
+			if (ast) return ast;
+		}
+	}
+};
+/** Lazy resolver for WeslBundle library modules. */
+var BundleResolver = class {
+	astCache = /* @__PURE__ */ new Map();
+	sources;
+	packageName;
+	debugWeslRoot;
+	constructor(bundle, debugWeslRoot) {
+		this.sources = bundle.modules;
+		this.packageName = bundle.name;
+		this.debugWeslRoot = normalizeDebugRoot(debugWeslRoot);
+	}
+	resolveModule(modulePath) {
+		const pkgPrefix = this.packageName + "::";
+		if (modulePath !== this.packageName && !modulePath.startsWith(pkgPrefix)) return;
+		const cached = this.astCache.get(modulePath);
+		if (cached) return cached;
+		const source = this.findSource(modulePath);
+		if (!source) return void 0;
+		const ast = parseSrcModule({
+			modulePath,
+			debugFilePath: this.modulePathToDebugPath(modulePath),
+			src: source
+		});
+		this.astCache.set(modulePath, ast);
+		return ast;
+	}
+	findSource(modulePath) {
+		const filePath = this.moduleToFilePath(modulePath);
+		if (modulePath === this.packageName) {
+			const libSrc = findInVariants(this.sources, "lib", ["wesl", "wgsl"]);
+			if (libSrc) return libSrc;
+		}
+		return findInVariants(this.sources, filePath);
+	}
+	moduleToFilePath(modulePath) {
+		return moduleToRelativePath(modulePath, this.packageName) ?? modulePath;
+	}
+	modulePathToDebugPath(modulePath) {
+		const filePath = this.moduleToFilePath(modulePath);
+		return this.debugWeslRoot + this.packageName + "/" + filePath + ".wesl";
+	}
+};
+/** Convert file path to module path (e.g., "foo/bar.wesl" to "package::foo::bar"). */
+function fileToModulePath(filePath, packageName, treatLibAsRoot) {
+	if (filePath.includes("::")) return filePath;
+	if (treatLibAsRoot && libRegex.test(filePath)) return packageName;
+	const moduleSuffix = noSuffix(normalize(filePath)).replaceAll("/", "::");
+	return packageName + "::" + moduleSuffix;
+}
+/** Try path variants with/without ./ prefix and extension suffixes. */
+function findInVariants(sources, basePath, extensions = ["wesl", "wgsl"]) {
+	for (const prefix of ["", "./"]) {
+		const path = prefix + basePath;
+		if (sources[path] !== void 0) return sources[path];
+		for (const ext of extensions) {
+			const withExt = `${path}.${ext}`;
+			if (sources[withExt] !== void 0) return sources[withExt];
+		}
+	}
 }
 
 //#endregion
@@ -3975,21 +4117,57 @@ function findUnboundRefs(resolver) {
 		mangler: minimalMangle,
 		packageName: "package",
 		unbound: [],
-		dontFollowDecls: true
+		dontFollowDecls: true,
+		discoveryMode: true
 	};
 	for (const [, ast] of resolver.allModules()) {
-		const rootDecls = findValidRootDecls(ast.rootScope, {});
+		const rootDecls = findAllRootDecls(ast.rootScope);
 		const liveDecls = {
 			decls: new Map(rootDecls.map((d) => [d.originalName, d])),
 			parent: null
 		};
-		filterMap(rootDecls, (decl) => decl.dependentScope).forEach((s) => {
-			bindIdentsRecursive(s, bindContext, makeLiveDecls(liveDecls));
-		});
+		for (const s of filterMap(rootDecls, (decl) => decl.dependentScope)) bindIdentsRecursive(s, bindContext, makeLiveDecls(liveDecls));
 		bindIdentsRecursive(ast.rootScope, bindContext, liveDecls);
 	}
 	return bindContext.unbound;
 }
+/** Thin decorator that records which modules were resolved. */
+var TrackingResolver = class {
+	visited = /* @__PURE__ */ new Set();
+	#inner;
+	constructor(inner) {
+		this.#inner = inner;
+	}
+	resolveModule(modulePath) {
+		const ast = this.#inner.resolveModule(modulePath);
+		if (ast) this.visited.add(modulePath);
+		return ast;
+	}
+};
+/**
+* Discover reachable modules and unbound external refs from a single root.
+*
+* Traces the import graph from `rootModuleName`, returning only the reachable
+* local modules in `weslSrc` and unresolved external references in `unbound`.
+*/
+function discoverModules(weslSrc, resolver, rootModuleName, packageName = "package") {
+	const tracking = new TrackingResolver(resolver);
+	const rootAst = tracking.resolveModule(rootModuleName);
+	if (!rootAst) throw new Error(`root module not found: '${rootModuleName}'`);
+	const result = bindIdents({
+		rootAst,
+		resolver: tracking,
+		accumulateUnbound: true,
+		discoveryMode: true
+	});
+	const moduleToKey = new Map(Object.keys(weslSrc).map((key) => [fileToModulePath(key, packageName, false), key]));
+	const reachable = [...tracking.visited].map((m) => moduleToKey.get(m)).filter((key) => key !== void 0).map((key) => [key, weslSrc[key]]);
+	const unbound = (result.unbound ?? []).map((ref) => ref.path);
+	return {
+		weslSrc: Object.fromEntries(reachable),
+		unbound
+	};
+}
 
 //#endregion
 //#region src/discovery/PackageNameUtils.ts
@@ -4160,155 +4338,6 @@ function compilationInfoToErrorMessage(compilationInfo, shaderModule) {
 	return result;
 }
 
-//#endregion
-//#region src/PathUtil.ts
-/** simplistic path manipulation utilities */
-/** return path with ./ and foo/.. elements removed */
-function normalize(path) {
-	const noDots = path.split("/").filter((s) => s !== ".");
-	const noDbl = [];
-	noDots.forEach((s) => {
-		if (s !== "") if (s === ".." && noDbl.length && noDbl[noDbl.length - 1] !== "..") noDbl.pop();
-		else noDbl.push(s);
-	});
-	return noDbl.join("/");
-}
-/** return path w/o a suffix.
-* e.g. /foo/bar.wgsl => /foo/bar */
-function noSuffix(path) {
-	const lastSlash = path.lastIndexOf("/");
-	const lastStart = lastSlash === -1 ? 0 : lastSlash + 1;
-	const suffix = path.indexOf(".", lastStart);
-	const suffixStart = suffix === -1 ? path.length : suffix;
-	return path.slice(0, suffixStart);
-}
-
-//#endregion
-//#region src/ModuleResolver.ts
-const libRegex = /^lib\.w[eg]sl$/i;
-/** Module resolver for in-memory source records. Lazy by default. */
-var RecordResolver = class {
-	astCache = /* @__PURE__ */ new Map();
-	sources;
-	packageName;
-	debugWeslRoot;
-	constructor(sources, options = {}) {
-		const { packageName = "package", debugWeslRoot } = options;
-		this.sources = sources;
-		this.packageName = packageName;
-		this.debugWeslRoot = normalizeDebugRoot(debugWeslRoot);
-	}
-	resolveModule(modulePath) {
-		const cached = this.astCache.get(modulePath);
-		if (cached) return cached;
-		const source = this.findSource(modulePath);
-		if (source === void 0) return void 0;
-		const ast = parseSrcModule({
-			modulePath,
-			debugFilePath: this.modulePathToDebugPath(modulePath),
-			src: source
-		});
-		this.astCache.set(modulePath, ast);
-		return ast;
-	}
-	findSource(modulePath) {
-		if (this.sources[modulePath] !== void 0) return this.sources[modulePath];
-		const filePath = this.moduleToFilePath(modulePath);
-		if (filePath === void 0) return void 0;
-		return findInVariants(this.sources, filePath);
-	}
-	/** Convert module path to file path, or undefined if not local. */
-	moduleToFilePath(modulePath) {
-		return moduleToRelativePath(modulePath, this.packageName);
-	}
-	modulePathToDebugPath(modulePath) {
-		const filePath = this.moduleToFilePath(modulePath) ?? modulePath;
-		return this.debugWeslRoot + filePath + ".wesl";
-	}
-	/** Parse all modules and return entries. */
-	allModules() {
-		for (const filePath of Object.keys(this.sources)) {
-			const treatLibAsRoot = this.packageName !== "package";
-			const modulePath = fileToModulePath(filePath, this.packageName, treatLibAsRoot);
-			this.resolveModule(modulePath);
-		}
-		return this.astCache.entries();
-	}
-};
-/** Composite resolver that tries each resolver in order until one succeeds. */
-var CompositeResolver = class {
-	resolvers;
-	constructor(resolvers) {
-		this.resolvers = resolvers;
-	}
-	resolveModule(modulePath) {
-		for (const resolver of this.resolvers) {
-			const ast = resolver.resolveModule(modulePath);
-			if (ast) return ast;
-		}
-	}
-};
-/** Lazy resolver for WeslBundle library modules. */
-var BundleResolver = class {
-	astCache = /* @__PURE__ */ new Map();
-	sources;
-	packageName;
-	debugWeslRoot;
-	constructor(bundle, debugWeslRoot) {
-		this.sources = bundle.modules;
-		this.packageName = bundle.name;
-		this.debugWeslRoot = normalizeDebugRoot(debugWeslRoot);
-	}
-	resolveModule(modulePath) {
-		const pkgPrefix = this.packageName + "::";
-		if (modulePath !== this.packageName && !modulePath.startsWith(pkgPrefix)) return;
-		const cached = this.astCache.get(modulePath);
-		if (cached) return cached;
-		const source = this.findSource(modulePath);
-		if (!source) return void 0;
-		const ast = parseSrcModule({
-			modulePath,
-			debugFilePath: this.modulePathToDebugPath(modulePath),
-			src: source
-		});
-		this.astCache.set(modulePath, ast);
-		return ast;
-	}
-	findSource(modulePath) {
-		const filePath = this.moduleToFilePath(modulePath);
-		if (modulePath === this.packageName) {
-			const libSrc = findInVariants(this.sources, "lib", ["wesl", "wgsl"]);
-			if (libSrc) return libSrc;
-		}
-		return findInVariants(this.sources, filePath);
-	}
-	moduleToFilePath(modulePath) {
-		return moduleToRelativePath(modulePath, this.packageName) ?? modulePath;
-	}
-	modulePathToDebugPath(modulePath) {
-		const filePath = this.moduleToFilePath(modulePath);
-		return this.debugWeslRoot + this.packageName + "/" + filePath + ".wesl";
-	}
-};
-/** Convert file path to module path (e.g., "foo/bar.wesl" to "package::foo::bar"). */
-function fileToModulePath(filePath, packageName, treatLibAsRoot) {
-	if (filePath.includes("::")) return filePath;
-	if (treatLibAsRoot && libRegex.test(filePath)) return packageName;
-	const moduleSuffix = noSuffix(normalize(filePath)).replaceAll("/", "::");
-	return packageName + "::" + moduleSuffix;
-}
-/** Try path variants with/without ./ prefix and extension suffixes. */
-function findInVariants(sources, basePath, extensions = ["wesl", "wgsl"]) {
-	for (const prefix of ["", "./"]) {
-		const path = prefix + basePath;
-		if (sources[path] !== void 0) return sources[path];
-		for (const ext of extensions) {
-			const withExt = `${path}.${ext}`;
-			if (sources[withExt] !== void 0) return sources[withExt];
-		}
-	}
-}
-
 //#endregion
 //#region src/SrcMap.ts
 /** map text ranges in multiple src texts to a single dest text */
@@ -4857,8 +4886,8 @@ function refersToBindingStruct(memberRef) {
 	};
 }
 /** If this identifier ultimately refers to a struct type, return the struct declaration */
-function traceToStruct(ident$1) {
-	const declElem = findDecl(ident$1).declElem;
+function traceToStruct(ident) {
+	const declElem = findDecl(ident).declElem;
 	if (declElem && declElem.kind === "param") {
 		const name = declElem.name.typeRef?.name;
 		if (typeof name !== "string") {
@@ -4963,4 +4992,4 @@ function makeWeslDevice(device) {
 }
 
 //#endregion
-export { BundleResolver, CompositeResolver, LinkedWesl, ParseError, RecordResolver, SrcMap, SrcMapBuilder, WeslParseError, WeslStream, _linkSync, astToString, attributeToString, bindAndTransform, bindIdents, bindIdentsRecursive, bindingStructsPlugin, childIdent, childScope, containsScope, debug, debugContentsToString, emptyScope, errorHighlight, fileToModulePath, filterMap, findMap, findRefsToBindingStructs, findUnboundIdents, findUnboundRefs, findValidRootDecls, flatImports, groupBy, grouped, identToString, last, lengthPrefixMangle, link, linkRegistry, liveDeclsToString, log, lowerBindingStructs, makeLiveDecls, makeWeslDevice, mapForward, mapValues, markBindingStructs, markEntryTypes, mergeScope, minimalMangle, minimallyMangledName, modulePartsToRelativePath, moduleToRelativePath, multiKeySet, multisampledTextureTypes, nextIdentId, noSuffix, normalize, normalizeDebugRoot, normalizeModuleName, npmNameVariations, offsetToLineNumber, overlapTail, parseSrcModule, partition, publicDecl, replaceWords, requestWeslDevice, resetScopeIds, resolveModulePath, sampledTextureTypes, sanitizePackageName, scan, scopeToString, scopeToStringLong, srcLog, stdEnumerant, stdEnumerants, stdFn, stdFns, stdType, stdTypes, textureStorageTypes, transformBindingReference, transformBindingStruct, underscoreMangle, validation, withLoggerAsync };
+export { BundleResolver, CompositeResolver, LinkedWesl, ParseError, RecordResolver, SrcMap, SrcMapBuilder, TrackingResolver, WeslParseError, WeslStream, _linkSync, astToString, attributeToString, bindAndTransform, bindIdents, bindIdentsRecursive, bindingStructsPlugin, childIdent, childScope, containsScope, debug, debugContentsToString, discoverModules, emptyScope, errorHighlight, fileToModulePath, filterMap, filterValidElements, findAllRootDecls, findMap, findRefsToBindingStructs, findUnboundIdents, findUnboundRefs, findValidRootDecls, flatImports, groupBy, grouped, identToString, last, lengthPrefixMangle, link, linkRegistry, liveDeclsToString, log, lowerBindingStructs, makeLiveDecls, makeWeslDevice, mapForward, mapValues, markBindingStructs, markEntryTypes, mergeScope, minimalMangle, minimallyMangledName, modulePartsToRelativePath, moduleToRelativePath, multiKeySet, multisampledTextureTypes, nextIdentId, noSuffix, normalize, normalizeDebugRoot, normalizeModuleName, npmNameVariations, offsetToLineNumber, overlapTail, parseSrcModule, partition, publicDecl, replaceWords, requestWeslDevice, resetScopeIds, resolveModulePath, sampledTextureTypes, sanitizePackageName, scan, scopeToString, scopeToStringLong, srcLog, stdEnumerant, stdEnumerants, stdFn, stdFns, stdType, stdTypes, textureStorageTypes, transformBindingReference, transformBindingStruct, underscoreMangle, validation, wgslStandardAttributes, withLoggerAsync };