import invariant from 'tiny-invariant'
import { type Address, toHex } from 'viem'
import { DEPOSIT_BALANCE_ROUTE } from '../constants.js'
import type { RenegadeConfig } from '../createConfig.js'
import { stringifyForWasm } from '../utils/bigJSON.js'
import { postRelayerWithAuth } from '../utils/http.js'
import { getBackOfQueueWallet } from './getBackOfQueueWallet.js'
import { getWalletId } from './getWalletId.js'

export type DepositParameters = {
  amount: bigint
  fromAddr: Address
  mint: Address
  newPublicKey?: string
  permit: string
  permitDeadline: bigint
  permitNonce: bigint
}

export type DepositReturnType = Promise<{ taskId: string }>

export async function deposit(
  config: RenegadeConfig,
  parameters: DepositParameters,
): DepositReturnType {
  const {
    amount,
    fromAddr,
    mint,
    newPublicKey,
    permit,
    permitDeadline,
    permitNonce,
  } = parameters

  const { getBaseUrl, utils, renegadeKeyType } = config

  const walletId = getWalletId(config)
  const wallet = await getBackOfQueueWallet(config)

  const seed = renegadeKeyType === 'internal' ? config.state.seed : undefined
  const signMessage =
    renegadeKeyType === 'external' ? config.signMessage : undefined

  if (renegadeKeyType === 'external') {
    invariant(
      signMessage !== undefined,
      'Sign message function is required for external key type',
    )
  }
  if (renegadeKeyType === 'internal') {
    invariant(seed !== undefined, 'Seed is required for internal key type')
  }

  const body = await utils.deposit(
    // TODO: Change Rust to accept Option<String>
    seed,
    stringifyForWasm(wallet),
    fromAddr,
    mint,
    toHex(amount),
    toHex(permitNonce),
    toHex(permitDeadline),
    permit,
    newPublicKey,
    signMessage,
  )

  try {
    const res = await postRelayerWithAuth(
      config,
      getBaseUrl(DEPOSIT_BALANCE_ROUTE(walletId)),
      body,
    )
    console.log(`task update-wallet(${res.task_id}): ${walletId}`)
    return { taskId: res.task_id }
  } catch (error) {
    console.error(`${walletId}`, {
      error,
    })
    throw error
  }
}