use-update-ai-copilot-config.tsx

import { useMutation, useQueryClient } from '@tanstack/react-query'
import posthog from 'posthog-js'
import { type AICopilotConfigResponse, devopsCopilot, mutations } from '@qovery/shared/devops-copilot/data-access'
import { toast } from '@qovery/shared/ui'

export interface UseUpdateAICopilotConfigProps {
  organizationId: string
  instructions?: string
}

export function useUpdateAICopilotConfig({ organizationId, instructions }: UseUpdateAICopilotConfigProps) {
  const queryClient = useQueryClient()

  return useMutation({
    mutationFn: ({
      enabled,
      readOnly,
      userEmail,
      tokenMode,
    }: {
      enabled: boolean
      readOnly: boolean
      userEmail?: string
      tokenMode?: 'jwt' | 'role'
    }) =>
      mutations.updateOrgConfig({
        organizationId,
        enabled,
        readOnly,
        instructions: instructions || '',
        userEmail,
        tokenMode,
      }),
    onMutate: async ({ readOnly, tokenMode }) => {
      await queryClient.cancelQueries({
        queryKey: devopsCopilot.config({ organizationId }).queryKey,
      })

      const previousConfig = queryClient.getQueryData(devopsCopilot.config({ organizationId }).queryKey)

      queryClient.setQueryData(
        devopsCopilot.config({ organizationId }).queryKey,
        (old: AICopilotConfigResponse | undefined) =>
          old
            ? {
                ...old,
                org_config: {
                  ...old.org_config,
                  read_only: readOnly,
                  ...(tokenMode !== undefined && { token_mode: tokenMode }),
                },
              }
            : old
      )

      return { previousConfig }
    },
    onSuccess: (_data, variables) => {
      if (variables.tokenMode !== undefined) {
        toast('success', 'Access source updated successfully')
        posthog.capture('ai-copilot-token-mode-changed', {
          organization_id: organizationId,
          token_mode: variables.tokenMode,
        })
      } else if (variables.enabled) {
        const modeName = variables.readOnly ? 'Read-Only' : 'Read-Write'
        toast('success', `AI Copilot enabled with ${modeName} mode`)
        posthog.capture('ai-copilot-enabled', {
          organization_id: organizationId,
          mode: modeName,
        })
      } else {
        toast('success', 'AI Copilot disabled successfully')
        posthog.capture('ai-copilot-disabled', {
          organization_id: organizationId,
        })
      }
    },
    onError: (err, _variables, context) => {
      if (context?.previousConfig) {
        queryClient.setQueryData(devopsCopilot.config({ organizationId }).queryKey, context.previousConfig)
      }
      const axiosErr = err as { response?: { data?: { error?: string } } }
      const message = axiosErr?.response?.data?.error ?? (err instanceof Error ? err.message : 'Please try again later')
      toast('error', 'Failed to update AI Copilot configuration', message)
    },
    onSettled: () => {
      queryClient.invalidateQueries({
        queryKey: devopsCopilot.config({ organizationId }).queryKey,
      })
    },
  })
}

export default useUpdateAICopilotConfig