mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-07-28 07:36:44 +00:00
91 lines
1.8 KiB
Go
91 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/ethereum/go-ethereum/command/flagset"
|
|
"github.com/ethereum/go-ethereum/command/server/proto"
|
|
)
|
|
|
|
// ChainSetHeadCommand is the command to group the peers commands
|
|
type ChainSetHeadCommand struct {
|
|
*Meta2
|
|
|
|
yes bool
|
|
}
|
|
|
|
// Help implements the cli.Command interface
|
|
func (c *ChainSetHeadCommand) Help() string {
|
|
return `Usage: bor chain sethead <number> [--yes]
|
|
|
|
This command sets the current chain to a certain block`
|
|
}
|
|
|
|
func (c *ChainSetHeadCommand) Flags() *flagset.Flagset {
|
|
flags := c.NewFlagSet("chain sethead")
|
|
|
|
flags.BoolFlag(&flagset.BoolFlag{
|
|
Name: "yes",
|
|
Usage: "Force set head",
|
|
Default: false,
|
|
Value: &c.yes,
|
|
})
|
|
return flags
|
|
}
|
|
|
|
// Synopsis implements the cli.Command interface
|
|
func (c *ChainSetHeadCommand) Synopsis() string {
|
|
return "Set the new head of the chain"
|
|
}
|
|
|
|
// Run implements the cli.Command interface
|
|
func (c *ChainSetHeadCommand) Run(args []string) int {
|
|
flags := c.Flags()
|
|
if err := flags.Parse(args); err != nil {
|
|
c.UI.Error(err.Error())
|
|
return 1
|
|
}
|
|
|
|
args = flags.Args()
|
|
if len(args) != 1 {
|
|
c.UI.Error("No number provided")
|
|
return 1
|
|
}
|
|
|
|
borClt, err := c.BorConn()
|
|
if err != nil {
|
|
c.UI.Error(err.Error())
|
|
return 1
|
|
}
|
|
|
|
arg := args[0]
|
|
fmt.Println(arg)
|
|
|
|
number, err := strconv.Atoi(arg)
|
|
if err != nil {
|
|
c.UI.Error(err.Error())
|
|
return 1
|
|
}
|
|
|
|
if !c.yes {
|
|
response, err := c.UI.Ask("Are you sure you want to reset the database? (y/n)")
|
|
if err != nil {
|
|
c.UI.Error(err.Error())
|
|
return 1
|
|
}
|
|
if response != "y" {
|
|
c.UI.Output("set head aborted")
|
|
return 0
|
|
}
|
|
}
|
|
|
|
if _, err := borClt.ChainSetHead(context.Background(), &proto.ChainSetHeadRequest{Number: uint64(number)}); err != nil {
|
|
c.UI.Error(err.Error())
|
|
return 1
|
|
}
|
|
|
|
c.UI.Output("Done!")
|
|
return 0
|
|
}
|