import { queryOptions } from '@tanstack/react-query'
import { apiJSON } from '@/lib/api-client'
import type {
  ModuleRow,
  ModulesResponse,
  Page,
} from '@/types/api'

export const modulesKeys = {
  all: ['modules'] as const,
  list: () => [...modulesKeys.all, 'list'] as const,
  detail: (id: string) => [...modulesKeys.all, 'detail', id] as const,
  domainEntries: (id: string) => [...modulesKeys.all, 'domain-entries', id] as const,
  asEntries: (id: string) => [...modulesKeys.all, 'as-entries', id] as const,
  cdnSources: (id: string) => [...modulesKeys.all, 'cdn-sources', id] as const,
  ipRangeEntries: (id: string) => [...modulesKeys.all, 'ip-range-entries', id] as const,
}

export function modulesListQueryOptions() {
  return queryOptions<ModulesResponse>({
    queryKey: modulesKeys.list(),
    queryFn: () => apiJSON<ModulesResponse>('/v1/modules?limit=200'),
  })
}

export function moduleDetailQueryOptions(id: string) {
  return queryOptions<ModuleRow>({
    queryKey: modulesKeys.detail(id),
    queryFn: () => apiJSON<ModuleRow>(`/v1/modules/${id}`),
  })
}

export type ModuleEntriesPage = Page<Record<string, unknown>>

export function moduleEntriesQueryOptions(id: string, type: ModuleRow['type']) {
  const pathByType: Record<ModuleRow['type'], string> = {
    DOMAINS: `/v1/modules/${id}/domain-entries?limit=500`,
    AS_PREFIXES: `/v1/modules/${id}/as-entries?limit=500`,
    CDN_CIDRS: `/v1/modules/${id}/cdn-sources?limit=500`,
    IP_RANGES: `/v1/modules/${id}/ip-range-entries?limit=500`,
  }
  const path = pathByType[type]
  const keyByType: Record<ModuleRow['type'], readonly string[]> = {
    DOMAINS: modulesKeys.domainEntries(id),
    AS_PREFIXES: modulesKeys.asEntries(id),
    CDN_CIDRS: modulesKeys.cdnSources(id),
    IP_RANGES: modulesKeys.ipRangeEntries(id),
  }
  return queryOptions<ModuleEntriesPage>({
    queryKey: keyByType[type],
    queryFn: () => apiJSON<ModuleEntriesPage>(path),
  })
}