cmd.go

package main

import (
	"fmt"
	"os"

	"github.com/abema/kubetool/kube"
	"github.com/alecthomas/kingpin"
	"github.com/fatih/color"
)

var (
	red = color.New(color.FgRed).SprintfFunc()

	app       = kingpin.New("kubetool", "kubernetes bulk task executor.")
	verbose   = app.Flag("verbose", "Enable verbose log.").Short('v').Bool()
	namespace = app.Flag("namespace", "Target namespace. default is all namespaces").String()
	yes       = app.Flag("yes", "Skip confirmation.").Short('y').Bool()
	force     = app.Flag("force", "Force reload pods. Ignores pod status while reloading.").Short('f').Bool()
	interval  = app.Flag("interval", "Reloading interval on restarting each pod.").Default("0").Int()
	minStable = app.Flag("min-stable", "Minimum value of available pod percentage to detect RC is stable or not. (0.0-1.0). Defalt is 0.8").Default("0.8").Float64()

	// command info
	info = app.Command("info", "Print cluster & version info about cluster.").Alias("i")

	// command rc
	rc = app.Command("rc", "Print all rc.")

	// command pod
	pod   = app.Command("pod", "Print all pods").Alias("pods").Alias("po")
	podRC = pod.Flag("rc", "rc name for pod target").String()

	// command reload
	reload     = app.Command("reload", "Reload all pods in rc.")
	reloadName = reload.Arg("rc-name", "Name of target RC.").Required().String()
	reloadOne  = reload.Flag("1", "Reload only 1 pod").Bool()

	// command set version
	update          = app.Command("update", "Update image version of rc")
	updateName      = update.Arg("rc-name", "Name of target RC.").Required().String()
	updateVersion   = update.Arg("version", "Version tag of image.").String()
	updateReload    = update.Flag("reload", "Reload pods after update.").Bool()
	updateReloadOne = update.Flag("1", "Reload only 1 pod after update.").Short('1').Bool()
	updateContainer = update.Flag("container", "Target container name. Default is first container in defs.").Short('c').String()

	fixVersion     = app.Command("fix-version", "Fix all pods to destroy all that has different version of RC ones.")
	fixVersionName = fixVersion.Arg("rc-name", "Name of target RC.").Required().String()
)

func init() {
	app.Version("0.1.0")
}

func main() {

	cmd := kingpin.MustParse(app.Parse(os.Args[1:]))

	ktool := kube.Tool{}
	ktool.SetYes(*yes)
	ktool.SetForce(*force)
	ktool.SetInterval(*interval)

	if namespace != nil {
		ktool.SetNamespace(*namespace)
	}

	if *minStable < 0 || *minStable > 1 {
		fmt.Fprintln(os.Stderr, "minimum stable rate must be in range of 0.0-1.0")
		os.Exit(1)
	}

	var err error

	switch cmd {
	case info.FullCommand():
		err = ktool.PrintInfo()
	case rc.FullCommand():
		err = ktool.PrintRCList()
	case pod.FullCommand():
		rcname := ""
		if podRC != nil {
			rcname = *podRC
		}
		err = ktool.PrintPodList(rcname)
	case reload.FullCommand():
		err = ktool.Reload(*reloadName, *reloadOne)
	case update.FullCommand():
		container := ""
		if updateContainer != nil {
			container = *updateContainer
		}
		err = ktool.Update(*updateName, container, *updateVersion)
		if *updateReload && err == nil {
			err = ktool.Reload(*updateName, *updateReloadOne)
		}
	case fixVersion.FullCommand():
		err = ktool.FixVersion(*fixVersionName)
	}
	if err != nil {
		fmt.Println(red(err.Error()))
		os.Exit(1)
	}

}