From 593069bddd932dc4048a4057b250a71bedf56121 Mon Sep 17 00:00:00 2001
From: Mubaris NK <mubaris@instadapp.io>
Date: Mon, 15 Mar 2021 22:16:22 +0530
Subject: [PATCH] Add snippet for enabling connectors after deployment

---
 scripts/deploy.js | 58 +++++++++++++++++++++++++++++------------------
 1 file changed, 36 insertions(+), 22 deletions(-)

diff --git a/scripts/deploy.js b/scripts/deploy.js
index bdcce559..e40f5fdb 100644
--- a/scripts/deploy.js
+++ b/scripts/deploy.js
@@ -4,29 +4,43 @@ const { ethers } = hre;
 const deployConnector = require("./deployConnector");
 
 async function main() {
-    const connectors = [
-      'ConnectV2OneInch',
-      'ConnectV2AaveV1',
-      'ConnectV2AaveV2',
-      'ConnectV2Auth',
-      'ConnectV2Basic',
-      'ConnectV2COMP',
-      'ConnectV2Compound',
-      'ConnectV2Dydx',
-      'ConnectV2Fee',
-      'ConnectV2Gelato',
-      'ConnectV2Maker',
-      'ConnectV2UniswapV2'
-    ]
-
-    for (const connector of connectors) {
-      await deployConnector(connector)
+    const accounts = await hre.ethers.getSigners()
+    const wallet = accounts[0]
+    
+    const connectMapping = {
+        '1inch': 'ConnectV2OneInch',
+        'aaveV1': 'ConnectV2AaveV1',
+        'aaveV2': 'ConnectV2AaveV2',
+        'auth': 'ConnectV2Auth',
+        'basic': 'ConnectV2Basic',
+        'comp': 'ConnectV2COMP',
+        'compound': 'ConnectV2Compound',
+        'dydx': 'ConnectV2Dydx',
+        'fee': 'ConnectV2Fee',
+        'gelato': 'ConnectV2Gelato',
+        'maker': 'ConnectV2Maker',
+        'uniswap': 'ConnectV2UniswapV2'
     }
+    
+    const addressMapping = {}
+    
+    for (const key in connectMapping) {
+        addressMapping[key] = await deployConnector(connectMapping[key])
+    }
+    
+    const connectorsAbi = [
+        "function addConnectors(string[] _connectorNames, address[] _connectors)"
+    ]
+    
+    // Replace the address with correct v2 connectors registry address
+    const connectorsContract = new ethers.Contract("0x84b457c6D31025d56449D5A01F0c34bF78636f67", connectorsAbi, wallet)
+    
+    await connectorsContract.addConnectors(Object.keys(addressMapping), Object.values(addressMapping))
 }
 
 main()
-  .then(() => process.exit(0))
-  .catch(error => {
-    console.error(error);
-    process.exit(1);
-  });
\ No newline at end of file
+    .then(() => process.exit(0))
+    .catch(error => {
+        console.error(error);
+        process.exit(1);
+    });
\ No newline at end of file