@huma-finance/widgets 0.0.48-beta.186 → 0.0.48-beta.188

package/dist/cjs/index.cjs

@@ -4799,6 +4799,7 @@ function CheckWithdrawable(_ref) {
 
 function ChooseAmount$1(_ref) {
   var lenderPosition = _ref.lenderPosition,
+    poolBalance = _ref.poolBalance,
     poolInfo = _ref.poolInfo;
   var dispatch = useAppDispatch();
   var poolUnderlyingToken = poolInfo.poolUnderlyingToken;
@@ -4808,7 +4809,9 @@ function ChooseAmount$1(_ref) {
     _useState2 = _slicedToArray(_useState, 2),
     currentAmount = _useState2[0],
     setCurrentAmount = _useState2[1];
-  var withdrawableAmount = shared.downScale(lenderPosition, decimals);
+  // Take the minimum of the pool balance or the lender position
+  var withdrawableAmountBN = lenderPosition.lt(poolBalance) ? lenderPosition : poolBalance;
+  var withdrawableAmount = shared.downScale(withdrawableAmountBN, decimals);
   var handleChangeAmount = React.useCallback(function (newAmount) {
     setCurrentAmount(newAmount);
     dispatch(setWithdrawAmount(newAmount));
@@ -4904,17 +4907,22 @@ function LendWithdraw(_ref) {
     _useLenderPosition2 = _slicedToArray(_useLenderPosition, 2),
     lenderPosition = _useLenderPosition2[0],
     refreshLenderPosition = _useLenderPosition2[1];
+  var _usePoolBalance = shared.usePoolBalance(poolName, poolType),
+    _usePoolBalance2 = _slicedToArray(_usePoolBalance, 2),
+    poolBalance = _usePoolBalance2[0],
+    refreshPoolBalance = _usePoolBalance2[1];
   React.useEffect(function () {
-    if (!step && lenderPosition) {
+    if (!step && lenderPosition && poolBalance) {
       dispatch(setStep(WIDGET_STEP.CheckWithdrawable));
     }
-  }, [dispatch, lenderPosition, step]);
+  }, [dispatch, lenderPosition, poolBalance, step]);
   var handleWithdrawSuccess = React.useCallback(function (blockNumber) {
     refreshLenderPosition();
+    refreshPoolBalance();
     if (handleSuccess) {
       handleSuccess(blockNumber);
     }
-  }, [handleSuccess, refreshLenderPosition]);
+  }, [handleSuccess, refreshLenderPosition, refreshPoolBalance]);
   if (!poolInfo) {
     return null;
   }
@@ -4927,7 +4935,8 @@ function LendWithdraw(_ref) {
       handleClose: handleClose
     }), step === WIDGET_STEP.ChooseAmount && jsxRuntime.jsx(ChooseAmount$1, {
       poolInfo: poolInfo,
-      lenderPosition: lenderPosition
+      lenderPosition: lenderPosition,
+      poolBalance: poolBalance
     }), step === WIDGET_STEP.Transfer && jsxRuntime.jsx(Transfer$1, {
       poolInfo: poolInfo
     }), step === WIDGET_STEP.Done && jsxRuntime.jsx(Success$1, {