@workos/oagen 0.24.3 → 0.24.5

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
 /**
@@ -1592,7 +1614,7 @@ function detectPagination(response, queryParams, dataPath) {
 }
 //#endregion
 //#region src/parser/responses.ts
-function classifyAndExtractResponse(schema, contextName) {
+function classifyAndExtractResponse(schema, contextName, componentSchemas) {
 	if (schema.$ref) return {
 		response: schemaToTypeRef(schema, contextName),
 		inlineModels: [],
@@ -1601,7 +1623,7 @@ function classifyAndExtractResponse(schema, contextName) {
 	const envelope = detectListEnvelope(schema);
 	if (envelope.isEnvelope) return extractListResponse(schema, contextName, envelope.dataPath);
 	if (isSingleResourceWrapper(schema)) return extractWrappedResource(schema, contextName);
-	return extractDirectResource(schema, contextName);
+	return extractDirectResource(schema, contextName, componentSchemas);
 }
 const KNOWN_DATA_PATHS = new Set([
 	"data",
@@ -1778,7 +1800,7 @@ function extractWrappedResource(schema, contextName) {
 		isPaginated: false
 	};
 }
-function extractDirectResource(schema, contextName) {
+function extractDirectResource(schema, contextName, componentSchemas) {
 	const inlineModels = [];
 	if (schema.properties && (schema.type === "object" || !schema.type)) {
 		const modelName = deriveModelName(schema, contextName);
@@ -1795,17 +1817,43 @@ function extractDirectResource(schema, contextName) {
 	if (schema.allOf) {
 		const mergedProperties = {};
 		const mergedRequired = [];
-		for (const sub of schema.allOf) {
-			if (sub.properties) Object.assign(mergedProperties, sub.properties);
-			if (sub.required) mergedRequired.push(...sub.required);
-		}
+		const unresolvedRefs = [];
+		const seenRefs = /* @__PURE__ */ new Set();
+		let mergedRefFields = false;
+		const mergeSubSchemas = (subs) => {
+			for (const sub of subs) {
+				let resolved = sub;
+				if (sub.$ref) {
+					if (seenRefs.has(sub.$ref)) continue;
+					seenRefs.add(sub.$ref);
+					const segments = sub.$ref.split("/");
+					const refName = segments[segments.length - 1];
+					const refSchema = refName ? componentSchemas?.[refName] : void 0;
+					if (!refSchema) {
+						unresolvedRefs.push(sub.$ref);
+						continue;
+					}
+					resolved = refSchema;
+					mergedRefFields = true;
+				}
+				if (resolved.allOf) mergeSubSchemas(resolved.allOf);
+				if (resolved.properties) Object.assign(mergedProperties, resolved.properties);
+				if (resolved.required) mergedRequired.push(...resolved.required);
+			}
+		};
+		mergeSubSchemas(schema.allOf);
+		if (unresolvedRefs.length === 1 && !mergedRefFields) return {
+			response: schemaToTypeRef({ $ref: unresolvedRefs[0] }, contextName),
+			inlineModels,
+			isPaginated: false
+		};
 		if (Object.keys(mergedProperties).length > 0) {
 			const merged = {
 				type: "object",
 				properties: mergedProperties,
 				required: mergedRequired
 			};
-			const modelName = deriveModelName(merged, contextName);
+			const modelName = mergedRefFields ? toPascalCase(contextName) : deriveModelName(merged, contextName);
 			inlineModels.push(...extractInlineModel(modelName, merged));
 			return {
 				response: {
@@ -2208,7 +2256,7 @@ function buildOperation(method, path, op, pathLevelParams, operationIdTransform,
 	const cookieParams = extractParams(allParams, "cookie", opContext, componentSchemas);
 	const reqBodyModels = [];
 	const { body: requestBody, encoding: requestBodyEncoding } = extractRequestBody(op.requestBody, op, reqBodyModels);
-	const { response, successResponses, errors, inlineModels, isPaginated, dataPath: responseDataPath, itemType: responseItemType } = extractResponses(op.responses, op, path, method);
+	const { response, successResponses, errors, inlineModels, isPaginated, dataPath: responseDataPath, itemType: responseItemType } = extractResponses(op.responses, op, path, method, componentSchemas);
 	inlineModels.push(...reqBodyModels);
 	const paginationFromParams = detectPagination(response, queryParams, responseDataPath);
 	let pagination;
@@ -2501,7 +2549,7 @@ function deriveResponseName(op, path, method) {
 	if (op?.operationId) return cleanSchemaName(toPascalCase(normalizeOperationIdForNaming(op.operationId)) + "Response");
 	return toPascalCase(path.split("/").filter(Boolean)[0] ?? "Unknown") + toPascalCase(method) + "Response";
 }
-function extractResponses(responses, op, path, method) {
+function extractResponses(responses, op, path, method, componentSchemas) {
 	const errors = [];
 	let response = {
 		kind: "primitive",
@@ -2528,7 +2576,7 @@ function extractResponses(responses, op, path, method) {
 			const jsonContent = resp.content?.["application/json"];
 			if (jsonContent?.schema) {
 				const contextName = deriveResponseName(op, path ?? "/", method ?? "get");
-				const result = classifyAndExtractResponse(jsonContent.schema, contextName);
+				const result = classifyAndExtractResponse(jsonContent.schema, contextName, componentSchemas);
 				extractedType = result.response;
 				inlineModels = result.inlineModels;
 				isPaginated = result.isPaginated;
@@ -6425,6 +6473,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;