import invariant from 'tiny-invariant'
import { parseAbiItem } from 'viem'
import { FIND_WALLET_ROUTE } from '../constants.js'
import type { Config, RenegadeConfig } from '../createConfig.js'
import { BaseError } from '../errors/base.js'
import { postRelayerRaw } from '../utils/http.js'
import { waitForWalletIndexing } from './waitForWalletIndexing.js'

export type LookupWalletReturnType = ReturnType<typeof waitForWalletIndexing>

export type LookupWalletParameters = {
  blinderSeed?: string
  shareSeed?: string
  skMatch?: string
}

export async function lookupWallet(
  config: RenegadeConfig,
  parameters: LookupWalletParameters = {},
): LookupWalletReturnType {
  const { getBaseUrl, utils } = config
  let body: string

  if (config.renegadeKeyType === 'internal') {
    const { seed } = config.state
    invariant(seed, 'seed is required')
    body = utils.find_wallet(seed)
  } else {
    const { blinderSeed, shareSeed, skMatch } = parameters
    const { walletId, publicKey, symmetricKey } = config
    invariant(blinderSeed, 'blinderSeed is required')
    invariant(shareSeed, 'shareSeed is required')
    invariant(skMatch, 'skMatch is required')
    body = await utils.find_external_wallet(
      walletId,
      blinderSeed,
      shareSeed,
      publicKey,
      skMatch,
      symmetricKey,
    )
  }
  const res = await postRelayerRaw(getBaseUrl(FIND_WALLET_ROUTE), body)

  if (res.task_id) {
    console.log(`task lookup-wallet(${res.task_id}): ${res.wallet_id}`, {
      status: 'looking up',
      walletId: res.wallet_id,
    })
    if (config.renegadeKeyType === 'internal') {
      config.setState((x) => ({ ...x, status: 'looking up' }))
    }
    return waitForWalletIndexing(config, {
      timeout: 300000,
      isLookup: true,
      onComplete(wallet) {
        if (config.renegadeKeyType === 'internal') {
          config.setState((x) => ({
            ...x,
            status: 'in relayer',
          }))
        }
        console.log(
          `task lookup-wallet(${res.task_id}) completed: ${wallet.id}`,
          {
            status: 'in relayer',
            walletId: wallet.id,
          },
        )
      },
      onFailure() {
        console.log(
          `task lookup-wallet(${res.task_id}) failed: ${res.wallet_id}`,
        )
        if (config.renegadeKeyType === 'internal') {
          config.setState({})
        }
      },
    })
  }
  return Promise.reject(new BaseError('Failed to lookup wallet'))
}

// Returns true iff the query successfully returns 0 logs
export async function checkForWalletUpdatesOnChain(config: Config) {
  try {
    const {
      utils,
      state: { seed },
    } = config
    invariant(seed, 'Seed is required')
    const blinderShare = utils.derive_blinder_share(seed)

    const logs = await config.viemClient.getLogs({
      address: config.darkPoolAddress,
      event: parseAbiItem(
        'event WalletUpdated(uint256 indexed wallet_blinder_share)',
      ),
      args: {
        wallet_blinder_share: blinderShare,
      },
      fromBlock: BigInt(
        process.env.FROM_BLOCK || process.env.NEXT_PUBLIC_FROM_BLOCK || 0,
      ),
    })
    return logs.length === 0
  } catch (error) {
    console.error(`Error looking up wallet on chain: ${error}`)
    return false
  }
}