Change Cosmos Validator list + delegate

LedgerHQ · May 13, 2022 · 525116f · 525116f
1 parent 5b012ce
commit 525116f
Show file tree

Hide file tree

Showing 10 changed files with 222 additions and 38 deletions.
diff --git a/src/families/cosmos/js-buildTransaction.ts b/src/families/cosmos/js-buildTransaction.ts
@@ -53,22 +53,24 @@ export const buildTransaction = async (
       if (!transaction.validators || transaction.validators.length < 1) {
         isComplete = false;
       } else {
-        transaction.validators.forEach((validator) => {
-          if (!validator.address || validator.amount.lte(0)) {
-            isComplete = false;
-          }
-
-          msg.push({
-            typeUrl: "/cosmos.staking.v1beta1.MsgDelegate",
-            value: {
-              delegatorAddress: account.freshAddress,
-              validatorAddress: validator.address,
-              amount: {
-                denom: account.currency.units[1].code,
-                amount: validator.amount.toString(),
-              },
+        const validator = transaction.validators[0];
+        if (!validator) {
+          isComplete = false;
+          break;
+        } else if (!validator.address || transaction.amount.lte(0)) {
+          isComplete = false;
+        }
+
+        msg.push({
+          typeUrl: "/cosmos.staking.v1beta1.MsgDelegate",
+          value: {
+            delegatorAddress: account.freshAddress,
+            validatorAddress: validator.address,
+            amount: {
+              denom: account.currency.units[1].code,
+              amount: transaction.amount.toString(),
             },
-          });
+          },
         });
       }
       break;

diff --git a/src/families/cosmos/js-getTransactionStatus.ts b/src/families/cosmos/js-getTransactionStatus.ts
@@ -146,25 +146,19 @@ const getDelegateTransactionStatus = async (
     errors.validators = new CosmosTooManyValidators();
   }
 
-  let amount = t.validators.reduce(
-    (old, current) => old.plus(current.amount),
-    new BigNumber(0)
-  );
-
-  if (amount.eq(0)) {
-    errors.amount = new AmountRequired();
-  }
-
   const estimatedFees = t.fees || new BigNumber(0);
 
-  if (!t.fees) {
+  if (!t.fees || !t.fees.gt(0)) {
     errors.fees = new FeeNotLoaded();
   }
 
-  let totalSpent = amount.plus(estimatedFees);
+  const amount = t.useAllAmount
+    ? getMaxEstimatedBalance(a, estimatedFees)
+    : t.amount;
+  const totalSpent = amount.plus(estimatedFees);
 
-  if (totalSpent.eq(a.spendableBalance)) {
-    warnings.delegate = new CosmosDelegateAllFundsWarning();
+  if (amount.eq(0)) {
+    errors.amount = new AmountRequired();
   }
 
   if (
@@ -173,8 +167,10 @@ const getDelegateTransactionStatus = async (
     (amount.lt(0) || totalSpent.gt(a.spendableBalance))
   ) {
     errors.amount = new NotEnoughBalance();
-    amount = new BigNumber(0);
-    totalSpent = new BigNumber(0);
+  }
+
+  if (!errors.amount && t.useAllAmount) {
+    warnings.amount = new CosmosDelegateAllFundsWarning();
   }
 
   return Promise.resolve({

diff --git a/src/families/cosmos/logic.ts b/src/families/cosmos/logic.ts
@@ -13,7 +13,7 @@ import type {
   CosmosRedelegation,
   CosmosMappedRedelegation,
 } from "./types";
-import type { Unit, Account } from "../../types";
+import type { Unit, Account, Transaction } from "../../types";
 import { calculateFees } from "./js-prepareTransaction";
 
 export const COSMOS_MAX_REDELEGATIONS = 7;
@@ -99,16 +99,21 @@ export function mapRedelegations(
 export const mapDelegationInfo = (
   delegations: CosmosDelegationInfo[],
   validators: CosmosValidatorItem[],
-  unit: Unit
+  unit: Unit,
+  transaction?: Transaction
 ): CosmosMappedDelegationInfo[] => {
   return delegations.map((d) => ({
     ...d,
     validator: validators.find((v) => v.validatorAddress === d.address),
-    formattedAmount: formatCurrencyUnit(unit, d.amount, {
-      disableRounding: true,
-      alwaysShowSign: false,
-      showCode: true,
-    }),
+    formattedAmount: formatCurrencyUnit(
+      unit,
+      transaction ? transaction.amount : d.amount,
+      {
+        disableRounding: true,
+        alwaysShowSign: false,
+        showCode: true,
+      }
+    ),
   }));
 };
 export const formatValue = (value: BigNumber, unit: Unit): number =>