import { defineConfig, defineTransformer } from 'orval'

const selectedTag = 'twitter-lite-e2e'
const selectedOperationIds = new Set([
  'getUserByScreenName',
  'getUserTweets',
  'getTweetDetail',
  'getSearchTimeline',
  'getListLatestTweetsTimeline',
])

const selectE2eOperations = defineTransformer((document) => {
  const foundOperationIds = new Set<string>()

  for (const pathItem of Object.values(document.paths ?? {})) {
    if (!pathItem || typeof pathItem !== 'object') continue

    for (const candidate of Object.values(pathItem)) {
      if (
        !candidate ||
        typeof candidate !== 'object' ||
        Array.isArray(candidate) ||
        !('operationId' in candidate)
      ) {
        continue
      }

      const operation = candidate as {
        operationId?: string
        tags?: string[]
      }

      if (
        operation.operationId &&
        selectedOperationIds.has(operation.operationId)
      ) {
        foundOperationIds.add(operation.operationId)
        operation.tags = [...(operation.tags ?? []), selectedTag]
      }
    }
  }

  const missingOperationIds = [...selectedOperationIds].filter(
    (operationId) => !foundOperationIds.has(operationId),
  )
  if (missingOperationIds.length > 0) {
    throw new Error(
      `twitter-openapi is missing selected operations: ${missingOperationIds.join(', ')}`,
    )
  }

  return document
})

export default defineConfig({
  twitterOpenApi: {
    input: {
      target:
        'https://raw.githubusercontent.com/fa0311/twitter-openapi/590dae5c9f8575abc91d3774946bfe6f23960aba/dist/docs/openapi-3.0.yaml',
      filters: {
        mode: 'include',
        tags: [selectedTag],
      },
      override: {
        transformer: selectE2eOperations,
      },
    },
    output: {
      clean: true,
      client: 'fetch',
      mode: 'single',
      target: './e2e/generated/twitter-openapi/client.ts',
    },
  },
})