package cli

import (
	"fmt"

	"github.com/ledgerwatch/erigon/cmd/lightclient/clparams"
	"github.com/ledgerwatch/erigon/cmd/sentinel_node/cli/flags"
	"github.com/urfave/cli"
)

type LightClientCliCfg struct {
	GenesisCfg     *clparams.GenesisConfig     `json:"genesisCfg"`
	BeaconCfg      *clparams.BeaconChainConfig `json:"beaconCfg"`
	NetworkCfg     *clparams.NetworkConfig     `json:"networkCfg"`
	Port           uint                        `json:"port"`
	Addr           string                      `json:"address"`
	ServerAddr     string                      `json:"serverAddr"`
	ServerProtocol string                      `json:"serverProtocol"`
	ServerTcpPort  uint                        `json:"serverTcpPort"`
	LogLvl         uint                        `json:"logLevel"`
	NoDiscovery    bool                        `json:"noDiscovery"`
	CheckpointUri  string                      `json:"checkpointUri"`
}

func SetUpLightClientCfg(ctx *cli.Context) (*LightClientCliCfg, error) {
	cfg := &LightClientCliCfg{}
	chainName := ctx.GlobalString(flags.LightClientChain.Name)
	var err error
	var network clparams.NetworkType
	cfg.GenesisCfg, cfg.NetworkCfg, cfg.BeaconCfg, network, err = clparams.GetConfigsByNetworkName(chainName)
	if err != nil {
		return nil, err
	}
	cfg.ServerAddr = fmt.Sprintf("%s:%d", ctx.GlobalString(flags.LightClientServerAddr.Name), ctx.GlobalInt(flags.LightClientServerPort.Name))
	cfg.ServerProtocol = ServerProtocolFromInt(ctx.GlobalUint(flags.LightClientServerProtocol.Name))

	cfg.Port = uint(ctx.GlobalInt(flags.LightClientPort.Name))
	cfg.Addr = ctx.GlobalString(flags.LightClientAddr.Name)

	cfg.LogLvl = ctx.GlobalUint(flags.LightClientVerbosity.Name)
	cfg.NoDiscovery = !ctx.GlobalBoolT(flags.LightClientDiscovery.Name)
	cfg.CheckpointUri = clparams.GetCheckpointSyncEndpoint(network)
	return cfg, nil
}

func ServerProtocolFromInt(n uint) string {
	switch n {
	case 1:
		return "tcp"
	case 2:
		return "udp"
	default:
		return "tcp"
	}
}