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
}