chore: allow enough balance check to return unknown state

apps/cowswap-frontend/src/api/gnosisProtocol/api.ts

@@ -14,7 +14,7 @@ import {
   PriceQuality,
   TotalSurplus,
   OrderQuoteSideKindBuy,
-  OrderQuoteSideKindSell
+  OrderQuoteSideKindSell,
 } from '@cowprotocol/cow-sdk'
 
 import { orderBookApi } from 'cowSdk'
@@ -225,7 +225,7 @@ export async function getProfileData(chainId: ChainId, address: string): Promise
   }
 }
 
-export function getPriceQuality(props: { fast?: boolean; verifyQuote: boolean }): PriceQuality {
+export function getPriceQuality(props: { fast?: boolean; verifyQuote: boolean | undefined }): PriceQuality {
   const { fast = false, verifyQuote } = props
   if (fast) {
     return PriceQuality.FAST

apps/cowswap-frontend/src/common/hooks/useNeedsApproval.ts

@@ -39,5 +39,6 @@ export function useNeedsApproval(amount: Nullish<CurrencyAmount<Currency>>): boo
     return false
   }
 
-  return !isEnoughAmount(amount, allowance)
+  const enoughBalance = isEnoughAmount(amount, allowance)
+  return enoughBalance === false
 }

apps/cowswap-frontend/src/legacy/state/orders/updaters/UnfillableOrdersUpdater.ts

@@ -145,8 +145,9 @@ export function UnfillableOrdersUpdater(): null {
 
         const currencyAmount = CurrencyAmount.fromRawAmount(order.inputToken, order.sellAmount)
         const enoughBalance = hasEnoughBalanceAndAllowance({ account, amount: currencyAmount, balances })
+        const verifiedQuote = verifiedQuotesEnabled && enoughBalance
 
-        _getOrderPrice(chainId, order, enoughBalance && verifiedQuotesEnabled, strategy)
+        _getOrderPrice(chainId, order, verifiedQuote, strategy)
           .then((quote) => {
             if (quote) {
               const [promisedPrice, promisedFee] = quote
@@ -207,7 +208,12 @@ export function UnfillableOrdersUpdater(): null {
 /**
  * Thin wrapper around `getBestPrice` that builds the params and returns null on failure
  */
-async function _getOrderPrice(chainId: ChainId, order: Order, verifyQuote: boolean, strategy: GpPriceStrategy) {
+async function _getOrderPrice(
+  chainId: ChainId,
+  order: Order,
+  verifyQuote: boolean | undefined,
+  strategy: GpPriceStrategy
+) {
   let baseToken, quoteToken
 
   const amount = getRemainderAmount(order.kind, order)

apps/cowswap-frontend/src/legacy/state/price/updater.ts

@@ -207,7 +207,7 @@ export default function FeesUpdater(): null {
       userAddress: account,
       validTo,
       isEthFlow,
-      priceQuality: getPriceQuality({ verifyQuote: enoughBalance && verifiedQuotesEnabled }),
+      priceQuality: getPriceQuality({ verifyQuote: verifiedQuotesEnabled && enoughBalance }),
     }
 
     // Don't refetch if offline.

apps/cowswap-frontend/src/modules/ordersTable/pure/OrdersTableContainer/OrderRow/index.tsx

@@ -174,7 +174,7 @@ export function OrderRow({
   const showCancellationModal = orderActions.getShowCancellationModal(order)
 
   const withWarning =
-    (!hasEnoughBalance || !hasEnoughAllowance) &&
+    (hasEnoughBalance === false || hasEnoughAllowance === false) &&
     // show the warning only for pending and scheduled orders
     (status === OrderStatus.PENDING || status === OrderStatus.SCHEDULED)
   const theme = useContext(ThemeContext)
@@ -353,10 +353,10 @@ export function OrderRow({
                     bgColor={theme.alert}
                     content={
                       <styledEl.WarningContent>
-                        {!hasEnoughBalance && (
+                        {hasEnoughBalance === false && (
                           <BalanceWarning symbol={inputTokenSymbol} isScheduled={isOrderScheduled} />
                         )}
-                        {!hasEnoughAllowance && (
+                        {hasEnoughAllowance === false && (
                           <AllowanceWarning symbol={inputTokenSymbol} isScheduled={isOrderScheduled} />
                         )}
                       </styledEl.WarningContent>

apps/cowswap-frontend/src/modules/ordersTable/pure/OrdersTableContainer/utils/getOrderParams.ts

@@ -12,8 +12,8 @@ export interface OrderParams {
   sellAmount: CurrencyAmount<Currency>
   buyAmount: CurrencyAmount<Currency>
   rateInfoParams: RateInfoParams
-  hasEnoughBalance: boolean
-  hasEnoughAllowance: boolean
+  hasEnoughBalance: boolean | undefined
+  hasEnoughAllowance: boolean | undefined
 }
 
 const PERCENTAGE_FOR_PARTIAL_FILLS = new Percent(5, 10000) // 0.05%
@@ -38,18 +38,20 @@ export function getOrderParams(
   const balance = balances[order.inputToken.address]?.value
   const allowance = allowances[order.inputToken.address]?.value
 
-  let hasEnoughBalance, hasEnoughAllowance
-
-  if (order.partiallyFillable) {
-    // When balance or allowance are undefined (loading state), show as true
-    // When loaded, check there's at least PERCENTAGE_FOR_PARTIAL_FILLS of balance/allowance to consider it as enough
-    const amount = sellAmount.multiply(PERCENTAGE_FOR_PARTIAL_FILLS)
-    hasEnoughBalance = balance === undefined || isEnoughAmount(amount, balance)
-    hasEnoughAllowance = allowance === undefined || isEnoughAmount(amount, allowance)
-  } else {
-    hasEnoughBalance = isEnoughAmount(sellAmount, balance)
-    hasEnoughAllowance = isEnoughAmount(sellAmount, allowance)
-  }
+  const [hasEnoughBalance, hasEnoughAllowance] = (() => {
+    if (order.partiallyFillable) {
+      // When balance or allowance are undefined (loading state), show as true
+      // When loaded, check there's at least PERCENTAGE_FOR_PARTIAL_FILLS of balance/allowance to consider it as enough
+      const amount = sellAmount.multiply(PERCENTAGE_FOR_PARTIAL_FILLS)
+      const hasEnoughBalance = isEnoughAmount(amount, balance)
+      const hasEnoughAllowance = isEnoughAmount(amount, allowance)
+      return [hasEnoughBalance, hasEnoughAllowance]
+    } else {
+      const hasEnoughBalance = isEnoughAmount(sellAmount, balance)
+      const hasEnoughAllowance = isEnoughAmount(sellAmount, allowance)
+      return [hasEnoughBalance, hasEnoughAllowance]
+    }
+  })()
 
   return {
     chainId,

apps/cowswap-frontend/src/modules/tokens/hooks/useEnoughBalance.ts

@@ -31,7 +31,7 @@ export interface UseEnoughBalanceParams {
  * @param params Parameters to check balance and optionally the allowance
  * @returns true if the account has enough balance (and allowance if it applies)
  */
-export function useEnoughBalanceAndAllowance(params: UseEnoughBalanceParams): boolean {
+export function useEnoughBalanceAndAllowance(params: UseEnoughBalanceParams): boolean | undefined {
   const { account, amount, checkAllowanceAddress } = params
   const isNativeCurrency = amount?.currency.isNative
   const token = amount?.currency.wrapped
@@ -75,22 +75,37 @@ export interface EnoughBalanceParams extends Omit<UseEnoughBalanceParams, 'check
  * @param params Parameters to check balance and optionally the allowance
  * @returns true if the account has enough balance (and allowance if it applies)
  */
-export function hasEnoughBalanceAndAllowance(params: EnoughBalanceParams): boolean {
+export function hasEnoughBalanceAndAllowance(params: EnoughBalanceParams): boolean | undefined {
   const { account, amount, balances, nativeBalance, allowances } = params
+
+  if (!account || !amount) {
+    return undefined
+  }
+
   const isNativeCurrency = amount?.currency.isNative
   const token = amount?.currency.wrapped
   const tokenAddress = getAddress(token)
 
-  const balance = tokenAddress ? balances[tokenAddress]?.value : undefined
-  const allowance = (tokenAddress && allowances && allowances[tokenAddress]?.value) || undefined
-
-  if (!account || !amount) {
-    return false
+  const enoughBalance = (() => {
+    const balance = tokenAddress ? balances[tokenAddress]?.value : undefined
+    const balanceAmount = isNativeCurrency ? nativeBalance : balance || undefined
+    return isEnoughAmount(amount, balanceAmount)
+  })()
+
+  const enoughAllowance = (() => {
+    if (!tokenAddress || !allowances) {
+      return undefined
+    }
+    if (isNativeCurrency) {
+      return true
+    }
+    const allowance = allowances[tokenAddress]?.value
+    return allowance && isEnoughAmount(amount, allowance)
+  })()
+
+  if (enoughBalance === undefined || enoughAllowance === undefined) {
+    return undefined
   }
 
-  const balanceAmount = isNativeCurrency ? nativeBalance : balance || undefined
-  const enoughBalance = isEnoughAmount(amount, balanceAmount)
-  const enoughAllowance = !allowances || isNativeCurrency || (allowance && isEnoughAmount(amount, allowance)) || false
-
   return enoughBalance && enoughAllowance
 }

apps/cowswap-frontend/src/modules/tradeQuote/hooks/useQuoteParams.ts

@@ -45,7 +45,7 @@ export function useQuoteParams(amount: string | null): LegacyFeeQuoteParams | un
       toDecimals,
       fromDecimals,
       isEthFlow: false,
-      priceQuality: getPriceQuality({ verifyQuote: enoughBalance && verifiedQuotesEnabled }),
+      priceQuality: getPriceQuality({ verifyQuote: verifiedQuotesEnabled && enoughBalance }),
     }
 
     return params

apps/cowswap-frontend/src/utils/isEnoughAmount.ts

@@ -3,10 +3,8 @@ import { Currency, CurrencyAmount } from '@uniswap/sdk-core'
 export function isEnoughAmount(
   sellAmount: CurrencyAmount<Currency>,
   targetAmount: CurrencyAmount<Currency> | undefined
-): boolean {
-  if (!targetAmount) return true
+): boolean | undefined {
+  if (!targetAmount) return undefined
 
-  if (targetAmount.equalTo(sellAmount)) return true
-
-  return sellAmount.lessThan(targetAmount)
+  return sellAmount.equalTo(targetAmount) || sellAmount.lessThan(targetAmount)
 }