Function extractEndpointInputs

scripts/generate-fal-image-field-map.ts:146–193

()

}

function extractEndpointInputs(): Map<string, EndpointFields> {
  const program = ts.createProgram([ENDPOINTS_DTS], {
    target: ts.ScriptTarget.ES2022,
    skipLibCheck: true,
  })
  const checker = program.getTypeChecker()
  const source = program.getSourceFile(ENDPOINTS_DTS)
  if (!source) throw new Error(`Could not load ${ENDPOINTS_DTS}`)

  let mapType: ts.Type | undefined
  source.forEachChild((node) => {
    if (
      ts.isTypeAliasDeclaration(node) &&
      node.name.text === 'EndpointTypeMap'
    ) {
      mapType = checker.getTypeAtLocation(node.name)
    }
  })
  if (!mapType) throw new Error('EndpointTypeMap not found in endpoints.d.ts')

  const endpoints = new Map<string, EndpointFields>()
  for (const endpoint of mapType.getProperties()) {
    const endpointType = checker.getTypeOfSymbol(endpoint)
    const inputSymbol = endpointType.getProperty('input')
    if (!inputSymbol) continue
    const inputType = checker.getTypeOfSymbol(inputSymbol)

    const fields = new Set<string>()
    const isList = new Map<string, boolean>()
    for (const field of inputType.getProperties()) {
      const name = field.getName()
      fields.add(name)
      const fieldType = checker.getTypeOfSymbol(field)
      isList.set(name, typeAcceptsArray(checker, fieldType))
    }

    const outputSymbol = endpointType.getProperty('output')
    const producesVideo = outputSymbol
      ? checker
          .getTypeOfSymbol(outputSymbol)
          .getProperties()
          .some((p) => p.getName() === 'video' || p.getName() === 'videos')
      : false

    endpoints.set(endpoint.getName(), { fields, isList, producesVideo })
  }
  return endpoints
}

function typeAcceptsArray(checker: ts.TypeChecker, type: ts.Type): boolean {
  const parts = type.isUnion() ? type.types : [type]