Check QUI balance before redemption

liquity · Apr 13, 2020 · 07cd822 · 07cd822
1 parent d5709f7
commit 07cd822
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 5 deletions.
diff --git a/packages/frontend/src/App.tsx b/packages/frontend/src/App.tsx
@@ -91,7 +91,7 @@ const LiquityFrontend: React.FC<LiquityFrontendProps> = ({ loader }) => {
               <>
                 <TroveManager {...{ liquity, trove, price, total, quiBalance }} />
                 <StabilityDepositManager {...{ liquity, deposit, trove, price, quiBalance }} />
-                <RedemptionManager {...{ liquity, price }} />
+                <RedemptionManager {...{ liquity, price, quiBalance }} />
               </>
             )}
           </Box>

diff --git a/packages/frontend/src/components/RedemptionManager.tsx b/packages/frontend/src/components/RedemptionManager.tsx
@@ -14,6 +14,7 @@ type RedemptionActionProps = {
   setExchangedQui: (exchangedQui: Decimal) => void;
   changePending: boolean;
   setChangePending: (isPending: boolean) => void;
+  quiBalance: Decimal;
 };
 
 const RedemptionAction: React.FC<RedemptionActionProps> = ({
@@ -22,7 +23,8 @@ const RedemptionAction: React.FC<RedemptionActionProps> = ({
   exchangedQui,
   setExchangedQui,
   changePending,
-  setChangePending
+  setChangePending,
+  quiBalance
 }) => {
   const myTransactionId = "redemption";
   const myTransactionState = useMyTransactionState(myTransactionId);
@@ -57,7 +59,11 @@ const RedemptionAction: React.FC<RedemptionActionProps> = ({
     </Flex>
   ) : changePending ? null : (
     <Flex mt={4} justifyContent="center">
-      <Transaction id={myTransactionId} {...{ send }}>
+      <Transaction
+        id={myTransactionId}
+        requires={[[quiBalance.gte(exchangedQui), "You don't have enough QUI"]]}
+        {...{ send }}
+      >
         <Button mx={2}>Exchange {exchangedQui.prettify()} QUI</Button>
       </Transaction>
     </Flex>
@@ -67,9 +73,14 @@ const RedemptionAction: React.FC<RedemptionActionProps> = ({
 type RedemptionManagerProps = {
   liquity: Liquity;
   price: Decimal;
+  quiBalance: Decimal;
 };
 
-export const RedemptionManager: React.FC<RedemptionManagerProps> = ({ liquity, price }) => {
+export const RedemptionManager: React.FC<RedemptionManagerProps> = ({
+  liquity,
+  price,
+  quiBalance
+}) => {
   const zero = Decimal.from(0);
   const [exchangedQui, setExchangedQui] = useState(zero);
   const [changePending, setChangePending] = useState(false);
@@ -140,7 +151,15 @@ export const RedemptionManager: React.FC<RedemptionManagerProps> = ({ liquity, p
       </Box>
 
       <RedemptionAction
-        {...{ liquity, price, exchangedQui, setExchangedQui, changePending, setChangePending }}
+        {...{
+          liquity,
+          price,
+          exchangedQui,
+          setExchangedQui,
+          changePending,
+          setChangePending,
+          quiBalance
+        }}
       />
     </>
   );