feat: change other values when changing user role and owned amount

iotaledger · Nov 22, 2023 · 69b8c22 · 69b8c22
1 parent 28c2770
commit 69b8c22
Showing 1 changed file with 30 additions and 1 deletion.
diff --git a/src/components/ManaCalculator/hooks/useManaState.ts b/src/components/ManaCalculator/hooks/useManaState.ts
@@ -153,6 +153,15 @@ export function useGivenManaState(
     setState({
       ...state,
       userType: value,
+      [getStakedOrDelegated(value)]: state.heldTokens,
+      ...(value === UserType.VALIDATOR
+        ? {
+            validator: {
+              ...state.validator,
+              attractedNewDelegatedStake: state.heldTokens,
+            },
+          }
+        : {}),
     });
   }
 
@@ -162,7 +171,27 @@ export function useGivenManaState(
   }
 
   function handleOwnHoldChange(value: number) {
-    setState({ ...state, heldTokens: value });
+    setState({
+      ...state,
+      ...getDerivedValuesFromHeldTokens(value),
+    });
+  }
+
+  function getDerivedValuesFromHeldTokens(
+    heldTokens: number,
+  ): Partial<ManaCalculatorProps> {
+    return {
+      heldTokens,
+      [getStakedOrDelegated(state.userType)]: heldTokens,
+      ...(state.userType === UserType.VALIDATOR
+        ? {
+            validator: {
+              ...state.validator,
+              attractedNewDelegatedStake: heldTokens,
+            },
+          }
+        : {}),
+    };
   }
 
   const congestionAmount = getNetworkCongestion(