@kubb/ast 5.0.0-alpha.15 → 5.0.0-alpha.16

package/dist/index.cjs

@@ -44,6 +44,16 @@ const schemaTypes = {
 	blob: "blob",
 	never: "never"
 };
+/**
+* Scalar primitive schema types used for union member simplification.
+*/
+const SCALAR_PRIMITIVE_TYPES = new Set([
+	"string",
+	"number",
+	"integer",
+	"bigint",
+	"boolean"
+]);
 const httpMethods = {
 	get: "GET",
 	post: "POST",
@@ -490,6 +500,13 @@ const isFunctionParametersNode = isKind("FunctionParameters");
 //#endregion
 //#region src/refs.ts
 /**
+* Extracts the final segment from a reference string.
+* Falls back to the original string when no slash exists.
+*/
+function extractRefName(ref) {
+	return ref.split("/").at(-1) ?? ref;
+}
+/**
 * Indexes named schemas from `root.schemas` by name. Unnamed schemas are skipped.
 */
 function buildRefMap(root) {
@@ -510,6 +527,72 @@ function refMapToObject(refMap) {
 	return Object.fromEntries(refMap);
 }
 //#endregion
+//#region src/transforms.ts
+/**
+* Replaces the discriminator property's schema inside an object node with
+* an enum of the provided values.
+*/
+function applyDiscriminatorEnum({ node, propertyName, values, enumName }) {
+	const objectNode = narrowSchema(node, "object");
+	if (!objectNode?.properties?.length) return node;
+	if (!objectNode.properties.some((prop) => prop.name === propertyName)) return node;
+	return createSchema({
+		...objectNode,
+		properties: objectNode.properties.map((prop) => {
+			if (prop.name !== propertyName) return prop;
+			return createProperty({
+				...prop,
+				schema: createSchema({
+					type: "enum",
+					primitive: "string",
+					enumValues: values,
+					name: enumName,
+					readOnly: prop.schema.readOnly,
+					writeOnly: prop.schema.writeOnly
+				})
+			});
+		})
+	});
+}
+/**
+* Merges adjacent anonymous object members into a single anonymous object.
+*/
+function mergeAdjacentAnonymousObjects(members) {
+	return members.reduce((acc, member) => {
+		const objectMember = narrowSchema(member, "object");
+		if (objectMember && !objectMember.name) {
+			const previous = acc.at(-1);
+			const previousObject = previous ? narrowSchema(previous, "object") : void 0;
+			if (previousObject && !previousObject.name) {
+				acc[acc.length - 1] = createSchema({
+					...previousObject,
+					properties: [...previousObject.properties ?? [], ...objectMember.properties ?? []]
+				});
+				return acc;
+			}
+		}
+		acc.push(member);
+		return acc;
+	}, []);
+}
+/**
+* Removes enum members subsumed by broader scalar members in the same union.
+*/
+function simplifyUnionMembers(members) {
+	const scalarPrimitives = new Set(members.filter((member) => SCALAR_PRIMITIVE_TYPES.has(member.type)).map((m) => m.type));
+	if (!scalarPrimitives.size) return members;
+	return members.filter((member) => {
+		const enumNode = narrowSchema(member, "enum");
+		if (!enumNode) return true;
+		const primitive = enumNode.primitive;
+		if (!primitive) return true;
+		if (!enumNode.enumType) return true;
+		if (scalarPrimitives.has(primitive)) return false;
+		if ((primitive === "integer" || primitive === "number") && (scalarPrimitives.has("integer") || scalarPrimitives.has("number"))) return false;
+		return true;
+	});
+}
+//#endregion
 //#region ../../internals/utils/dist/index.js
 /**
 * Shared implementation for camelCase and PascalCase conversion.
@@ -1067,6 +1150,8 @@ function collect(node, options) {
 	return results;
 }
 //#endregion
+exports.SCALAR_PRIMITIVE_TYPES = SCALAR_PRIMITIVE_TYPES;
+exports.applyDiscriminatorEnum = applyDiscriminatorEnum;
 exports.applyParamsCasing = applyParamsCasing;
 exports.buildRefMap = buildRefMap;
 exports.collect = collect;
@@ -1081,6 +1166,7 @@ exports.createResponse = createResponse;
 exports.createRoot = createRoot;
 exports.createSchema = createSchema;
 exports.definePrinter = definePrinter;
+exports.extractRefName = extractRefName;
 exports.functionPrinter = functionPrinter;
 exports.httpMethods = httpMethods;
 exports.isFunctionParameterNode = isFunctionParameterNode;
@@ -1094,11 +1180,13 @@ exports.isResponseNode = isResponseNode;
 exports.isRootNode = isRootNode;
 exports.isSchemaNode = isSchemaNode;
 exports.mediaTypes = mediaTypes;
+exports.mergeAdjacentAnonymousObjects = mergeAdjacentAnonymousObjects;
 exports.narrowSchema = narrowSchema;
 exports.nodeKinds = nodeKinds;
 exports.refMapToObject = refMapToObject;
 exports.resolveRef = resolveRef;
 exports.schemaTypes = schemaTypes;
+exports.simplifyUnionMembers = simplifyUnionMembers;
 exports.syncPropertySchema = syncPropertySchema;
 exports.transform = transform;
 exports.walk = walk;