@workos/oagen 0.24.2 → 0.24.4

package/dist/cli/index.mjs

@@ -252,6 +252,28 @@ function walkTypeRef(ref, visitor) {
 		default: assertNever(ref);
 	}
 }
+/**
+* Validate model-placement hints against the FULL spec. Run once at the start of
+* generation — where the complete model/service universe is known — so typos in
+* `modelHints` fail loud even though {@link assignModelsToServices} is
+* intentionally lenient about the partial slices it may be handed.
+*/
+function validateModelHints(models, services, hints) {
+	const modelNames = new Set(models.map((m) => m.name));
+	const serviceNames = new Set(services.map((s) => s.name));
+	for (const [modelName, serviceName] of Object.entries(hints)) {
+		if (!modelNames.has(modelName)) {
+			const err = /* @__PURE__ */ new Error(`modelHints: unknown model "${modelName}". Hint keys must match an IR model name (post-cleanSchemaName/schemaNameTransform).`);
+			err.name = "ConfigError";
+			throw err;
+		}
+		if (!serviceNames.has(serviceName)) {
+			const err = /* @__PURE__ */ new Error(`modelHints: unknown service "${serviceName}" for model "${modelName}". Hint values must match an IR service name (PascalCase, derived from the operation's first tag).`);
+			err.name = "ConfigError";
+			throw err;
+		}
+	}
+}
 //#endregion
 //#region src/utils/naming.ts
 /**
@@ -5837,13 +5859,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;
 	}
@@ -6420,6 +6447,7 @@ async function removeEmptyParents(startDir, stopDir) {
 //#endregion
 //#region src/engine/orchestrator.ts
 async function generate(spec, emitter, options) {
+	if (options.modelHints) validateModelHints(spec.models, spec.services, options.modelHints);
 	const scopedServices = options.services && options.services.length > 0 ? resolveScopedServices(spec, options.services, options.mountRules) : void 0;
 	const scoped = scopedServices !== void 0;
 	const noPrune = options.noPrune === true || scoped;