import {
  type Config,
  type GetChainIdReturnType,
  getChainId,
  type ResolvedRegister,
  watchChainId,
} from '@wagmi/core'
import type { ConfigParameter } from '@wagmi/core/internal'
import { onScopeDispose, type Ref, readonly, ref } from 'vue'
import { useConfig } from './useConfig.js'

export type UseChainIdParameters<config extends Config = Config> =
  ConfigParameter<config>

export type UseChainIdReturnType<config extends Config = Config> = Ref<
  GetChainIdReturnType<config>
>

/** https://wagmi.sh/vue/api/composables/useChainId */
export function useChainId<config extends Config = ResolvedRegister['config']>(
  parameters: UseChainIdParameters<config> = {},
): UseChainIdReturnType<config> {
  const config = useConfig(parameters)

  const chainId = ref<GetChainIdReturnType>(getChainId(config))
  const unsubscribe = watchChainId(config, {
    onChange(data) {
      chainId.value = data
    },
  })
  onScopeDispose(() => unsubscribe())

  return readonly(chainId)
}