import { OpenAPIRegistry, OpenApiGeneratorV3 } from '@asteasolutions/zod-to-openapi'
import type { ReferenceObject, SchemaObject } from 'openapi3-ts/oas30'
import type { ZodSchema } from './types.js'

function oneOf<T extends ZodSchema>(schemas: T[]): (ReferenceObject | SchemaObject)[] {
  const registry = new OpenAPIRegistry()

  schemas.forEach((schema, index) => {
    registry.register(index.toString(), schema)
  })

  const generator = new OpenApiGeneratorV3(registry.definitions)
  const components = generator.generateComponents()

  return components.components?.schemas ? Object.values(components.components.schemas) : []
}

export default oneOf