Merge pull request #249 from blocknative/feature/ledger-transport

Add custom ledger transport option

Author: lnbc1QWFyb24

src/interfaces.ts

@@ -150,6 +150,7 @@ export interface TrezorOptions {
 
 export interface LedgerOptions {
   rpcUrl: string
+  LedgerTransport?: any
 }
 
 export interface TorusOptions {
@@ -161,13 +162,12 @@ export interface TorusOptions {
   enabledVerifiers: TorusVerifierStatus
 }
 
-
 interface TorusVerifierStatus {
-  google?: boolean;
-  facebook?: boolean;
-  reddit?: boolean;
-  twitch?: boolean;
-  discord?: boolean;
+  google?: boolean
+  facebook?: boolean
+  reddit?: boolean
+  twitch?: boolean
+  discord?: boolean
 }
 
 export interface AuthereumOptions {

src/modules/select/wallets/ledger.ts

@@ -15,7 +15,15 @@ import * as EthereumTx from 'ethereumjs-tx'
 import buffer from 'buffer'
 
 function ledger(options: LedgerOptions & CommonWalletOptions): WalletModule {
-  const { rpcUrl, networkId, preferred, label, iconSrc, svg } = options
+  const {
+    rpcUrl,
+    LedgerTransport,
+    networkId,
+    preferred,
+    label,
+    iconSrc,
+    svg
+  } = options
 
   return {
     name: label || 'Ledger',
@@ -27,6 +35,7 @@ function ledger(options: LedgerOptions & CommonWalletOptions): WalletModule {
       const provider = await ledgerProvider({
         rpcUrl,
         networkId,
+        LedgerTransport,
         BigNumber,
         networkName
       })
@@ -36,7 +45,7 @@ function ledger(options: LedgerOptions & CommonWalletOptions): WalletModule {
         interface: {
           name: 'Ledger',
           connect: provider.enable,
-          disconnect: () => provider.stop(),
+          disconnect: provider.disconnect,
           address: {
             get: async () => provider.getPrimaryAddress()
           },
@@ -61,10 +70,11 @@ function ledger(options: LedgerOptions & CommonWalletOptions): WalletModule {
 async function ledgerProvider(options: {
   networkId: number
   rpcUrl: string
+  LedgerTransport: any
   BigNumber: any
   networkName: (id: number) => string
 }) {
-  const { networkId, rpcUrl, BigNumber, networkName } = options
+  const { networkId, rpcUrl, LedgerTransport, BigNumber, networkName } = options
   const basePath = networkIdToDerivationPath(networkId)
 
   let addressToPath = new Map()
@@ -90,6 +100,27 @@ async function ledgerProvider(options: {
   provider.setPrimaryAccount = setPrimaryAccount
   provider.getBalance = getBalance
   provider.send = provider.sendAsync
+  provider.disconnect = disconnect
+
+  let transport: any
+  let eth: any
+
+  try {
+    transport = LedgerTransport
+      ? await LedgerTransport.create()
+      : await TransportU2F.create()
+
+    eth = new Eth(transport)
+  } catch (error) {
+    throw new Error('Error connecting to Ledger wallet')
+  }
+
+  function disconnect() {
+    transport.close()
+    addressToPath = new Map()
+    enabled = false
+    provider.stop()
+  }
 
   function enable() {
     enabled = true
@@ -161,16 +192,6 @@ async function ledgerProvider(options: {
         paths.push(`${basePath}/0'/0`)
       }
 
-      let transport
-      let eth
-
-      try {
-        transport = await TransportU2F.create()
-        eth = new Eth(transport)
-      } catch (error) {
-        reject({ message: 'Error connecting to Ledger wallet' })
-      }
-
       for (const path of paths) {
         try {
           const { address } = await eth.getAddress(path)
@@ -215,15 +236,6 @@ async function ledgerProvider(options: {
 
   async function signTransaction(transactionData: any) {
     const path = [...addressToPath.values()][0]
-    let transport
-    let eth
-
-    try {
-      transport = await TransportU2F.create()
-      eth = new Eth(transport)
-    } catch (error) {
-      throw new Error('Error connecting to Ledger wallet')
-    }
 
     try {
       const transaction = new EthereumTx.Transaction(transactionData, {

src/validation.ts

@@ -527,6 +527,7 @@ export function validateWalletInit(
     appUrl,
     email,
     rpcUrl,
+    LedgerTransport,
     buildEnv,
     buttonPosition,
     enableLogging,
@@ -551,6 +552,7 @@ export function validateWalletInit(
       'appUrl',
       'email',
       'rpcUrl',
+      'LedgerTransport',
       'buildEnv',
       'buttonPosition',
       'enableLogging',
@@ -619,11 +621,17 @@ export function validateWalletInit(
   validateType({
     name: 'walletInit.appUrl',
     value: appUrl,
-
     type: 'string',
     optional: true
   })
 
+  validateType({
+    name: 'walletInit.LedgerTransport',
+    value: LedgerTransport,
+    type: 'function',
+    optional: true
+  })
+
   validateType({
     name: 'walletInit.email',
     value: email,
@@ -660,7 +668,6 @@ export function validateWalletInit(
     optional: true
   })
 
-
   validateType({
     name: 'walletInit.enabledVerifiers',
     value: enabledVerifiers,