From ba68afc1fcf8bd0ff5b7a97c2ee5603c681fb8ba Mon Sep 17 00:00:00 2001 From: David Racero Date: Fri, 18 Jun 2021 09:38:29 +0200 Subject: [PATCH] feat: optional aave collector from curve script --- tasks/deployments/deploy-CurveTreasury.ts | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tasks/deployments/deploy-CurveTreasury.ts b/tasks/deployments/deploy-CurveTreasury.ts index 3872363a..df6cc303 100644 --- a/tasks/deployments/deploy-CurveTreasury.ts +++ b/tasks/deployments/deploy-CurveTreasury.ts @@ -10,14 +10,18 @@ import { waitForTx } from '../../helpers/misc-utils'; task(`deploy-curve-treasury`, `Deploys the CurveTreasury contract`) .addParam('proxyAdmin') .addParam('treasuryAdmin') - .addParam('pool') + .addOptionalParam('pool') .addFlag('verify', `Verify contract via Etherscan API.`) .setAction(async ({ verify, proxyAdmin, treasuryAdmin, pool }, localBRE) => { await localBRE.run('set-DRE'); - const poolConfig = loadPoolConfig(pool); const net = localBRE.network.name; console.log(`\n- Curve Treasury deployment`); + let aaveCollector = ZERO_ADDRESS; + + if (pool) { + aaveCollector = loadPoolConfig(pool).ReserveFactorTreasuryAddress[net]; + } // Deploy implementation const implementation = await deployCurveTreasury( @@ -25,7 +29,7 @@ task(`deploy-curve-treasury`, `Deploys the CurveTreasury contract`) CRV_TOKEN[net], CURVE_CONFIG.curveFeeDistributor[net], CURVE_CONFIG.gaugeController[net], - poolConfig.ReserveFactorTreasuryAddress[net], + aaveCollector, verify );