Merge pull request #33 from neutral-protocol/miotto/lila-6102-update-…

…pools-to-keep-track-of-tco2-supply

[LILA-6102] use new TCO2 supply method from Pool

src/FeeCalculator.sol

@@ -10,6 +10,7 @@ import "@openzeppelin/contracts/access/Ownable.sol";
 import {SD59x18, sd, intoUint256} from "@prb/math/src/SD59x18.sol";
 
 import "./interfaces/IFeeCalculator.sol";
+import "./interfaces/IPool.sol";
 
 /// @title FeeCalculator
 /// @author Neutral Labs Inc.
@@ -230,7 +231,7 @@ contract FeeCalculator is IFeeCalculator, Ownable {
     /// @param pool The address of the pool.
     /// @return The total supply of the pool.
     function getTotalSupply(address pool) private view returns (uint256) {
-        uint256 totalSupply = IERC20(pool).totalSupply();
+        uint256 totalSupply = IPool(pool).totalTCO2Supply();
         return totalSupply;
     }
 

src/interfaces/IPool.sol

@@ -0,0 +1,16 @@
+// SPDX-FileCopyrightText: 2023 Neutral Labs Inc.
+//
+// SPDX-License-Identifier: UNLICENSED
+
+// If you encounter a vulnerability or an issue, please contact <[email protected]>
+pragma solidity ^0.8.13;
+
+/// @title IPool
+/// @author Neutral Labs Inc.
+/// @notice This interface defines methods exposed by the Pool
+interface IPool {
+    /// @notice Exposes the total TCO2 supply, tracked as the aggregation of deposit,
+    /// redemmption and bridge actions
+    /// @return supply Current supply
+    function totalTCO2Supply() external view returns (uint256 supply);
+}

test/TestUtilities.sol

@@ -24,6 +24,10 @@ contract MockPool is IERC20 {
         return _totalSupply;
     }
 
+    function totalTCO2Supply() external view returns (uint256) {
+        return _totalSupply;
+    }
+
     function setTotalSupply(uint256 ts) public {
         _totalSupply = ts;
     }