From c3dbd98460217e361908e70ba306acbc8a88dbd6 Mon Sep 17 00:00:00 2001
From: Pedro Rezende <pedrorezende@users.noreply.github.com>
Date: Thu, 15 Aug 2024 19:33:27 -0300
Subject: [PATCH] feat(namadillo): adding validation to unstake amounts (#994)

---
 apps/namadillo/src/App/Staking/Unstake.tsx            |  6 ++++++
 .../namadillo/src/App/Staking/UnstakeBondingTable.tsx | 11 +++++++++--
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/apps/namadillo/src/App/Staking/Unstake.tsx b/apps/namadillo/src/App/Staking/Unstake.tsx
index e99b9f86b..c28ab4487 100644
--- a/apps/namadillo/src/App/Staking/Unstake.tsx
+++ b/apps/namadillo/src/App/Staking/Unstake.tsx
@@ -138,6 +138,12 @@ const Unstake = (): JSX.Element => {
 
   const validationMessage = ((): string => {
     if (totalStakedAmount.lt(totalUpdatedAmount)) return "Invalid amount";
+
+    for (const address in updatedAmountByAddress) {
+      if (stakedAmountByAddress[address].lt(updatedAmountByAddress[address])) {
+        return "Invalid amount";
+      }
+    }
     return "";
   })();
 
diff --git a/apps/namadillo/src/App/Staking/UnstakeBondingTable.tsx b/apps/namadillo/src/App/Staking/UnstakeBondingTable.tsx
index 018b10d26..1d749e5a4 100644
--- a/apps/namadillo/src/App/Staking/UnstakeBondingTable.tsx
+++ b/apps/namadillo/src/App/Staking/UnstakeBondingTable.tsx
@@ -70,6 +70,7 @@ export const UnstakeBondingTable = ({
       updatedAmountByAddress[validator.address] ?? new BigNumber(0);
 
     const hasNewAmounts = amountToUnstake.gt(0);
+    const newAmount = stakedAmount.minus(amountToUnstake);
 
     return {
       className: "",
@@ -114,9 +115,15 @@ export const UnstakeBondingTable = ({
             <NamCurrency amount={stakedAmount} />
           </span>
           {hasNewAmounts && (
-            <span className="text-orange text-sm">
+            <span
+              className={twMerge(
+                clsx("text-orange text-sm", {
+                  "text-fail": newAmount.lt(0),
+                })
+              )}
+            >
               =
-              <NamCurrency amount={stakedAmount.minus(amountToUnstake)} />
+              <NamCurrency amount={newAmount} />
             </span>
           )}
         </div>,