package apply

// Author: Weisen Pan
// Date: 2023-10-24


import (
	"fmt"
	"os"

	applypkg "github.com/hkust-adsl/kubernetes-scheduler-simulator/knets_pkg/apply"
	log "github.com/sirupsen/logrus"
	"github.com/spf13/cobra"
)

// options stores the command line options.
var options = applypkg.Options{}

// ApplyCmd represents the 'apply' command.
var ApplyCmd = &cobra.Command{
	Use:   "apply",
	Short: "Make a reasonable cluster capacity planning based on application resource requirements",
	Run: func(cmd *cobra.Command, args []string) {
		// Create a new Applier instance with the specified options.
		applier := applypkg.NewApplier(options)

		// Run the Applier to perform cluster capacity planning.
		if err := applier.Run(); err != nil {
			fmt.Printf("apply error: %s", err.Error())
			os.Exit(1)
		}
	},
}

func init() {
	// Define command-line flags for the 'apply' command.
	ApplyCmd.Flags().StringVarP(&options.SimonConfig, "simon-config", "f", options.SimonConfig, "path to the cluster kube-config file used to connect cluster, one of both kube-config and cluster-config must exist.")
	ApplyCmd.Flags().StringVarP(&options.DefaultSchedulerConfigFile, "default-scheduler-config", "s", options.DefaultSchedulerConfigFile, "path to JSON or YAML file containing scheduler configuration.")
	ApplyCmd.Flags().BoolVar(&options.UseGreed, "use-greed", false, "use greedy algorithm when queue pods")
	ApplyCmd.Flags().BoolVarP(&options.Interactive, "interactive", "i", false, "interactive mode")
	ApplyCmd.Flags().StringSliceVarP(&options.ExtendedResources, "extended-resources", "e", nil, "show extended resources when reporting, e.g. open-local, gpu")

	// Mark the 'simon-config' flag as required.
	if err := ApplyCmd.MarkFlagRequired("simon-config"); err != nil {
		log.Fatal("failed to init ApplyCmd on simon-config flag")
	}
}