import { useMemo } from 'react'
import FloatingElement from './FloatingElement'
import { ElementTitle } from './styles'
import useMangoStore, { mangoClient } from '../stores/useMangoStore'
import {
  ceilToDecimal,
  floorToDecimal,
  i80f48ToPercent,
  formatUsdValue,
  tokenPrecision,
} from '../utils/index'
import { LinkButton } from './Button'
import Tooltip from './Tooltip'
import SideBadge from './SideBadge'
import {
  getMarketIndexBySymbol,
  nativeI80F48ToUi,
  PerpAccount,
  PerpMarket,
  QUOTE_INDEX,
  ZERO_BN,
  ZERO_I80F48,
} from '@blockworks-foundation/mango-client'
import useTradeHistory from '../hooks/useTradeHistory'
import { getAvgEntryPrice, getBreakEvenPrice } from './PerpPositionsTable'
import { notify } from '../utils/notifications'

const handleSettlePnl = async (
  perpMarket: PerpMarket,
  perpAccount: PerpAccount
) => {
  const mangoAccount = useMangoStore.getState().selectedMangoAccount.current
  const mangoGroup = useMangoStore.getState().selectedMangoGroup.current
  const mangoCache = useMangoStore.getState().selectedMangoGroup.cache
  const wallet = useMangoStore.getState().wallet.current
  const actions = useMangoStore.getState().actions
  const marketIndex = mangoGroup.getPerpMarketIndex(perpMarket.publicKey)

  try {
    const txid = await mangoClient.settlePnl(
      mangoGroup,
      mangoCache,
      mangoAccount,
      perpMarket,
      mangoGroup.rootBankAccounts[QUOTE_INDEX],
      mangoCache.priceCache[marketIndex].price,
      wallet
    )
    actions.fetchMangoAccounts()
    notify({
      title: 'Successfully settled PNL',
      description: '',
      txid,
    })
  } catch (e) {
    console.log('Error settling PNL: ', `${e}`, `${perpAccount}`)
    notify({
      title: 'Error settling PNL',
      description: e.message,
      txid: e.txid,
      type: 'error',
    })
  }
}

export function SettlePnlTooltip() {
  return (
    <div>
      Settling will update your USDC balance to reflect the unsettled PnL
      amount.{' '}
      <a
        href="https://docs.mango.markets/mango-v3/overview#settle-pnl"
        target="_blank"
        rel="noopener noreferrer"
      >
        Learn more
      </a>
    </div>
  )
}

export default function MarketPosition() {
  const mangoGroup = useMangoStore((s) => s.selectedMangoGroup.current)
  const mangoGroupConfig = useMangoStore((s) => s.selectedMangoGroup.config)
  const mangoGroupCache = useMangoStore((s) => s.selectedMangoGroup.cache)
  const mangoAccount = useMangoStore((s) => s.selectedMangoAccount.current)
  const selectedMarket = useMangoStore((s) => s.selectedMarket.current)
  const marketConfig = useMangoStore((s) => s.selectedMarket.config)
  const connected = useMangoStore((s) => s.wallet.connected)
  const isLoading = useMangoStore((s) => s.selectedMangoAccount.initialLoad)
  const setMangoStore = useMangoStore((s) => s.set)
  const baseSymbol = marketConfig.baseSymbol
  const marketName = marketConfig.name
  const tradeHistory = useTradeHistory()
  const perpTradeHistory = tradeHistory?.filter(
    (t) => t.marketName === marketName
  )

  const marketIndex = useMemo(() => {
    return getMarketIndexBySymbol(mangoGroupConfig, baseSymbol)
  }, [mangoGroupConfig, baseSymbol])

  const perpAccount = useMemo(() => {
    if (marketName.includes('PERP') && mangoAccount) {
      return mangoAccount.perpAccounts[marketIndex]
    }
  }, [marketName, mangoAccount, marketIndex])

  const handleSizeClick = (size) => {
    setMangoStore((state) => {
      state.tradeForm.baseSize = size
    })
  }

  if (!mangoGroup) return null

  return selectedMarket instanceof PerpMarket ? (
    <FloatingElement showConnect>
      <div className={!connected ? 'filter blur-sm' : null}>
        <ElementTitle>Position</ElementTitle>
        <div className={`flex items-center justify-between pb-3`}>
          <div className="font-normal text-th-fgd-3 leading-4">Side</div>
          {isLoading ? (
            <DataLoader />
          ) : perpAccount && !perpAccount.basePosition.eq(ZERO_BN) ? (
            <SideBadge
              side={perpAccount.basePosition.gt(ZERO_BN) ? 'long' : 'short'}
            />
          ) : (
            '--'
          )}
        </div>
        <div className={`flex justify-between pb-3`}>
          <div className="font-normal text-th-fgd-3 leading-4">
            Position size
          </div>
          <div className={`text-th-fgd-1`}>
            {isLoading ? (
              <DataLoader />
            ) : perpAccount &&
              Math.abs(
                selectedMarket.baseLotsToNumber(perpAccount.basePosition)
              ) > 0 ? (
              <span
                className="cursor-pointer underline hover:no-underline"
                onClick={() =>
                  handleSizeClick(
                    Math.abs(
                      selectedMarket.baseLotsToNumber(perpAccount.basePosition)
                    )
                  )
                }
              >
                {`${Math.abs(
                  selectedMarket.baseLotsToNumber(perpAccount.basePosition)
                )} ${baseSymbol}`}
              </span>
            ) : (
              `0 ${baseSymbol}`
            )}
          </div>
        </div>
        <div className={`flex justify-between pb-3`}>
          <div className="font-normal text-th-fgd-3 leading-4">
            Notional size
          </div>
          <div className={`text-th-fgd-1`}>
            {isLoading ? (
              <DataLoader />
            ) : perpAccount ? (
              formatUsdValue(
                selectedMarket.baseLotsToNumber(perpAccount.basePosition) *
                  mangoGroup.getPrice(marketIndex, mangoGroupCache).toNumber()
              )
            ) : (
              0
            )}
          </div>
        </div>
        <div className={`flex justify-between pb-3`}>
          <div className="font-normal text-th-fgd-3 leading-4">
            Avg entry price
          </div>
          <div className={`text-th-fgd-1`}>
            {isLoading ? (
              <DataLoader />
            ) : perpAccount ? (
              getAvgEntryPrice(
                mangoAccount,
                perpAccount,
                selectedMarket,
                perpTradeHistory
              )
            ) : (
              0
            )}
          </div>
        </div>
        <div className={`flex justify-between pb-3`}>
          <div className="font-normal text-th-fgd-3 leading-4">
            Break-even price
          </div>
          <div className={`text-th-fgd-1`}>
            {isLoading ? (
              <DataLoader />
            ) : perpAccount ? (
              getBreakEvenPrice(
                mangoAccount,
                perpAccount,
                selectedMarket,
                perpTradeHistory
              )
            ) : (
              0
            )}
          </div>
        </div>
        <div className={`flex justify-between pb-3`}>
          <Tooltip content={<SettlePnlTooltip />}>
            <Tooltip.Content className="font-normal text-th-fgd-3 leading-4">
              Unsettled PnL
            </Tooltip.Content>
          </Tooltip>
          <div className={`flex items-center text-th-fgd-1`}>
            {isLoading ? (
              <DataLoader />
            ) : perpAccount ? (
              formatUsdValue(
                +nativeI80F48ToUi(
                  perpAccount.getPnl(
                    mangoGroup.perpMarkets[marketIndex],
                    mangoGroupCache.perpMarketCache[marketIndex],
                    mangoGroupCache.priceCache[marketIndex].price
                  ),
                  marketConfig.quoteDecimals
                )
              )
            ) : (
              '0'
            )}
            <LinkButton
              onClick={() => handleSettlePnl(selectedMarket, perpAccount)}
              className="ml-2 text-th-primary text-xs disabled:cursor-not-allowed disabled:opacity-60 disabled:hover:underline"
              disabled={
                perpAccount
                  ? perpAccount
                      .getPnl(
                        mangoGroup.perpMarkets[marketIndex],
                        mangoGroupCache.perpMarketCache[marketIndex],
                        mangoGroupCache.priceCache[marketIndex].price
                      )
                      .eq(ZERO_I80F48)
                  : true
              }
            >
              Settle
            </LinkButton>
          </div>
        </div>
      </div>
    </FloatingElement>
  ) : (
    <>
      <FloatingElement showConnect>
        <div className={!connected ? 'filter blur' : null}>
          <ElementTitle>Balances</ElementTitle>
          {mangoGroup ? (
            <div className="grid grid-cols-2 grid-rows-1 gap-4 pt-2">
              {mangoGroupConfig.tokens
                .filter((t) => t.symbol === baseSymbol || t.symbol === 'USDC')
                .reverse()
                .map(({ symbol, mintKey }) => {
                  const tokenIndex = mangoGroup.getTokenIndex(mintKey)
                  const deposit = mangoAccount
                    ? mangoAccount
                        .getUiDeposit(
                          mangoGroupCache.rootBankCache[tokenIndex],
                          mangoGroup,
                          tokenIndex
                        )
                        .toNumber()
                    : null
                  const borrow = mangoAccount
                    ? mangoAccount
                        .getUiBorrow(
                          mangoGroupCache.rootBankCache[tokenIndex],
                          mangoGroup,
                          tokenIndex
                        )
                        .toNumber()
                    : null
                  return (
                    <div
                      className="border border-th-bkg-4 p-4 rounded-md"
                      key={mintKey.toString()}
                    >
                      <div className="flex items-center justify-between mb-4">
                        <div className="flex items-center">
                          <img
                            alt=""
                            src={`/assets/icons/${symbol.toLowerCase()}.svg`}
                            className={`h-5 mr-2.5 w-auto`}
                          />
                          <span className="text-th-fgd-2">{symbol}</span>
                        </div>
                      </div>
                      <div className="pb-3">
                        <div className="pb-0.5 text-th-fgd-3 text-xs">
                          Balance
                        </div>
                        <div className={`text-th-fgd-1`}>
                          {isLoading ? (
                            <DataLoader />
                          ) : mangoAccount ? (
                            deposit > 0 ? (
                              floorToDecimal(
                                deposit,
                                mangoGroup.tokens[tokenIndex].decimals
                              ).toFixed(tokenPrecision[symbol])
                            ) : borrow > 0 ? (
                              `-${ceilToDecimal(
                                borrow,
                                mangoGroup.tokens[tokenIndex].decimals
                              ).toFixed(tokenPrecision[symbol])}`
                            ) : (
                              0
                            )
                          ) : (
                            0
                          )}
                        </div>
                      </div>
                      <div className="pb-3">
                        <Tooltip content="Available to withdraw after accounting for collateral and open orders">
                          <div className="pb-0.5 text-th-fgd-3 text-xs">
                            Available Balance
                          </div>
                        </Tooltip>
                        <div className={`text-th-fgd-1`}>
                          {isLoading ? (
                            <DataLoader />
                          ) : mangoAccount ? (
                            nativeI80F48ToUi(
                              mangoAccount.getAvailableBalance(
                                mangoGroup,
                                mangoGroupCache,
                                tokenIndex
                              ),
                              mangoGroup.tokens[tokenIndex].decimals
                            ).toFixed(tokenPrecision[symbol])
                          ) : (
                            0
                          )}
                        </div>
                      </div>
                      <div>
                        <Tooltip content="Deposit APY / Borrow APR">
                          <div
                            className={`cursor-help font-normal pb-0.5 text-th-fgd-3 default-transition text-xs hover:border-th-bkg-2 hover:text-th-fgd-3`}
                          >
                            Interest Rates
                          </div>
                          <div className={`text-th-fgd-1`}>
                            <span className={`text-th-green`}>
                              {i80f48ToPercent(
                                mangoGroup.getDepositRate(tokenIndex)
                              ).toFixed(2)}
                              %
                            </span>
                            <span className={`text-th-fgd-4`}>{'  /  '}</span>
                            <span className={`text-th-red`}>
                              {i80f48ToPercent(
                                mangoGroup.getBorrowRate(tokenIndex)
                              ).toFixed(2)}
                              %
                            </span>
                          </div>
                        </Tooltip>
                      </div>
                    </div>
                  )
                })}
            </div>
          ) : null}
        </div>
      </FloatingElement>
    </>
  )
}

export const DataLoader = () => (
  <div className="animate-pulse bg-th-bkg-3 h-5 w-10 rounded-sm" />
)