mirrors/bitcoin-s
1
0
Fork 0
mirror of https://github.com/bitcoin-s/bitcoin-s.git synced 2025-02-22 22:36:34 +01:00
Code Issues Projects Releases Packages Wiki Activity

Create all root level accounts on wallet creation

Browse source
This commit is contained in:
Torkel Rogstad 2019-06-05 20:01:59 +02:00
parent 42c161d3e3
commit 8a338d542f
1 changed files with 39 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

54
wallet/src/main/scala/org/bitcoins/wallet/Wallet.scala
View file

@ -166,24 +166,26 @@ object Wallet extends CreateWalletApi with BitcoinSLogger {
encrypted <- encryptedMnemonicE
} yield {
val wallet = WalletImpl(mnemonic)
val coin =
HDCoin(config.defaultAccountKind, HDUtil.getCoinType(config.network))
val account = HDAccount(coin, 0)
val xpriv = wallet.xprivForPurpose(config.defaultAccountKind)
// safe since we're deriving from a priv
val xpub = xpriv.deriveChildPubKey(account).get
val accountDb = AccountDb(xpub, account)
val mnemonicPath =
WalletStorage.writeMnemonicToDisk(encrypted)
logger.debug(s"Saved encrypted wallet mnemonic to $mnemonicPath")
for {
_ <- config.initialize()
_ <- wallet.accountDAO
.create(accountDb)
.map(_ => logger.trace(s"Saved account to DB"))
_ = {
val mnemonicPath =
WalletStorage.writeMnemonicToDisk(encrypted)
logger.debug(s"Saved encrypted wallet mnemonic to $mnemonicPath")
}
_ <- {
// We want to make sure all level 0 accounts are created,
// so the user can change the default account kind later
// and still have their wallet work
val createAccountFutures =
HDPurposes.all.map(createRootAccount(wallet, _))
Future
.sequence(createAccountFutures)
.map(_ => logger.debug(s"Created root level accounts for wallet"))
}
} yield wallet
}
@ -197,4 +199,26 @@ object Wallet extends CreateWalletApi with BitcoinSLogger {
case Left(err) => err
}
}
/** Creates the level 0 account for the given HD purpose */
private def createRootAccount(wallet: Wallet, purpose: HDPurpose)(
implicit config: WalletAppConfig,
ec: ExecutionContext): Future[AccountDb] = {
val coin =
HDCoin(purpose, HDUtil.getCoinType(config.network))
val account = HDAccount(coin, 0)
val xpriv = wallet.xprivForPurpose(purpose)
// safe since we're deriving from a priv
val xpub = xpriv.deriveChildPubKey(account).get
val accountDb = AccountDb(xpub, account)
logger.debug(s"Creating account with constant prefix $purpose")
wallet.accountDAO
.create(accountDb)
.map { written =>
logger.debug(s"Saved account with constant prefix $purpose to DB")
written
}
}
}