mirror of
https://github.com/OffchainLabs/prysm.git
synced 2026-01-10 07:58:22 -05:00
6fddd13cb2
* http resolver * Redo * Revert "Redo" This reverts commit5437c44ac2. * Revert "http resolver" This reverts commit206207b530. * Add host change to ValidatorClient + Validator * Update mockgen * Tidy * Add mock validator * Update gomock * Gaz * Solve interface issues * Fix host * Fix test * Add tests * Add endpoint change log * Fix log * Gen mock * Fix test * Fix deepsource * Lint + deepsource * Move to healthCheckRoutine * Fix build errors * Switch host to string * Forgot a couple * Radek' review * Add PushProposerSettings to goroutine * Radek' review * James' review + test fix * Radek' suggestion Co-authored-by: Radosław Kapka <rkapka@wp.pl> * Check if new node is healthy * Fix linter errors * Add host switch logic to ChangeHost * Lint + comment * Fix messy merge * rename ChangeHost to SetHost * improve log * remove log * switch one node * rename param --------- Co-authored-by: Radosław Kapka <rkapka@wp.pl> Co-authored-by: rkapka <radoslaw.kapka@gmail.com>
99 lines
3.1 KiB
Go
99 lines
3.1 KiB
Go
package accounts
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
grpcutil "github.com/prysmaticlabs/prysm/v5/api/grpc"
|
|
"github.com/prysmaticlabs/prysm/v5/crypto/bls"
|
|
"github.com/prysmaticlabs/prysm/v5/validator/accounts/wallet"
|
|
beaconApi "github.com/prysmaticlabs/prysm/v5/validator/client/beacon-api"
|
|
iface "github.com/prysmaticlabs/prysm/v5/validator/client/iface"
|
|
nodeClientFactory "github.com/prysmaticlabs/prysm/v5/validator/client/node-client-factory"
|
|
validatorClientFactory "github.com/prysmaticlabs/prysm/v5/validator/client/validator-client-factory"
|
|
validatorHelpers "github.com/prysmaticlabs/prysm/v5/validator/helpers"
|
|
"github.com/prysmaticlabs/prysm/v5/validator/keymanager"
|
|
"github.com/prysmaticlabs/prysm/v5/validator/keymanager/derived"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
// NewCLIManager allows for managing validator accounts via CLI commands.
|
|
func NewCLIManager(opts ...Option) (*CLIManager, error) {
|
|
acc := &CLIManager{
|
|
mnemonicLanguage: derived.DefaultMnemonicLanguage,
|
|
inputReader: os.Stdin,
|
|
}
|
|
for _, opt := range opts {
|
|
if err := opt(acc); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return acc, nil
|
|
}
|
|
|
|
// CLIManager defines a struct capable of performing various validator
|
|
// wallet & account operations via the command line.
|
|
type CLIManager struct {
|
|
wallet *wallet.Wallet
|
|
keymanager keymanager.IKeymanager
|
|
keymanagerKind keymanager.Kind
|
|
showPrivateKeys bool
|
|
listValidatorIndices bool
|
|
deletePublicKeys bool
|
|
importPrivateKeys bool
|
|
readPasswordFile bool
|
|
skipMnemonicConfirm bool
|
|
dialOpts []grpc.DialOption
|
|
grpcHeaders []string
|
|
beaconRPCProvider string
|
|
walletKeyCount int
|
|
privateKeyFile string
|
|
passwordFilePath string
|
|
keysDir string
|
|
mnemonicLanguage string
|
|
backupsDir string
|
|
backupsPassword string
|
|
filteredPubKeys []bls.PublicKey
|
|
rawPubKeys [][]byte
|
|
formattedPubKeys []string
|
|
exitJSONOutputPath string
|
|
walletDir string
|
|
walletPassword string
|
|
mnemonic string
|
|
numAccounts int
|
|
mnemonic25thWord string
|
|
beaconApiEndpoint string
|
|
beaconApiTimeout time.Duration
|
|
inputReader io.Reader
|
|
}
|
|
|
|
func (acm *CLIManager) prepareBeaconClients(ctx context.Context) (*iface.ValidatorClient, *iface.NodeClient, error) {
|
|
if acm.dialOpts == nil {
|
|
return nil, nil, errors.New("failed to construct dial options for beacon clients")
|
|
}
|
|
|
|
ctx = grpcutil.AppendHeaders(ctx, acm.grpcHeaders)
|
|
grpcConn, err := grpc.DialContext(ctx, acm.beaconRPCProvider, acm.dialOpts...)
|
|
if err != nil {
|
|
return nil, nil, errors.Wrapf(err, "could not dial endpoint %s", acm.beaconRPCProvider)
|
|
}
|
|
conn := validatorHelpers.NewNodeConnection(
|
|
grpcConn,
|
|
acm.beaconApiEndpoint,
|
|
acm.beaconApiTimeout,
|
|
)
|
|
|
|
restHandler := beaconApi.NewBeaconApiJsonRestHandler(
|
|
http.Client{Timeout: acm.beaconApiTimeout},
|
|
acm.beaconApiEndpoint,
|
|
)
|
|
validatorClient := validatorClientFactory.NewValidatorClient(conn, restHandler)
|
|
nodeClient := nodeClientFactory.NewNodeClient(conn, restHandler)
|
|
|
|
return &validatorClient, &nodeClient, nil
|
|
}
|