import type { QueryOptions } from '@tanstack/query-core'

import {
  type GetTokenErrorType,
  type GetTokenParameters,
  type GetTokenReturnType,
  getToken,
} from '../actions/getToken.js'
import type { Config } from '../createConfig.js'
import type { ScopeKeyParameter } from '../types/properties.js'
import type { Compute, ExactPartial } from '../types/utils.js'
import { filterQueryOptions } from './utils.js'

export type GetTokenOptions<config extends Config> = Compute<
  ExactPartial<GetTokenParameters<config>> & ScopeKeyParameter
>

export function getTokenQueryOptions<config extends Config>(
  config: config,
  options: GetTokenOptions<config> = {},
) {
  return {
    async queryFn({ queryKey }) {
      const { address, scopeKey: _, ...parameters } = queryKey[1]
      if (!address) throw new Error('address is required')
      return getToken(config, { ...parameters, address })
    },
    queryKey: getTokenQueryKey(options),
  } as const satisfies QueryOptions<
    GetTokenQueryFnData,
    GetTokenErrorType,
    GetTokenData,
    GetTokenQueryKey<config>
  >
}

export type GetTokenQueryFnData = GetTokenReturnType

export type GetTokenData = GetTokenQueryFnData

export function getTokenQueryKey<config extends Config>(
  options: GetTokenOptions<config> = {},
) {
  return ['token', filterQueryOptions(options)] as const
}

export type GetTokenQueryKey<config extends Config> = ReturnType<
  typeof getTokenQueryKey<config>
>