kodadot/nft-gallery

import { web3Enable } from '@polkadot/extension-dapp'
import type { SubmittableResult } from '@polkadot/api'
import {
  Chain,
  allowedTransitions,
  chainToPrefixMap,
  prefixToChainMap,
} from '@/utils/teleport'
import { getss58AddressByPrefix } from '@/utils/account'
import { txCb } from '@/utils/transactionExecutor'

export type TeleportParams = {
  amount: number
  from: Chain
  to: Chain
  fromAddress: string
  toAddress: string
  currency: string
}

export default function (fetchBalancePeriodically: boolean = false) {
  const isError = ref<boolean>(false)
  const txId = ref<string | null>(null)

  const { isLoading, status, initTransactionLoader, stopLoader }
    = useTransactionStatus()
  const { accountId } = useAuth()
  const { assets } = usePrefix()
  const { $i18n, $consola } = useNuxtApp()
  const { decimalsOf } = useChain()
  const { urlPrefix } = usePrefix()
  const { fetchMultipleBalance, chainBalances } = useMultipleBalance(
    fetchBalancePeriodically,
  )

  const chain = computed<Chain | null>(
    () => prefixToChainMap[urlPrefix.value] || null,
  )
  const isAvailable = computed(() =>
    Object.keys(allowedTransitions).includes(chain.value || ''),
  )

  const teleport = async ({
    amount,
    from,
    to,
    fromAddress,
    toAddress,
    currency,
    onSuccess,
    onError,
  }: TeleportParams & {
    onSuccess?: (blockHash) => void
    onError?: () => void
  }) => {
    if (!amount || amount < 0) {
      return
    }

    await web3Enable('Kodadot')
    let isFirstStatus = true
    initTransactionLoader()
    isError.value = false

    const transactionHandler = txCb(
      ({ blockHash }) => {
        $consola.log(`Transaction finalized at blockHash ${blockHash}`)
        status.value = TransactionStatus.Finalized
        isLoading.value = false

        if (onSuccess) {
          onSuccess(blockHash)
        }
      },
      (dispatchError) => {
        warningMessage(dispatchError.toString())
        stopLoader()
        isError.value = true

        if (onError) {
          onError()
        }
      },
      (submittableResult: SubmittableResult) => {
        const { txHash } = submittableResult

        if (isFirstStatus) {
          $consola.log(`Transaction hash is ${txHash.toHex()}`)
          txId.value = txHash.toHex()
          status.value = TransactionStatus.Block
          isFirstStatus = false
        }
      },
    )

    const errorHandler = () => {
      warningMessage($i18n.t('general.tx.cancelled'), { reportable: false })

      isLoading.value = false
      status.value = TransactionStatus.Cancelled
      isError.value = true

      if (onError) {
        onError()
      }
    }

    const promise = await getTransaction({
      amount: amount,
      from: from,
      to: to,
      address: toAddress,
      currency: currency,
    })

    if (promise === undefined) {
      errorHandler()
      return
    }

    const injector = await getAddress(toDefaultAddress(fromAddress))

    promise
      .signAndSend(fromAddress, { signer: injector.signer }, transactionHandler)
      .catch(errorHandler)
  }

  const getAddressByChain = (chain: Chain) => {
    return (
      accountId.value
      && getss58AddressByPrefix(accountId.value, chainToPrefixMap[chain])
    )
  }

  const getChainTokenDecimals = (chain: Chain) => {
    switch (chain) {
      case Chain.KUSAMA:
      case Chain.ASSETHUBKUSAMA:
        return assets(5).decimals
      case Chain.POLKADOT:
        return decimalsOf('dot')
      case Chain.ASSETHUBPOLKADOT:
        return decimalsOf('ahp')
    }
  }

  const fetchChainsBalances = (chains: Chain[]) => {
    const chainPrefixes = chains.map(chain => chainToPrefixMap[chain])
    return fetchMultipleBalance(chainPrefixes)
  }

  return {
    chain,
    txId,
    status,
    isError,
    isLoading,
    isAvailable,
    chainBalances,
    teleport,
    getAddressByChain,
    getChainTokenDecimals,
    fetchChainsBalances,
  }
}