import {isUnion} from '../helpers'
import {
  type ApiCustomizationOptions,
  type ApiSpecification,
  type ConvertedType,
  type GeneratedApiSpecification,
  type InputObjectType,
} from '../types'
import {generateTypeFilters} from './generateTypeFilters'
import {generateTypeQueries} from './generateTypeQueries'
import {generateTypeSortings} from './generateTypeSortings'

export default (
  extracted: ApiSpecification,
  options?: ApiCustomizationOptions,
): GeneratedApiSpecification => {
  const documentInterface = extracted.interfaces.find((iface) => iface.name === 'Document')
  if (!documentInterface || isUnion(documentInterface)) {
    throw new Error('Failed to find document interface')
  }

  const types = [...extracted.types, documentInterface as ConvertedType]

  const filters = generateTypeFilters(types, options)
  const sortings = generateTypeSortings(types)
  const queries = generateTypeQueries(
    types,
    sortings.filter((node): node is InputObjectType => node.kind === 'InputObject'),
    options,
  )
  const graphqlTypes = [...extracted.types, ...filters, ...sortings]

  return {types: graphqlTypes, queries, interfaces: extracted.interfaces, generation: 'gen3'}
}