lnwallet: nuke most of setup.go, no longer needed w/ wallet.Loader
This commit is contained in:
parent
b3cdc6167f
commit
daeecd5fc0
@ -18,35 +18,13 @@
|
|||||||
package lnwallet
|
package lnwallet
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
|
||||||
"bytes"
|
|
||||||
"encoding/hex"
|
|
||||||
"fmt"
|
|
||||||
"os"
|
|
||||||
"path/filepath"
|
"path/filepath"
|
||||||
"strings"
|
|
||||||
|
|
||||||
"golang.org/x/crypto/ssh/terminal"
|
|
||||||
|
|
||||||
"github.com/btcsuite/btcd/chaincfg"
|
"github.com/btcsuite/btcd/chaincfg"
|
||||||
"github.com/btcsuite/btcd/wire"
|
"github.com/btcsuite/btcd/wire"
|
||||||
"github.com/btcsuite/btcutil/hdkeychain"
|
|
||||||
"github.com/btcsuite/btcwallet/waddrmgr"
|
|
||||||
"github.com/btcsuite/btcwallet/wallet"
|
|
||||||
"github.com/btcsuite/btcwallet/walletdb"
|
|
||||||
_ "github.com/btcsuite/btcwallet/walletdb/bdb"
|
_ "github.com/btcsuite/btcwallet/walletdb/bdb"
|
||||||
)
|
)
|
||||||
|
|
||||||
// filesExists reports whether the named file or directory exists.
|
|
||||||
func fileExists(name string) bool {
|
|
||||||
if _, err := os.Stat(name); err != nil {
|
|
||||||
if os.IsNotExist(err) {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
// networkDir returns the directory name of a network directory to hold wallet
|
// networkDir returns the directory name of a network directory to hold wallet
|
||||||
// files.
|
// files.
|
||||||
func networkDir(dataDir string, chainParams *chaincfg.Params) string {
|
func networkDir(dataDir string, chainParams *chaincfg.Params) string {
|
||||||
@ -63,162 +41,3 @@ func networkDir(dataDir string, chainParams *chaincfg.Params) string {
|
|||||||
|
|
||||||
return filepath.Join(dataDir, netname)
|
return filepath.Join(dataDir, netname)
|
||||||
}
|
}
|
||||||
|
|
||||||
// checkCreateDir checks that the path exists and is a directory.
|
|
||||||
// If path does not exist, it is created.
|
|
||||||
func checkCreateDir(path string) error {
|
|
||||||
if fi, err := os.Stat(path); err != nil {
|
|
||||||
if os.IsNotExist(err) {
|
|
||||||
// Attempt data directory creation
|
|
||||||
if err = os.MkdirAll(path, 0700); err != nil {
|
|
||||||
return fmt.Errorf("cannot create directory: %s", err)
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
return fmt.Errorf("error checking directory: %s", err)
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
if !fi.IsDir() {
|
|
||||||
return fmt.Errorf("path '%s' is not a directory", path)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// createWallet generates a new wallet. The new wallet will reside at the
|
|
||||||
// provided path.
|
|
||||||
// TODO(roasbeef): maybe pass in config after all for testing purposes?
|
|
||||||
func createWallet(privPass, pubPass, userSeed []byte,
|
|
||||||
dbPath string, activeNet *chaincfg.Params) error {
|
|
||||||
// TODO(roasbeef): replace with tadge's seed format?
|
|
||||||
hdSeed := userSeed
|
|
||||||
var seedErr error
|
|
||||||
if userSeed == nil {
|
|
||||||
hdSeed, seedErr = hdkeychain.GenerateSeed(hdkeychain.RecommendedSeedLen)
|
|
||||||
if seedErr != nil {
|
|
||||||
return seedErr
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Create the wallet.
|
|
||||||
fmt.Println("Creating the wallet...")
|
|
||||||
|
|
||||||
// Create the wallet database backed by bolt db.
|
|
||||||
db, err := walletdb.Create("bdb", dbPath)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Create the address manager.
|
|
||||||
namespace, err := db.Namespace(waddrmgrNamespaceKey)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
manager, err := waddrmgr.Create(namespace, hdSeed, []byte(pubPass),
|
|
||||||
[]byte(privPass), activeNet, nil)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := manager.Close(); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
if err := db.Close(); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
fmt.Println("The lnwallet has been created successfully.")
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// openDb opens and returns a walletdb.DB (boltdb here) given the directory and
|
|
||||||
// dbname
|
|
||||||
func openDb(directory string, dbname string) (walletdb.DB, error) {
|
|
||||||
dbPath := filepath.Join(directory, dbname)
|
|
||||||
|
|
||||||
// Ensure that the network directory exists.
|
|
||||||
if err := checkCreateDir(directory); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Open the database using the boltdb backend.
|
|
||||||
wdb, err := walletdb.Open("bdb", dbPath)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
return wdb, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// promptSeed is used to prompt for the wallet seed which maybe required during
|
|
||||||
// upgrades.
|
|
||||||
func promptSeed() ([]byte, error) {
|
|
||||||
reader := bufio.NewReader(os.Stdin)
|
|
||||||
for {
|
|
||||||
fmt.Print("Enter existing wallet seed: ")
|
|
||||||
seedStr, err := reader.ReadString('\n')
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
seedStr = strings.TrimSpace(strings.ToLower(seedStr))
|
|
||||||
|
|
||||||
seed, err := hex.DecodeString(seedStr)
|
|
||||||
if err != nil || len(seed) < hdkeychain.MinSeedBytes ||
|
|
||||||
len(seed) > hdkeychain.MaxSeedBytes {
|
|
||||||
|
|
||||||
fmt.Printf("Invalid seed specified. Must be a "+
|
|
||||||
"hexadecimal value that is at least %d bits and "+
|
|
||||||
"at most %d bits\n", hdkeychain.MinSeedBytes*8,
|
|
||||||
hdkeychain.MaxSeedBytes*8)
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
return seed, nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// promptPrivPassPhrase is used to prompt for the private passphrase which maybe
|
|
||||||
// required during upgrades.
|
|
||||||
func promptPrivPassPhrase() ([]byte, error) {
|
|
||||||
prompt := "Enter the private passphrase of your wallet: "
|
|
||||||
for {
|
|
||||||
fmt.Print(prompt)
|
|
||||||
pass, err := terminal.ReadPassword(int(os.Stdin.Fd()))
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
fmt.Print("\n")
|
|
||||||
pass = bytes.TrimSpace(pass)
|
|
||||||
if len(pass) == 0 {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
return pass, nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// openWallet returns a wallet. The function handles opening an existing wallet
|
|
||||||
// database, the address manager and the transaction store and uses the values
|
|
||||||
// to open a wallet.Wallet
|
|
||||||
func openWallet(pubPass []byte, dbDir string, activeNet *chaincfg.Params) (*wallet.Wallet, walletdb.DB, error) {
|
|
||||||
db, err := openDb(dbDir, walletDbName)
|
|
||||||
if err != nil {
|
|
||||||
return nil, nil, fmt.Errorf("Failed to open database: %v", err)
|
|
||||||
}
|
|
||||||
addrMgrNS, err := db.Namespace(waddrmgrNamespaceKey)
|
|
||||||
if err != nil {
|
|
||||||
return nil, nil, err
|
|
||||||
}
|
|
||||||
txMgrNS, err := db.Namespace(wtxmgrNamespaceKey)
|
|
||||||
if err != nil {
|
|
||||||
return nil, nil, err
|
|
||||||
}
|
|
||||||
// TODO(roasbeef): pass these in as funcs instead, priv pass already
|
|
||||||
// loaded into memory, use tadge's format to read HD seed.
|
|
||||||
cbs := &waddrmgr.OpenCallbacks{
|
|
||||||
ObtainSeed: promptSeed,
|
|
||||||
ObtainPrivatePass: promptPrivPassPhrase,
|
|
||||||
}
|
|
||||||
w, err := wallet.Open(pubPass, activeNet, db, addrMgrNS, txMgrNS,
|
|
||||||
cbs)
|
|
||||||
return w, db, err
|
|
||||||
}
|
|
||||||
|
Loading…
Reference in New Issue
Block a user