518 lines
18 KiB
TypeScript
518 lines
18 KiB
TypeScript
import {
|
|
ArrowPathIcon,
|
|
ChevronDownIcon,
|
|
ExclamationCircleIcon,
|
|
} from '@heroicons/react/20/solid'
|
|
import { useWallet } from '@solana/wallet-adapter-react'
|
|
import { useTranslation } from 'next-i18next'
|
|
import React, { useCallback, useEffect, useMemo, useState } from 'react'
|
|
import NumberFormat, {
|
|
NumberFormatValues,
|
|
SourceInfo,
|
|
} from 'react-number-format'
|
|
import mangoStore from '@store/mangoStore'
|
|
import { notify } from '../utils/notifications'
|
|
import { formatTokenSymbol } from '../utils/tokens'
|
|
// import ActionTokenList from './account/ActionTokenList'
|
|
import Label from './forms/Label'
|
|
import Button, { IconButton } from './shared/Button'
|
|
import Loading from './shared/Loading'
|
|
import MaxAmountButton from '@components/shared/MaxAmountButton'
|
|
import Tooltip from '@components/shared/Tooltip'
|
|
import SolBalanceWarnings from '@components/shared/SolBalanceWarnings'
|
|
import useSolBalance from 'hooks/useSolBalance'
|
|
import { floorToDecimal, withValueLimit } from 'utils/numbers'
|
|
import BankAmountWithValue from './shared/BankAmountWithValue'
|
|
// import useBanksWithBalances from 'hooks/useBanksWithBalances'
|
|
import { isMangoError } from 'types'
|
|
// import TokenListButton from './shared/TokenListButton'
|
|
import TokenLogo from './shared/TokenLogo'
|
|
import SecondaryConnectButton from './shared/SecondaryConnectButton'
|
|
import useMangoAccountAccounts from 'hooks/useMangoAccountAccounts'
|
|
import InlineNotification from './shared/InlineNotification'
|
|
import Link from 'next/link'
|
|
import useMangoGroup from 'hooks/useMangoGroup'
|
|
import FormatNumericValue from './shared/FormatNumericValue'
|
|
import useMangoAccount from 'hooks/useMangoAccount'
|
|
import { unstakeAndSwap, withdrawAndClose } from 'utils/transactions'
|
|
import { NUMBERFORMAT_CLASSES } from './StakeForm'
|
|
import ButtonGroup from './forms/ButtonGroup'
|
|
import Decimal from 'decimal.js'
|
|
import { Disclosure } from '@headlessui/react'
|
|
import { sleep } from 'utils'
|
|
import useIpAddress from 'hooks/useIpAddress'
|
|
import {
|
|
ClientContextKeys,
|
|
JLP_BORROW_TOKEN,
|
|
LST_BORROW_TOKEN,
|
|
} from 'utils/constants'
|
|
|
|
const set = mangoStore.getState().set
|
|
|
|
interface UnstakeFormProps {
|
|
token: string
|
|
clientContext: ClientContextKeys
|
|
}
|
|
|
|
function UnstakeForm({
|
|
token: selectedToken,
|
|
clientContext,
|
|
}: UnstakeFormProps) {
|
|
const { t } = useTranslation(['common', 'account'])
|
|
const [inputAmount, setInputAmount] = useState('')
|
|
const [submitting, setSubmitting] = useState(false)
|
|
const [refreshingWalletTokens, setRefreshingWalletTokens] = useState(false)
|
|
const [sizePercentage, setSizePercentage] = useState('')
|
|
const { maxSolDeposit } = useSolBalance()
|
|
const { usedTokens, totalTokens } = useMangoAccountAccounts()
|
|
const { jlpGroup, lstGroup } = useMangoGroup()
|
|
const { mangoAccount } = useMangoAccount()
|
|
const { ipAllowed } = useIpAddress()
|
|
|
|
const [stakeBank, borrowBank] = useMemo(() => {
|
|
const stakeBank =
|
|
clientContext === 'jlp'
|
|
? jlpGroup?.banksMapByName.get(selectedToken)?.[0]
|
|
: lstGroup?.banksMapByName.get(selectedToken)?.[0]
|
|
const borrowBank =
|
|
clientContext === 'jlp'
|
|
? jlpGroup?.banksMapByName.get(JLP_BORROW_TOKEN)?.[0]
|
|
: lstGroup?.banksMapByName.get(LST_BORROW_TOKEN)?.[0]
|
|
return [stakeBank, borrowBank]
|
|
}, [selectedToken, jlpGroup, lstGroup, clientContext])
|
|
|
|
const tokenPositionsFull = useMemo(() => {
|
|
if (!stakeBank || !usedTokens.length || !totalTokens.length) return false
|
|
const hasTokenPosition = usedTokens.find(
|
|
(token) => token.tokenIndex === stakeBank.tokenIndex,
|
|
)
|
|
return hasTokenPosition ? false : usedTokens.length >= totalTokens.length
|
|
}, [stakeBank, usedTokens, totalTokens])
|
|
|
|
const { connected, publicKey } = useWallet()
|
|
|
|
const stakeBankAmount =
|
|
mangoAccount && stakeBank && mangoAccount.getTokenBalance(stakeBank)
|
|
|
|
const borrowBankAmount =
|
|
mangoAccount && borrowBank && mangoAccount.getTokenBalance(borrowBank)
|
|
|
|
const leverage = useMemo(() => {
|
|
try {
|
|
if (
|
|
stakeBankAmount &&
|
|
borrowBankAmount &&
|
|
borrowBankAmount.toNumber() < 0
|
|
) {
|
|
const stakeAmountValue = stakeBankAmount.mul(stakeBank.getAssetPrice())
|
|
const lev = stakeAmountValue
|
|
.div(
|
|
stakeAmountValue.sub(
|
|
borrowBankAmount.abs().mul(borrowBank.getAssetPrice()),
|
|
),
|
|
)
|
|
.toNumber()
|
|
|
|
return Math.sign(lev) !== -1 ? lev : 1
|
|
}
|
|
return 1
|
|
} catch (e) {
|
|
console.log(e)
|
|
return 1
|
|
}
|
|
}, [stakeBankAmount, borrowBankAmount, stakeBank, borrowBank])
|
|
|
|
const tokenMax = useMemo(() => {
|
|
if (!stakeBank || !mangoAccount) return { maxAmount: 0.0, maxDecimals: 6 }
|
|
return {
|
|
maxAmount: mangoAccount.getTokenBalanceUi(stakeBank) / leverage,
|
|
maxDecimals: stakeBank.mintDecimals,
|
|
}
|
|
}, [stakeBank, mangoAccount, leverage])
|
|
|
|
const setMax = useCallback(() => {
|
|
const max = floorToDecimal(tokenMax.maxAmount, tokenMax.maxDecimals)
|
|
setInputAmount(max.toFixed())
|
|
setSizePercentage('100')
|
|
}, [tokenMax])
|
|
|
|
const handleSizePercentage = useCallback(
|
|
(percentage: string) => {
|
|
if (!stakeBank) return
|
|
setSizePercentage(percentage)
|
|
const amount = floorToDecimal(
|
|
new Decimal(percentage).div(100).mul(tokenMax.maxAmount),
|
|
stakeBank.mintDecimals,
|
|
)
|
|
setInputAmount(amount.toFixed())
|
|
},
|
|
[tokenMax, stakeBank],
|
|
)
|
|
|
|
const handleRefreshWalletBalances = useCallback(async () => {
|
|
if (!publicKey) return
|
|
const actions = mangoStore.getState().actions
|
|
setRefreshingWalletTokens(true)
|
|
await actions.fetchMangoAccounts(publicKey)
|
|
setRefreshingWalletTokens(false)
|
|
}, [publicKey])
|
|
|
|
const borrowed = useMemo(() => {
|
|
if (!borrowBank || !mangoAccount) return 0.0
|
|
return mangoAccount.getTokenBalanceUi(borrowBank)
|
|
}, [borrowBank, mangoAccount])
|
|
|
|
const handleWithdraw = useCallback(async () => {
|
|
if (!ipAllowed || !stakeBank || !borrowBank || !publicKey) {
|
|
return
|
|
}
|
|
const client = mangoStore.getState().client
|
|
const group = mangoStore.getState().group[clientContext]
|
|
const actions = mangoStore.getState().actions
|
|
let mangoAccount = mangoStore.getState().mangoAccount.current
|
|
|
|
if (!group || !mangoAccount) return
|
|
|
|
setSubmitting(true)
|
|
try {
|
|
if (mangoAccount.getTokenBalanceUi(borrowBank) < 0) {
|
|
notify({
|
|
title: 'Sending transaction 1 of 2',
|
|
type: 'info',
|
|
})
|
|
console.log(
|
|
'unstake and swap',
|
|
mangoAccount.getTokenBalanceUi(borrowBank),
|
|
)
|
|
|
|
const stakeAmountToRepay = (leverage - 1) * Number(inputAmount)
|
|
|
|
const { signature: tx } = await unstakeAndSwap(
|
|
client[clientContext],
|
|
group,
|
|
mangoAccount,
|
|
stakeBank.mint,
|
|
stakeAmountToRepay,
|
|
)
|
|
|
|
notify({
|
|
title: 'Swap Transaction confirmed.',
|
|
type: 'success',
|
|
txid: tx,
|
|
})
|
|
await sleep(100)
|
|
await actions.fetchMangoAccounts(publicKey)
|
|
await actions.reloadMangoAccount(clientContext)
|
|
await actions.fetchWalletTokens(publicKey)
|
|
mangoAccount = mangoStore.getState().mangoAccount.current
|
|
notify({
|
|
title: 'Sending transaction 2 of 2',
|
|
type: 'info',
|
|
})
|
|
}
|
|
if (!mangoAccount) return
|
|
const { signature: tx2 } = await withdrawAndClose(
|
|
client[clientContext],
|
|
group,
|
|
mangoAccount,
|
|
stakeBank.mint,
|
|
Number(inputAmount),
|
|
)
|
|
notify({
|
|
title: 'Withdraw transaction confirmed.',
|
|
type: 'success',
|
|
txid: tx2,
|
|
})
|
|
setSubmitting(false)
|
|
setInputAmount('')
|
|
setSizePercentage('')
|
|
await sleep(100)
|
|
await actions.fetchMangoAccounts(publicKey)
|
|
await actions.reloadMangoAccount(clientContext)
|
|
await actions.fetchWalletTokens(publicKey)
|
|
} catch (e) {
|
|
console.error('Error withdrawing:', e)
|
|
notify({
|
|
title: 'Error withdrawing',
|
|
description: `${e}`,
|
|
type: 'error',
|
|
})
|
|
setSubmitting(false)
|
|
if (!isMangoError(e)) return
|
|
notify({
|
|
title: 'Transaction failed',
|
|
description: e.message,
|
|
txid: e?.txid,
|
|
type: 'error',
|
|
})
|
|
}
|
|
}, [
|
|
ipAllowed,
|
|
stakeBank,
|
|
borrowBank,
|
|
publicKey,
|
|
inputAmount,
|
|
leverage,
|
|
clientContext,
|
|
])
|
|
|
|
const maxWithdraw = useMemo(() => {
|
|
if (!mangoAccount || !stakeBank) return 0
|
|
const group = clientContext === 'jlp' ? jlpGroup : lstGroup
|
|
if (!group) return 0
|
|
try {
|
|
return mangoAccount.getMaxWithdrawWithBorrowForTokenUi(
|
|
group,
|
|
stakeBank.mint,
|
|
)
|
|
} catch (e) {
|
|
return 0
|
|
}
|
|
}, [jlpGroup, lstGroup, mangoAccount, stakeBank, clientContext])
|
|
|
|
const availableVaultBalance = useMemo(() => {
|
|
if (!stakeBank) return 0
|
|
const group = clientContext === 'jlp' ? jlpGroup : lstGroup
|
|
if (!group) return 0
|
|
const vaultBalance = group.getTokenVaultBalanceByMintUi(stakeBank.mint)
|
|
const vaultDeposits = stakeBank.uiDeposits()
|
|
const available =
|
|
vaultBalance - vaultDeposits * stakeBank.minVaultToDepositsRatio
|
|
return available
|
|
}, [stakeBank, jlpGroup, lstGroup, clientContext])
|
|
|
|
const showInsufficientBalance =
|
|
tokenMax.maxAmount < Number(inputAmount) ||
|
|
(selectedToken === 'USDC' && maxSolDeposit <= 0)
|
|
|
|
const lowVaultBalance = maxWithdraw > availableVaultBalance
|
|
|
|
// const lowVaultBalance =
|
|
// Math.floor(tokenMax.maxAmount * 100000) <
|
|
// Math.floor(Number(inputAmount) * 100000) &&
|
|
// Number(inputAmount) > maxWithdraw
|
|
|
|
useEffect(() => {
|
|
const group = mangoStore.getState().group[clientContext]
|
|
set((state) => {
|
|
state.swap.outputBank = group?.banksMapByName.get(selectedToken)?.[0]
|
|
})
|
|
}, [selectedToken, clientContext])
|
|
|
|
return (
|
|
<>
|
|
<div className="flex flex-col justify-between">
|
|
<div className="pb-8">
|
|
<SolBalanceWarnings
|
|
amount={inputAmount}
|
|
className="mb-4"
|
|
setAmount={setInputAmount}
|
|
selectedToken={selectedToken}
|
|
/>
|
|
<div className="grid grid-cols-2">
|
|
<div className="col-span-2 flex justify-between">
|
|
<Label text="Amount" />
|
|
<div className="mb-2 flex items-center space-x-2">
|
|
<MaxAmountButton
|
|
decimals={tokenMax.maxDecimals}
|
|
label={t('balance')}
|
|
onClick={setMax}
|
|
value={tokenMax.maxAmount}
|
|
/>
|
|
<Tooltip content="Refresh Balance">
|
|
<IconButton
|
|
className={refreshingWalletTokens ? 'animate-spin' : ''}
|
|
onClick={handleRefreshWalletBalances}
|
|
hideBg
|
|
>
|
|
<ArrowPathIcon className="h-5 w-5" />
|
|
</IconButton>
|
|
</Tooltip>
|
|
</div>
|
|
</div>
|
|
<div className="col-span-2">
|
|
<div className="relative">
|
|
<NumberFormat
|
|
name="amountIn"
|
|
id="amountIn"
|
|
inputMode="decimal"
|
|
thousandSeparator=","
|
|
allowNegative={false}
|
|
isNumericString={true}
|
|
decimalScale={stakeBank?.mintDecimals || 6}
|
|
className={NUMBERFORMAT_CLASSES}
|
|
placeholder="0.00"
|
|
value={inputAmount}
|
|
onValueChange={(e: NumberFormatValues, info: SourceInfo) => {
|
|
setInputAmount(
|
|
!Number.isNaN(Number(e.value)) ? e.value : '',
|
|
)
|
|
if (info.source === 'event') {
|
|
setSizePercentage('')
|
|
}
|
|
}}
|
|
isAllowed={withValueLimit}
|
|
/>
|
|
<div className="absolute left-4 top-1/2 -translate-y-1/2">
|
|
<TokenLogo bank={stakeBank} size={24} />
|
|
</div>
|
|
<div className="absolute right-4 top-1/2 -translate-y-1/2">
|
|
<span className="font-bold text-th-fgd-1">
|
|
{formatTokenSymbol(selectedToken)}
|
|
</span>
|
|
</div>
|
|
</div>
|
|
</div>
|
|
<div className="col-span-2 mt-2">
|
|
<ButtonGroup
|
|
activeValue={sizePercentage}
|
|
onChange={(p) => handleSizePercentage(p)}
|
|
values={['10', '25', '50', '75', '100']}
|
|
unit="%"
|
|
/>
|
|
</div>
|
|
</div>
|
|
{stakeBank && borrowBank ? (
|
|
<div className={`${stakeBank.name == 'USDC' ? '' : 'pt-8'}`}>
|
|
<Disclosure>
|
|
{({ open }) => (
|
|
<>
|
|
{stakeBank.name == 'USDC' ? (
|
|
<></>
|
|
) : (
|
|
<>
|
|
<Disclosure.Button
|
|
className={`w-full rounded-xl border-2 border-th-bkg-3 px-4 py-3 text-left focus:outline-none ${
|
|
open ? 'rounded-b-none border-b-0' : ''
|
|
}`}
|
|
>
|
|
<div className="flex items-center justify-between">
|
|
<p className="font-medium">Staked Amount</p>
|
|
<div className="flex items-center space-x-2">
|
|
<span className="text-lg font-bold text-th-fgd-1">
|
|
<FormatNumericValue
|
|
value={tokenMax.maxAmount}
|
|
decimals={stakeBank.mintDecimals}
|
|
/>
|
|
</span>
|
|
<ChevronDownIcon
|
|
className={`${
|
|
open ? 'rotate-180' : ''
|
|
} h-6 w-6 shrink-0 text-th-fgd-1`}
|
|
/>
|
|
</div>
|
|
</div>
|
|
</Disclosure.Button>
|
|
<Disclosure.Panel className="space-y-2 rounded-xl rounded-t-none border-2 border-t-0 border-th-bkg-3 px-4 pb-3">
|
|
<div className="flex justify-between">
|
|
<p className="text-th-fgd-4">Staked Amount</p>
|
|
<span className="font-bold text-th-fgd-1">
|
|
<BankAmountWithValue
|
|
amount={tokenMax.maxAmount}
|
|
bank={stakeBank}
|
|
/>
|
|
</span>
|
|
</div>
|
|
<div className="flex justify-between">
|
|
<p className="text-th-fgd-4">
|
|
Staked Amount with borrow
|
|
</p>
|
|
<span className="font-bold text-th-fgd-1">
|
|
<BankAmountWithValue
|
|
amount={tokenMax.maxAmount * leverage}
|
|
bank={stakeBank}
|
|
/>
|
|
</span>
|
|
</div>
|
|
<div className="flex justify-between">
|
|
<p className="text-th-fgd-4">Leverage</p>
|
|
<span className="font-bold text-th-fgd-1">
|
|
<FormatNumericValue
|
|
value={leverage}
|
|
decimals={2}
|
|
/>
|
|
x
|
|
</span>
|
|
</div>
|
|
<div className="flex justify-between">
|
|
<p className="text-th-fgd-4">
|
|
{borrowBank.name} borrowed
|
|
</p>
|
|
{borrowBank ? (
|
|
<span
|
|
className={`font-bold ${
|
|
borrowed > 0.001
|
|
? 'text-th-fgd-1'
|
|
: 'text-th-bkg-4'
|
|
}`}
|
|
>
|
|
<FormatNumericValue
|
|
value={borrowed}
|
|
decimals={3}
|
|
/>
|
|
</span>
|
|
) : null}
|
|
</div>
|
|
</Disclosure.Panel>
|
|
</>
|
|
)}
|
|
</>
|
|
)}
|
|
</Disclosure>
|
|
</div>
|
|
) : null}
|
|
</div>
|
|
{connected ? (
|
|
<Button
|
|
onClick={handleWithdraw}
|
|
className="w-full"
|
|
disabled={
|
|
connected &&
|
|
(!inputAmount || showInsufficientBalance || lowVaultBalance)
|
|
}
|
|
size="large"
|
|
>
|
|
{submitting ? (
|
|
<Loading className="mr-2 h-5 w-5" />
|
|
) : showInsufficientBalance ? (
|
|
<div className="flex items-center">
|
|
<ExclamationCircleIcon className="icon-shadow mr-2 h-5 w-5 shrink-0" />
|
|
{t('swap:insufficient-balance', {
|
|
symbol: formatTokenSymbol(selectedToken),
|
|
})}
|
|
</div>
|
|
) : (
|
|
`Unboost ${inputAmount} ${formatTokenSymbol(selectedToken)}`
|
|
)}
|
|
</Button>
|
|
) : (
|
|
<SecondaryConnectButton className="w-full" isLarge />
|
|
)}
|
|
{tokenPositionsFull ? (
|
|
<InlineNotification
|
|
type="error"
|
|
desc={
|
|
<>
|
|
{t('error-token-positions-full')}{' '}
|
|
<Link href="/settings" shallow>
|
|
{t('manage')}
|
|
</Link>
|
|
</>
|
|
}
|
|
/>
|
|
) : null}
|
|
{lowVaultBalance ? (
|
|
<div className="mt-4">
|
|
<InlineNotification
|
|
type="error"
|
|
desc={`The available ${selectedToken} vault balance is low. ${selectedToken} deposits are required to unboost your full position.`}
|
|
/>
|
|
</div>
|
|
) : null}
|
|
</div>
|
|
</>
|
|
)
|
|
}
|
|
|
|
export default UnstakeForm
|