package wwctl

import (
	"os"

	"github.com/spf13/cobra"
	"github.com/warewulf/warewulf/internal/app/wwctl/clean"
	"github.com/warewulf/warewulf/internal/app/wwctl/configure"
	"github.com/warewulf/warewulf/internal/app/wwctl/genconf"
	"github.com/warewulf/warewulf/internal/app/wwctl/image"
	"github.com/warewulf/warewulf/internal/app/wwctl/node"
	"github.com/warewulf/warewulf/internal/app/wwctl/overlay"
	"github.com/warewulf/warewulf/internal/app/wwctl/power"
	"github.com/warewulf/warewulf/internal/app/wwctl/profile"
	"github.com/warewulf/warewulf/internal/app/wwctl/server"
	"github.com/warewulf/warewulf/internal/app/wwctl/ssh"
	"github.com/warewulf/warewulf/internal/app/wwctl/upgrade"
	"github.com/warewulf/warewulf/internal/app/wwctl/version"
	warewulfconf "github.com/warewulf/warewulf/internal/pkg/config"
	"github.com/warewulf/warewulf/internal/pkg/help"
	"github.com/warewulf/warewulf/internal/pkg/wwlog"
)

var (
	rootCmd = &cobra.Command{
		DisableFlagsInUseLine: true,
		Use:                   "wwctl COMMAND [OPTIONS]",
		Short:                 "Warewulf Control",
		Long:                  "Control interface to the Warewulf Cluster Provisioning System.",
		PersistentPreRunE:     rootPersistentPreRunE,
		SilenceUsage:          true,
		SilenceErrors:         true,
		Args:                  cobra.NoArgs,
	}
	verboseArg      bool
	DebugFlag       bool
	LogLevel        int
	WarewulfConfArg string
	AllowEmptyConf  bool
)

func init() {
	rootCmd.CompletionOptions.HiddenDefaultCmd = true
	rootCmd.PersistentFlags().BoolVarP(&verboseArg, "verbose", "v", false, "Run with increased verbosity.")
	rootCmd.PersistentFlags().BoolVarP(&DebugFlag, "debug", "d", false, "Run with debugging messages enabled.")
	rootCmd.PersistentFlags().IntVar(&LogLevel, "loglevel", wwlog.INFO, "Set log level to given string")
	_ = rootCmd.PersistentFlags().MarkHidden("loglevel")
	rootCmd.PersistentFlags().StringVar(&WarewulfConfArg, "warewulfconf", "", "Set the warewulf configuration file")
	rootCmd.PersistentFlags().BoolVar(&AllowEmptyConf, "emptyconf", false, "Allow empty configuration")
	_ = rootCmd.PersistentFlags().MarkHidden("emptyconf")
	rootCmd.SetUsageTemplate(help.UsageTemplate)
	rootCmd.SetHelpTemplate(help.HelpTemplate)
	rootCmd.AddCommand(overlay.GetCommand())
	rootCmd.AddCommand(image.GetCommand())
	rootCmd.AddCommand(node.GetCommand())
	rootCmd.AddCommand(power.GetCommand())
	rootCmd.AddCommand(profile.GetCommand())
	rootCmd.AddCommand(configure.GetCommand())
	rootCmd.AddCommand(server.GetCommand())
	rootCmd.AddCommand(version.GetCommand())
	rootCmd.AddCommand(ssh.GetCommand())
	rootCmd.AddCommand(genconf.GetCommand())
	rootCmd.AddCommand(clean.GetCommand())
	rootCmd.AddCommand(upgrade.GetCommand())
}

// GetRootCommand returns the root cobra.Command for the application.
func GetRootCommand() *cobra.Command {
	return rootCmd
}

func rootPersistentPreRunE(cmd *cobra.Command, args []string) (err error) {
	if DebugFlag {
		wwlog.SetLogLevel(wwlog.DEBUG)
	} else if verboseArg {
		wwlog.SetLogLevel(wwlog.VERBOSE)
	} else {
		wwlog.SetLogLevel(wwlog.INFO)
	}
	if LogLevel != wwlog.INFO {
		wwlog.SetLogLevel(LogLevel)
	}

	if WarewulfConfArg != "" {
		warewulfconf.ConfigFile = WarewulfConfArg
	} else if os.Getenv("WAREWULFCONF") != "" {
		warewulfconf.ConfigFile = os.Getenv("WAREWULFCONF")
	}

	conf := warewulfconf.Get()
	if !AllowEmptyConf && !conf.InitializedFromFile() {
		if err = conf.Read(warewulfconf.ConfigFile, true); err != nil {
			wwlog.Error("error reading config file: %s", err)
			return
		}
	}
	return
}