use-update-access-code.ts

import type { AccessCodesUpdateParameters } from '@seamapi/http/connect'
import type { AccessCode } from '@seamapi/types/connect'
import { useQueryClient } from '@tanstack/react-query'

import {
  useSeamMutation,
  type UseSeamMutationResult,
} from '../use-seam-mutation.js'

export type UseUpdateAccessCodeParams = never

export type UseUpdateAccessCodeData = undefined

export type UseUpdateAccessCodeMutationVariables = Pick<
  AccessCodesUpdateParameters,
  | 'device_id'
  | 'access_code_id'
  | 'code'
  | 'name'
  | 'starts_at'
  | 'ends_at'
  | 'type'
>

export function useUpdateAccessCode(): UseSeamMutationResult<'/access_codes/update'> {
  const queryClient = useQueryClient()

  return useSeamMutation('/access_codes/update', {
    onSuccess: (_data, variables) => {
      if (variables == null) return

      queryClient.setQueryData<AccessCode | null>(
        ['access_codes', 'get', { access_code_id: variables.access_code_id }],
        (accessCode) => {
          if (accessCode == null) {
            return
          }

          return {
            ...accessCode,
            code: variables.code ?? accessCode.code,
            name: variables.name ?? accessCode.name,
            starts_at: variables.starts_at ?? accessCode.starts_at,
            ends_at: variables.ends_at ?? accessCode.ends_at,
            type: variables.type ?? accessCode.type,
            status: 'setting',
          }
        }
      )

      queryClient.setQueryData<AccessCode[]>(
        ['access_codes', 'list'],
        (accessCodes) => {
          if (accessCodes == null) {
            return accessCodes
          }

          return accessCodes.map((accessCode) => {
            if (accessCode.access_code_id === variables.access_code_id) {
              return {
                ...accessCode,
                code: variables.code ?? accessCode.code,
                name: variables.name ?? accessCode.name,
                status: 'setting',
              }
            }

            return accessCode
          })
        }
      )
    },
  })
}