feat: update updatePool function

primitivefinance · Dec 12, 2023 · 20a059e · 20a059e
1 parent 8ff06bd
commit 20a059e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/contracts/strategies/G3MStrategy.sol b/contracts/strategies/G3MStrategy.sol
@@ -67,7 +67,14 @@ contract G3MStrategy is IG3MStrategy {
     ) external onlyPortfolio {
         if (caller != configs[poolId].controller) revert NotController();
 
-        // TODO: Add some actual features here
+        (uint256 endWeightX, uint256 endUpdate) =
+            abi.decode(data, (uint256, uint256));
+        (uint256 weightX,) = computeWeights(poolId);
+
+        configs[poolId].startWeightX = weightX;
+        configs[poolId].endWeightX = block.timestamp;
+        configs[poolId].endWeightX = endWeightX;
+        configs[poolId].endUpdate = endUpdate;
     }
 
     /// @inheritdoc IStrategy