@kubb/ast 5.1.0 → 5.2.0

package/dist/index.cjs

@@ -1536,6 +1536,89 @@ function findCircularSchemas(schemas) {
 	return findCircularSchemasMemo(schemas);
 }
 //#endregion
+//#region src/utils/schemaProperties.ts
+/**
+* Finds every matching property exposed by an object, resolved reference, or intersection.
+*
+* References are followed through their parsed `schema`. Circular references terminate without
+* revisiting a node. Unions are not traversed because a property on one branch does not describe
+* every branch.
+*
+* @example Faker discriminator narrowing
+* ```ts
+* const values = resolveSchemaProperties({ node: member, propertyName: union.discriminatorPropertyName }).flatMap(({ schema }) =>
+*   getSchemaLiteralValues(schema),
+* )
+* const typeName = `Extract<Pet, { type: ${values.map(JSON.stringify).join(' | ')} }>`
+* ```
+*
+* @example Zod discriminated unions
+* ```ts
+* const canDiscriminate = union.members?.every(
+*   (member) => resolveSchemaProperties({ node: member, propertyName: 'type' }).length > 0,
+* )
+* const expression = canDiscriminate ? `z.discriminatedUnion('type', members)` : `z.union(members)`
+* ```
+*
+* @example TypeScript property lookup
+* ```ts
+* const properties = resolveSchemaProperties({ node: schema, propertyName: 'status' })
+* const statusSchemas = properties.map((property) => print(property.schema))
+* ```
+*/
+function resolveSchemaProperties({ node, propertyName }) {
+	const properties = [];
+	const visited = /* @__PURE__ */ new WeakSet();
+	function visit(current) {
+		if (!current || visited.has(current)) return;
+		visited.add(current);
+		if (current.type === "object") {
+			const property = current.properties.find((candidate) => candidate.name === propertyName);
+			if (property) properties.push(property);
+			return;
+		}
+		if (current.type === "ref") {
+			visit(current.schema);
+			return;
+		}
+		if (current.type === "intersection") for (const member of current.members ?? []) visit(member);
+	}
+	visit(node);
+	return properties;
+}
+/**
+* Reads the unique literal values represented by an enum, resolved reference, or union.
+*
+* Named enum values take precedence over simple enum values, matching how renderers consume
+* `EnumSchemaNode`.
+*
+* @example
+* ```ts
+* const values = getSchemaLiteralValues(enumNode)
+* // ['cat', 'dog']
+* ```
+*/
+function getSchemaLiteralValues(node) {
+	const values = /* @__PURE__ */ new Set();
+	const visited = /* @__PURE__ */ new WeakSet();
+	function visit(current) {
+		if (!current || visited.has(current)) return;
+		visited.add(current);
+		if (current.type === "enum") {
+			const enumValues = current.namedEnumValues?.map(({ value }) => value) ?? current.enumValues ?? [];
+			for (const value of enumValues) values.add(value);
+			return;
+		}
+		if (current.type === "ref") {
+			visit(current.schema);
+			return;
+		}
+		if (current.type === "union") for (const member of current.members ?? []) visit(member);
+	}
+	visit(node);
+	return [...values];
+}
+//#endregion
 //#region src/factory.ts
 var factory_exports = /* @__PURE__ */ __exportAll({
 	createArrowFunction: () => createArrowFunction,
@@ -1606,6 +1689,7 @@ var exports_exports = /* @__PURE__ */ __exportAll({
 	findCircularSchemas: () => findCircularSchemas,
 	findCircularSchemasFromGraph: () => findCircularSchemasFromGraph,
 	functionDef: () => functionDef,
+	getSchemaLiteralValues: () => getSchemaLiteralValues,
 	importDef: () => importDef,
 	inputDef: () => inputDef,
 	isHttpOperationNode: () => isHttpOperationNode,
@@ -1619,6 +1703,7 @@ var exports_exports = /* @__PURE__ */ __exportAll({
 	propertyDef: () => propertyDef,
 	requestBodyDef: () => requestBodyDef,
 	resolveRefName: () => resolveRefName,
+	resolveSchemaProperties: () => resolveSchemaProperties,
 	responseDef: () => responseDef,
 	schemaDef: () => schemaDef,
 	schemaTypes: () => schemaTypes,
@@ -1659,6 +1744,7 @@ exports.fileDef = fileDef;
 exports.findCircularSchemas = findCircularSchemas;
 exports.findCircularSchemasFromGraph = findCircularSchemasFromGraph;
 exports.functionDef = functionDef;
+exports.getSchemaLiteralValues = getSchemaLiteralValues;
 exports.importDef = importDef;
 exports.inputDef = inputDef;
 exports.isHttpOperationNode = isHttpOperationNode;
@@ -1672,6 +1758,7 @@ exports.parameterDef = parameterDef;
 exports.propertyDef = propertyDef;
 exports.requestBodyDef = requestBodyDef;
 exports.resolveRefName = resolveRefName;
+exports.resolveSchemaProperties = resolveSchemaProperties;
 exports.responseDef = responseDef;
 exports.schemaDef = schemaDef;
 exports.schemaTypes = schemaTypes;