services: add checks flag

* checks flags allows to define http and tcp healthchecks

Author: antham

docs/reference.md

@@ -225,6 +225,7 @@ koyeb apps init NAME [flags]
 ### Options
 
 ```
+      --checks strings                          HTTP healthcheck (<port>:http:<path>) and TCP healthcheck (<port>:tcp)
       --docker string                           Docker image
       --docker-args strings                     Docker args
       --docker-command string                   Docker command
@@ -898,6 +899,7 @@ koyeb services create NAME [flags]
 
 ```
   -a, --app string                              App
+      --checks strings                          HTTP healthcheck (<port>:http:<path>) and TCP healthcheck (<port>:tcp)
       --docker string                           Docker image
       --docker-args strings                     Docker args
       --docker-command string                   Docker command
@@ -1179,6 +1181,7 @@ koyeb services update NAME [flags]
 ### Options
 
 ```
+      --checks strings                          HTTP healthcheck (<port>:http:<path>) and TCP healthcheck (<port>:tcp)
       --docker string                           Docker image
       --docker-args strings                     Docker args
       --docker-command string                   Docker command

go.mod

@@ -9,7 +9,7 @@ require (
 	github.com/gofrs/uuid v4.3.0+incompatible
 	github.com/gorilla/websocket v1.5.0
 	github.com/iancoleman/strcase v0.2.0
-	github.com/koyeb/koyeb-api-client-go v0.0.0-20221026085506-07c8c0a7b717
+	github.com/koyeb/koyeb-api-client-go v0.0.0-20230130102659-26de8f39c858
 	github.com/logrusorgru/aurora v2.0.3+incompatible
 	github.com/manifoldco/promptui v0.9.0
 	github.com/mitchellh/go-homedir v1.1.0

go.sum

@@ -241,6 +241,10 @@ github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxv
 github.com/konsorten/go-windows-terminal-sequences v1.0.2/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
 github.com/koyeb/koyeb-api-client-go v0.0.0-20221026085506-07c8c0a7b717 h1:/x2pFgpZtC936I+VV+DFEt0XnWTio6cO2/HuCd0Vn38=
 github.com/koyeb/koyeb-api-client-go v0.0.0-20221026085506-07c8c0a7b717/go.mod h1:+oQfFj2WL3gi9Pb+UHbob4D7xaT52mPfKyH1UvWa4PQ=
+github.com/koyeb/koyeb-api-client-go v0.0.0-20221103082152-736202b94397 h1:/w+PLg5VsAh485pwQrVefY+/So6sSNDdkDZre13bjP8=
+github.com/koyeb/koyeb-api-client-go v0.0.0-20221103082152-736202b94397/go.mod h1:+oQfFj2WL3gi9Pb+UHbob4D7xaT52mPfKyH1UvWa4PQ=
+github.com/koyeb/koyeb-api-client-go v0.0.0-20230130102659-26de8f39c858 h1:BcXBg0dWO84Ng1/yWlasGPfz0OA1cg3vwOdDqk1bht4=
+github.com/koyeb/koyeb-api-client-go v0.0.0-20230130102659-26de8f39c858/go.mod h1:+oQfFj2WL3gi9Pb+UHbob4D7xaT52mPfKyH1UvWa4PQ=
 github.com/kr/fs v0.1.0/go.mod h1:FFnZGqtBN9Gxj7eW1uZ42v5BccTP0vu6NEaFoC2HwRg=
 github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
 github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=

pkg/koyeb/services.go

@@ -2,6 +2,7 @@ package koyeb
 
 import (
 	"context"
+	"fmt"
 	"strconv"
 	"strings"
 
@@ -197,6 +198,7 @@ func addServiceDefinitionFlags(flags *pflag.FlagSet) {
 	flags.String("instance-type", "nano", "Instance type")
 	flags.Int64("min-scale", 1, "Min scale")
 	flags.Int64("max-scale", 1, "Max scale")
+	flags.StringSlice("checks", []string{""}, "HTTP healthcheck (<port>:http:<path>) and TCP healthcheck (<port>:tcp)")
 }
 
 func parseServiceDefinitionFlags(flags *pflag.FlagSet, definition *koyeb.DeploymentDefinition, useDefault bool) error {
@@ -292,6 +294,46 @@ func parseServiceDefinitionFlags(flags *pflag.FlagSet, definition *koyeb.Deploym
 		definition.SetScalings([]koyeb.DeploymentScaling{*scaling})
 	}
 
+	if flags.Lookup("checks").Changed {
+		checks, _ := flags.GetStringSlice("checks")
+		healthchecks := []koyeb.DeploymentHealthCheck{}
+
+		for _, c := range checks {
+			healthcheck := koyeb.NewDeploymentHealthCheck()
+			components := strings.Split(c, ":")
+			componentsCount := len(components)
+			if componentsCount < 2 || componentsCount > 3 {
+				return fmt.Errorf(`Invalid checks: "%s", must be either "<port>:http:<path>" or "<port>:tcp"`, c)
+			}
+
+			healthcheckType := components[1]
+			portStr := components[0]
+			port, err := strconv.Atoi(portStr)
+			if err != nil {
+				return errors.Errorf(`Invalid port: "%s"`, portStr)
+			}
+
+			switch healthcheckType {
+			case "http":
+				if componentsCount < 3 {
+					return errors.New("Missing path definition for http check")
+				}
+				HTTPHealthCheck := koyeb.NewHTTPHealthCheck()
+				HTTPHealthCheck.Port = koyeb.PtrInt64(int64(port))
+				HTTPHealthCheck.Path = koyeb.PtrString(components[2])
+				healthcheck.SetHttp(*HTTPHealthCheck)
+			case "tcp":
+				TCPHealthCheck := koyeb.NewTCPHealthCheck()
+				TCPHealthCheck.Port = koyeb.PtrInt64(int64(port))
+				healthcheck.SetTcp(*TCPHealthCheck)
+			default:
+				return fmt.Errorf(`Invalid healthcheck: "%s", must be either "http" or "tcp"`, healthcheckType)
+			}
+			healthchecks = append(healthchecks, *healthcheck)
+		}
+		definition.SetHealthChecks(healthchecks)
+	}
+
 	// Docker
 	if useDefault && !flags.Lookup("git").Changed || flags.Lookup("docker").Changed && !flags.Lookup("git").Changed {
 		createDockerSource := koyeb.NewDockerSourceWithDefaults()