cmd.go

package mesos

import (
	"strings"

	"github.com/ovh/cds/sdk"
	"github.com/ovh/cds/sdk/hatchery"
	"github.com/spf13/cobra"
	"github.com/spf13/viper"
)

func init() {
	hatcheryMesos = &HatcheryMesos{}
	Cmd.Flags().StringVar(&hatcheryMesos.marathonHost, "marathon-host", "", "marathon-host")
	viper.BindPFlag("marathon-host", Cmd.Flags().Lookup("marathon-host"))

	Cmd.Flags().StringVar(&hatcheryMesos.marathonID, "marathon-id", "", "marathon-id")
	viper.BindPFlag("marathon-id", Cmd.Flags().Lookup("marathon-id"))

	Cmd.Flags().StringVar(&hatcheryMesos.marathonVHOST, "marathon-vhost", "", "marathon-vhost")
	viper.BindPFlag("marathon-vhost", Cmd.Flags().Lookup("marathon-vhost"))

	Cmd.Flags().StringVar(&hatcheryMesos.marathonUser, "marathon-user", "", "marathon-user")
	viper.BindPFlag("marathon-user", Cmd.Flags().Lookup("marathon-user"))

	Cmd.Flags().StringVar(&hatcheryMesos.marathonPassword, "marathon-password", "", "marathon-password")
	viper.BindPFlag("marathon-password", Cmd.Flags().Lookup("marathon-password"))

	Cmd.Flags().StringVar(&hatcheryMesos.marathonLabelsString, "marathon-labels", "", "marathon-labels")
	viper.BindPFlag("marathon-labels", Cmd.Flags().Lookup("marathon-labels"))

	Cmd.Flags().IntVar(&hatcheryMesos.defaultMemory, "worker-memory", 1024, "Worker default memory")
	viper.BindPFlag("worker-memory", Cmd.Flags().Lookup("worker-memory"))

	Cmd.Flags().IntVar(&hatcheryMesos.workerTTL, "worker-ttl", 10, "Worker TTL (minutes)")
	viper.BindPFlag("worker-ttl", Cmd.Flags().Lookup("worker-ttl"))
}

// Cmd configures comamnd for HatcheryLocal
var Cmd = &cobra.Command{
	Use:   "mesos",
	Short: "Hatchery mesos commands: hatchery mesos --help",
	Long: `Hatchery mesos commands: hatchery mesos <command>
Start worker model instances on a mesos cluster

$ cds generate token --group shared.infra --expiration persistent
2706bda13748877c57029598b915d46236988c7c57ea0d3808524a1e1a3adef4

$ hatchery mesos --api=https://<api.domain> --token=<token>

	`,
	Run: func(cmd *cobra.Command, args []string) {
		hatchery.Born(hatcheryMesos, viper.GetString("api"), viper.GetString("token"), viper.GetInt("provision"), viper.GetInt("request-api-timeout"), viper.GetBool("insecure"))
	},
	PreRun: func(cmd *cobra.Command, args []string) {
		hatcheryMesos.token = viper.GetString("token")

		hatcheryMesos.marathonHost = viper.GetString("marathon-host")
		if hatcheryMesos.marathonHost == "" {
			sdk.Exit("flag or environmnent variable marathon-host not provided, aborting\n")
		}

		hatcheryMesos.marathonID = viper.GetString("marathon-id")
		if hatcheryMesos.marathonID == "" {
			sdk.Exit("flag or environmnent variable marathon-id not provided, aborting\n")
		}

		hatcheryMesos.marathonVHOST = viper.GetString("marathon-vhost")
		if hatcheryMesos.marathonVHOST == "" {
			sdk.Exit("flag or environmnent variable marathon-vhost not provided, aborting\n")
		}

		hatcheryMesos.marathonUser = viper.GetString("marathon-user")
		if hatcheryMesos.marathonUser == "" {
			sdk.Exit("flag or environmnent variable marathon-user not provided, aborting\n")
		}

		hatcheryMesos.marathonPassword = viper.GetString("marathon-password")
		if hatcheryMesos.marathonPassword == "" {
			sdk.Exit("flag or environmnent variable marathon-password not provided, aborting\n")
		}

		hatcheryMesos.marathonLabelsString = viper.GetString("marathon-labels")
		hatcheryMesos.marathonLabels = map[string]string{}
		if hatcheryMesos.marathonLabelsString != "" {
			array := strings.Split(hatcheryMesos.marathonLabelsString, ",")
			for _, s := range array {
				if !strings.Contains(s, "=") {
					continue
				}
				tuple := strings.Split(s, "=")
				if len(tuple) != 2 {
					sdk.Exit("malformatted flag or environmnent variable marathon-labels")
				}
				hatcheryMesos.marathonLabels[tuple[0]] = tuple[1]
			}
		}

	},
}