309 lines
11 KiB
TypeScript
309 lines
11 KiB
TypeScript
import { HealthType } from '@blockworks-foundation/mango-v4'
|
|
import {
|
|
ArrowUpTrayIcon,
|
|
ChevronDownIcon,
|
|
ExclamationCircleIcon,
|
|
} from '@heroicons/react/20/solid'
|
|
import Decimal from 'decimal.js'
|
|
import { useTranslation } from 'next-i18next'
|
|
import Image from 'next/legacy/image'
|
|
import { useCallback, useMemo, useState } from 'react'
|
|
import NumberFormat, { NumberFormatValues } from 'react-number-format'
|
|
|
|
import mangoStore from '@store/mangoStore'
|
|
import { ModalProps } from '../../types/modal'
|
|
import { INPUT_TOKEN_DEFAULT } from '../../utils/constants'
|
|
import { notify } from '../../utils/notifications'
|
|
import { formatFixedDecimals } from '../../utils/numbers'
|
|
import ActionTokenList from '../account/ActionTokenList'
|
|
import ButtonGroup from '../forms/ButtonGroup'
|
|
import Label from '../forms/Label'
|
|
import Button from '../shared/Button'
|
|
import InlineNotification from '../shared/InlineNotification'
|
|
import Loading from '../shared/Loading'
|
|
import Modal from '../shared/Modal'
|
|
import { EnterBottomExitBottom, FadeInFadeOut } from '../shared/Transitions'
|
|
import { withValueLimit } from '../swap/SwapForm'
|
|
import { getMaxWithdrawForBank } from '../swap/useTokenMax'
|
|
import MaxAmountButton from '@components/shared/MaxAmountButton'
|
|
import HealthImpactTokenChange from '@components/HealthImpactTokenChange'
|
|
import useMangoAccount from 'hooks/useMangoAccount'
|
|
import useJupiterMints from 'hooks/useJupiterMints'
|
|
|
|
interface WithdrawModalProps {
|
|
token?: string
|
|
}
|
|
|
|
type ModalCombinedProps = WithdrawModalProps & ModalProps
|
|
|
|
function WithdrawModal({ isOpen, onClose, token }: ModalCombinedProps) {
|
|
const { t } = useTranslation(['common', 'trade'])
|
|
const group = mangoStore((s) => s.group)
|
|
const [inputAmount, setInputAmount] = useState('')
|
|
const [submitting, setSubmitting] = useState(false)
|
|
const [selectedToken, setSelectedToken] = useState(
|
|
token || INPUT_TOKEN_DEFAULT
|
|
)
|
|
const [showTokenList, setShowTokenList] = useState(false)
|
|
const [sizePercentage, setSizePercentage] = useState('')
|
|
const { mangoTokens } = useJupiterMints()
|
|
const { mangoAccount } = useMangoAccount()
|
|
|
|
const bank = useMemo(() => {
|
|
const group = mangoStore.getState().group
|
|
return group?.banksMapByName.get(selectedToken)![0]
|
|
}, [selectedToken])
|
|
|
|
const logoUri = useMemo(() => {
|
|
let logoURI
|
|
if (mangoTokens.length) {
|
|
logoURI = mangoTokens.find(
|
|
(t) => t.address === bank?.mint.toString()
|
|
)!.logoURI
|
|
}
|
|
return logoURI
|
|
}, [bank?.mint, mangoTokens])
|
|
|
|
const tokenMax = useMemo(() => {
|
|
if (!bank || !mangoAccount || !group) return new Decimal(0)
|
|
const amount = getMaxWithdrawForBank(group, bank, mangoAccount)
|
|
|
|
return amount && amount.gt(0)
|
|
? amount.toDecimalPlaces(bank.mintDecimals)
|
|
: new Decimal(0)
|
|
}, [mangoAccount, bank, group])
|
|
|
|
const handleSizePercentage = useCallback(
|
|
(percentage: string) => {
|
|
setSizePercentage(percentage)
|
|
const amount = tokenMax.mul(Number(percentage) / 100)
|
|
setInputAmount(amount.toFixed())
|
|
},
|
|
[tokenMax]
|
|
)
|
|
|
|
const handleWithdraw = async () => {
|
|
const client = mangoStore.getState().client
|
|
const group = mangoStore.getState().group
|
|
const mangoAccount = mangoStore.getState().mangoAccount.current
|
|
const actions = mangoStore.getState().actions
|
|
if (!mangoAccount || !group) return
|
|
setSubmitting(true)
|
|
try {
|
|
const tx = await client.tokenWithdraw(
|
|
group,
|
|
mangoAccount,
|
|
bank!.mint,
|
|
parseFloat(inputAmount),
|
|
false
|
|
)
|
|
notify({
|
|
title: 'Transaction confirmed',
|
|
type: 'success',
|
|
txid: tx,
|
|
})
|
|
actions.reloadMangoAccount()
|
|
} catch (e: any) {
|
|
console.error(e)
|
|
notify({
|
|
title: 'Transaction failed',
|
|
description: e.message,
|
|
txid: e?.txid,
|
|
type: 'error',
|
|
})
|
|
} finally {
|
|
setSubmitting(false)
|
|
onClose()
|
|
}
|
|
}
|
|
|
|
const handleSelectToken = (token: string) => {
|
|
setSelectedToken(token)
|
|
setShowTokenList(false)
|
|
}
|
|
|
|
const withdrawBanks = useMemo(() => {
|
|
if (mangoAccount) {
|
|
const banks = group?.banksMapByName
|
|
? Array.from(group?.banksMapByName, ([key, value]) => {
|
|
const accountBalance = getMaxWithdrawForBank(
|
|
group,
|
|
value[0],
|
|
mangoAccount
|
|
)
|
|
return {
|
|
key,
|
|
value,
|
|
accountBalance: accountBalance ? accountBalance.toNumber() : 0,
|
|
accountBalanceValue:
|
|
accountBalance && value[0]?.uiPrice
|
|
? accountBalance.toNumber() * value[0]?.uiPrice
|
|
: 0,
|
|
}
|
|
})
|
|
: []
|
|
return banks
|
|
}
|
|
return []
|
|
}, [mangoAccount, group])
|
|
|
|
const initHealth = useMemo(() => {
|
|
return group && mangoAccount
|
|
? mangoAccount.getHealthRatioUi(group, HealthType.init)
|
|
: 100
|
|
}, [mangoAccount])
|
|
|
|
const showInsufficientBalance = Number(inputAmount)
|
|
? tokenMax.lt(inputAmount)
|
|
: false
|
|
|
|
return (
|
|
<Modal isOpen={isOpen} onClose={onClose}>
|
|
<div className="">
|
|
<EnterBottomExitBottom
|
|
className="absolute bottom-0 left-0 z-20 h-full w-full overflow-auto rounded-lg bg-th-bkg-1 p-6"
|
|
show={showTokenList}
|
|
>
|
|
<h2 className="mb-4 text-center">{t('select-token')}</h2>
|
|
<div className="grid auto-cols-fr grid-flow-col px-4 pb-2">
|
|
<div className="">
|
|
<p className="text-xs">{t('token')}</p>
|
|
</div>
|
|
<div className="flex justify-end">
|
|
<p className="text-xs">{t('available-balance')}</p>
|
|
</div>
|
|
</div>
|
|
<ActionTokenList
|
|
banks={withdrawBanks}
|
|
onSelect={handleSelectToken}
|
|
sortByKey="accountBalanceValue"
|
|
valueKey="accountBalance"
|
|
/>
|
|
</EnterBottomExitBottom>
|
|
<FadeInFadeOut
|
|
className="flex h-full flex-col justify-between"
|
|
show={isOpen}
|
|
>
|
|
<div>
|
|
<h2 className="mb-4 text-center">{t('withdraw')}</h2>
|
|
{initHealth! <= 0 ? (
|
|
<div className="mb-4">
|
|
<InlineNotification
|
|
type="error"
|
|
desc="You have no available collateral to withdraw."
|
|
/>
|
|
</div>
|
|
) : null}
|
|
<div className="grid grid-cols-2">
|
|
<div className="col-span-2 flex justify-between">
|
|
<Label text={t('token')} />
|
|
<MaxAmountButton
|
|
className="mb-2"
|
|
label={t('max')}
|
|
onClick={() => handleSizePercentage('100')}
|
|
value={tokenMax.toString()}
|
|
/>
|
|
</div>
|
|
<div className="col-span-1 rounded-lg rounded-r-none border border-r-0 border-th-bkg-4 bg-th-bkg-1">
|
|
<button
|
|
onClick={() => setShowTokenList(true)}
|
|
className="default-transition flex h-full w-full items-center rounded-lg rounded-r-none py-2 px-3 text-th-fgd-2 hover:cursor-pointer hover:bg-th-bkg-2 hover:text-th-fgd-1"
|
|
>
|
|
<div className="mr-2.5 flex min-w-[24px] items-center">
|
|
<Image
|
|
alt=""
|
|
width="24"
|
|
height="24"
|
|
src={
|
|
logoUri || `/icons/${selectedToken.toLowerCase()}.svg`
|
|
}
|
|
/>
|
|
</div>
|
|
<div className="flex w-full items-center justify-between">
|
|
<div className="text-xl font-bold">{selectedToken}</div>
|
|
<ChevronDownIcon className="h-6 w-6" />
|
|
</div>
|
|
</button>
|
|
</div>
|
|
<div className="col-span-1">
|
|
<NumberFormat
|
|
name="amountIn"
|
|
id="amountIn"
|
|
inputMode="decimal"
|
|
thousandSeparator=","
|
|
allowNegative={false}
|
|
isNumericString={true}
|
|
decimalScale={bank?.mintDecimals || 6}
|
|
className="w-full rounded-lg rounded-l-none border border-th-bkg-4 bg-th-bkg-1 p-3 text-right font-mono text-xl tracking-wider text-th-fgd-1 focus:outline-none"
|
|
placeholder="0.00"
|
|
value={inputAmount}
|
|
onValueChange={(e: NumberFormatValues) =>
|
|
setInputAmount(
|
|
!Number.isNaN(Number(e.value)) ? e.value : ''
|
|
)
|
|
}
|
|
isAllowed={withValueLimit}
|
|
/>
|
|
</div>
|
|
<div className="col-span-2 mt-2">
|
|
<ButtonGroup
|
|
activeValue={sizePercentage}
|
|
className="font-mono"
|
|
onChange={(p) => handleSizePercentage(p)}
|
|
values={['10', '25', '50', '75', '100']}
|
|
unit="%"
|
|
/>
|
|
</div>
|
|
</div>
|
|
<div className="my-6 space-y-2 border-y border-th-bkg-3 px-2 py-4">
|
|
<HealthImpactTokenChange
|
|
mintPk={bank!.mint}
|
|
uiAmount={Number(inputAmount)}
|
|
/>
|
|
<div className="flex justify-between">
|
|
<p>{t('withdrawal-value')}</p>
|
|
<p className="font-mono text-th-fgd-1">
|
|
{bank?.uiPrice
|
|
? formatFixedDecimals(
|
|
bank.uiPrice * Number(inputAmount),
|
|
true
|
|
)
|
|
: '-'}
|
|
</p>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
<div className="flex justify-center">
|
|
<Button
|
|
onClick={handleWithdraw}
|
|
className="flex w-full items-center justify-center"
|
|
size="large"
|
|
disabled={
|
|
!inputAmount || showInsufficientBalance || initHealth! <= 0
|
|
}
|
|
>
|
|
{submitting ? (
|
|
<Loading className="mr-2 h-5 w-5" />
|
|
) : showInsufficientBalance ? (
|
|
<div className="flex items-center">
|
|
<ExclamationCircleIcon className="mr-2 h-5 w-5 flex-shrink-0" />
|
|
{t('swap:insufficient-balance', {
|
|
symbol: selectedToken,
|
|
})}
|
|
</div>
|
|
) : (
|
|
<div className="flex items-center">
|
|
<ArrowUpTrayIcon className="mr-2 h-5 w-5" />
|
|
{t('withdraw')}
|
|
</div>
|
|
)}
|
|
</Button>
|
|
</div>
|
|
</FadeInFadeOut>
|
|
</div>
|
|
</Modal>
|
|
)
|
|
}
|
|
|
|
export default WithdrawModal
|