mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
4c916403e9
* Enable dupword linter & fix findings * Correct an incorrect fix * Add nolint comment * Add another nolint comment * Revert unintended change to bazel version --------- Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com> Co-authored-by: Radosław Kapka <rkapka@wp.pl>
82 lines
2.8 KiB
Go
82 lines
2.8 KiB
Go
package historycmd
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/prysmaticlabs/prysm/v4/cmd"
|
|
"github.com/prysmaticlabs/prysm/v4/cmd/validator/flags"
|
|
"github.com/prysmaticlabs/prysm/v4/io/file"
|
|
"github.com/prysmaticlabs/prysm/v4/validator/accounts/userprompt"
|
|
"github.com/prysmaticlabs/prysm/v4/validator/db/kv"
|
|
slashingprotection "github.com/prysmaticlabs/prysm/v4/validator/slashing-protection-history"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
// Reads an input slashing protection EIP-3076
|
|
// standard JSON file and attempts to insert its data into our validator DB.
|
|
//
|
|
// Steps:
|
|
// 1. Parse a path to the validator's datadir from the CLI context.
|
|
// 2. Open the validator database.
|
|
// 3. Read the JSON file from user input.
|
|
// 4. Call the function which actually imports the data from
|
|
// the standard slashing protection JSON file into our database.
|
|
func importSlashingProtectionJSON(cliCtx *cli.Context) error {
|
|
var err error
|
|
dataDir := cliCtx.String(cmd.DataDirFlag.Name)
|
|
if !cliCtx.IsSet(cmd.DataDirFlag.Name) {
|
|
dataDir, err = userprompt.InputDirectory(cliCtx, userprompt.DataDirDirPromptText, cmd.DataDirFlag)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "could not read directory value from input")
|
|
}
|
|
}
|
|
// ensure that the validator.db is found under the specified dir or its subdirectories
|
|
found, _, err := file.RecursiveFileFind(kv.ProtectionDbFileName, dataDir)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "error finding validator database at path %s", dataDir)
|
|
}
|
|
if !found {
|
|
log.Infof(
|
|
"Did not find existing validator.db inside of %s, creating a new one",
|
|
dataDir,
|
|
)
|
|
} else {
|
|
log.Infof("Found existing validator.db inside of %s", dataDir)
|
|
}
|
|
valDB, err := kv.NewKVStore(cliCtx.Context, dataDir, &kv.Config{})
|
|
if err != nil {
|
|
return errors.Wrapf(err, "could not access validator database at path: %s", dataDir)
|
|
}
|
|
defer func() {
|
|
if err := valDB.Close(); err != nil {
|
|
log.WithError(err).Errorf("Could not close validator DB")
|
|
}
|
|
}()
|
|
protectionFilePath, err := userprompt.InputDirectory(cliCtx, userprompt.SlashingProtectionJSONPromptText, flags.SlashingProtectionJSONFileFlag)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not get slashing protection json file")
|
|
}
|
|
if protectionFilePath == "" {
|
|
return fmt.Errorf(
|
|
"no path to a slashing_protection.json file specified, please retry or "+
|
|
"you can also specify it with the %s flag",
|
|
flags.SlashingProtectionJSONFileFlag.Name,
|
|
)
|
|
}
|
|
enc, err := file.ReadFileAsBytes(protectionFilePath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
log.Infof("Starting import of slashing protection file %s", protectionFilePath)
|
|
buf := bytes.NewBuffer(enc)
|
|
if err := slashingprotection.ImportStandardProtectionJSON(
|
|
cliCtx.Context, valDB, buf,
|
|
); err != nil {
|
|
return err
|
|
}
|
|
log.Infof("Slashing protection JSON successfully imported into %s", dataDir)
|
|
return nil
|
|
}
|