mango-ui-v3/components/WithdrawModal.tsx

555 lines
19 KiB
TypeScript

import React, { FunctionComponent, useEffect, useMemo, useState } from 'react'
import Modal from './Modal'
import Input, { Label } from './Input'
import { ElementTitle } from './styles'
import useMangoStore from '../stores/useMangoStore'
import { floorToDecimal, tokenPrecision } from '../utils/index'
import Loading from './Loading'
import Button, { LinkButton } from './Button'
import Switch from './Switch'
import Tooltip from './Tooltip'
import {
ExclamationCircleIcon,
InformationCircleIcon,
} from '@heroicons/react/outline'
import Select from './Select'
import { withdraw } from '../utils/mango'
import {
ZERO_I80F48,
I80F48,
MangoAccount,
nativeI80F48ToUi,
} from '@blockworks-foundation/mango-client'
import { notify } from '../utils/notifications'
import { useTranslation } from 'next-i18next'
import { ExpandableRow } from './TableElements'
interface WithdrawModalProps {
onClose: () => void
isOpen: boolean
title?: string
tokenSymbol?: string
borrow?: boolean
}
const WithdrawModal: FunctionComponent<WithdrawModalProps> = ({
isOpen,
onClose,
tokenSymbol = '',
borrow = false,
title,
}) => {
const { t } = useTranslation('common')
const [withdrawTokenSymbol, setWithdrawTokenSymbol] = useState(
tokenSymbol || 'USDC'
)
const [inputAmount, setInputAmount] = useState('')
const [invalidAmountMessage, setInvalidAmountMessage] = useState('')
const [maxAmount, setMaxAmount] = useState(0)
const [submitting, setSubmitting] = useState(false)
const [includeBorrow, setIncludeBorrow] = useState(borrow)
const [simulation, setSimulation] = useState(null)
const [showSimulation, setShowSimulation] = useState(false)
const actions = useMangoStore((s) => s.actions)
const mangoGroup = useMangoStore((s) => s.selectedMangoGroup.current)
const mangoAccount = useMangoStore((s) => s.selectedMangoAccount.current)
const mangoCache = useMangoStore((s) => s.selectedMangoGroup.cache)
const mangoGroupConfig = useMangoStore((s) => s.selectedMangoGroup.config)
const tokens = useMemo(() => mangoGroupConfig.tokens, [mangoGroupConfig])
const token = useMemo(
() => tokens.find((t) => t.symbol === withdrawTokenSymbol),
[withdrawTokenSymbol, tokens]
)
const tokenIndex = mangoGroup.getTokenIndex(token.mintKey)
useEffect(() => {
if (!mangoGroup || !mangoAccount || !withdrawTokenSymbol) return
const mintDecimals = mangoGroup.tokens[tokenIndex].decimals
const tokenDeposits = mangoAccount.getUiDeposit(
mangoCache.rootBankCache[tokenIndex],
mangoGroup,
tokenIndex
)
const tokenBorrows = mangoAccount.getUiBorrow(
mangoCache.rootBankCache[tokenIndex],
mangoGroup,
tokenIndex
)
const maxWithoutBorrows = nativeI80F48ToUi(
mangoAccount
.getAvailableBalance(mangoGroup, mangoCache, tokenIndex)
.floor(),
mangoGroup.tokens[tokenIndex].decimals
)
const maxWithBorrows = mangoAccount
.getMaxWithBorrowForToken(mangoGroup, mangoCache, tokenIndex)
.add(maxWithoutBorrows)
.mul(I80F48.fromString('0.995')) // handle rounding errors when borrowing
// get max withdraw amount
let maxWithdraw = maxWithoutBorrows
if (includeBorrow) {
maxWithdraw = maxWithoutBorrows.gt(maxWithBorrows)
? maxWithoutBorrows
: maxWithBorrows
}
if (maxWithdraw.gt(I80F48.fromNumber(0))) {
setMaxAmount(
floorToDecimal(parseFloat(maxWithdraw.toFixed()), token.decimals)
)
} else {
setMaxAmount(0)
}
// simulate change to deposits & borrow based on input amount
const parsedInputAmount = inputAmount
? I80F48.fromString(inputAmount)
: ZERO_I80F48
let newDeposit = tokenDeposits.sub(parsedInputAmount)
newDeposit = newDeposit.gt(ZERO_I80F48) ? newDeposit : ZERO_I80F48
let newBorrow = parsedInputAmount.sub(tokenDeposits)
newBorrow = newBorrow.gt(ZERO_I80F48) ? newBorrow : ZERO_I80F48
newBorrow = newBorrow.add(tokenBorrows)
// clone MangoAccount and arrays to not modify selectedMangoAccount
const simulation = new MangoAccount(null, mangoAccount)
simulation.deposits = [...mangoAccount.deposits]
simulation.borrows = [...mangoAccount.borrows]
// update with simulated values
simulation.deposits[tokenIndex] = newDeposit
.mul(I80F48.fromNumber(Math.pow(10, mintDecimals)))
.div(mangoCache.rootBankCache[tokenIndex].depositIndex)
simulation.borrows[tokenIndex] = newBorrow
.mul(I80F48.fromNumber(Math.pow(10, mintDecimals)))
.div(mangoCache.rootBankCache[tokenIndex].borrowIndex)
const liabsVal = simulation
.getLiabsVal(mangoGroup, mangoCache, 'Init')
.toNumber()
const leverage = simulation.getLeverage(mangoGroup, mangoCache).toNumber()
const equity = simulation.computeValue(mangoGroup, mangoCache).toNumber()
const initHealthRatio = simulation
.getHealthRatio(mangoGroup, mangoCache, 'Init')
.toNumber()
setSimulation({
initHealthRatio,
liabsVal,
leverage,
equity,
})
}, [
includeBorrow,
inputAmount,
tokenIndex,
mangoAccount,
mangoGroup,
mangoCache,
])
const handleWithdraw = () => {
setSubmitting(true)
withdraw({
amount: Number(inputAmount),
token: mangoGroup.tokens[tokenIndex].mint,
allowBorrow: includeBorrow,
})
.then((txid: string) => {
setSubmitting(false)
actions.reloadMangoAccount()
actions.fetchWalletTokens()
notify({
title: t('withdraw-success'),
type: 'success',
txid,
})
onClose()
})
.catch((err) => {
setSubmitting(false)
console.error('Error withdrawing:', err)
notify({
title: t('withdraw-error'),
description: err.message,
txid: err.txid,
type: 'error',
})
onClose()
})
}
const handleSetSelectedAsset = (symbol) => {
setInputAmount('')
setWithdrawTokenSymbol(symbol)
}
const getDepositsForSelectedAsset = (): I80F48 => {
return mangoAccount
? mangoAccount.getUiDeposit(
mangoCache.rootBankCache[tokenIndex],
mangoGroup,
tokenIndex
)
: ZERO_I80F48
}
const getBorrowAmount = () => {
const tokenBalance = getDepositsForSelectedAsset()
const borrowAmount = I80F48.fromString(inputAmount).sub(tokenBalance)
return borrowAmount.gt(ZERO_I80F48) ? borrowAmount : 0
}
const getAccountStatusColor = (
initHealthRatio: number,
isRisk?: boolean,
isStatus?: boolean
) => {
if (initHealthRatio < 1) {
return isRisk ? (
<div className="text-th-red">{t('high')}</div>
) : isStatus ? (
'bg-th-red'
) : (
'ring-th-red text-th-red'
)
} else if (initHealthRatio > 1 && initHealthRatio < 10) {
return isRisk ? (
<div className="text-th-orange">{t('moderate')}</div>
) : isStatus ? (
'bg-th-orange'
) : (
'ring-th-orange text-th-orange'
)
} else {
return isRisk ? (
<div className="text-th-green">{t('low')}</div>
) : isStatus ? (
'bg-th-green'
) : (
'ring-th-green text-th-green'
)
}
}
const handleIncludeBorrowSwitch = (checked) => {
setIncludeBorrow(checked)
setInputAmount('')
setInvalidAmountMessage('')
}
const onChangeAmountInput = (amount: string) => {
setInputAmount(amount)
setInvalidAmountMessage('')
}
const validateAmountInput = (amount) => {
const parsedAmount = Number(amount)
if (
(getDepositsForSelectedAsset() === ZERO_I80F48 ||
getDepositsForSelectedAsset().lt(I80F48.fromNumber(parsedAmount))) &&
!includeBorrow
) {
setInvalidAmountMessage(t('insufficient-balance-withdraw'))
}
}
useEffect(() => {
if (simulation && simulation.initHealthRatio < 0 && includeBorrow) {
setInvalidAmountMessage(t('leverage-too-high'))
}
}, [simulation])
const getTokenBalances = () => {
const mangoCache = useMangoStore.getState().selectedMangoGroup.cache
const mangoGroup = useMangoStore.getState().selectedMangoGroup.current
return tokens.map((token) => {
const tokenIndex = mangoGroup.getTokenIndex(token.mintKey)
return {
symbol: token.symbol,
balance: mangoAccount
?.getUiDeposit(
mangoCache.rootBankCache[tokenIndex],
mangoGroup,
tokenIndex
)
?.toFixed(tokenPrecision[token.symbol]),
}
})
}
if (!withdrawTokenSymbol) return null
return (
<Modal isOpen={isOpen} onClose={onClose}>
<>
{!showSimulation ? (
<>
<Modal.Header>
<ElementTitle noMarginBottom>
{title ? title : t('withdraw-funds')}
</ElementTitle>
</Modal.Header>
<Label>{t('asset')}</Label>
<Select
value={
withdrawTokenSymbol && mangoAccount ? (
<div className="flex items-center justify-between w-full">
<div className="flex items-center">
<img
alt=""
width="20"
height="20"
src={`/assets/icons/${withdrawTokenSymbol.toLowerCase()}.svg`}
className={`mr-2.5`}
/>
{withdrawTokenSymbol}
</div>
{mangoAccount
?.getUiDeposit(
mangoCache.rootBankCache[tokenIndex],
mangoGroup,
tokenIndex
)
?.toFixed(tokenPrecision[withdrawTokenSymbol])}
</div>
) : (
<span className="text-th-fgd-4">{t('select-asset')}</span>
)
}
onChange={(asset) => handleSetSelectedAsset(asset)}
>
{getTokenBalances().map(({ symbol, balance }) => (
<Select.Option key={symbol} value={symbol}>
<div className="flex items-center justify-between">
<div className="flex items-center">
<img
alt=""
width="16"
height="16"
src={`/assets/icons/${symbol.toLowerCase()}.svg`}
className={`mr-2`}
/>
<span>{symbol}</span>
</div>
{balance}
</div>
</Select.Option>
))}
</Select>
<div className="flex items-center jusitfy-between text-th-fgd-1 mt-4 p-2 rounded-md bg-th-bkg-3">
<div className="flex items-center text-fgd-1 pr-4">
<span>{t('borrow-funds')}</span>
<Tooltip content={t('tooltip-interest-charged')}>
<InformationCircleIcon
className={`h-5 w-5 ml-2 text-th-primary cursor-help`}
/>
</Tooltip>
</div>
<Switch
checked={includeBorrow}
className="ml-auto"
onChange={(checked) => handleIncludeBorrowSwitch(checked)}
/>
</div>
<div className="flex justify-between pt-4">
<Label>{t('amount')}</Label>
<LinkButton
className="mb-1.5"
onClick={() => setInputAmount(maxAmount.toString())}
>
{includeBorrow ? t('max-with-borrow') : t('max')}
</LinkButton>
</div>
<div className="flex">
<Input
disabled={!withdrawTokenSymbol}
type="number"
min="0"
error={!!invalidAmountMessage}
placeholder="0.00"
value={inputAmount}
onBlur={(e) => validateAmountInput(e.target.value)}
onChange={(e) => onChangeAmountInput(e.target.value)}
suffix={withdrawTokenSymbol}
/>
{simulation ? (
<Tooltip
placement="right"
content={t('tooltip-projected-leverage')}
className="py-1"
>
<span
className={`${getAccountStatusColor(
simulation.initHealthRatio
)} bg-th-bkg-1 ring-1 ring-inset flex h-10 items-center justify-center ml-1 px-2 rounded`}
>
{simulation.leverage.toFixed(2)}x
</span>
</Tooltip>
) : null}
</div>
{invalidAmountMessage ? (
<div className="flex items-center pt-1.5 text-th-red">
<ExclamationCircleIcon className="h-4 w-4 mr-1.5" />
{invalidAmountMessage}
</div>
) : null}
<div className={`flex justify-center pt-6`}>
<Button
onClick={() => setShowSimulation(true)}
disabled={
Number(inputAmount) <= 0 || simulation?.initHealthRatio < 0
}
className="w-full"
>
{t('next')}
</Button>
</div>
</>
) : null}
{showSimulation && simulation ? (
<>
<Modal.Header>
<ElementTitle noMarginBottom>
{t('confirm-withdraw')}
</ElementTitle>
</Modal.Header>
{simulation.initHealthRatio < 0 ? (
<div className="border border-th-red mb-4 p-2 rounded">
<div className="flex items-center text-th-red">
<ExclamationCircleIcon className="h-4 w-4 mr-1.5 flex-shrink-0" />
{t('prices-changed')}
</div>
</div>
) : null}
<div className="bg-th-bkg-1 p-4 rounded-lg text-th-fgd-1 text-center">
<div className="text-th-fgd-3 pb-1">{t('about-to-withdraw')}</div>
<div className="flex items-center justify-center">
<div className="font-semibold relative text-xl">
{inputAmount}
<span className="absolute bottom-0.5 font-normal ml-1.5 text-xs text-th-fgd-4">
{withdrawTokenSymbol}
</span>
</div>
</div>
{getBorrowAmount() > 0 ? (
<div className="pt-2 text-th-fgd-4">{`${t(
'includes-borrow'
)} ~${getBorrowAmount().toFixed(
mangoGroup.tokens[tokenIndex].decimals
)} ${withdrawTokenSymbol}`}</div>
) : null}
</div>
<div className="border-b border-th-bkg-4 pt-4">
<ExpandableRow
buttonTemplate={
<div className="flex items-center justify-between">
<div className="flex items-center">
<span className="flex h-2 w-2 mr-2.5 relative">
<span
className={`animate-ping absolute inline-flex h-full w-full rounded-full ${getAccountStatusColor(
simulation.initHealthRatio,
false,
true
)} opacity-75`}
></span>
<span
className={`relative inline-flex rounded-full h-2 w-2 ${getAccountStatusColor(
simulation.initHealthRatio,
false,
true
)}`}
></span>
</span>
{t('health-check')}
<Tooltip content={t('tooltip-after-withdrawal')}>
<InformationCircleIcon
className={`h-5 w-5 ml-2 text-th-primary cursor-help`}
/>
</Tooltip>
</div>
</div>
}
panelTemplate={
simulation ? (
<div>
<div className="flex justify-between pb-2">
<p className="mb-0">{t('account-value')}</p>
<div className="text-th-fgd-1">
$
{simulation.equity.toLocaleString(undefined, {
maximumFractionDigits: 2,
})}
</div>
</div>
<div className="flex justify-between pb-2">
<p className="mb-0">{t('account-risk')}</p>
<div className="text-th-fgd-1">
{getAccountStatusColor(
simulation.initHealthRatio,
true
)}
</div>
</div>
<div className="flex justify-between pb-2">
<p className="mb-0">{t('leverage')}</p>
<div className="text-th-fgd-1">
{simulation.leverage.toFixed(2)}x
</div>
</div>
<div className="flex justify-between">
<p className="mb-0">{t('borrow-value')}</p>
<div className="text-th-fgd-1">
$
{simulation.liabsVal.toLocaleString(undefined, {
maximumFractionDigits: 2,
})}
</div>
</div>
</div>
) : null
}
/>
</div>
<div className={`mt-6 flex flex-col items-center`}>
<Button
onClick={handleWithdraw}
disabled={
Number(inputAmount) <= 0 ||
simulation.initHealthRatio < 0 ||
submitting
}
className="w-full"
>
<div className={`flex items-center justify-center`}>
{submitting && <Loading className="-ml-1 mr-3" />}
{t('withdraw')}
</div>
</Button>
<LinkButton
className="mt-4"
onClick={() => setShowSimulation(false)}
>
{t('cancel')}
</LinkButton>
</div>
</>
) : null}
</>
</Modal>
)
}
export default React.memo(WithdrawModal)