BitcoinLedgerProvider: Implement getConnectedNetwork

Author: monokh

src/providers/LedgerProvider.js

@@ -7,11 +7,12 @@ export default class LedgerProvider extends WalletProvider {
     return Transport.isSupported()
   }
 
-  constructor (App, baseDerivationPath) {
-    super()
+  constructor (App, baseDerivationPath, network) {
+    super(network)
 
     this._App = App
     this._baseDerivationPath = baseDerivationPath
+    this._network = network
     this._addressCache = {}
   }
 

src/providers/bitcoin/BitcoinLedgerProvider.js

@@ -3,14 +3,14 @@ import Bitcoin from '@ledgerhq/hw-app-btc'
 
 import { BigNumber } from 'bignumber.js'
 import { base58, padHexStart } from '../../crypto'
-import { pubKeyToAddress, addressToPubKeyHash, compressPubKey } from './BitcoinUtil'
+import { pubKeyToAddress, addressToPubKeyHash, compressPubKey, getAddressNetwork } from './BitcoinUtil'
 import Address from '../../Address'
 import networks from './networks'
 import bip32 from 'bip32'
 
 export default class BitcoinLedgerProvider extends LedgerProvider {
   constructor (chain = { network: networks.bitcoin, segwit: false }, numberOfBlockConfirmation = 1) {
-    super(Bitcoin, `${chain.segwit ? '49' : '44'}'/${chain.network.coinType}'/0'/`)
+    super(Bitcoin, `${chain.segwit ? '49' : '44'}'/${chain.network.coinType}'/0'/`, chain.network)
     this._derivationPath = `${chain.segwit ? '49' : '44'}'/${chain.network.coinType}'/0'/`
     this._network = chain.network
     this._bjsnetwork = chain.network.name.replace('bitcoin_', '') // for bitcoin js
@@ -502,4 +502,10 @@ export default class BitcoinLedgerProvider extends LedgerProvider {
   async getAddresses (startingIndex = 0, numAddresses = 1, change = false) {
     return this.getLedgerAddresses(startingIndex, numAddresses, change)
   }
+
+  async getConnectedNetwork () {
+    const walletPubKey = await this.getWalletPublicKey(this._baseDerivationPath)
+    const network = getAddressNetwork(walletPubKey.bitcoinAddress)
+    return network
+  }
 }

src/providers/bitcoin/BitcoinUtil.js

@@ -1,3 +1,5 @@
+import _ from 'lodash'
+
 import {
   ensureBuffer,
   hash160,
@@ -112,6 +114,17 @@ function toHexInt (number) {
     toHexDigit(number & 0xff)
   )
 }
+/**
+ * Get a network object from an address
+ * @param {string} address The bitcoin address
+ * @return {Network}
+ */
+function getAddressNetwork (address) {
+  const prefix = base58.decode(address).toString('hex').substring(0, 2).toUpperCase()
+  const networkKey = _.findKey(networks,
+    network => [network.pubKeyHash, network.scriptHash].includes(prefix))
+  return networks[networkKey]
+}
 
 export {
   toHexInt,
@@ -120,5 +133,6 @@ export {
   pubKeyHashToAddress,
   addressToPubKeyHash,
   reverseBuffer,
-  scriptNumEncode
+  scriptNumEncode,
+  getAddressNetwork
 }