package morph

import (
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/ape"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/balance"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/config"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/container"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/contract"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/frostfsid"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/generate"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/netmap"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/node"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/notary"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/policy"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/proxy"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-adm/internal/modules/morph/util"
	"github.com/spf13/cobra"
	"github.com/spf13/viper"
)

const (
	maxObjectSizeCLIFlag = "max-object-size"

	epochDurationCLIFlag = "epoch-duration"

	containerFeeCLIFlag      = "container-fee"
	containerAliasFeeCLIFlag = "container-alias-fee"

	candidateFeeCLIFlag = "candidate-fee"

	homomorphicHashDisabledCLIFlag = "homomorphic-disabled"

	withdrawFeeCLIFlag = "withdraw-fee"
)

var (
	// RootCmd is a root command of config section.
	RootCmd = &cobra.Command{
		Use:   "morph",
		Short: "Section for morph network configuration commands",
	}

	initCmd = &cobra.Command{
		Use:   "init",
		Short: "Initialize side chain network with smart-contracts and network settings",
		PreRun: func(cmd *cobra.Command, _ []string) {
			_ = viper.BindPFlag(util.AlphabetWalletsFlag, cmd.Flags().Lookup(util.AlphabetWalletsFlag))
			_ = viper.BindPFlag(util.EndpointFlag, cmd.Flags().Lookup(util.EndpointFlag))
			_ = viper.BindPFlag(util.EpochDurationInitFlag, cmd.Flags().Lookup(epochDurationCLIFlag))
			_ = viper.BindPFlag(util.MaxObjectSizeInitFlag, cmd.Flags().Lookup(maxObjectSizeCLIFlag))
			_ = viper.BindPFlag(util.HomomorphicHashDisabledInitFlag, cmd.Flags().Lookup(homomorphicHashDisabledCLIFlag))
			_ = viper.BindPFlag(util.CandidateFeeInitFlag, cmd.Flags().Lookup(candidateFeeCLIFlag))
			_ = viper.BindPFlag(util.ContainerFeeInitFlag, cmd.Flags().Lookup(containerFeeCLIFlag))
			_ = viper.BindPFlag(util.ContainerAliasFeeInitFlag, cmd.Flags().Lookup(containerAliasFeeCLIFlag))
			_ = viper.BindPFlag(util.WithdrawFeeInitFlag, cmd.Flags().Lookup(withdrawFeeCLIFlag))
			_ = viper.BindPFlag(util.ProtoConfigPath, cmd.Flags().Lookup(util.ProtoConfigPath))
		},
		RunE: initializeSideChainCmd,
	}
)

func init() {
	RootCmd.AddCommand(generate.RefillGasCmd)
	initInitCmd()
	RootCmd.AddCommand(contract.DeployCmd)
	RootCmd.AddCommand(generate.GenerateStorageCmd)
	RootCmd.AddCommand(netmap.ForceNewEpoch)
	RootCmd.AddCommand(node.RemoveCmd)
	RootCmd.AddCommand(policy.Set)
	RootCmd.AddCommand(policy.Dump)
	RootCmd.AddCommand(contract.DumpHashesCmd)
	RootCmd.AddCommand(config.SetCmd)
	RootCmd.AddCommand(config.DumpCmd)
	RootCmd.AddCommand(balance.DumpCmd)
	RootCmd.AddCommand(contract.UpdateCmd)
	RootCmd.AddCommand(container.ListCmd)
	RootCmd.AddCommand(container.RestoreCmd)
	RootCmd.AddCommand(container.DumpCmd)
	RootCmd.AddCommand(generate.GenerateAlphabetCmd)
	RootCmd.AddCommand(notary.DepositCmd)
	RootCmd.AddCommand(netmap.CandidatesCmd)

	RootCmd.AddCommand(ape.Cmd)
	RootCmd.AddCommand(proxy.AddAccountCmd)
	RootCmd.AddCommand(proxy.RemoveAccountCmd)

	RootCmd.AddCommand(frostfsid.Cmd)
}

func initInitCmd() {
	RootCmd.AddCommand(initCmd)
	initCmd.Flags().String(util.AlphabetWalletsFlag, "", util.AlphabetWalletsFlagDesc)
	initCmd.Flags().StringP(util.EndpointFlag, util.EndpointFlagShort, "", util.EndpointFlagDesc)
	initCmd.Flags().String(util.ContractsInitFlag, "", util.ContractsInitFlagDesc)
	initCmd.Flags().String(util.ContractsURLFlag, "", util.ContractsURLFlagDesc)
	initCmd.Flags().Uint(epochDurationCLIFlag, 240, "Amount of side chain blocks in one FrostFS epoch")
	initCmd.Flags().Uint(maxObjectSizeCLIFlag, 67108864, "Max single object size in bytes")
	initCmd.Flags().Bool(homomorphicHashDisabledCLIFlag, false, "Disable object homomorphic hashing")
	// Defaults are taken from neo-preodolenie.
	initCmd.Flags().Uint64(containerFeeCLIFlag, 1000, "Container registration fee")
	initCmd.Flags().Uint64(containerAliasFeeCLIFlag, 500, "Container alias fee")
	initCmd.Flags().String(util.ProtoConfigPath, "", "Path to the consensus node configuration")
	initCmd.Flags().String(util.LocalDumpFlag, "", "Path to the blocks dump file")
	initCmd.MarkFlagsMutuallyExclusive(util.ContractsInitFlag, util.ContractsURLFlag)
}