mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-03 08:37:37 +00:00
790a09f9b1
* removing timeout on wait for activation, instead switched to an event driven approach * fixing unit tests * linting * simplifying return * adding sleep for the remaining slot to avoid cpu spikes * removing ifstatement on log * removing ifstatement on log * improving switch statement * removing the loop entirely * fixing unit test * fixing manu's reported issue with deletion of json file * missed change around writefile at path * gofmt * fixing deepsource issue with reading file * trying to clean file to avoid deepsource issue * still getting error trying a different approach * fixing stream loop * fixing unit test * Update validator/keymanager/local/keymanager.go Co-authored-by: Manu NALEPA <enalepa@offchainlabs.com> * fixing linting --------- Co-authored-by: Manu NALEPA <enalepa@offchainlabs.com>
94 lines
2.8 KiB
Go
94 lines
2.8 KiB
Go
package accounts
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/prysmaticlabs/prysm/v4/validator/accounts/wallet"
|
|
"github.com/prysmaticlabs/prysm/v4/validator/keymanager"
|
|
"github.com/prysmaticlabs/prysm/v4/validator/keymanager/derived"
|
|
"github.com/prysmaticlabs/prysm/v4/validator/keymanager/local"
|
|
)
|
|
|
|
// WalletCreate creates wallet specified by configuration options.
|
|
func (acm *CLIManager) WalletCreate(ctx context.Context) (*wallet.Wallet, error) {
|
|
w := wallet.New(&wallet.Config{
|
|
WalletDir: acm.walletDir,
|
|
KeymanagerKind: acm.keymanagerKind,
|
|
WalletPassword: acm.walletPassword,
|
|
})
|
|
var err error
|
|
switch w.KeymanagerKind() {
|
|
case keymanager.Local:
|
|
if err := w.SaveWallet(); err != nil {
|
|
return nil, errors.Wrap(err, "could not initialize wallet: could not save wallet to disk")
|
|
}
|
|
accountsKeystore, err := local.CreateEmptyKeyStoreRepresentationForNewWallet(ctx, w.Password())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
encodedAccounts, err := json.MarshalIndent(accountsKeystore, "", "\t")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
_, err = w.WriteFileAtPath(ctx, local.AccountsPath, local.AccountsKeystoreFileName, encodedAccounts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
log.WithField("--wallet-dir", acm.walletDir).Info(
|
|
"Successfully created wallet with ability to import keystores",
|
|
)
|
|
case keymanager.Derived:
|
|
if err = createDerivedKeymanagerWallet(
|
|
ctx,
|
|
w,
|
|
acm.mnemonic25thWord,
|
|
acm.mnemonicLanguage,
|
|
acm.skipMnemonicConfirm,
|
|
acm.numAccounts,
|
|
); err != nil {
|
|
return nil, errors.Wrap(err, "could not initialize wallet")
|
|
}
|
|
log.WithField("--wallet-dir", acm.walletDir).Info(
|
|
"Successfully created HD wallet from mnemonic and regenerated accounts",
|
|
)
|
|
case keymanager.Web3Signer:
|
|
return nil, errors.New("web3signer keymanager does not require persistent wallets.")
|
|
default:
|
|
return nil, errors.Wrapf(err, errKeymanagerNotSupported, w.KeymanagerKind())
|
|
}
|
|
return w, nil
|
|
}
|
|
|
|
func createDerivedKeymanagerWallet(
|
|
ctx context.Context,
|
|
wallet *wallet.Wallet,
|
|
mnemonicPassphrase string,
|
|
mnemonicLanguage string,
|
|
skipMnemonicConfirm bool,
|
|
numAccounts int,
|
|
) error {
|
|
if wallet == nil {
|
|
return errors.New("nil wallet")
|
|
}
|
|
if err := wallet.SaveWallet(); err != nil {
|
|
return errors.Wrap(err, "could not save wallet to disk")
|
|
}
|
|
km, err := derived.NewKeymanager(ctx, &derived.SetupConfig{
|
|
Wallet: wallet,
|
|
ListenForChanges: true,
|
|
})
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not initialize HD keymanager")
|
|
}
|
|
mnemonic, err := derived.GenerateAndConfirmMnemonic(mnemonicLanguage, skipMnemonicConfirm)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not confirm mnemonic")
|
|
}
|
|
if err := km.RecoverAccountsFromMnemonic(ctx, mnemonic, mnemonicLanguage, mnemonicPassphrase, numAccounts); err != nil {
|
|
return errors.Wrap(err, "could not recover accounts from mnemonic")
|
|
}
|
|
return nil
|
|
}
|