Add health command to configure HTTP health checks

Author: collimarco

lib/cuber/cli.rb

@@ -7,6 +7,7 @@
 require 'json'
 require 'shellwords'
 require 'time'
+require 'uri'
 
 module Cuber
   class CLI

lib/cuber/cuberfile_parser.rb

@@ -15,6 +15,7 @@ def initialize
       @cron = {}
       @secrets = {}
       @env = {}
+      @health = nil
       @lb = {}
       @ingress = nil
       @ssl = nil
@@ -75,6 +76,10 @@ def cron name, schedule, cmd
     def env key, value, secret: false
       secret ? (@secrets[key] = value) : (@env[key] = value)
     end
+    
+    def health url
+      @health = url
+    end
 
     def lb key, value
       @lb[key] = value

lib/cuber/cuberfile_validator.rb

@@ -20,6 +20,7 @@ def validate
       validate_procs
       validate_cron
       validate_env
+      validate_health
       validate_lb
       validate_ingress
       validate_ssl
@@ -103,6 +104,11 @@ def validate_env
         @errors << "env \"#{key}\" name can only include uppercase letters, digits or underscores" if key !~ /\A[A-Z_]+[A-Z0-9_]*\z/
       end
     end
+    
+    def validate_health
+      return unless @options[:health]
+      @errors << 'health checks must be an http url' unless URI.parse(@options[:health]).kind_of? URI::HTTP
+    end
 
     def validate_lb
       @options[:lb].each do |key, value|

lib/cuber/templates/deployment.yml.erb

@@ -164,10 +164,15 @@ spec:
           value: "8080"
         ports:
         - containerPort: 8080
+        <%- end -%>
+        <%- if procname.to_s == 'web' && @options[:health] -%>
         readinessProbe:
           httpGet:
-            path: /
+            path: <%= URI.parse(@options[:health]).path.to_json %>
             port: 8080
+            httpHeaders:
+              - name: Host
+                value: <%= URI.parse(@options[:health]).host.to_json %>
         <%- end -%>
       <%- if @options[:migrate] && @options[:migrate][:check] -%>
       initContainers:

lib/cuber/version.rb

@@ -1,3 +1,3 @@
 module Cuber
-  VERSION = '1.6.0'.freeze
+  VERSION = '1.7.0'.freeze
 end