Merge pull request #83 from bacongobbler/deis-healthchecks

feat(deis.go): implement `deis healthchecks` client commands

Author: Matthew Fisher

cmd/config.go

@@ -90,6 +90,16 @@ func ConfigSet(appID string, configVars []string) error {
 		configMap["SSH_KEY"] = base64.StdEncoding.EncodeToString([]byte(sshKey))
 	}
 
+	// NOTE(bacongobbler): check if the user is using the old way to set healthchecks. If so,
+	// send them a deprecation notice.
+	for key, _ := range configMap {
+		if strings.Contains(key, "HEALTHCHECK_") {
+			fmt.Println(`Hey there! We've noticed that you're using 'deis config:set HEALTHCHECK_URL'
+to set up healthchecks. This functionality has been deprecated. In the future, please use
+'deis healthchecks' to set up application health checks. Thanks!`)
+		}
+	}
+
 	fmt.Print("Creating config... ")
 
 	quit := progress()

cmd/healthchecks.go

@@ -0,0 +1,106 @@
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/deis/controller-sdk-go/api"
+	"github.com/deis/controller-sdk-go/config"
+)
+
+// HealthchecksList lists an app's healthchecks.
+func HealthchecksList(appID string) error {
+	c, appID, err := load(appID)
+
+	if err != nil {
+		return err
+	}
+
+	config, err := config.List(c, appID)
+
+	if err != nil {
+		return err
+	}
+
+	fmt.Printf("=== %s Healthchecks\n\n", appID)
+
+	fmt.Println("--- Liveness")
+	if livenessProbe, found := config.Healthcheck["livenessProbe"]; found {
+		fmt.Println(livenessProbe)
+	} else {
+		fmt.Println("No liveness probe configured.")
+	}
+
+	fmt.Println("\n--- Readiness")
+	if readinessProbe, found := config.Healthcheck["readinessProbe"]; found {
+		fmt.Println(readinessProbe)
+	} else {
+		fmt.Println("No readiness probe configured.")
+	}
+	return nil
+}
+
+// HealthchecksSet sets an app's healthchecks.
+func HealthchecksSet(appID, healthcheckType string, probe *api.Healthcheck) error {
+	c, appID, err := load(appID)
+
+	if err != nil {
+		return err
+	}
+
+	fmt.Printf("Applying %s healthcheck... ", healthcheckType)
+
+	quit := progress()
+	configObj := api.Config{}
+	configObj.Healthcheck = make(map[string]*api.Healthcheck)
+
+	configObj.Healthcheck[healthcheckType] = probe
+
+	_, err = config.Set(c, appID, configObj)
+
+	quit <- true
+	<-quit
+
+	if err != nil {
+		return err
+	}
+
+	fmt.Print("done\n\n")
+
+	return HealthchecksList(appID)
+}
+
+// HealthchecksUnset removes an app's healthchecks.
+func HealthchecksUnset(appID string, healthchecks []string) error {
+	c, appID, err := load(appID)
+
+	if err != nil {
+		return err
+	}
+
+	fmt.Print("Removing healthchecks... ")
+
+	quit := progress()
+
+	configObj := api.Config{}
+
+	healthcheckMap := make(map[string]*api.Healthcheck)
+
+	for _, healthcheck := range healthchecks {
+		healthcheckMap[healthcheck] = nil
+	}
+
+	configObj.Healthcheck = healthcheckMap
+
+	_, err = config.Set(c, appID, configObj)
+
+	quit <- true
+	<-quit
+
+	if err != nil {
+		return err
+	}
+
+	fmt.Print("done\n\n")
+
+	return HealthchecksList(appID)
+}

deis.go

@@ -43,6 +43,7 @@ Subcommands, use 'deis help [subcommand]' to learn more::
   config        manage environment variables that define app config
   domains       manage and assign domain names to your applications
   git           manage git for applications
+  healthchecks  manage healthchecks for applications
   keys          manage ssh keys used for 'git push' deployments
   limits        manage resource limits for your application
   perms         manage permissions for applications
@@ -98,6 +99,8 @@ Use 'git push deis master' to deploy to an application.
 		err = parser.Domains(argv)
 	case "git":
 		err = parser.Git(argv)
+	case "healthchecks":
+		err = parser.Healthchecks(argv)
 	case "help":
 		fmt.Print(usage)
 		return 0