@roochnetwork/rooch-sdk-kit/src/hooks/wallet/useConnectWallet.ts

Rooch SDK Kit

// Copyright (c) RoochNetwork
// SPDX-License-Identifier: Apache-2.0

import type { UseMutationOptions, UseMutationResult } from '@tanstack/react-query'
import { useMutation } from '@tanstack/react-query'
import { ThirdPartyAddress, Session } from '@roochnetwork/rooch-sdk'

import { useWalletStore } from './useWalletStore.js'
import { walletMutationKeys } from '../../constants/index.js'
import { Wallet } from '../../wellet/index.js'
import { useSessionStore } from '../useSessionsStore.js'
import { useSessions } from '../useSessions.js'
import { useTriggerError } from '../../provider/globalProvider.js'

type ConnectWalletArgs = {
  wallet: Wallet
}
type ConnectWalletResult = ThirdPartyAddress[]

type UseConnectWalletMutationOptions = Omit<
  UseMutationOptions<ConnectWalletResult, Error, ConnectWalletArgs, unknown>,
  'mutationFn'
>

/**
 * Mutation hook for establishing a connection to a specific wallet.
 *
 */
export function useConnectWallet({
  mutationKey,
  ...mutationOptions
}: UseConnectWalletMutationOptions = {}): UseMutationResult<
  ConnectWalletResult,
  Error,
  ConnectWalletArgs,
  unknown
> {
  const sessions = useSessions()
  const setCurrentSession = useSessionStore((state) => state.setCurrentSession)
  const setWalletConnected = useWalletStore((state) => state.setWalletConnected)
  const setWalletDisconnected = useWalletStore((state) => state.setWalletDisconnected)
  const setConnectionStatus = useWalletStore((state) => state.setConnectionStatus)
  const triggerError = useTriggerError()
  return useMutation({
    mutationKey: walletMutationKeys.connectWallet(mutationKey),
    mutationFn: async ({ wallet }) => {
      try {
        setConnectionStatus('connecting')

        const connectAddress = await wallet.connect()
        const selectedAddress = connectAddress[0]

        setWalletConnected(wallet, connectAddress, selectedAddress)

        const cur = sessions.find(
          (item: Session) =>
            item.getRoochAddress().toStr() === selectedAddress?.genRoochAddress().toStr(),
        )

        setCurrentSession(cur)

        return connectAddress
      } catch (error: any) {
        // TODO: one key, A retry will be performed after the rejection
        setWalletDisconnected()
        if ('code' in error && 'message' in error) {
          triggerError(error)
        }
        throw error
      }
    },
    ...mutationOptions,
  })
}