Add max balance for trading based on trasanction cost

apps/main/src/modules/borrow/BorrowContextProvider.tsx

@@ -15,7 +15,9 @@ import { ApyProvider } from "@/modules/borrow/context/ApyContext"
 import { useExternalApyData } from "@/modules/borrow/hooks/useExternalApyData"
 import { useFormatReserve } from "@/modules/borrow/hooks/useFormatReserve"
 import { useCreateBatchTx } from "@/modules/transactions/hooks/useBatchTx"
+import { useMaxBalance } from "@/modules/transactions/hooks/useMaxBalance"
 import { transformEvmCallToPapiTx } from "@/modules/transactions/utils/tx"
+import { useAssets } from "@/providers/assetsProvider"
 import { useRpcProvider } from "@/providers/rpcProvider"
 import { useTransactionsStore } from "@/states/transactions"
 
@@ -37,6 +39,7 @@ export const BorrowContextProvider: React.FC<PropsWithChildren> = ({
   const createBatchTx = useCreateBatchTx()
   const { evm, dataEnv, papi } = useRpcProvider()
   const squidClient = useSquidClient()
+  const { getRelatedAToken } = useAssets()
   const { market } = useSearch({ from: "/borrow" })
 
   const selectedMarket = market || defaultMarketByEnv[dataEnv]
@@ -61,18 +64,25 @@ export const BorrowContextProvider: React.FC<PropsWithChildren> = ({
     [createTransaction, createBatchTx, papi],
   )
 
+  const getRelatedATokenId = useCallback(
+    (id: string) => getRelatedAToken(id)?.id,
+    [getRelatedAToken],
+  )
+
   return (
     <ApyProvider>
       <MoneyMarketProvider
         market={selectedMarket}
         provider={evm.transport}
         squidClient={squidClient}
         onCreateTransaction={createTx}
+        useMaxBalance={useMaxBalance}
         formatCurrency={createFormatterFn(t, "currency")}
         formatNumber={createFormatterFn(t, "number")}
         formatPercent={createFormatterFn(t, "percent")}
         formatReserve={useFormatReserve()}
         externalApyData={useExternalApyData()}
+        getRelatedATokenId={getRelatedATokenId}
       >
         {children}
       </MoneyMarketProvider>

apps/main/src/modules/liquidity/components/AddIsolatedLiquidity/AddIsolatedLiquidity.tsx

@@ -69,12 +69,11 @@ export const AddIsolatedLiquidityForm = ({
 
   const {
     form,
+    getMaxBalance,
     reserveA,
     reserveB,
     ratio,
     mutation,
-    assetABalance,
-    assetBBalance,
     assetAMeta,
     assetBMeta,
     shares,
@@ -127,7 +126,7 @@ export const AddIsolatedLiquidityForm = ({
               assetFieldName="assetA"
               amountFieldName="amountA"
               assets={[]}
-              maxBalance={assetABalance}
+              maxBalance={getMaxBalance(assetAMeta)}
               disabledAssetSelector
               onAmountChange={(value) => {
                 form.setValue("lastUpdated", "assetA")
@@ -166,7 +165,7 @@ export const AddIsolatedLiquidityForm = ({
               assetFieldName="assetB"
               amountFieldName="amountB"
               assets={[]}
-              maxBalance={assetBBalance}
+              maxBalance={getMaxBalance(assetBMeta)}
               disabledAssetSelector
               onAmountChange={(value) => {
                 form.setValue("lastUpdated", "assetB")

apps/main/src/modules/liquidity/components/AddIsolatedLiquidity/AddIsolatedLiquidity.utils.ts

@@ -15,12 +15,15 @@ import { PoolToken } from "@/api/pools"
 import { spotPriceQuery } from "@/api/spotPrice"
 import { TXYKConsts, XYKPoolWithLiquidity } from "@/api/xyk"
 import { useXYKFarmMinShares } from "@/modules/liquidity/components/JoinFarms/JoinFarms.utils"
+import {
+  useFormMaxBalanceWithFee,
+  ValidateFormMaxBalanceWithFee,
+} from "@/modules/transactions/hooks/useFormMaxBalanceWithFee"
 import { TShareToken } from "@/providers/assetsProvider"
 import { useRpcProvider } from "@/providers/rpcProvider"
-import { useAccountBalances } from "@/states/account"
 import { useTransactionsStore } from "@/states/transactions"
 import { scale, scaleHuman } from "@/utils/formatting"
-import { positive, required, validateFieldMaxBalance } from "@/utils/validators"
+import { positive, required } from "@/utils/validators"
 
 export const orders = ["assetA", "assetB"] as const
 type Order = (typeof orders)[number]
@@ -35,17 +38,16 @@ export type TAddIsolatedLiquidityFormValues = {
 }
 
 export const useAddIsolatedLiquidityZod = (
-  balanceA: string,
-  balanceB: string,
+  validateBalance: ValidateFormMaxBalanceWithFee,
   minTradingLimit: bigint,
   minPoolLiquidity: bigint,
 ) => {
   const { t } = useTranslation("liquidity")
 
   return z
     .object({
-      amountA: required.pipe(positive).check(validateFieldMaxBalance(balanceA)),
-      amountB: required.check(validateFieldMaxBalance(balanceB)),
+      amountA: required.pipe(positive),
+      amountB: required,
       assetA: z.custom<TAssetData>(),
       assetB: z.custom<TAssetData>(),
       lastUpdated: z.literal(["assetA", "assetB"]),
@@ -55,6 +57,10 @@ export const useAddIsolatedLiquidityZod = (
           error: t("liquidity.add.modal.validation.minPoolLiquidity"),
         }),
     })
+    .check(
+      validateBalance("amountA", (form) => [form.assetA, form.amountA]),
+      validateBalance("amountB", (form) => [form.assetB, form.amountB]),
+    )
     .refine(
       ({ amountA, assetA }) => {
         const minAssetATradingLimit = Big(
@@ -104,7 +110,6 @@ export const useAddIsolatedLiquidity = ({
   const { t } = useTranslation("liquidity")
   const { papi } = useRpcProvider()
   const createTransaction = useTransactionsStore((s) => s.createTransaction)
-  const { getTransferableBalance } = useAccountBalances()
   const { data: farms } = useIsolatedPoolFarms(shareTokenMeta.poolAddress)
   const activeFarms = farms?.filter((farm) => farm.apr !== "0") ?? []
   const isFarms = activeFarms.length > 0
@@ -117,22 +122,22 @@ export const useAddIsolatedLiquidity = ({
 
   const [assetAMeta, assetBMeta] = shareTokenMeta.assets
 
-  const assetABalance = scaleHuman(
-    getTransferableBalance(assetAMeta.id),
-    assetAMeta.decimals,
-  )
-  const assetBBalance = scaleHuman(
-    getTransferableBalance(assetBMeta.id),
-    assetBMeta.decimals,
-  )
-
   const { data: spotPriceData, isPending: isSpotPricePending } = useQuery(
     spotPriceQuery(rpc, assetAMeta.id, assetBMeta.id),
   )
 
+  const addLiquidityTx = papi.tx.XYK.add_liquidity({
+    asset_a: Number(assetAMeta.id),
+    asset_b: Number(assetBMeta.id),
+    amount_a: BigInt(scale("1", assetAMeta.decimals)),
+    amount_b_max_limit: BigInt(scale("1", assetBMeta.decimals)),
+  })
+
+  const { validateBalance, getMaxBalance } =
+    useFormMaxBalanceWithFee(addLiquidityTx)
+
   const zodSchema = useAddIsolatedLiquidityZod(
-    assetABalance,
-    assetBBalance,
+    validateBalance,
     pool.minTradingLimit,
     consts.minPoolLiquidity,
   )
@@ -236,12 +241,11 @@ export const useAddIsolatedLiquidity = ({
 
   return {
     form,
+    getMaxBalance,
     reserveA,
     reserveB,
     ratio,
     mutation,
-    assetABalance,
-    assetBBalance,
     assetAMeta,
     assetBMeta,
     shares,

apps/main/src/modules/liquidity/components/AddLiquidity/AddLiqudity.utils.ts

@@ -21,15 +21,15 @@ import {
   useOmnipoolAssetsData,
   useOmnipoolMinLiquidity,
 } from "@/api/omnipool"
-import i18n from "@/i18n"
 import { useMinOmnipoolFarmJoin } from "@/modules/liquidity/components/JoinFarms/JoinFarms.utils"
+import { useFormMaxBalanceWithFee } from "@/modules/transactions/hooks/useFormMaxBalanceWithFee"
+import { AnyTransaction } from "@/modules/transactions/types"
 import { useAssets } from "@/providers/assetsProvider"
 import { useRpcProvider } from "@/providers/rpcProvider"
-import { useAccountBalances } from "@/states/account"
 import { useTradeSettings } from "@/states/tradeSettings"
 import { useTransactionsStore } from "@/states/transactions"
-import { scale, scaleHuman, toDecimal } from "@/utils/formatting"
-import { positive, required, validateMaxBalance } from "@/utils/validators"
+import { scale, scaleHuman } from "@/utils/formatting"
+import { positive, required } from "@/utils/validators"
 
 export type TAddLiquidityFormValues = { amount: string; asset: TAssetData }
 
@@ -200,38 +200,30 @@ export const useAddLiquidityForm = ({
   initialAmount,
   selectedAsset,
   rule,
+  tx,
 }: {
   selectedAsset: TAssetData
   initialAmount?: string
   rule?: ZodType<string, string> | undefined
+  tx: AnyTransaction | null
 }) => {
-  const { getTransferableBalance } = useAccountBalances()
+  const { validateBalance, getMaxBalance } = useFormMaxBalanceWithFee(tx, 3)
 
   const form = useForm<TAddLiquidityFormValues>({
     mode: "onChange",
     defaultValues: { amount: initialAmount ?? "", asset: selectedAsset },
     resolver: rule
       ? standardSchemaResolver(
-          z.object({ amount: rule, asset: z.custom<TAssetData>() }).refine(
-            (values) => {
-              return validateMaxBalance(
-                toDecimal(
-                  getTransferableBalance(values.asset.id),
-                  values.asset.decimals,
-                ),
-                values.amount,
-              )
-            },
-            {
-              message: i18n.t("error.maxBalance"),
-              path: ["amount"],
-            },
-          ),
+          z
+            .object({ amount: rule, asset: z.custom<TAssetData>() })
+            .check(
+              validateBalance("amount", (form) => [form.asset, form.amount]),
+            ),
         )
       : undefined,
   })
 
-  return form
+  return { form, getMaxBalance }
 }
 
 export const useAddLiquidity = ({
@@ -264,19 +256,26 @@ export const useAddLiquidity = ({
     : false
 
   const addLiquidityZod = useAddToOmnipoolZod(poolId)
-  const underlyingAssetId = isErc20Asset
-    ? getErc20AToken(poolId)?.underlyingAssetId
-    : undefined
 
-  const form = useAddLiquidityForm({
+  const addLiquidityTx = papi.tx.Omnipool.add_liquidity_with_limit({
+    asset: Number(poolId),
+    amount: BigInt(scale("1", poolMeta.decimals)),
+    min_shares_limit: 0n,
+  })
+
+  const { form, getMaxBalance } = useAddLiquidityForm({
     initialAmount: "",
     selectedAsset: poolMeta,
     rule: addLiquidityZod,
+    tx: addLiquidityTx,
   })
 
   const amount = Big(form.watch("amount") || "0")
   const selectedAsset = form.watch("asset")
   const isErc20SelectedAsset = isErc20AToken(selectedAsset)
+  const underlyingAssetId = isErc20Asset
+    ? getErc20AToken(poolId)?.underlyingAssetId
+    : undefined
 
   const { isJoinFarms, joinFarmErrorMessage, activeFarms } =
     useCheckJoinOmnipoolFarm({
@@ -380,6 +379,7 @@ export const useAddLiquidity = ({
 
   return {
     form,
+    getMaxBalance,
     liquidityShares,
     isFarms,
     activeFarms,

apps/main/src/modules/liquidity/components/AddLiquidity/AddLiquidity.tsx

@@ -47,6 +47,7 @@ export const AddLiquidity: FC<AddLiquidityProps> = ({
 
   const {
     form,
+    getMaxBalance,
     liquidityShares,
     poolMeta,
     activeFarms,
@@ -58,7 +59,7 @@ export const AddLiquidity: FC<AddLiquidityProps> = ({
     healthFactor,
   } = useAddLiquidity({ poolId: id, onSubmitted })
 
-  const { formState, handleSubmit } = form
+  const { formState, handleSubmit, watch } = form
   const customErrors = getCustomErrors(formState.errors.amount)
 
   return (
@@ -75,6 +76,7 @@ export const AddLiquidity: FC<AddLiquidityProps> = ({
             assetFieldName="asset"
             amountFieldName="amount"
             assets={underlyingAssetMeta ? [underlyingAssetMeta, poolMeta] : []}
+            maxBalance={getMaxBalance(watch("asset"))}
             sx={{ pt: 0 }}
             disabledAssetSelector={!underlyingAssetMeta}
           />