Add a getShare(token) view to TrueMultiFarm (#890)

Author: yuchenlintt

contracts/truefi2/TrueMultiFarm.sol

@@ -175,6 +175,14 @@ contract TrueMultiFarm is ITrueMultiFarm, UpgradeableClaimable {
         }
     }
 
+    /*
+     * What proportional share of rewards get distributed to this token?
+     * The denominator is visible in the public `shares()` view.
+     */
+    function getShare(IERC20 token) external view returns (uint256) {
+        return shares.staked[address(token)];
+    }
+
     /**
      * @dev Set shares for farms
      * Example: setShares([DAI, USDC], [1, 2]) will ensure that 33.(3)% of rewards will go to DAI farm and rest to USDC farm

test/truefi2/TrueMultiFarm.test.ts

@@ -321,6 +321,11 @@ describe('TrueMultiFarm', () => {
       await farm.setShares([secondToken.address], [3], txArgs)
     })
 
+    it('getShare shows pool shares', async () => {
+      expect(await farm.getShare(firstToken.address)).to.equal(1)
+      expect(await farm.getShare(secondToken.address)).to.equal(3)
+    })
+
     describe('one staker', () => {
       beforeEach(async () => {
         await timeTravelTo(provider, start)