import type { Config } from '../createConfig.js'
import type { BaseErrorType } from '../errors/base.js'
import { createWallet } from './createWallet.js'
import { getWalletFromRelayer } from './getWalletFromRelayer.js'
import { getWalletId } from './getWalletId.js'
import { checkForWalletUpdatesOnChain, lookupWallet } from './lookupWallet.js'

export type ConnectParameters = {
  isCreateWallet?: boolean
}

export type ConnectReturnType = {
  isLookup: boolean
  job: Promise<void>
} | void

export type ConnectErrorType = BaseErrorType

export async function connect(
  config: Config,
  params: ConnectParameters = {},
): Promise<ConnectReturnType> {
  try {
    const { isCreateWallet = false } = params
    const walletId = getWalletId(config)
    config.setState((x) => ({ ...x, id: walletId }))

    console.log('Attempting to connect wallet', { walletId: config.state.id })

    try {
      const wallet = await getWalletFromRelayer(config)
      if (wallet) {
        config.setState((x) => ({ ...x, status: 'in relayer' }))
        console.log('Wallet found in relayer', {
          status: 'in relayer',
          walletId: config.state.id,
        })
        return
      }
    } catch (error) {
      console.error('Wallet not found in relayer', {
        error,
        walletId: config.state.id,
      })
    }

    // Create wallet iff no WalletUpdated events found onchain
    const shouldCreateWallet = await checkForWalletUpdatesOnChain(config)
    if (shouldCreateWallet || isCreateWallet) {
      return Promise.resolve({
        isLookup: false,
        job: createWallet(config),
      })
    }
    return Promise.resolve({
      isLookup: true,
      job: lookupWallet(config),
    })
  } catch (error) {
    console.error('Could not connect wallet', {
      error,
      walletId: config.state.id,
    })
    config.setState({})
  }
}