add: HD Wallet class and key type #root#

Author: Tom Wang

index.js

@@ -3,8 +3,8 @@
 let MnemonicHD = require("./lib/utils/mnemonic.hd")
 let SeedHD = require("./lib/utils/seed.hd")
 let Base58HD = require("./lib/utils/base58.hd")
-let HdWallet = require("./lib/base/hd.wallet.class")
 
+let HdWallet = require("./lib/base/hd.wallet.class")
 let AddressClass = require("./lib/base/address.class")
 
 let Address = {
@@ -20,7 +20,8 @@ module.exports = {
         "base58": "base58",
         "root": "root"
     },
-    AddressClass: AddressClass,
+    AddressClass,
+    HdWallet,
     HD: (key, keyType = "mnemonic", pwd) => {
         let hd
         if (keyType == "mnemonic") {
@@ -30,7 +31,7 @@ module.exports = {
         } else if (keyType == "base58") {
             hd = new Base58HD(key)
         } else if (keyType == "root") {
-            return HdWallet
+            hd = new HdWallet(key)
         } else {
             throw "key type unsupported"
         }

lib/utils/base58.hd.js

@@ -2,6 +2,7 @@
 
 let HdChainCode = require("../base/hd.wallet.class")
 const bip32 = require('bip32');
+const bip39 = require('bip39');
 
 /**
  * BIP44 提出了5層的路徑建議 m / purpose' / coin_type' / account' / change / address_index
@@ -27,14 +28,14 @@ module.exports = class Base58HD extends HdChainCode {
         const seed = bip39.mnemonicToSeedSync(mnemonic, pwd)
         const root = bip32.fromSeed(seed)
         const base58 = root.toBase58()
-        return pwd ? {root, base58, mnemonic, pwd} : {base58, mnemonic}
+        return pwd ? {root, base58, mnemonic, pwd} : {root,base58, mnemonic}
     }
 
     static mnemonicToBase58(mnemonic, pwd) {
         const seed = bip39.mnemonicToSeedSync(mnemonic, pwd)
         const root = bip32.fromSeed(seed)
         const base58 = root.toBase58()
-        return pwd ? {root, base58, mnemonic, pwd} : {base58, mnemonic}
+        return pwd ? {root, base58, mnemonic, pwd} : {root,base58, mnemonic}
     }
 };
 

lib/utils/seed.hd.js

@@ -1,8 +1,9 @@
 'use strict';
 
+let HdWallet = require("../base/hd.wallet.class")
 const bip32 = require('bip32');
 const bip39 = require('bip39');
-let HdWallet = require("../base/hd.wallet.class")
+
 
 let seedBuff
 
@@ -21,15 +22,13 @@ module.exports = class SeedHD extends HdWallet {
         const mnemonic = bip39.generateMnemonic(strength)
         let seed = bip39.mnemonicToSeedSync(mnemonic, pwd)
         const root = bip32.fromSeed(seed)
-        seed = seed.toString("hex")
-        return pwd ? {root, seed, mnemonic, pwd} : {mnemonic, seed}
+        return pwd ? {root, seed, mnemonic, pwd} : {root,mnemonic, seed}
     }
 
     static mnemonicToSeed(mnemonic, pwd) {
         let seed = bip39.mnemonicToSeedSync(mnemonic, pwd)
         const root = bip32.fromSeed(seed)
-        seed = seed.toString("hex")
-        return pwd ? {root, seed, mnemonic, pwd} : {mnemonic, seed}
+        return pwd ? {root, seed, mnemonic, pwd} : {root,mnemonic, seed}
     }
 };
 

test/index.test.js

@@ -1,15 +1,12 @@
+let hdAddress = require("../index")
+
 let logInfo = (hdCoin, hdIndex) => {
     let addr = hdCoin.getAddress(hdIndex)
     console.log(`${hdCoin.coin}:"${addr.address}"`)
 }
 
-(async () => {
-    let hdData = require("./data").hd
-    let hdIndex = hdData.index
-    const mnemonic = hdData.mnemonic
-    let hdAddress = require("../index")
-
-    let hd = hdAddress.HD(mnemonic)
+let getAddress = (hd) => {
+    let hdIndex = 6677
     logInfo(hd.BTC, hdIndex)
     logInfo(hd.BTC_TEST, hdIndex)
     logInfo(hd.BCH, hdIndex)
@@ -18,4 +15,57 @@ let logInfo = (hdCoin, hdIndex) => {
     logInfo(hd.LTC_TEST, hdIndex)
     logInfo(hd.ETH, hdIndex)
     logInfo(hd.TRX, hdIndex)
-})()
+}
+const mnemonic = "star star star star star star"
+const pwd = "star"
+const seed = "03d0be996b63e90c7625dd3f5319c3bc11669d3d35ae5dc345595e5e59be74084f"
+const base58 = "xprv9s21ZrQH143K3QTDL4LXw2F7HEK3wJUD2nW2nRk4stbPy6cq3jPPqjiChkVvvNKmPGJxWUtg6LnF5kejMRNNU3TGtRBeJgk33yuGBxrMPHi"
+
+describe("address init", () => {
+
+    it(" getRandomMnemonic", () => {
+        const mnemonic = hdAddress.mnemonic.getRandomMnemonic()
+        let hd = hdAddress.HD(mnemonic)
+        getAddress(hd)
+    })
+
+    it(" from mnemonic str", () => {
+        let hd = hdAddress.HD(mnemonic, hdAddress.keyType.mnemonic, pwd)
+        getAddress(hd)
+    })
+
+    it(" getRandomSeed", () => {
+        const {seed} = hdAddress.seed.getRandomSeed()
+        let hd = hdAddress.HD(seed, hdAddress.keyType.seed)
+        getAddress(hd)
+    })
+
+    it(" from seed string", () => {
+        // Seed should be at least 128 bits and most 512 bits
+        const seedBuf = Buffer.from(seed, "hex")
+        let hd = hdAddress.HD(seedBuf, hdAddress.keyType.seed)
+        getAddress(hd)
+    })
+
+    it(" getRandomBase58", () => {
+        const {base58} = hdAddress.base58.getRandomBase58()
+        let hd = hdAddress.HD(base58, hdAddress.keyType.base58)
+        getAddress(hd)
+    })
+
+
+    it("from base58 str", () => {
+        let hd = hdAddress.HD(base58, hdAddress.keyType.base58)
+        getAddress(hd)
+    })
+
+    it(" getHDWallet", () => {
+        const {root} = hdAddress.seed.mnemonicToSeed(mnemonic)
+        let hd = hdAddress.HD(root, hdAddress.keyType.root)
+        let {path, pub} = hd.wallet.getCoinKeyPair("BTC", 0, 0, 0)
+        console.log(path)
+        let {address} = hd.TRX.getAddress(1)
+        console.log(address)
+    })
+
+})