feat: upgradeRouter script

.gitmodules

@@ -1,3 +1,6 @@
 [submodule "lib/forge-std"]
 	path = lib/forge-std
 	url = https://github.com/foundry-rs/forge-std
+[submodule "lib/utils"]
+	path = lib/utils
+	url = https://github.com/AngleProtocol/utils

contracts/BaseRouter.sol

@@ -155,7 +155,11 @@ abstract contract BaseRouter is Initializable, IDepositWithReferral {
     );
 
     /// @notice Deploys the router contract on a chain
-    function initializeRouter(address _core, address _uniswapRouter, address _oneInch) public initializer {
+    function initializeRouter(
+        address _core,
+        address _uniswapRouter,
+        address _oneInch
+    ) public initializer {
         if (_core == address(0)) revert ZeroAddress();
         core = ICoreBorrow(_core);
         uniswapV3Router = IUniswapV3Router(_uniswapRouter);
@@ -409,7 +413,11 @@ abstract contract BaseRouter is Initializable, IDepositWithReferral {
     /// @param tokenOut Token to sweep
     /// @param minAmountOut Minimum amount of tokens to recover
     /// @param to Address to which tokens should be sent
-    function _sweep(address tokenOut, uint256 minAmountOut, address to) internal virtual {
+    function _sweep(
+        address tokenOut,
+        uint256 minAmountOut,
+        address to
+    ) internal virtual {
         uint256 balanceToken = IERC20(tokenOut).balanceOf(address(this));
         _slippageCheck(balanceToken, minAmountOut);
         if (balanceToken != 0) {
@@ -580,7 +588,11 @@ abstract contract BaseRouter is Initializable, IDepositWithReferral {
     /// @param token Address of the token to change allowance
     /// @param spender Address to change the allowance of
     /// @param amount Amount allowed
-    function _changeAllowance(IERC20 token, address spender, uint256 amount) internal {
+    function _changeAllowance(
+        IERC20 token,
+        address spender,
+        uint256 amount
+    ) internal {
         uint256 currentAllowance = token.allowance(address(this), spender);
         // In case `currentAllowance < type(uint256).max / 2` and we want to increase it:
         // Do nothing (to handle tokens that need reapprovals to 0 and save gas)

remappings.txt

@@ -4,3 +4,4 @@
 @uniswap/=node_modules/@uniswap/
 ds-test/=lib/forge-std/lib/ds-test/src/
 forge-std/=lib/forge-std/src/
+utils/=lib/utils/

scripts/foundry/UpgradeRouter.s.sol

@@ -0,0 +1,45 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity ^0.8.17;
+
+import "forge-std/Script.sol";
+import "utils/src/CommonUtils.sol";
+import { AngleRouterMainnet } from "contracts/implementations/mainnet/AngleRouterMainnet.sol";
+import { AngleRouterArbitrum } from "contracts/implementations/arbitrum/AngleRouterArbitrum.sol";
+import { AngleRouterOptimism } from "contracts/implementations/optimism/AngleRouterOptimism.sol";
+import { AngleRouterAvalanche } from "contracts/implementations/avalanche/AngleRouterAvalanche.sol";
+import { AngleRouterBase } from "contracts/implementations/base/AngleRouterBase.sol";
+import { AngleRouterCelo } from "contracts/implementations/celo/AngleRouterCelo.sol";
+import { AngleRouterGnosis } from "contracts/implementations/gnosis/AngleRouterGnosis.sol";
+import { AngleRouterLinea } from "contracts/implementations/linea/AngleRouterLinea.sol";
+import { AngleRouterPolygon } from "contracts/implementations/polygon/AngleRouterPolygon.sol";
+
+contract UpgradeRouterScript is Script, CommonUtils {
+
+function run() public {
+    uint256 chainId = vm.envUint("CHAIN_ID");
+
+    address routerImpl;
+    if (chainId == CHAIN_ETHEREUM) {
+        routerImpl = address(new AngleRouterMainnet());
+    } else if (chainId == CHAIN_ARBITRUM) {
+        routerImpl = address(new AngleRouterArbitrum());
+    } else if (chainId == CHAIN_OPTIMISM) {
+        routerImpl = address(new AngleRouterOptimism());
+    } else if (chainId == CHAIN_AVALANCHE) {
+        routerImpl = address(new AngleRouterAvalanche());
+    } else if (chainId == CHAIN_BASE) {
+        routerImpl = address(new AngleRouterBase());
+    } else if (chainId == CHAIN_CELO) {
+        routerImpl = address(new AngleRouterCelo());
+    } else if (chainId == CHAIN_GNOSIS) {
+        routerImpl = address(new AngleRouterGnosis());
+    } else if (chainId == CHAIN_LINEA) {
+        routerImpl = address(new AngleRouterLinea());
+    } else if (chainId == CHAIN_POLYGON) {
+        routerImpl = address(new AngleRouterPolygon());
+    }
+
+    console.log("Deployed router implementation at address: %s", routerImpl);
+}
+
+}