feat: add caller param to updatePool

primitivefinance · Dec 4, 2023 · 9334c05 · 9334c05
1 parent c41b120
commit 9334c05
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 3 deletions.
diff --git a/contracts/Portfolio.sol b/contracts/Portfolio.sol
@@ -299,7 +299,7 @@ contract Portfolio is ERC1155, IPortfolio {
     }
 
     function updatePool(uint64 poolId, bytes memory data) external {
-        IStrategy(getStrategy(poolId)).updatePool(poolId, data);
+        IStrategy(getStrategy(poolId)).updatePool(poolId, msg.sender, data);
     }
 
     /// @inheritdoc IPortfolioActions

diff --git a/contracts/interfaces/IStrategy.sol b/contracts/interfaces/IStrategy.sol
@@ -86,5 +86,9 @@ interface IStrategy is IPortfolioStrategy {
         uint256 reserveY
     ) external view returns (bool, int256);
 
-    function updatePool(uint64 poolId, bytes memory data) external;
+    function updatePool(
+        uint64 poolId,
+        address caller,
+        bytes memory data
+    ) external;
 }
diff --git a/contracts/strategies/G3MStrategy.sol b/contracts/strategies/G3MStrategy.sol
@@ -44,6 +44,8 @@ contract G3MStrategy is IG3MStrategy {
         return configs[poolId].weightX != 0;
     }
 
+    function updatePool(uint64, address, bytes memory) external { }
+
     /// @inheritdoc IStrategy
     function beforeSwap(
         uint64 poolId,

diff --git a/contracts/strategies/NormalStrategy.sol b/contracts/strategies/NormalStrategy.sol
@@ -107,7 +107,7 @@ contract NormalStrategy is INormalStrategy {
     error CannotUpdatePool();
 
     /// @inheritdoc IStrategy
-    function updatePool(uint64, bytes memory) external {
+    function updatePool(uint64, address, bytes memory) external {
         revert CannotUpdatePool();
     }