Balancer Contract Registry initialization contract (#1327)

Author: EndymionJkb

pkg/governance-scripts/contracts/BalancerContractRegistryInitializer.sol

@@ -0,0 +1,121 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+pragma solidity ^0.8.24;
+
+import { IAuthentication } from "@balancer-labs/v3-interfaces/contracts/solidity-utils/helpers/IAuthentication.sol";
+import { IBasicAuthorizer } from "@balancer-labs/v3-interfaces/contracts/governance-scripts/IBasicAuthorizer.sol";
+import {
+    IBalancerContractRegistry,
+    ContractType
+} from "@balancer-labs/v3-interfaces/contracts/standalone-utils/IBalancerContractRegistry.sol";
+import { IVault } from "@balancer-labs/v3-interfaces/contracts/vault/IVault.sol";
+
+import { SingletonAuthentication } from "@balancer-labs/v3-vault/contracts/SingletonAuthentication.sol";
+import { InputHelpers } from "@balancer-labs/v3-solidity-utils/contracts/helpers/InputHelpers.sol";
+
+// Associated with `20250221-balancer-registry-initializer`.
+contract BalancerContractRegistryInitializer {
+    IBalancerContractRegistry public immutable balancerContractRegistry;
+
+    // IAuthorizer with interface for granting/revoking roles.
+    IBasicAuthorizer internal immutable _authorizer;
+
+    // Set to true when operation is complete.
+    bool private _initialized;
+
+    string[] private _routerNames;
+    address[] private _routerAddresses;
+
+    string[] private _poolFactoryNames;
+    address[] private _poolFactoryAddresses;
+
+    string[] private _aliasNames;
+    address[] private _aliasAddresses;
+
+    /// @notice The initialization can only be done once.
+    error AlreadyInitialized();
+
+    /// @notice The Vault passed in as a sanity check doesn't match the Vault associated with the registry.
+    error VaultMismatch();
+
+    constructor(
+        IVault vault,
+        IBalancerContractRegistry balancerContractRegistry_,
+        string[] memory routerNames,
+        address[] memory routerAddresses,
+        string[] memory poolFactoryNames,
+        address[] memory poolFactoryAddresses,
+        string[] memory aliasNames,
+        address[] memory aliasAddresses
+    ) {
+        InputHelpers.ensureInputLengthMatch(_routerNames.length, _routerAddresses.length);
+        InputHelpers.ensureInputLengthMatch(_poolFactoryNames.length, _poolFactoryAddresses.length);
+        InputHelpers.ensureInputLengthMatch(_aliasNames.length, _aliasAddresses.length);
+
+        // Extract the Vault (also indirectly verifying the registry contract is valid).
+        IVault registryVault = SingletonAuthentication(address(balancerContractRegistry_)).getVault();
+        if (registryVault != vault) {
+            revert VaultMismatch();
+        }
+
+        balancerContractRegistry = balancerContractRegistry_;
+
+        _routerNames = routerNames;
+        _routerAddresses = routerAddresses;
+        _poolFactoryNames = poolFactoryNames;
+        _poolFactoryAddresses = poolFactoryAddresses;
+        _aliasNames = aliasNames;
+        _aliasAddresses = aliasAddresses;
+
+        _authorizer = IBasicAuthorizer(address(vault.getAuthorizer()));
+    }
+
+    function initializeBalancerContractRegistry() external {
+        // Explicitly ensure this can only be called once.
+        if (_initialized) {
+            revert AlreadyInitialized();
+        }
+
+        _initialized = true;
+
+        // Grant permissions to register contracts and add aliases.
+        bytes32 registerContractRole = IAuthentication(address(balancerContractRegistry)).getActionId(
+            IBalancerContractRegistry.registerBalancerContract.selector
+        );
+        bytes32 addAliasRole = IAuthentication(address(balancerContractRegistry)).getActionId(
+            IBalancerContractRegistry.addOrUpdateBalancerContractAlias.selector
+        );
+
+        _authorizer.grantRole(registerContractRole, address(this));
+        _authorizer.grantRole(addAliasRole, address(this));
+
+        // Add Routers.
+        for (uint256 i = 0; i < _routerNames.length; ++i) {
+            balancerContractRegistry.registerBalancerContract(
+                ContractType.ROUTER,
+                _routerNames[i],
+                _routerAddresses[i]
+            );
+        }
+
+        // Add Pool Factories.
+        for (uint256 i = 0; i < _poolFactoryNames.length; ++i) {
+            balancerContractRegistry.registerBalancerContract(
+                ContractType.POOL_FACTORY,
+                _poolFactoryNames[i],
+                _poolFactoryAddresses[i]
+            );
+        }
+
+        // Add (pool factory) aliases.
+        for (uint256 i = 0; i < _aliasNames.length; ++i) {
+            balancerContractRegistry.addOrUpdateBalancerContractAlias(_aliasNames[i], _aliasAddresses[i]);
+        }
+
+        // Renounce all roles.
+        _authorizer.renounceRole(registerContractRole, address(this));
+        _authorizer.renounceRole(addAliasRole, address(this));
+
+        _authorizer.renounceRole(_authorizer.DEFAULT_ADMIN_ROLE(), address(this));
+    }
+}