package main

import (
	"flag"
	"kube-forge/pkg/config"
	"kube-forge/pkg/kubespray"
	"kube-forge/pkg/templates"
	"os"
)

func parseArgs() (string, string, string) {
	var password, configPath, dataDir string
	flag.StringVar(&password, "p", "", "Password to access hosts")
	flag.StringVar(&configPath, "c", "/etc/kube-forge/config.yaml", "Path to config file")
	flag.StringVar(&dataDir, "d", "/var/lib/kube-forge", "Path to data dir")
	flag.Parse()
	return password, configPath, dataDir
}

func main() {
	password, configPath, dataDir := parseArgs()
	config := config.CreateConfig(configPath, dataDir, password)

	repositories, releases := templates.GetHelmAppsConfigData()
	config.Repositories = repositories
	config.Releases = releases

	templates.ApplyTemplates()

	for _, cmd := range os.Args {
		switch cmd {
		case "apply":
			kubespray.InstallCluster("")
			kubespray.CopyK8SAdminConfig("k8s-admin.conf")
		case "apply-modules":
			kubespray.InstallCluster("helm-apps")
		case "upgrade":
			kubespray.UpgradeCluster("")
		case "scale":
			kubespray.ScaleCluster()
		}
	}
}