feat: add name param to strategy contracts

primitivefinance · Dec 15, 2023 · e3594aa · e3594aa
1 parent 17da0fb
commit e3594aa
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/contracts/interfaces/IStrategy.sol b/contracts/interfaces/IStrategy.sol
@@ -19,6 +19,8 @@ interface IStrategy is IPortfolioStrategy {
     // ====== Required ====== //
     // Includes: IPortfolioStrategy
 
+    function name() external view returns (string memory);
+
     /**
      * @notice
      * Called by the Portfolio contract after a pool is created.

diff --git a/contracts/strategies/G3MStrategy.sol b/contracts/strategies/G3MStrategy.sol
@@ -13,6 +13,9 @@ contract G3MStrategy is IG3MStrategy {
 
     address public immutable portfolio;
 
+    /// @inheritdoc IStrategy
+    string public constant name = "G3MStrategy";
+
     mapping(uint64 => IG3MStrategy.Config) public configs;
 
     modifier onlyPortfolio() {

diff --git a/contracts/strategies/NormalStrategy.sol b/contracts/strategies/NormalStrategy.sol
@@ -31,6 +31,9 @@ contract NormalStrategy is INormalStrategy {
     /// @dev Canonical Portfolio smart contract.
     address public immutable portfolio;
 
+    /// @inheritdoc IStrategy
+    string public constant name = "NormalStrategy";
+
     /// @dev Tracks each pool strategy configuration.
     mapping(uint64 poolId => PortfolioConfig config) public configs;