mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-27 05:38:55 +00:00
9fab9df61e
* add functional options accounts delete * bazel run //:gazelle -- fix Co-authored-by: james-prysm <90280386+james-prysm@users.noreply.github.com> Co-authored-by: Raul Jordan <raul@prysmaticlabs.com>
64 lines
1.9 KiB
Go
64 lines
1.9 KiB
Go
package accounts
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/prysmaticlabs/prysm/cmd"
|
|
"github.com/prysmaticlabs/prysm/cmd/validator/flags"
|
|
"github.com/prysmaticlabs/prysm/validator/accounts"
|
|
"github.com/prysmaticlabs/prysm/validator/accounts/userprompt"
|
|
"github.com/prysmaticlabs/prysm/validator/client"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
func accountsDelete(c *cli.Context) error {
|
|
w, km, err := walletWithKeymanager(c)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
dialOpts := client.ConstructDialOptions(
|
|
c.Int(cmd.GrpcMaxCallRecvMsgSizeFlag.Name),
|
|
c.String(flags.CertFlag.Name),
|
|
c.Uint(flags.GrpcRetriesFlag.Name),
|
|
c.Duration(flags.GrpcRetryDelayFlag.Name),
|
|
)
|
|
grpcHeaders := strings.Split(c.String(flags.GrpcHeadersFlag.Name), ",")
|
|
|
|
opts := []accounts.Option{
|
|
accounts.WithWallet(w),
|
|
accounts.WithKeymanager(km),
|
|
accounts.WithGRPCDialOpts(dialOpts),
|
|
accounts.WithBeaconRPCProvider(c.String(flags.BeaconRPCProviderFlag.Name)),
|
|
accounts.WithGRPCHeaders(grpcHeaders),
|
|
}
|
|
|
|
// Get full set of public keys from the keymanager.
|
|
validatingPublicKeys, err := km.FetchValidatingPublicKeys(c.Context)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(validatingPublicKeys) == 0 {
|
|
return errors.New("wallet is empty, no accounts to delete")
|
|
}
|
|
// Filter keys either from CLI flag or from interactive session.
|
|
filteredPubKeys, err := accounts.FilterPublicKeysFromUserInput(
|
|
c,
|
|
flags.DeletePublicKeysFlag,
|
|
validatingPublicKeys,
|
|
userprompt.SelectAccountsDeletePromptText,
|
|
)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not filter public keys for deletion")
|
|
}
|
|
opts = append(opts, accounts.WithFilteredPubKeys(filteredPubKeys))
|
|
opts = append(opts, accounts.WithWalletKeyCount(len(validatingPublicKeys)))
|
|
opts = append(opts, accounts.WithDeletePublicKeys(c.IsSet(flags.DeletePublicKeysFlag.Name)))
|
|
|
|
acc, err := accounts.NewCLIManager(opts...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return acc.Delete(c.Context)
|
|
}
|