EvoBGP/apps/web/src/queries/api-keys.ts

import { queryOptions, useMutation, useQueryClient } from '@tanstack/react-query'
import { toast } from 'sonner'

import { apiJSON, apiMutate } from '@/lib/api-client'
import type { ApiKey, ApiKeyCreate, ApiKeyCreated, ApiKeysResponse } from '@/types/api'

export const apiKeysKeys = {
  all: ['api-keys'] as const,
  list: () => [...apiKeysKeys.all, 'list'] as const,
}

export function apiKeysQueryOptions() {
  return queryOptions<ApiKey[]>({
    queryKey: apiKeysKeys.list(),
    queryFn: async () => {
      const page = await apiJSON<ApiKeysResponse>('/v1/api-keys?limit=500')
      return page.items ?? []
    },
    staleTime: 60_000,
  })
}

function invalidateApiKeysList(qc: ReturnType<typeof useQueryClient>) {
  void qc.invalidateQueries({ queryKey: apiKeysKeys.list() })
}

export function useCreateApiKeyMutation() {
  const qc = useQueryClient()
  return useMutation({
    mutationFn: (body: ApiKeyCreate) =>
      apiMutate<ApiKeyCreated>('/v1/api-keys', 'POST', body, { idempotent: false }),
    onSuccess: () => {
      toast.success('Ключ создан')
      invalidateApiKeysList(qc)
    },
    onError: (e) => toast.error(e instanceof Error ? e.message : 'Не удалось создать ключ'),
  })
}

export function useRevokeApiKeyMutation() {
  const qc = useQueryClient()
  return useMutation({
    mutationFn: (id: string) =>
      apiMutate(`/v1/api-keys/${id}`, 'DELETE', undefined, { idempotent: false }),
    onSuccess: () => {
      toast.success('Ключ отозван')
      invalidateApiKeysList(qc)
    },
    onError: (e) => toast.error(e instanceof Error ? e.message : 'Не удалось отозвать'),
  })
}

export function useRotateApiKeyMutation() {
  const qc = useQueryClient()
  return useMutation({
    mutationFn: (id: string) =>
      apiMutate<ApiKeyCreated>(`/v1/api-keys/${id}/rotate`, 'POST', undefined, {
        idempotent: false,
      }),
    onSuccess: () => {
      toast.success('Ключ ротирован')
      invalidateApiKeysList(qc)
    },
    onError: (e) => toast.error(e instanceof Error ? e.message : 'Не удалось ротировать'),
  })
}