import { Bank, Serum3Market } from '@blockworks-foundation/mango-v4'
import useJupiterMints from 'hooks/useJupiterMints'
import { QuestionMarkCircleIcon } from '@heroicons/react/20/solid'
import mangoStore from '@store/mangoStore'
import Decimal from 'decimal.js'
import useMangoAccount from 'hooks/useMangoAccount'
import { useViewport } from 'hooks/useViewport'
import { useTranslation } from 'next-i18next'
import Image from 'next/legacy/image'
import { useRouter } from 'next/router'
import { useMemo } from 'react'
import {
  floorToDecimal,
  formatDecimal,
  formatFixedDecimals,
  getDecimalCount,
  trimDecimals,
} from 'utils/numbers'
import { breakpoints } from 'utils/theme'
import { calculateMarketPrice } from 'utils/tradeForm'
import { LinkButton } from './Button'
import { Table, Td, Th, TrBody, TrHead } from './TableElements'
import useSelectedMarket from 'hooks/useSelectedMarket'
import useMangoGroup from 'hooks/useMangoGroup'

const BalancesTable = () => {
  const { t } = useTranslation(['common', 'trade'])
  const { mangoAccount } = useMangoAccount()
  const spotBalances = mangoStore((s) => s.mangoAccount.spotBalances)
  const { group } = useMangoGroup()
  const { mangoTokens } = useJupiterMints()
  const { width } = useViewport()
  const showTableView = width ? width > breakpoints.md : false

  const banks = useMemo(() => {
    if (!group) return []

    const rawBanks = Array.from(group?.banksMapByName, ([key, value]) => ({
      key,
      value,
      balance: mangoAccount?.getTokenBalanceUi(value[0]),
    }))
    const sortedBanks = mangoAccount
      ? rawBanks
          .sort(
            (a, b) =>
              Math.abs(b.balance! * b.value[0].uiPrice) -
              Math.abs(a.balance! * a.value[0].uiPrice)
          )
          .filter(
            (c) =>
              Math.abs(
                floorToDecimal(c.balance!, c.value[0].mintDecimals).toNumber()
              ) > 0
          )
      : rawBanks

    return sortedBanks
  }, [group, mangoAccount])

  return banks?.length ? (
    showTableView ? (
      <Table>
        <thead>
          <TrHead>
            <Th className="bg-th-bkg-1 text-left">{t('token')}</Th>
            <Th className="bg-th-bkg-1 text-right">{t('balance')}</Th>
            <Th className="bg-th-bkg-1 text-right">{t('trade:in-orders')}</Th>
            <Th className="bg-th-bkg-1 text-right" id="trade-step-ten">
              {t('trade:unsettled')}
            </Th>
          </TrHead>
        </thead>
        <tbody>
          {banks.map(({ key, value }) => {
            const bank = value[0]

            let logoURI
            if (mangoTokens.length) {
              logoURI = mangoTokens.find(
                (t) => t.address === bank.mint.toString()
              )?.logoURI
            }

            const inOrders = spotBalances[bank.mint.toString()]?.inOrders || 0
            const unsettled = spotBalances[bank.mint.toString()]?.unsettled || 0

            return (
              <TrBody key={key} className="text-sm">
                <Td>
                  <div className="flex items-center">
                    <div className="mr-2.5 flex flex-shrink-0 items-center">
                      {logoURI ? (
                        <Image alt="" width="20" height="20" src={logoURI} />
                      ) : (
                        <QuestionMarkCircleIcon className="h-7 w-7 text-th-fgd-3" />
                      )}
                    </div>
                    <span>{bank.name}</span>
                  </div>
                </Td>
                <Td className="text-right">
                  <Balance bank={bank} />
                  <p className="text-sm text-th-fgd-4">
                    {mangoAccount
                      ? `${formatFixedDecimals(
                          mangoAccount.getTokenBalanceUi(bank) * bank.uiPrice,
                          true
                        )}`
                      : '$0.00'}
                  </p>
                </Td>
                <Td className="text-right font-mono">
                  <p>{inOrders}</p>
                  <p className="text-sm text-th-fgd-4">
                    {formatFixedDecimals(inOrders * bank.uiPrice, true)}
                  </p>
                </Td>
                <Td className="text-right font-mono">
                  <p>{unsettled ? unsettled.toFixed(bank.mintDecimals) : 0}</p>
                  <p className="text-sm text-th-fgd-4">
                    {formatFixedDecimals(unsettled * bank.uiPrice, true)}
                  </p>
                </Td>
              </TrBody>
            )
          })}
        </tbody>
      </Table>
    ) : (
      <>
        {banks.map(({ key, value }) => {
          const bank = value[0]

          let logoURI
          if (mangoTokens.length) {
            logoURI = mangoTokens.find(
              (t) => t.address === bank.mint.toString()
            )?.logoURI
          }

          const inOrders = spotBalances[bank.mint.toString()]?.inOrders || 0
          const unsettled = spotBalances[bank.mint.toString()]?.unsettled || 0

          return (
            <div
              className="flex items-center justify-between border-b border-th-bkg-3 p-4"
              key={key}
            >
              <div className="flex items-center">
                <div className="mr-2.5 flex flex-shrink-0 items-center">
                  {logoURI ? (
                    <Image alt="" width="20" height="20" src={logoURI} />
                  ) : (
                    <QuestionMarkCircleIcon className="h-7 w-7 text-th-fgd-3" />
                  )}
                </div>
                <span>{bank.name}</span>
              </div>
              <div className="text-right">
                <div className="mb-0.5 flex justify-end space-x-1.5">
                  <Balance bank={bank} />
                  <span className="text-sm text-th-fgd-4">
                    (
                    {mangoAccount
                      ? `${formatFixedDecimals(
                          mangoAccount.getTokenBalanceUi(bank) * bank.uiPrice,
                          true
                        )}`
                      : '$0.00'}
                    )
                  </span>
                </div>
                <div className="flex space-x-2">
                  <p className="text-xs text-th-fgd-4">
                    {t('trade:in-orders')}:{' '}
                    <span className="font-mono text-th-fgd-3">{inOrders}</span>
                  </p>
                  <p className="text-xs text-th-fgd-4">
                    {t('trade:unsettled')}:{' '}
                    <span className="font-mono text-th-fgd-3">
                      {unsettled ? unsettled.toFixed(bank.mintDecimals) : 0}
                    </span>
                  </p>
                </div>
              </div>
            </div>
          )
        })}
      </>
    )
  ) : (
    <div className="flex flex-col items-center p-8">
      <p>{t('trade:no-balances')}</p>
    </div>
  )
}

export default BalancesTable

const Balance = ({ bank }: { bank: Bank }) => {
  const { mangoAccount } = useMangoAccount()
  const { selectedMarket } = useSelectedMarket()
  const { asPath } = useRouter()

  const handleBalanceClick = (balance: number, type: 'base' | 'quote') => {
    const set = mangoStore.getState().set
    const group = mangoStore.getState().group
    const tradeForm = mangoStore.getState().tradeForm

    if (!group || !selectedMarket) return

    let price: number
    if (tradeForm.tradeType === 'Market') {
      const orderbook = mangoStore.getState().selectedMarket.orderbook
      const side =
        (balance > 0 && type === 'quote') || (balance < 0 && type === 'base')
          ? 'buy'
          : 'sell'
      price = calculateMarketPrice(orderbook, balance, side)
    } else {
      price = new Decimal(tradeForm.price).toNumber()
    }
    let minOrderDecimals: number
    let tickSize: number
    if (selectedMarket instanceof Serum3Market) {
      const market = group.getSerum3ExternalMarket(
        selectedMarket.serumMarketExternal
      )
      minOrderDecimals = getDecimalCount(market.minOrderSize)
      tickSize = getDecimalCount(market.tickSize)
    } else {
      minOrderDecimals = getDecimalCount(selectedMarket.minOrderSize)
      tickSize = getDecimalCount(selectedMarket.tickSize)
    }

    if (type === 'quote') {
      const trimmedBalance = trimDecimals(balance, tickSize)
      const baseSize = trimDecimals(trimmedBalance / price, minOrderDecimals)
      const quoteSize = trimDecimals(baseSize * price, tickSize)
      set((s) => {
        s.tradeForm.baseSize = baseSize.toString()
        s.tradeForm.quoteSize = quoteSize.toString()
      })
    } else {
      const baseSize = trimDecimals(balance, minOrderDecimals)
      const quoteSize = trimDecimals(baseSize * price, tickSize)
      set((s) => {
        s.tradeForm.baseSize = baseSize.toString()
        s.tradeForm.quoteSize = quoteSize.toString()
      })
    }
  }

  const balance = useMemo(() => {
    return mangoAccount ? mangoAccount.getTokenBalanceUi(bank) : 0
  }, [mangoAccount])

  const isBaseOrQuote = useMemo(() => {
    if (selectedMarket instanceof Serum3Market && asPath.includes('/trade')) {
      if (bank.tokenIndex === selectedMarket.baseTokenIndex) {
        return 'base'
      } else if (bank.tokenIndex === selectedMarket.quoteTokenIndex) {
        return 'quote'
      } else return ''
    }
  }, [bank, selectedMarket])

  return (
    <p className="flex justify-end">
      {balance ? (
        isBaseOrQuote ? (
          <LinkButton
            className="font-normal"
            onClick={() =>
              handleBalanceClick(
                parseFloat(formatDecimal(balance, bank.mintDecimals)),
                isBaseOrQuote
              )
            }
          >
            {formatDecimal(balance, bank.mintDecimals)}
          </LinkButton>
        ) : (
          formatDecimal(balance, bank.mintDecimals)
        )
      ) : (
        0
      )}
    </p>
  )
}