From 6e1a549fcb5cc9257ee984a906db78dd2568042b Mon Sep 17 00:00:00 2001
From: Amir Bandeali <abandeali1@gmail.com>
Date: Wed, 27 Jun 2018 15:13:57 -0700
Subject: Use ledger subprovider

---
 packages/migrations/src/migrate.ts | 35 ++++++++++++++++++++++-------------
 1 file changed, 22 insertions(+), 13 deletions(-)

(limited to 'packages/migrations/src/migrate.ts')

diff --git a/packages/migrations/src/migrate.ts b/packages/migrations/src/migrate.ts
index 35e086382..d9cc1ae01 100644
--- a/packages/migrations/src/migrate.ts
+++ b/packages/migrations/src/migrate.ts
@@ -6,13 +6,15 @@ import { Provider } from 'ethereum-types';
 import * as yargs from 'yargs';
 
 import { runV1MigrationsAsync } from './1.0.0/migration';
-import { runV2BetaKovanMigrationsAsync } from './2.0.0-beta-kovan/migration';
+import { runV2TestnetMigrationsAsync } from './2.0.0-beta-testnet/migration';
 import { runV2MigrationsAsync } from './2.0.0/migration';
 
+import { providerFactory } from './utils/provider_factory';
+
 enum ContractVersions {
     V1 = '1.0.0',
     V2 = '2.0.0',
-    V2BetaKovan = '2.0.0-beta-kovan',
+    V2Testnet = '2.0.0-beta-testnet',
 }
 const args = yargs.argv;
 
@@ -24,17 +26,24 @@ const args = yargs.argv;
     };
     const contractsVersion = args.contractsVersion;
     const artifactsDir = `artifacts/${contractsVersion}`;
-    if (contractsVersion === ContractVersions.V1) {
-        await runV1MigrationsAsync(provider, artifactsDir, txDefaults);
-    } else if (contractsVersion === ContractVersions.V2) {
-        await runV2MigrationsAsync(provider, artifactsDir, txDefaults);
-    } else {
-        const web3Wrapper = new Web3Wrapper(provider);
-        const accounts = await web3Wrapper.getAvailableAddressesAsync();
-        const kovanTxDefaults = {
-            from: accounts[0],
-        };
-        await runV2BetaKovanMigrationsAsync(provider, artifactsDir, kovanTxDefaults);
+    switch (contractsVersion) {
+        case ContractVersions.V1:
+            await runV1MigrationsAsync(provider, artifactsDir, txDefaults);
+            break;
+        case ContractVersions.V2:
+            await runV2MigrationsAsync(provider, artifactsDir, txDefaults);
+            break;
+        case ContractVersions.V2Testnet:
+            const ledgerProvider = await providerFactory.getLedgerProviderAsync();
+            const web3Wrapper = new Web3Wrapper(ledgerProvider);
+            const accounts = await web3Wrapper.getAvailableAddressesAsync();
+            const testnetTxDefaults = {
+                from: accounts[0],
+            };
+            await runV2TestnetMigrationsAsync(ledgerProvider, artifactsDir, testnetTxDefaults);
+            break;
+        default:
+            throw new Error(`Unsupported contract version: ${contractsVersion}`);
     }
     process.exit(0);
 })().catch(err => {
-- 
cgit