sort sell token list by max amount

components/swap/SwapForm.tsx

@@ -229,6 +229,7 @@ const SwapForm = () => {
                 : handleTokenOutSelect
             }
             type={showTokenSelect}
+            useMargin={useMargin}
           />
         </EnterBottomExitBottom>
         <EnterBottomExitBottom

components/swap/SwapFormTokenList.tsx

@@ -10,6 +10,8 @@ import {
   XMarkIcon,
 } from '@heroicons/react/20/solid'
 import { useTranslation } from 'next-i18next'
+import { floorToDecimal } from '../../utils/numbers'
+import Decimal from 'decimal.js'
 
 const generateSearchTerm = (item: Token, searchValue: string) => {
   const normalizedSearchValue = searchValue.toLowerCase()
@@ -40,17 +42,27 @@ const startSearch = (items: Token[], searchValue: string) => {
 const TokenItem = ({
   token,
   onSubmit,
+  useMargin,
+  type,
 }: {
   token: Token
   onSubmit: (x: string) => void
+  useMargin: boolean
+  type: string
 }) => {
   const { address, symbol, logoURI, name } = token
+
+  const isDisabled =
+    (type === 'input' && useMargin && token.maxAmountWithBorrow!.eq(0)) ||
+    (type === 'input' && !useMargin && token.maxAmount!.eq(0))
+
   return (
     <div>
       <button
         key={address}
-        className="flex w-full cursor-pointer items-center justify-between rounded-md p-2 font-normal focus:bg-th-bkg-3 focus:outline-none md:hover:bg-th-bkg-4"
+        className="default-transition flex w-full cursor-pointer items-start justify-between rounded-md p-2 font-normal focus:bg-th-bkg-3 focus:outline-none disabled:cursor-not-allowed disabled:opacity-60 md:hover:bg-th-bkg-4"
         onClick={() => onSubmit(address)}
+        disabled={isDisabled}
       >
         <div className="flex items-center">
           <picture>
@@ -59,9 +71,18 @@ const TokenItem = ({
           </picture>
           <div className="ml-2.5">
             <div className="text-left text-th-fgd-2">{symbol || 'unknown'}</div>
-            <div className="text-left text-th-fgd-4">{name || 'unknown'}</div>
+            <div className="text-left text-xs text-th-fgd-4">
+              {name || 'unknown'}
+            </div>
           </div>
         </div>
+        {type === 'input' ? (
+          <p className="text-sm text-th-fgd-2">
+            {useMargin
+              ? token.maxAmountWithBorrow!.toString()
+              : token.maxAmount!.toString()}
+          </p>
+        ) : null}
       </button>
     </div>
   )
@@ -73,10 +94,12 @@ const SwapFormTokenList = ({
   onClose,
   onTokenSelect,
   type,
+  useMargin,
 }: {
   onClose: () => void
   onTokenSelect: (x: string) => void
   type: string
+  useMargin: boolean
 }) => {
   const { t } = useTranslation()
   const [search, setSearch] = useState('')
@@ -85,6 +108,8 @@ const SwapFormTokenList = ({
   const jupiterTokens = mangoStore((s) => s.jupiterTokens)
   const inputBank = mangoStore((s) => s.swap.inputBank)
   const outputBank = mangoStore((s) => s.swap.outputBank)
+  const mangoAccount = mangoStore((s) => s.mangoAccount.current)
+  const group = mangoStore((s) => s.group)
 
   const popularTokens = useMemo(() => {
     return tokens.filter((token) => {
@@ -105,27 +130,70 @@ const SwapFormTokenList = ({
   }, [onClose])
 
   const tokenInfos = useMemo(() => {
-    if (tokens?.length) {
-      const filteredTokens = tokens.filter((token) => {
-        if (type === 'input') {
-          return token.symbol === outputBank?.name ? false : true
-        } else {
-          return token.symbol === inputBank?.name ? false : true
-        }
-      })
-      if (walletTokens?.length) {
-        const walletMints = walletTokens.map((tok) => tok.mint.toString())
-        return filteredTokens.sort(
-          (a, b) =>
-            walletMints.indexOf(b.address) - walletMints.indexOf(a.address)
+    if (
+      tokens?.length &&
+      group &&
+      mangoAccount &&
+      outputBank &&
+      type === 'input'
+    ) {
+      const filteredSortedTokens = tokens
+        .map((token) => {
+          const bank = group.banksMapByMint.get(token.address)![0]
+          const tokenBalance = floorToDecimal(
+            mangoAccount.getTokenBalanceUi(bank),
+            bank.mintDecimals
+          )
+          const maxAmountWithoutMargin = tokenBalance.gt(0)
+            ? tokenBalance
+            : new Decimal(0)
+          const maxUiAmountWithBorrow = floorToDecimal(
+            mangoAccount?.getMaxSourceUiForTokenSwap(
+              group,
+              bank.mint,
+              outputBank.mint,
+              1
+            )!,
+            bank.mintDecimals
+          )
+          const inputBankVaultBalance = group.getTokenVaultBalanceByMintUi(
+            bank.mint
+          )
+          const maxAmount = useMargin
+            ? Decimal.min(
+                maxAmountWithoutMargin,
+                inputBankVaultBalance,
+                maxUiAmountWithBorrow!
+              )
+            : Decimal.min(maxAmountWithoutMargin, inputBankVaultBalance)
+
+          const maxAmountWithBorrow = Decimal.min(
+            maxUiAmountWithBorrow!,
+            inputBankVaultBalance
+          )
+          return { ...token, maxAmount, maxAmountWithBorrow }
+        })
+        .filter((token) => (token.symbol === outputBank?.name ? false : true))
+        .sort((a, b) =>
+          useMargin
+            ? Number(b.maxAmountWithBorrow) - Number(a.maxAmountWithBorrow)
+            : Number(b.maxAmount) - Number(a.maxAmount)
         )
-      } else {
-        return filteredTokens
-      }
+
+      return filteredSortedTokens
+    } else if (tokens?.length) {
+      const filteredTokens = tokens
+        .map((token) => ({
+          ...token,
+          maxAmount: new Decimal(0),
+          maxAmountWithBorrow: new Decimal(0),
+        }))
+        .filter((token) => (token.symbol === inputBank?.name ? false : true))
+      return filteredTokens
     } else {
       return []
     }
-  }, [tokens, walletTokens, inputBank, outputBank])
+  }, [tokens, walletTokens, inputBank, outputBank, mangoAccount, group])
 
   const handleUpdateSearch = (e: ChangeEvent<HTMLInputElement>) => {
     setSearch(e.target.value)
@@ -139,7 +207,9 @@ const SwapFormTokenList = ({
       <IconButton className="absolute top-2 right-2" onClick={onClose} hideBg>
         <XMarkIcon className="h-5 w-5" />
       </IconButton>
-      <div className="flex items-center text-th-fgd-4">
+      {/* No need for search/popular tokens until we have more tokens */}
+
+      {/* <div className="flex items-center text-th-fgd-4">
         <Input
           type="text"
           placeholder="Search by token or paste address"
@@ -180,14 +250,22 @@ const SwapFormTokenList = ({
             )
           })}
         </div>
-      ) : null}
-      <div className="my-2 border-t border-th-bkg-4"></div>
+      ) : null} */}
+      {/* <div className="my-2 border-t border-th-bkg-4"></div> */}
+      <div className="mb-2 flex justify-between rounded bg-th-bkg-3 p-2">
+        <p className="text-xs text-th-fgd-4">{t('token')}</p>
+        {type === 'input' ? (
+          <p className="text-xs text-th-fgd-4">{t('max')}</p>
+        ) : null}
+      </div>
       <div className="overflow-auto">
         {sortedTokens.map((token) => (
           <TokenItem
             key={token.address}
             token={token}
             onSubmit={onTokenSelect}
+            useMargin={useMargin}
+            type={type}
           />
         ))}
       </div>

types/jupiter.ts

@@ -1,4 +1,5 @@
 import { RouteInfo } from '@jup-ag/core'
+import Decimal from 'decimal.js'
 
 export type Routes = {
   routesInfos: RouteInfo[]
@@ -16,4 +17,6 @@ export interface Token {
   extensions?: {
     coingeckoId: string
   }
+  maxAmount?: Decimal
+  maxAmountWithBorrow?: Decimal
 }