-
Notifications
You must be signed in to change notification settings - Fork 12
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[backport] feat: add new cmd fpd commit-pubrand (#153)
- Loading branch information
Showing
5 changed files
with
209 additions
and
5 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,103 @@ | ||
| package daemon | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "math" | ||
| "path/filepath" | ||
| "strconv" | ||
|
|
||
| bbntypes "github.com/babylonlabs-io/babylon/types" | ||
| fpcc "github.com/babylonlabs-io/finality-provider/clientcontroller" | ||
| eotsclient "github.com/babylonlabs-io/finality-provider/eotsmanager/client" | ||
| fpcfg "github.com/babylonlabs-io/finality-provider/finality-provider/config" | ||
| "github.com/babylonlabs-io/finality-provider/finality-provider/service" | ||
| "github.com/babylonlabs-io/finality-provider/finality-provider/store" | ||
| "github.com/babylonlabs-io/finality-provider/log" | ||
| "github.com/babylonlabs-io/finality-provider/metrics" | ||
| "github.com/babylonlabs-io/finality-provider/util" | ||
| "github.com/cosmos/cosmos-sdk/client" | ||
| "github.com/spf13/cobra" | ||
| ) | ||
|
|
||
| // CommandCommitPubRand returns the commit-pubrand command | ||
| func CommandCommitPubRand() *cobra.Command { | ||
| var cmd = &cobra.Command{ | ||
| Use: "unsafe-commit-pubrand [fp-eots-pk-hex] [target-height]", | ||
| Aliases: []string{"unsafe-cpr"}, | ||
| Short: "[UNSAFE] Manually trigger public randomness commitment for a finality provider", | ||
| Long: `[UNSAFE] Manually trigger public randomness commitment for a finality provider. | ||
| WARNING: this can drain the finality provider's balance if the target height is too high.`, | ||
| Example: `fpd unsafe-commit-pubrand --home /home/user/.fpd [fp-eots-pk-hex] [target-height]`, | ||
| Args: cobra.ExactArgs(2), | ||
| RunE: runCommandCommitPubRand, | ||
| } | ||
| cmd.Flags().Uint64("start-height", math.MaxUint64, "The block height to start committing pubrand from (optional)") | ||
| return cmd | ||
| } | ||
|
|
||
| func runCommandCommitPubRand(cmd *cobra.Command, args []string) error { | ||
| fpPk, err := bbntypes.NewBIP340PubKeyFromHex(args[0]) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| targetHeight, err := strconv.ParseUint(args[1], 10, 64) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| startHeight, err := cmd.Flags().GetUint64("start-height") | ||
| if err != nil { | ||
| return err | ||
| } | ||
|
|
||
| // Get homePath from context like in start.go | ||
| clientCtx := client.GetClientContextFromCmd(cmd) | ||
| homePath, err := filepath.Abs(clientCtx.HomeDir) | ||
| if err != nil { | ||
| return err | ||
| } | ||
| homePath = util.CleanAndExpandPath(homePath) | ||
|
|
||
| cfg, err := fpcfg.LoadConfig(homePath) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to load configuration: %w", err) | ||
| } | ||
|
|
||
| logger, err := log.NewRootLoggerWithFile(fpcfg.LogFile(homePath), cfg.LogLevel) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to initialize the logger: %w", err) | ||
| } | ||
|
|
||
| db, err := cfg.DatabaseConfig.GetDBBackend() | ||
| if err != nil { | ||
| return fmt.Errorf("failed to create db backend: %w", err) | ||
| } | ||
|
|
||
| fpStore, err := store.NewFinalityProviderStore(db) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to initiate finality provider store: %w", err) | ||
| } | ||
| pubRandStore, err := store.NewPubRandProofStore(db) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to initiate public randomness store: %w", err) | ||
| } | ||
| cc, err := fpcc.NewClientController(cfg.ChainType, cfg.BabylonConfig, &cfg.BTCNetParams, logger) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to create rpc client for the Babylon chain: %w", err) | ||
| } | ||
| em, err := eotsclient.NewEOTSManagerGRpcClient(cfg.EOTSManagerAddress) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to create EOTS manager client: %w", err) | ||
| } | ||
|
|
||
| fp, err := service.NewFinalityProviderInstance( | ||
| fpPk, cfg, fpStore, pubRandStore, cc, em, metrics.NewFpMetrics(), "", | ||
| make(chan<- *service.CriticalError), logger) | ||
| if err != nil { | ||
| return fmt.Errorf("failed to create finality-provider %s instance: %w", fpPk.MarshalHex(), err) | ||
| } | ||
|
|
||
| if startHeight == math.MaxUint64 { | ||
| return fp.TestCommitPubRand(targetHeight) | ||
| } | ||
| return fp.TestCommitPubRandWithStartHeight(startHeight, targetHeight) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters