Adding new verbs and fleshing out the skeleton. Drain works while Kill has an error tha requires a deeper dive.

cmd/stop.go

@@ -0,0 +1,57 @@
+package cmd
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/paypal/gorealis/gen-go/apache/aurora"
+	"github.com/spf13/cobra"
+)
+
+func init() {
+	rootCmd.AddCommand(stopCmd)
+
+	// Stop subcommands
+	stopCmd.AddCommand(stopMaintCmd)
+}
+
+var stopCmd = &cobra.Command{
+	Use:   "stop",
+	Short: "Stop a service or maintenance on a host (DRAIN).",
+}
+
+var stopMaintCmd = &cobra.Command{
+	Use:   "drain [space separated host list]",
+	Short: "Stop maintenance on a host (move to NONE).",
+	Long:  `Transition a list of hosts currently in a maintenance status out of it.`,
+	Run:   endMaintenance,
+}
+
+func endMaintenance(cmd *cobra.Command, args []string) {
+	fmt.Println("Setting hosts to NONE maintenance status.")
+	fmt.Println(args)
+	_, result, err := client.EndMaintenance(args...)
+	if err != nil {
+		fmt.Printf("error: %+v\n", err.Error())
+		os.Exit(1)
+	}
+
+	// Monitor change to NONE mode
+	hostResult, err := monitor.HostMaintenance(
+		args,
+		[]aurora.MaintenanceMode{aurora.MaintenanceMode_NONE},
+		5,
+		10)
+	if err != nil {
+		for host, ok := range hostResult {
+			if !ok {
+				fmt.Printf("Host %s did not transtion into desired mode(s)\n", host)
+			}
+		}
+
+		fmt.Printf("error: %+v\n", err.Error())
+		return
+	}
+
+	fmt.Print(result.String())
+}