@workos/oagen 0.24.2 → 0.24.3

package/dist/cli/index.mjs

@@ -5837,13 +5837,18 @@ function buildEmitterContext(spec, options) {
 	let scopedModelNames;
 	let scopedEnumNames;
 	if (options.scopedServices && options.scopedServices.size > 0) {
-		const scope = options.scopedServices;
-		const present = options.presentServiceKeys ?? /* @__PURE__ */ new Set();
-		const postMountOf = (service) => resolvedOperations.find((r) => r.service.name === service.name)?.mountOn ?? service.name;
-		const reachable = collectReferencedNames(spec.services.filter((s) => {
-			const postMount = postMountOf(s);
-			return scope.has(postMount) || present.has(canonicalServiceKey(postMount));
-		}), spec.models, { preserveAllDiscriminated: false });
+		const selected = options.scopedServices;
+		const selectedOpsByService = /* @__PURE__ */ new Map();
+		for (const r of resolvedOperations) {
+			if (!selected.has(r.mountOn)) continue;
+			const ops = selectedOpsByService.get(r.service.name) ?? [];
+			ops.push(r.operation);
+			selectedOpsByService.set(r.service.name, ops);
+		}
+		const reachable = collectReferencedNames(spec.services.filter((s) => selectedOpsByService.has(s.name)).map((s) => ({
+			...s,
+			operations: selectedOpsByService.get(s.name)
+		})), spec.models, { preserveAllDiscriminated: false });
 		scopedModelNames = reachable.models;
 		scopedEnumNames = reachable.enums;
 	}