@workos/oagen 0.5.0 → 0.7.0

package/dist/scripts/smoke/baseline.mjs

@@ -550,7 +550,7 @@ function findSchemaByName(pascalName, schemas) {
 	for (const [k, v] of Object.entries(schemas)) if (resolveSchemaName(k) === pascalName) return v;
 	return null;
 }
-/** Walk into a schema to find a field's sub-schema, resolving allOf. */
+/** Walk into a schema to find a field's sub-schema, resolving allOf and oneOf/anyOf. */
 function findNestedFieldSchema(fieldName, parentSchema, schemas) {
 	if (parentSchema.properties?.[fieldName]) return parentSchema.properties[fieldName];
 	if (parentSchema.allOf) for (const sub of parentSchema.allOf) {
@@ -562,6 +562,15 @@ function findNestedFieldSchema(fieldName, parentSchema, schemas) {
 		}
 		if (resolved.properties?.[fieldName]) return resolved.properties[fieldName];
 	}
+	for (const variant of [...parentSchema.oneOf ?? [], ...parentSchema.anyOf ?? []]) {
+		let resolved = variant;
+		if (variant.$ref) {
+			const segments = variant.$ref.split("/");
+			const refName = segments[segments.length - 1];
+			if (refName && schemas[refName]) resolved = schemas[refName];
+		}
+		if (resolved.properties?.[fieldName]) return resolved.properties[fieldName];
+	}
 	return null;
 }
 /** Extract a model (and nested models) from an inline field schema. */
@@ -629,6 +638,86 @@ function buildFieldFromSchema(fieldName, fieldSchema, contextName, requiredSet)
 		example: fieldSchema.example
 	};
 }
+/**
+* If every non-null oneOf/anyOf variant is a string-valued const, return the
+* list of those const values so the union can collapse into a single enum.
+* Returns null when any variant isn't a pure string-const (model ref,
+* number const, nullable, etc.).
+*/
+function collectLiteralStringConsts(variants) {
+	if (variants.length === 0) return null;
+	const values = [];
+	for (const v of variants) {
+		if (typeof v.const === "string") {
+			values.push(v.const);
+			continue;
+		}
+		if (Array.isArray(v.enum) && v.enum.length === 1 && typeof v.enum[0] === "string") {
+			values.push(v.enum[0]);
+			continue;
+		}
+		return null;
+	}
+	return values;
+}
+/** Build an EnumRef the same way the regular enum path does for naming. */
+function buildSyntheticEnumRef(values, contextName, parentModelName) {
+	const baseName = toPascalCase(contextName ?? "UnknownEnum");
+	const cleanParent = parentModelName ? stripListItemMarkers(parentModelName) : void 0;
+	return {
+		kind: "enum",
+		name: cleanParent && !baseName.startsWith(cleanParent) ? `${cleanParent}${baseName}` : baseName,
+		values
+	};
+}
+/**
+* Detect an implicit discriminator on an oneOf where every variant is an
+* object schema that pins the same property to a const value. Returns
+* { property, mapping } so the IR can represent the union as discriminated
+* without needing an explicit `discriminator:` key in the spec.
+*
+* Both shapes are supported:
+*   { properties: { event: { const: "x" } }, required: ["event"] }
+*   { properties: { event: { type: "string", enum: ["x"] } } }
+*
+* Returns null when any variant doesn't fit the pattern or when variants
+* disagree on which property carries the discriminator.
+*/
+function detectConstPropertyDiscriminator(variants) {
+	if (variants.length < 2) return null;
+	let candidateProperty = null;
+	const candidates = Object.keys(variants[0]?.properties ?? {});
+	for (const propName of candidates) if (variants.every((v) => getConstPropertyValue(v, propName) !== null)) {
+		candidateProperty = propName;
+		break;
+	}
+	if (!candidateProperty) return null;
+	const mapping = {};
+	for (const v of variants) {
+		const value = getConstPropertyValue(v, candidateProperty);
+		if (value === null) return null;
+		const variantName = resolveVariantModelName(v);
+		if (!variantName) return null;
+		mapping[value] = variantName;
+	}
+	return {
+		property: candidateProperty,
+		mapping
+	};
+}
+function getConstPropertyValue(schema, property) {
+	const prop = schema.properties?.[property];
+	if (!prop) return null;
+	if (typeof prop.const === "string") return prop.const;
+	if (Array.isArray(prop.enum) && prop.enum.length === 1 && typeof prop.enum[0] === "string") return prop.enum[0];
+	return null;
+}
+function resolveVariantModelName(schema) {
+	if (schema.$ref) return resolveSchemaName(schema.$ref.replace(/^#\/components\/schemas\//, ""));
+	const title = typeof schema["title"] === "string" ? schema["title"] : null;
+	if (title) return resolveSchemaName(title);
+	return null;
+}
 function extractEnum(name, schema) {
 	return {
 		name,
@@ -647,6 +736,19 @@ function extractModel(name, schema, schemas) {
 		if (!fieldSchema) continue;
 		fields.push(buildFieldFromSchema(fieldName, fieldSchema, name, requiredSet));
 	}
+	if (fields.length === 0 && (schema.oneOf || schema.anyOf)) {
+		const variants = schema.oneOf ?? schema.anyOf ?? [];
+		const emptyRequired = /* @__PURE__ */ new Set();
+		const seenFieldNames = /* @__PURE__ */ new Set();
+		for (const variant of variants) {
+			if (!variant.properties) continue;
+			for (const [fieldName, fieldSchema] of Object.entries(variant.properties)) {
+				if (!fieldSchema || seenFieldNames.has(fieldName)) continue;
+				seenFieldNames.add(fieldName);
+				fields.push(buildFieldFromSchema(fieldName, fieldSchema, name, emptyRequired));
+			}
+		}
+	}
 	if (schema.additionalProperties && typeof schema.additionalProperties === "object" && schema.properties) {
 		if (Object.keys(schema.additionalProperties).length > 0) {
 			const valueType = schemaToTypeRef(schema.additionalProperties, "additionalProperties", name);
@@ -686,6 +788,18 @@ function extractAllOfModel(name, schema, schemas) {
 				});
 				if (f.required) requiredSet.add(f.name);
 			}
+		} else if (resolved.oneOf || resolved.anyOf) {
+			const variants = resolved.oneOf ?? resolved.anyOf ?? [];
+			const emptyRequired = /* @__PURE__ */ new Set();
+			const seenFieldNames = new Set(fields.map((f) => f.name));
+			for (const variant of variants) {
+				if (!variant.properties) continue;
+				for (const [fieldName, fieldSchema] of Object.entries(variant.properties)) {
+					if (!fieldSchema || seenFieldNames.has(fieldName)) continue;
+					seenFieldNames.add(fieldName);
+					fields.push(buildFieldFromSchema(fieldName, fieldSchema, name, emptyRequired));
+				}
+			}
 		} else {
 			if (resolved.required) for (const r of resolved.required) requiredSet.add(r);
 			if (resolved.properties) {
@@ -852,6 +966,15 @@ function schemaToTypeRef(schema, contextName, parentModelName) {
 			kind: "nullable",
 			inner: schemaToTypeRef(nonNullVariants[0], contextName, parentModelName)
 		};
+		const literalStrings = collectLiteralStringConsts(nonNullVariants);
+		if (literalStrings !== null && literalStrings.length >= 2) {
+			const enumRef = buildSyntheticEnumRef(literalStrings, contextName, parentModelName);
+			return nullVariant ? {
+				kind: "nullable",
+				inner: enumRef
+			} : enumRef;
+		}
+		const syntheticDiscriminator = !schema.discriminator && compositionKind === "oneOf" ? detectConstPropertyDiscriminator(nonNullVariants) : null;
 		const variants = rawVariants.filter((v) => v.type !== "null").map((v) => schemaToTypeRef(v, contextName, parentModelName));
 		const hasNull = !!nullVariant;
 		const union = {
@@ -861,7 +984,7 @@ function schemaToTypeRef(schema, contextName, parentModelName) {
 			...schema.discriminator ? { discriminator: {
 				property: schema.discriminator.propertyName,
 				mapping: Object.fromEntries(Object.entries(schema.discriminator.mapping ?? {}).map(([k, v]) => [k, v.replace(/^#\/components\/schemas\//, "")]))
-			} } : {}
+			} } : syntheticDiscriminator ? { discriminator: syntheticDiscriminator } : {}
 		};
 		return hasNull ? {
 			kind: "nullable",
@@ -1533,6 +1656,94 @@ function inferDeeperContext(path) {
 	if (nonParams.length >= 3) return toPascalCase(nonParams.slice(1, -1).join("_"));
 	return null;
 }
+/**
+* Extract mutually-exclusive parameter groups from the `x-mutually-exclusive-parameter-groups`
+* operation extension. Cross-references grouped parameter names against the already-extracted
+* IR parameter arrays so emitters get object-identity references.
+*/
+function extractParameterGroups(op, allIRParams, operationContext) {
+	const raw = op["x-mutually-exclusive-parameter-groups"];
+	if (!raw || typeof raw !== "object") return void 0;
+	const paramByName = /* @__PURE__ */ new Map();
+	for (const p of allIRParams) paramByName.set(p.name, p);
+	const groups = [];
+	for (const [groupName, groupDef] of Object.entries(raw)) {
+		if (!groupDef || typeof groupDef !== "object") throw new Error(`Malformed x-mutually-exclusive-parameter-groups.${groupName} in ${operationContext}: expected an object with "optional" and "variants".`);
+		if (typeof groupDef.optional !== "boolean") throw new Error(`Malformed x-mutually-exclusive-parameter-groups.${groupName}.optional in ${operationContext}: expected a boolean.`);
+		if (!groupDef.variants || typeof groupDef.variants !== "object") throw new Error(`Malformed x-mutually-exclusive-parameter-groups.${groupName}.variants in ${operationContext}: expected an object mapping variant names to parameter name arrays.`);
+		if (Object.keys(groupDef.variants).length === 0) throw new Error(`Malformed x-mutually-exclusive-parameter-groups.${groupName} in ${operationContext}: group has zero variants.`);
+		const variants = Object.entries(groupDef.variants).map(([variantName, paramNames]) => {
+			if (!Array.isArray(paramNames) || paramNames.length === 0) throw new Error(`Malformed x-mutually-exclusive-parameter-groups.${groupName}.variants.${variantName} in ${operationContext}: expected a non-empty array of parameter names.`);
+			return {
+				name: variantName,
+				parameters: paramNames.map((pName) => {
+					const irParam = paramByName.get(pName);
+					if (!irParam) throw new Error(`x-mutually-exclusive-parameter-groups.${groupName}.variants.${variantName} references parameter "${pName}" which does not exist in the operation's parameters[] (${operationContext}).`);
+					return irParam;
+				})
+			};
+		});
+		groups.push({
+			name: groupName,
+			optional: groupDef.optional,
+			variants
+		});
+	}
+	return groups.length > 0 ? groups : void 0;
+}
+/**
+* Extract mutually-exclusive body parameter groups from the
+* `x-mutually-exclusive-body-groups` operation extension. Unlike query
+* parameter groups (whose parameters live in operation.queryParams), body
+* group parameters are synthetic — built from the oneOf variant schemas in
+* the request body. Emitters use the group's presence to generate sum-type
+* interfaces and custom JSON marshalling instead of flat optional fields.
+*/
+function extractBodyParameterGroups(op, operationContext) {
+	const raw = op["x-mutually-exclusive-body-groups"];
+	if (!raw || typeof raw !== "object") return void 0;
+	const bodySchema = op.requestBody?.content?.["application/json"]?.schema;
+	const variantFieldSchemas = /* @__PURE__ */ new Map();
+	if (bodySchema) {
+		for (const sub of bodySchema.allOf ?? []) for (const variant of sub.oneOf ?? []) if (variant.properties) {
+			for (const [name, fieldSchema] of Object.entries(variant.properties)) if (fieldSchema) variantFieldSchemas.set(name, fieldSchema);
+		}
+		for (const variant of bodySchema.oneOf ?? []) if (variant.properties) {
+			for (const [name, fieldSchema] of Object.entries(variant.properties)) if (fieldSchema) variantFieldSchemas.set(name, fieldSchema);
+		}
+	}
+	const groups = [];
+	for (const [groupName, groupDef] of Object.entries(raw)) {
+		if (!groupDef || typeof groupDef !== "object") continue;
+		if (typeof groupDef.optional !== "boolean") continue;
+		if (!groupDef.variants || typeof groupDef.variants !== "object") continue;
+		if (Object.keys(groupDef.variants).length === 0) continue;
+		const variants = Object.entries(groupDef.variants).map(([variantName, paramNames]) => {
+			if (!Array.isArray(paramNames) || paramNames.length === 0) throw new Error(`Malformed x-mutually-exclusive-body-groups.${groupName}.variants.${variantName} in ${operationContext}: expected a non-empty array of parameter names.`);
+			return {
+				name: variantName,
+				parameters: paramNames.map((pName) => {
+					const fieldSchema = variantFieldSchemas.get(pName);
+					return {
+						name: pName,
+						type: fieldSchema ? schemaToTypeRef(fieldSchema, toPascalCase(pName)) : {
+							kind: "primitive",
+							type: "string"
+						},
+						required: false,
+						description: fieldSchema?.description
+					};
+				})
+			};
+		});
+		groups.push({
+			name: groupName,
+			optional: groupDef.optional,
+			variants
+		});
+	}
+	return groups.length > 0 ? groups : void 0;
+}
 function buildOperation(method, path, op, pathLevelParams, operationIdTransform, serviceName) {
 	const allParams = [...pathLevelParams, ...op.parameters ?? []];
 	const hasIdempotencyHeader = allParams.some((p) => p.in === "header" && p.name.toLowerCase() === "idempotency-key");
@@ -1558,6 +1769,16 @@ function buildOperation(method, path, op, pathLevelParams, operationIdTransform,
 		};
 	} else if (paginationFromParams) pagination = paginationFromParams;
 	const security = extractOperationSecurity(op.security);
+	const allIRParams = [
+		...pathParams,
+		...queryParams,
+		...headerParams,
+		...cookieParams
+	];
+	const opLabel = op.operationId ?? `${method.toUpperCase()} ${path}`;
+	const queryParamGroups = extractParameterGroups(op, allIRParams, opLabel);
+	const bodyParamGroups = extractBodyParameterGroups(op, opLabel);
+	const parameterGroups = queryParamGroups || bodyParamGroups ? [...queryParamGroups ?? [], ...bodyParamGroups ?? []] : void 0;
 	return {
 		operation: {
 			name: inferOperationName(method, path, op.operationId, operationIdTransform),
@@ -1577,7 +1798,8 @@ function buildOperation(method, path, op, pathLevelParams, operationIdTransform,
 			injectIdempotencyKey: hasIdempotencyHeader,
 			deprecated: op.deprecated || void 0,
 			async: op["x-oagen-async"],
-			security
+			security,
+			parameterGroups
 		},
 		inlineModels
 	};