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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kubb/ast",
-  "version": "5.0.0-alpha.15",
+  "version": "5.0.0-alpha.17",
   "description": "Spec-agnostic AST layer for Kubb. Defines nodes, visitor pattern, and factory functions used across codegen plugins.",
   "keywords": [
     "kubb",

package/src/constants.ts

@@ -5,7 +5,7 @@ import type { ParameterLocation } from './nodes/parameter.ts'
 import type { SchemaType } from './nodes/schema.ts'
 
 /**
- * Depth for schema traversal in visitor functions.
+ * Traversal depth used by AST visitor utilities.
  */
 export type VisitorDepth = 'shallow' | 'deep'
 
@@ -26,7 +26,21 @@ export const nodeKinds = {
   functionParameters: 'FunctionParameters',
 } as const satisfies Record<string, NodeKind>
 
+/**
+ * Canonical schema type strings used by AST schema nodes.
+ *
+ * These values are used across the AST as stable discriminators
+ * (for example `schema.type === schemaTypes.object`).
+ *
+ * The map is grouped by intent:
+ * - primitives (`string`, `number`, `boolean`, ...)
+ * - structural/composite (`object`, `array`, `union`, ...)
+ * - special OpenAPI-oriented types (`ref`, `datetime`, `uuid`, ...)
+ */
 export const schemaTypes = {
+  /**
+   * Text value.
+   */
   string: 'string',
   /**
    * Floating-point number (`float`, `double`).
@@ -40,28 +54,93 @@ export const schemaTypes = {
    * 64-bit integer (`int64`). Only used when `integerType` is set to `'bigint'`.
    */
   bigint: 'bigint',
+  /**
+   * Boolean value
+   */
   boolean: 'boolean',
+  /**
+   * Explicit null value.
+   */
   null: 'null',
+  /**
+   * Any value (no type restriction).
+   */
   any: 'any',
+  /**
+   * Unknown value (must be narrowed before usage).
+   */
   unknown: 'unknown',
+  /**
+   * No return value (`void`).
+   */
   void: 'void',
+  /**
+   * Object with named properties.
+   */
   object: 'object',
+  /**
+   * Sequential list of items.
+   */
   array: 'array',
+  /**
+   * Fixed-length list with position-specific items.
+   */
   tuple: 'tuple',
+  /**
+   * "One of" multiple schema members.
+   */
   union: 'union',
+  /**
+   * "All of" multiple schema members.
+   */
   intersection: 'intersection',
+  /**
+   * Enum schema.
+   */
   enum: 'enum',
+  /**
+   * Reference to another schema.
+   */
   ref: 'ref',
+  /**
+   * Calendar date (for example `2026-03-24`).
+   */
   date: 'date',
+  /**
+   * Date-time value (for example `2026-03-24T09:00:00Z`).
+   */
   datetime: 'datetime',
+  /**
+   * Time-only value (for example `09:00:00`).
+   */
   time: 'time',
+  /**
+   * UUID value.
+   */
   uuid: 'uuid',
+  /**
+   * Email address value.
+   */
   email: 'email',
+  /**
+   * URL value.
+   */
   url: 'url',
+  /**
+   * Binary/blob value.
+   */
   blob: 'blob',
+  /**
+   * Impossible value (`never`).
+   */
   never: 'never',
 } as const satisfies Record<SchemaType, SchemaType>
 
+/**
+ * Primitive scalar schema types used when simplifying union members.
+ */
+export const SCALAR_PRIMITIVE_TYPES = new Set(['string', 'number', 'integer', 'bigint', 'boolean'] as const)
+
 export const httpMethods = {
   get: 'GET',
   post: 'POST',
@@ -81,12 +160,22 @@ export const parameterLocations = {
 } as const satisfies Record<ParameterLocation, ParameterLocation>
 
 /**
- * Default max concurrent visitor calls in `walk`.
+ * Default maximum number of concurrent callbacks used by `walk`.
+ *
+ * @example
+ * ```ts
+ * walk(root, { concurrency: WALK_CONCURRENCY, root: () => {} })
+ * ```
  */
 export const WALK_CONCURRENCY = 30
 
 /**
- * Fallback status code string for API spec responses.
+ * Fallback response status code used for catch-all responses.
+ *
+ * @example
+ * ```ts
+ * const status = DEFAULT_STATUS_CODE // 'default'
+ * ```
  */
 export const DEFAULT_STATUS_CODE = 'default' as const
 

package/src/factory.ts

@@ -1,3 +1,4 @@
+import type { InferSchemaNode } from './infer.ts'
 import type {
   FunctionParameterNode,
   FunctionParametersNode,
@@ -10,14 +11,39 @@ import type {
   RootNode,
   SchemaNode,
 } from './nodes/index.ts'
+import { syncOptionality } from './utils.ts'
 
 /**
- * Distributive variant of `Omit` that preserves union members.
+ * Distributive `Omit` that preserves each member of a union.
+ *
+ * @example
+ * ```ts
+ * type A = { kind: 'a'; keep: string; drop: number }
+ * type B = { kind: 'b'; keep: boolean; drop: number }
+ * type Result = DistributiveOmit<A | B, 'drop'>
+ * // -> { kind: 'a'; keep: string } | { kind: 'b'; keep: boolean }
+ * ```
  */
 export type DistributiveOmit<T, K extends PropertyKey> = T extends unknown ? Omit<T, K> : never
 
+type CreateSchemaObjectInput = Omit<ObjectSchemaNode, 'kind' | 'properties'> & { properties?: Array<PropertyNode> }
+type CreateSchemaInput = CreateSchemaObjectInput | DistributiveOmit<Exclude<SchemaNode, ObjectSchemaNode>, 'kind'>
+type CreateSchemaOutput<T extends CreateSchemaInput> = InferSchemaNode<T> & { kind: 'Schema' }
+
 /**
- * Creates a `RootNode`.
+ * Creates a `RootNode` with stable defaults for `schemas` and `operations`.
+ *
+ * @example
+ * ```ts
+ * const root = createRoot()
+ * // { kind: 'Root', schemas: [], operations: [] }
+ * ```
+ *
+ * @example
+ * ```ts
+ * const root = createRoot({ schemas: [petSchema] })
+ * // keeps default operations: []
+ * ```
  */
 export function createRoot(overrides: Partial<Omit<RootNode, 'kind'>> = {}): RootNode {
   return {
@@ -29,7 +55,27 @@ export function createRoot(overrides: Partial<Omit<RootNode, 'kind'>> = {}): Roo
 }
 
 /**
- * Creates an `OperationNode`.
+ * Creates an `OperationNode` with default empty arrays for `tags`, `parameters`, and `responses`.
+ *
+ * @example
+ * ```ts
+ * const operation = createOperation({
+ *   operationId: 'getPetById',
+ *   method: 'GET',
+ *   path: '/pet/{petId}',
+ * })
+ * // tags, parameters, and responses are []
+ * ```
+ *
+ * @example
+ * ```ts
+ * const operation = createOperation({
+ *   operationId: 'findPets',
+ *   method: 'GET',
+ *   path: '/pet/findByStatus',
+ *   tags: ['pet'],
+ * })
+ * ```
  */
 export function createOperation(
   props: Pick<OperationNode, 'operationId' | 'method' | 'path'> & Partial<Omit<OperationNode, 'kind' | 'operationId' | 'method' | 'path'>>,
@@ -45,38 +91,63 @@ export function createOperation(
 
 /**
  * Creates a `SchemaNode`, narrowed to the variant of `props.type`.
- * For object schemas, `properties` defaults to `[]` when not provided.
+ * For object schemas, `properties` defaults to an empty array.
+ *
+ * @example
+ * ```ts
+ * const scalar = createSchema({ type: 'string' })
+ * // { kind: 'Schema', type: 'string' }
+ * ```
+ *
+ * @example
+ * ```ts
+ * const object = createSchema({ type: 'object' })
+ * // { kind: 'Schema', type: 'object', properties: [] }
+ * ```
+ *
+ * @example
+ * ```ts
+ * const enumSchema = createSchema({
+ *   type: 'enum',
+ *   primitive: 'string',
+ *   enumValues: ['available', 'pending'],
+ * })
+ * ```
  */
-export function createSchema<T extends Omit<ObjectSchemaNode, 'kind' | 'properties'> & { properties?: Array<PropertyNode> }>(
-  props: T,
-): Omit<T, 'properties'> & { properties: Array<PropertyNode>; kind: 'Schema' }
-export function createSchema<T extends DistributiveOmit<Exclude<SchemaNode, ObjectSchemaNode>, 'kind'>>(props: T): T & { kind: 'Schema' }
-export function createSchema(props: DistributiveOmit<SchemaNode, 'kind'>): SchemaNode
-export function createSchema(props: Record<string, unknown>): Record<string, unknown> {
+export function createSchema<T extends CreateSchemaInput>(props: T): CreateSchemaOutput<T>
+export function createSchema(props: CreateSchemaInput): SchemaNode
+export function createSchema(props: CreateSchemaInput): SchemaNode {
   if (props['type'] === 'object') {
-    return { properties: [], ...props, kind: 'Schema' }
+    return { properties: [], ...props, kind: 'Schema' } as CreateSchemaOutput<typeof props>
   }
 
-  return { ...props, kind: 'Schema' }
+  return { ...props, kind: 'Schema' } as CreateSchemaOutput<typeof props>
 }
 
 /**
- * Derives `schema.optional` and `schema.nullish` from `required` and `schema.nullable`.
- * This keeps `PropertyNode.required` as the single source of truth for optionality.
- */
-export function syncPropertySchema(required: boolean, schema: SchemaNode): SchemaNode {
-  const nullable = schema.nullable ?? false
-
-  return {
-    ...schema,
-    optional: !required && !nullable ? true : undefined,
-    nullish: !required && nullable ? true : undefined,
-  }
-}
-
-/**
- * Creates a `PropertyNode`. `required` defaults to `false`.
- * `schema.optional` and `schema.nullish` are auto-derived from `required` and `schema.nullable`.
+ * Creates a `PropertyNode`.
+ *
+ * `required` defaults to `false`.
+ * `schema.optional` and `schema.nullish` are derived from `required` and `schema.nullable`.
+ *
+ * @example
+ * ```ts
+ * const property = createProperty({
+ *   name: 'status',
+ *   schema: createSchema({ type: 'string' }),
+ * })
+ * // required=false, schema.optional=true
+ * ```
+ *
+ * @example
+ * ```ts
+ * const property = createProperty({
+ *   name: 'status',
+ *   required: true,
+ *   schema: createSchema({ type: 'string', nullable: true }),
+ * })
+ * // required=true, no optional/nullish
+ * ```
  */
 export function createProperty(props: Pick<PropertyNode, 'name' | 'schema'> & Partial<Omit<PropertyNode, 'kind' | 'name' | 'schema'>>): PropertyNode {
   const required = props.required ?? false
@@ -85,13 +156,35 @@ export function createProperty(props: Pick<PropertyNode, 'name' | 'schema'> & Pa
     ...props,
     kind: 'Property',
     required,
-    schema: syncPropertySchema(required, props.schema),
+    schema: syncOptionality(required, props.schema),
   }
 }
 
 /**
- * Creates a `ParameterNode`. `required` defaults to `false`.
- * `schema.optional` is auto-derived from `required` and `schema.nullable`.
+ * Creates a `ParameterNode`.
+ *
+ * `required` defaults to `false`.
+ * Nested schema flags are set from `required` and `schema.nullable`.
+ *
+ * @example
+ * ```ts
+ * const param = createParameter({
+ *   name: 'petId',
+ *   in: 'path',
+ *   required: true,
+ *   schema: createSchema({ type: 'string' }),
+ * })
+ * ```
+ *
+ * @example
+ * ```ts
+ * const param = createParameter({
+ *   name: 'status',
+ *   in: 'query',
+ *   schema: createSchema({ type: 'string', nullable: true }),
+ * })
+ * // required=false, schema.nullish=true
+ * ```
  */
 export function createParameter(
   props: Pick<ParameterNode, 'name' | 'in' | 'schema'> & Partial<Omit<ParameterNode, 'kind' | 'name' | 'in' | 'schema'>>,
@@ -101,12 +194,21 @@ export function createParameter(
     ...props,
     kind: 'Parameter',
     required,
-    schema: syncPropertySchema(required, props.schema),
+    schema: syncOptionality(required, props.schema),
   }
 }
 
 /**
  * Creates a `ResponseNode`.
+ *
+ * @example
+ * ```ts
+ * const response = createResponse({
+ *   statusCode: '200',
+ *   description: 'Success',
+ *   schema: createSchema({ type: 'object', properties: [] }),
+ * })
+ * ```
  */
 export function createResponse(
   props: Pick<ResponseNode, 'statusCode' | 'schema'> & Partial<Omit<ResponseNode, 'kind' | 'statusCode' | 'schema'>>,
@@ -118,7 +220,36 @@ export function createResponse(
 }
 
 /**
- * Creates a `FunctionParameterNode`. `optional` defaults to `false`.
+ * Creates a single-property object schema used as a discriminator literal.
+ *
+ * @example
+ * ```ts
+ * createDiscriminantNode({ propertyName: 'type', value: 'dog' })
+ * // -> { type: 'object', properties: [{ name: 'type', required: true, schema: enum('dog') }] }
+ * ```
+ */
+export function createDiscriminantNode({ propertyName, value }: { propertyName: string; value: string }): SchemaNode {
+  return createSchema({
+    type: 'object',
+    primitive: 'object',
+    properties: [
+      createProperty({
+        name: propertyName,
+        schema: createSchema({
+          type: 'enum',
+          primitive: 'string',
+          enumValues: [value],
+        }),
+        required: true,
+      }),
+    ],
+  })
+}
+
+/**
+ * Creates a `FunctionParameterNode`.
+ *
+ * `optional` defaults to `false`.
  *
  * @example Required typed param
  * ```ts
@@ -132,7 +263,7 @@ export function createResponse(
  * // → params?: QueryParams
  * ```
  *
- * @example Param with default (implicitly optional — cannot combine with `optional: true`)
+ * @example Param with default (implicitly optional; cannot combine with `optional: true`)
  * ```ts
  * createFunctionParameter({ name: 'config', type: 'RequestConfig', default: '{}' })
  * // → config: RequestConfig = {}
@@ -149,7 +280,7 @@ export function createFunctionParameter(
 }
 
 /**
- * Creates an `ObjectBindingParameterNode` — an object-destructured parameter group.
+ * Creates an `ObjectBindingParameterNode` for object-destructured parameter groups.
  *
  * @example Destructured object param
  * ```ts
@@ -164,7 +295,7 @@ export function createFunctionParameter(
  * // call        → { id, name }
  * ```
  *
- * @example Inline — children emitted as individual top-level params
+ * @example Inline mode — children emitted as individual top-level parameters
  * ```ts
  * createObjectBindingParameter({
  *   properties: [createFunctionParameter({ name: 'petId', type: 'string', optional: false })],
@@ -184,7 +315,7 @@ export function createObjectBindingParameter(
 }
 
 /**
- * Creates a `FunctionParametersNode` from an ordered list of params.
+ * Creates a `FunctionParametersNode` from an ordered list of parameters.
  *
  * @example
  * ```ts
@@ -195,6 +326,12 @@ export function createObjectBindingParameter(
  *   ],
  * })
  * ```
+ *
+ * @example
+ * ```ts
+ * const empty = createFunctionParameters()
+ * // { kind: 'FunctionParameters', params: [] }
+ * ```
  */
 export function createFunctionParameters(props: Partial<Omit<FunctionParametersNode, 'kind'>> = {}): FunctionParametersNode {
   return {

package/src/guards.ts

@@ -14,7 +14,13 @@ import type {
 } from './nodes/index.ts'
 
 /**
- * Narrows a `SchemaNode` to the specific variant matching `type`.
+ * Narrows a `SchemaNode` to the variant that matches `type`.
+ *
+ * @example
+ * ```ts
+ * const schema = createSchema({ type: 'string' })
+ * const stringNode = narrowSchema(schema, 'string') // StringSchemaNode | undefined
+ * ```
  */
 export function narrowSchema<T extends SchemaNode['type']>(node: SchemaNode | undefined, type: T): SchemaNodeByType[T] | undefined {
   return node?.type === type ? (node as SchemaNodeByType[T]) : undefined
@@ -25,46 +31,67 @@ function isKind<T extends Node>(kind: NodeKind) {
 }
 
 /**
- * Type guard for `RootNode`.
+ * Returns `true` when the input is a `RootNode`.
+ *
+ * @example
+ * ```ts
+ * if (isRootNode(node)) {
+ *   console.log(node.schemas.length)
+ * }
+ * ```
  */
 export const isRootNode = isKind<RootNode>('Root')
 
 /**
- * Type guard for `OperationNode`.
+ * Returns `true` when the input is an `OperationNode`.
+ *
+ * @example
+ * ```ts
+ * if (isOperationNode(node)) {
+ *   console.log(node.operationId)
+ * }
+ * ```
  */
 export const isOperationNode = isKind<OperationNode>('Operation')
 
 /**
- * Type guard for `SchemaNode`.
+ * Returns `true` when the input is a `SchemaNode`.
+ *
+ * @example
+ * ```ts
+ * if (isSchemaNode(node)) {
+ *   console.log(node.type)
+ * }
+ * ```
  */
 export const isSchemaNode = isKind<SchemaNode>('Schema')
 
 /**
- * Type guard for `PropertyNode`.
+ * Returns `true` when the input is a `PropertyNode`.
  */
 export const isPropertyNode = isKind<PropertyNode>('Property')
 
 /**
- * Type guard for `ParameterNode`.
+ * Returns `true` when the input is a `ParameterNode`.
  */
 export const isParameterNode = isKind<ParameterNode>('Parameter')
 
 /**
- * Type guard for `ResponseNode`.
+ * Returns `true` when the input is a `ResponseNode`.
  */
 export const isResponseNode = isKind<ResponseNode>('Response')
 
 /**
- * Type guard for `FunctionParameterNode`.
+ * Returns `true` when the input is a `FunctionParameterNode`.
  */
 export const isFunctionParameterNode = isKind<FunctionParameterNode>('FunctionParameter')
 
 /**
- * Type guard for `ObjectBindingParameterNode`.
+ * Returns `true` when the input is an `ObjectBindingParameterNode`.
  */
 export const isObjectBindingParameterNode = isKind<ObjectBindingParameterNode>('ObjectBindingParameter')
 
 /**
- * Type guard for `FunctionParametersNode`.
+ * Returns `true` when the input is a `FunctionParametersNode`.
  */
 export const isFunctionParametersNode = isKind<FunctionParametersNode>('FunctionParameters')

package/src/index.ts

@@ -1,5 +1,6 @@
-export { httpMethods, mediaTypes, nodeKinds, schemaTypes } from './constants.ts'
+export { httpMethods, mediaTypes, nodeKinds, SCALAR_PRIMITIVE_TYPES, schemaTypes } from './constants.ts'
 export {
+  createDiscriminantNode,
   createFunctionParameter,
   createFunctionParameters,
   createObjectBindingParameter,
@@ -9,9 +10,7 @@ export {
   createResponse,
   createRoot,
   createSchema,
-  syncPropertySchema,
 } from './factory.ts'
-export { functionPrinter } from './functionPrinter.ts'
 export {
   isFunctionParameterNode,
   isFunctionParametersNode,
@@ -24,7 +23,10 @@ export {
   isSchemaNode,
   narrowSchema,
 } from './guards.ts'
-export { definePrinter } from './printer.ts'
-export { buildRefMap, refMapToObject, resolveRef } from './refs.ts'
-export { applyParamsCasing, isPlainStringType } from './utils.ts'
+export type { InferSchema, InferSchemaNode, ParserOptions } from './infer.ts'
+export { defineFunctionPrinter, definePrinter, functionPrinter } from './printers/index.ts'
+export { buildRefMap, extractRefName, refMapToObject, resolveRef } from './refs.ts'
+export { childName, collectImports, enumPropName, findDiscriminator } from './resolvers.ts'
+export { mergeAdjacentObjects, resolveNames, setDiscriminatorEnum, setEnumName, simplifyUnion } from './transformers.ts'
+export { caseParams, isStringType, syncOptionality } from './utils.ts'
 export { collect, composeTransformers, transform, walk } from './visitor.ts'