import { TransactionRequest } from '@ethersproject/providers'
import { ChainId } from '@sushiswap/chain'
import { Amount, Type } from '@sushiswap/currency'
import { ZERO } from '@sushiswap/math'
import { NotificationData } from '@sushiswap/ui'
import { Dispatch, SetStateAction, useCallback } from 'react'
import { useAccount } from 'wagmi'
import { SendTransactionResult } from 'wagmi/actions'

import { useSendTransaction } from './useSendTransaction'
import { useWETH9Contract } from './useWETH9Contract'

export enum WrapType {
  Wrap = 'Wrap',
  Unwrap = 'Unwrap',
}

type UseWrapCallbackParams = {
  chainId: ChainId | undefined
  wrapType: WrapType
  onSuccess?(data: NotificationData): void
  amount: Amount<Type> | undefined
}

type UseWrapCallback = (params: UseWrapCallbackParams) => ReturnType<typeof useSendTransaction>

export const useWrapCallback: UseWrapCallback = ({ chainId, wrapType, amount, onSuccess }) => {
  const { address } = useAccount()
  const contract = useWETH9Contract(chainId)

  const onSettled = useCallback(
    (data: SendTransactionResult | undefined) => {
      if (data && onSuccess && amount && chainId) {
        const ts = new Date().getTime()
        onSuccess({
          type: wrapType === WrapType.Wrap ? 'enterBar' : 'leaveBar',
          chainId,
          txHash: data.hash,
          promise: data.wait(),
          summary: {
            pending: `${wrapType === WrapType.Wrap ? 'Wrapping' : 'Unwrapping'} ${amount.toSignificant(6)} ${
              amount.currency.symbol
            }`,
            completed: `Successfully ${wrapType === WrapType.Wrap ? 'wrapped' : 'unwrapped'} ${amount.toSignificant(
              6
            )} ${amount.currency.symbol}`,
            failed: `Something went wrong when trying to ${wrapType === WrapType.Wrap ? 'wrap' : 'unwrap'}`,
          },
          groupTimestamp: ts,
          timestamp: ts,
        })
      }
    },
    [amount, chainId, onSuccess, wrapType]
  )

  const prepare = useCallback(
    (setRequest: Dispatch<SetStateAction<Partial<TransactionRequest & { to: string }>>>) => {
      if (!contract || !chainId || !address || !amount || !amount.greaterThan(ZERO)) return

      if (wrapType === WrapType.Wrap) {
        setRequest({
          from: address,
          to: contract.address,
          data: contract.interface.encodeFunctionData('deposit'),
          value: amount.quotient.toString(),
        })
      }

      if (wrapType === WrapType.Unwrap) {
        setRequest({
          from: address,
          to: contract.address,
          data: contract.interface.encodeFunctionData('withdraw', [amount.quotient.toString()]),
        })
      }
    },
    [address, amount, chainId, contract, wrapType]
  )

  return useSendTransaction({
    chainId,
    prepare,
    onSettled,
    enabled: contract && chainId && address && amount && amount?.greaterThan(ZERO),
  })
}