remove fieldsToUpdate undefined

bancorprotocol · Dec 18, 2023 · daa2d2b · daa2d2b
1 parent 894498d
commit daa2d2b
Showing 1 changed file with 23 additions and 8 deletions.
diff --git a/src/components/strategies/useUpdateStrategy.ts b/src/components/strategies/useUpdateStrategy.ts
@@ -1,3 +1,4 @@
+import { StrategyUpdate } from '@bancor/carbon-sdk';
 import { MarginalPriceOptions } from '@bancor/carbon-sdk/strategy-management';
 import { useNotifications } from 'hooks/useNotifications';
 import {
@@ -173,14 +174,21 @@ export const useUpdateStrategy = () => {
       throw new Error('error in withdraw strategy budget: missing data ');
     }
 
+    const fieldsToUpdate: StrategyUpdate = Object.assign({});
+
+    if (buyMarginalPrice) {
+      fieldsToUpdate.buyBudget = order0.balance;
+    }
+
+    if (sellMarginalPrice) {
+      fieldsToUpdate.sellBudget = order1.balance;
+    }
+
     updateMutation.mutate(
       {
         id,
         encoded,
-        fieldsToUpdate: {
-          buyBudget: order0.balance,
-          sellBudget: order1.balance,
-        },
+        fieldsToUpdate,
         buyMarginalPrice,
         sellMarginalPrice,
       },
@@ -219,14 +227,21 @@ export const useUpdateStrategy = () => {
       throw new Error('error in deposit strategy budget: missing data');
     }
 
+    const fieldsToUpdate: StrategyUpdate = Object.assign({});
+
+    if (buyMarginalPrice) {
+      fieldsToUpdate.buyBudget = order0.balance;
+    }
+
+    if (sellMarginalPrice) {
+      fieldsToUpdate.sellBudget = order1.balance;
+    }
+
     updateMutation.mutate(
       {
         id,
         encoded,
-        fieldsToUpdate: {
-          buyBudget: order0.balance,
-          sellBudget: order1.balance,
-        },
+        fieldsToUpdate,
         buyMarginalPrice,
         sellMarginalPrice,
       },