fixes requested

dOrgTech · Feb 3, 2024 · 9c73493 · 9c73493
1 parent 5beadc2
commit 9c73493
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 13 deletions.
diff --git a/components/choices/index.js b/components/choices/index.js
@@ -39,7 +39,7 @@ const updateChoiceById = async (req, response) => {
   let i = 0;
 
   try {
-    const {votesData, isXTZ} = getInputFromSigPayload(payloadBytes);
+    const { votesData } = getInputFromSigPayload(payloadBytes);
     let db_connect = dbo.getDb("Lite");
 
     const pollID = votesData[0].pollID;
@@ -95,7 +95,7 @@ const updateChoiceById = async (req, response) => {
       token.tokenID,
       block,
       address,
-      isXTZ
+      poll.isXTZ
     );
 
     if (!total) {

diff --git a/components/tokens/index.js b/components/tokens/index.js
@@ -75,7 +75,8 @@ const getVotingPowerAtLevel = async (req, response) => {
       daoContract,
       tokenID,
       level,
-      userAddress
+      userAddress,
+      isXTZ = false
     );
 
     const votingXTZWeight = await getUserTotalVotingPowerAtReferenceBlock(
@@ -84,7 +85,8 @@ const getVotingPowerAtLevel = async (req, response) => {
       daoContract,
       tokenID,
       level,
-      userAddress
+      userAddress,
+      isXTZ = true
     );
 
     response.json({ votingWeight, votingXTZWeight });

diff --git a/utils.js b/utils.js
@@ -138,18 +138,15 @@ const getUserTotalVotingPowerAtReferenceBlock = async (
   tokenID,
   level,
   userAddress,
-  isXTZ
+  isXTZ = false
 ) => {
   try {
-
+    let userVotingPower = new BigNumber(0);
     if (!isXTZ) {
-      let userVotingPower = new BigNumber(0);
-
       const isTokenDelegation = await isTokenDelegationSupported(
         network,
         address
       );
-
       if (isTokenDelegation) {
         const userVotePower = await getUserTotalVotingWeightAtBlock(
           network,
@@ -184,15 +181,12 @@ const getUserTotalVotingPowerAtReferenceBlock = async (
 
       return userVotingPower;
     } else {
-      let userVotingPower = new BigNumber(0);
-
       const selfBalance = await getUserXTZBalanceAtLevel(
         network,
         level,
         userAddress
       );
-      userVotingPower = userVotingPower.plus(selfBalance);
-      return userVotingPower;
+      return userVotingPower.plus(selfBalance);
     }
   } catch (error) {
     console.log("error: ", error);