mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
80 lines
2.8 KiB
Go
80 lines
2.8 KiB
Go
package slashingprotection
|
|
|
|
import (
|
|
"encoding/json"
|
|
"path/filepath"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/prysmaticlabs/prysm/shared/cmd"
|
|
"github.com/prysmaticlabs/prysm/shared/fileutil"
|
|
"github.com/prysmaticlabs/prysm/validator/accounts/prompt"
|
|
"github.com/prysmaticlabs/prysm/validator/db/kv"
|
|
"github.com/prysmaticlabs/prysm/validator/flags"
|
|
export "github.com/prysmaticlabs/prysm/validator/slashing-protection/local/standard-protection-format"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
const (
|
|
jsonExportFileName = "slashing_protection.json"
|
|
)
|
|
|
|
// ExportSlashingProtectionJSONCli extracts a validator's slashing protection
|
|
// history from their database and formats it into an EIP-3076 standard JSON
|
|
// file via a CLI entrypoint to make it easy to migrate machines or eth2 clients.
|
|
//
|
|
// Steps:
|
|
// 1. Parse a path to the validator's datadir from the CLI context.
|
|
// 2. Open the validator database.
|
|
// 3. Call the function which actually exports the data from
|
|
// from the validator's db into an EIP standard slashing protection format
|
|
// 4. Format and save the JSON file to a user's specified output directory.
|
|
func ExportSlashingProtectionJSONCli(cliCtx *cli.Context) error {
|
|
var err error
|
|
dataDir := cliCtx.String(cmd.DataDirFlag.Name)
|
|
if !cliCtx.IsSet(cmd.DataDirFlag.Name) {
|
|
dataDir, err = prompt.InputDirectory(cliCtx, prompt.DataDirDirPromptText, cmd.DataDirFlag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
validatorDB, 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 := validatorDB.Close(); err != nil {
|
|
log.WithError(err).Errorf("Could not close validator DB")
|
|
}
|
|
}()
|
|
eipJSON, err := export.ExportStandardProtectionJSON(cliCtx.Context, validatorDB)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not export slashing protection history")
|
|
}
|
|
outputDir, err := prompt.InputDirectory(
|
|
cliCtx,
|
|
"Enter your desired output directory for your slashing protection history",
|
|
flags.SlashingProtectionExportDirFlag,
|
|
)
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not get slashing protection json file")
|
|
}
|
|
if outputDir == "" {
|
|
return errors.New("output directory not specified")
|
|
}
|
|
exists, err := fileutil.HasDir(outputDir)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "could not check if output directory %s already exists", outputDir)
|
|
}
|
|
if !exists {
|
|
if err := fileutil.MkdirAll(outputDir); err != nil {
|
|
return errors.Wrapf(err, "could not create output directory %s", outputDir)
|
|
}
|
|
}
|
|
outputFilePath := filepath.Join(outputDir, jsonExportFileName)
|
|
encoded, err := json.MarshalIndent(eipJSON, "", "\t")
|
|
if err != nil {
|
|
return errors.Wrap(err, "could not JSON marshal slashing protection history")
|
|
}
|
|
return fileutil.WriteFile(outputFilePath, encoded)
|
|
}
|