From 1f7e2984dd173511c5ba5a03045d9dab514e581b Mon Sep 17 00:00:00 2001
From: Lecky Lao <leckylao@gmail.com>
Date: Sun, 19 Jul 2020 22:22:00 +1000
Subject: [PATCH] adding totalSupply for staked and token

---
 contracts/protocols/staking.sol | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/contracts/protocols/staking.sol b/contracts/protocols/staking.sol
index c47fc46..9556a25 100644
--- a/contracts/protocols/staking.sol
+++ b/contracts/protocols/staking.sol
@@ -9,6 +9,7 @@ interface TokenInterface {
 
 interface IStakingRewards {
   function balanceOf(address) external view returns (uint256);
+  function totalSupply() external view returns (uint256) {
   function earned(address) external view returns (uint256);
 }
 
@@ -36,14 +37,18 @@ contract Resolver is CurveStakingHelpers {
 
   function getPosition(address user, string memory stakingPoolName) public view returns (
     uint stakedBal,
+    uint stakedTotalSupply,
     uint rewardsEarned,
-    uint tokenBal
+    uint tokenBal,
+    uint tokenTotalSupply
   ) {
     Staking staking = Staking(getStakingAddr());
     (IStakingRewards stakingContract, TokenInterface stakingToken, bytes32 stakingType) = staking.getStakingData(stakingPoolName);
     stakedBal = stakingContract.balanceOf(user);
+    stakedBal = stakingContract.totalSupply();
     rewardsEarned = stakingContract.earned(user);
     tokenBal = stakingToken.balanceOf(user);
+    tokenBal = stakingToken.totalSupply();
   }
 }