kubes 0.4.4 → 0.5.1

data/lib/kubes/compiler/shared/custom_helpers.rb

@@ -0,0 +1,17 @@
+module Kubes::Compiler::Shared
+  module CustomHelpers
+    # Load custom helper methods from project
+    @@custom_helpers_loaded = false
+    def load_custom_helpers
+      return if @@custom_helpers_loaded
+      paths = Dir.glob("#{Kubes.root}/.kubes/helpers/**/*.rb")
+      paths.sort_by! { |p| p.size } # so namespaces are loaded first
+      paths.each do |path|
+        filename = path.sub(%r{.*.kubes/helpers/},'').sub('.rb','')
+        module_name = filename.camelize
+        self.class.send :include, module_name.constantize
+      end
+      @@custom_helpers_loaded = true
+    end
+  end
+end

data/lib/kubes/compiler/shared/helpers.rb

@@ -5,12 +5,21 @@ module Kubes::Compiler::Shared
     extend Kubes::Compiler::Dsl::Core::Fields
     fields "name"
 
-    def built_image
+    def docker_image
       return @options[:image] if @options[:image] # override
+      return Kubes.config.image if Kubes.config.image
+      built_image_helper
+    end
+
+    def built_image
+      Deprecated.new.built_image
+      built_image_helper
+    end
 
+    def built_image_helper
       path = Kubes.config.state.docker_image_path
       unless File.exist?(path)
-        raise "Missing file with docker image built by kubes: #{path}. Try first running: kubes docker build"
+        raise Kubes::MissingDockerImage.new("Missing file with docker image built by kubes: #{path}. Try first running: kubes docker build")
       end
       IO.read(path)
     end
@@ -25,7 +34,7 @@ module Kubes::Compiler::Shared
     end
 
     def encode64(v)
-      Base64.strict_encode64(v).strip
+      Base64.strict_encode64(v.to_s).strip
     end
     alias_method :base64, :encode64
 

data/lib/kubes/compiler/shared/helpers/deprecated.rb

@@ -0,0 +1,37 @@
+module Kubes::Compiler::Shared::Helpers
+  class Deprecated
+    def built_image
+      puts "DEPRECATED: built_image is deprecated, use docker_image helper instead.".color(:yellow)
+      print_source
+    end
+
+    def error_info
+      error_info = caller.find { |l| l.include?('.kubes/resources') }
+      path, line_number, _ = error_info.split(':')
+      {path: path, line_number: line_number}
+    end
+
+    def print_source
+      info = error_info
+      path = info[:path]
+      line_number = info[:line_number].to_i
+
+      pretty_path = path.sub("#{Kubes.root}/",'')
+      puts "Here's the line in #{pretty_path} that calls built_image:\n\n"
+
+      contents = IO.read(path)
+      content_lines = contents.split("\n")
+      context = 5 # lines of context
+      top, bottom = [line_number-context-1, 0].max, line_number+context-1
+      lpad = content_lines.size.to_s.size
+      content_lines[top..bottom].each_with_index do |line_content, index|
+        current_line = top+index+1
+        if current_line == line_number
+          printf("%#{lpad}d %s\n".color(:red), current_line, line_content)
+        else
+          printf("%#{lpad}d %s\n", current_line, line_content)
+        end
+      end
+    end
+  end
+end

data/lib/kubes/compiler/strategy/base.rb

@@ -2,6 +2,7 @@ class Kubes::Compiler::Strategy
   class Base
     include Kubes::Logging
     include Kubes::Compiler::Util::SaveFile
+    include Kubes::Compiler::Shared::CustomHelpers
 
     def initialize(options={})
       @options = options

data/lib/kubes/compiler/strategy/dsl.rb

@@ -3,6 +3,7 @@ class Kubes::Compiler::Strategy
     include Kubes::Compiler::Util::Normalize
 
     def run
+      load_custom_helpers
       dsl = dsl_class.new(@options) # Deployment, Service, etc
       data = dsl.run
       Result.new(@save_file, data)

data/lib/kubes/compiler/strategy/erb.rb

@@ -4,10 +4,12 @@ class Kubes::Compiler::Strategy
   class Erb < Base
     extend Kubes::Compiler::Dsl::Core::Fields
     include Kubes::Compiler::Dsl::Core::Helpers
+    include Kubes::Compiler::Shared::CustomHelpers
     include Kubes::Compiler::Shared::Helpers
     include Kubes::Compiler::Layering
 
     def run
+      load_custom_helpers
       @data = {}
 
       render_files(pre_layers)

data/lib/kubes/config.rb

@@ -32,7 +32,7 @@ module Kubes
 
       config.repo = nil # expected to be set by .kubes/config.rb
 
-      config.logger = Logger.new($stdout)
+      config.logger = Logger.new($stderr)
       config.logger.level = ENV['KUBES_LOG_LEVEL'] || :info
 
       config.skip = []

data/lib/kubes/core.rb

@@ -28,5 +28,11 @@ module Kubes
     def kustomize?
       Kubectl::Kustomize.detect?
     end
+
+    def check_project!
+      return if File.exist?("#{Kubes.root}/.kubes/config.rb")
+      logger.error "ERROR: It doesnt look like this is a kubes project. Are you sure you are in a kubes project?".color(:red)
+      ENV['TS_TEST'] ? raise : exit(1)
+    end
   end
 end

data/lib/kubes/docker/strategy/image_name.rb

@@ -30,7 +30,7 @@ module Kubes::Docker::Strategy
       return "tongueroo/demo-kubes:kubes-12345678" if ENV['TEST']
 
       unless File.exist?(image_state_path)
-        puts "Unable to find #{image_state_path} which contains the last docker image name built with kubes build.  Please run `kubes docker build` first."
+        logger.error "ERROR: Unable to find #{image_state_path} which contains the last docker image name built with kubes build.  Please run `kubes docker build` first."
         exit 1
       end
       IO.read(image_state_path).strip

data/lib/kubes/kubectl.rb

@@ -17,13 +17,11 @@ module Kubes
       params = args.flatten.join(' ')
       args = "#{@name} #{params}" # @name: apply or delete
 
-      switch_context do
-        run_hooks("kubectl.rb", name: @name, file: @options[:file]) do
-          if options[:capture]
-            self.class.capture(args, options) # already includes kubectl
-          else
-            self.class.execute(args, options)
-          end
+      run_hooks("kubectl.rb", name: @name, file: @options[:file]) do
+        if options[:capture]
+          self.class.capture(args, options) # already includes kubectl
+        else
+          self.class.execute(args, options)
         end
       end
     end
@@ -44,22 +42,6 @@ module Kubes
       Kubes.config.kubectl.exit_on_fail[@name]
     end
 
-    def switch_context(&block)
-      kubectl = Kubes.config.kubectl
-      context = kubectl.context
-      if context
-        previous_context = capture("kubectl config current-context")
-        sh("kubectl config use-context #{context}", mute: true)
-        result = block.call
-        if !previous_context.blank? && !kubectl.context_keep
-          sh("kubectl config use-context #{previous_context}", mute: true)
-        end
-        result
-      else
-        block.call
-      end
-    end
-
     def args
       # base at end in case of redirection. IE: command > /path
       custom.args + default.args

data/lib/kubes/kubectl/batch.rb

@@ -3,6 +3,7 @@ class Kubes::Kubectl
     include Kubes::Hooks::Concern
     include Kubes::Logging
     include Kubes::Util::Consider
+    include Kubes::Util::Sh
     include Ordering
 
     def initialize(name, options={})
@@ -12,48 +13,41 @@ class Kubes::Kubectl
     def run
       # @options[:preview] is really only used for kubectl delete
       logger.info "Will run:" if @options[:preview]
-      run_hooks("kubes.rb", name: @name) do
-        sorted_files.each do |file|
-          if @options[:preview]
-            logger.info "    kubectl #{@name} -f #{file}"
-          else
-            Kubes::Kubectl.run(@name, @options.merge(file: file))
+      switch_context do
+        run_hooks("kubes.rb", name: @name) do
+          sorted_files.each do |file|
+            if @options[:preview]
+              logger.info "    kubectl #{@name} -f #{file}"
+            else
+              Kubes::Kubectl.run(@name, @options.merge(file: file))
+            end
           end
         end
       end
     end
 
-    # kubes apply                   # {role: nil, resource: nil}
-    # kubes apply clock             # {role: "clock", resource: nil}
-    # kubes apply clock deployment  # {role: "clock", resource: "deployment"}
-    def search_expr
-      role, resource = @options[:role], @options[:resource]
-      if role && resource
-        "#{Kubes.root}/.kubes/output/#{role}/#{resource}.yaml"
-      elsif role
-        "#{Kubes.root}/.kubes/output/#{role}/*.yaml"
-      else
-        "#{Kubes.root}/.kubes/output/**/*.yaml"
-      end
-    end
+    def switch_context(&block)
+      kubectl = Kubes.config.kubectl
+      context = kubectl.context
 
-    def files
-      files = []
-      Dir.glob(search_expr).each do |path|
-        next unless process?(path)
-        file = path.sub("#{Kubes.root}/", '')
-        files << file
+      unless context
+        block.call
+        return
       end
-      files
-    end
 
-    def process?(path)
-      consider?(path) && two_levels_deep?(path)
-    end
+      previous_context = sh_capture("kubectl config current-context")
+      if previous_context == context
+        block.call
+        return
+      end
 
-    def two_levels_deep?(path)
-      rel_path = path.sub(%r{.*\.kubes/output/},'')
-      rel_path.split('/').size == 2
+      logger.debug "Switching kubectl context to: #{context}"
+      sh("kubectl config use-context #{context}", mute: true)
+      result = block.call
+      if !previous_context.blank? && !kubectl.context_keep
+        sh("kubectl config use-context #{previous_context}", mute: true)
+      end
+      result
     end
   end
 end

data/lib/kubes/kubectl/ordering.rb

@@ -35,5 +35,47 @@ class Kubes::Kubectl
       i = index.to_s.rjust(3, "0") # pad with 0
       "#{i}-#{value}" # append name so that terms with same index get order alphabetically
     end
+
+    # kubes apply                   # {role: nil, resource: nil}
+    # kubes apply clock             # {role: "clock", resource: nil}
+    # kubes apply clock deployment  # {role: "clock", resource: "deployment"}
+    def search_expr
+      role, resource = @options[:role], @options[:resource]
+      if role && resource
+        "#{Kubes.root}/.kubes/output/#{role}/#{resource}.yaml"
+      elsif role
+        "#{Kubes.root}/.kubes/output/#{role}/*.yaml"
+      else
+        "#{Kubes.root}/.kubes/output/**/*.yaml"
+      end
+    end
+
+    def files
+      files = []
+      Dir.glob(search_expr).each do |path|
+        next unless process?(path)
+        file = path.sub("#{Kubes.root}/", '')
+        files << file
+      end
+      files
+    end
+
+    # Only considering files 2 layers deep. So:
+    #
+    #    Yes = web/deployment.yaml
+    #    No = web/deployment/dev.yaml
+    #
+    def process?(path)
+      if Kubes.kustomize?
+        File.file?(path)
+      else
+        consider?(path) && two_levels_deep?(path)
+      end
+    end
+
+    def two_levels_deep?(path)
+      rel_path = path.sub(%r{.*\.kubes/(resources|output)/},'')
+      rel_path.split('/').size == 2
+    end
   end
 end

data/lib/kubes/version.rb

@@ -1,3 +1,3 @@
 module Kubes
-  VERSION = "0.4.4"
+  VERSION = "0.5.1"
 end

data/lib/templates/base/.kubes/config.rb.tt

@@ -2,7 +2,7 @@ Kubes.configure do |config|
   config.repo = "<%= @options[:repo] %>"
   config.logger.level = "info"
   # auto-switching
-  # config.kubectl.context = "dev-services"
+  # config.kubectl.context = "dev-cluster"
   # config.kubectl.context_keep = true # keep context after switching
   # config.kubectl.exit_on_fail_for_apply  = true # whether or not continue if the kubectl command fails
   # config.kubectl.exit_on_fail_for_delete = false # whether or not continue if the kubectl command fails

data/lib/templates/base/.kubes/config/env/dev.rb

@@ -2,5 +2,5 @@
 #
 # Example:
 # Kubes.configure do |config|
-#   config.kubectl.context = "dev-services"
+#   config.kubectl.context = "dev-cluster"
 # end

data/lib/templates/base/.kubes/config/env/prod.rb

@@ -2,5 +2,5 @@
 #
 # Example:
 # Kubes.configure do |config|
-#   config.kubectl.context = "prod-services"
+#   config.kubectl.context = "prod-cluster"
 # end

data/lib/templates/dsl/.kubes/resources/web/deployment.rb

@@ -2,7 +2,7 @@ name "web"
 labels(role: "web")
 
 replicas 1 # overridden on a env basis
-image built_image # IE: user/<%= app %>:kubes-2020-06-13T19-55-16-43afc6e
+image docker_image # IE: user/<%= app %>:kubes-2020-06-13T19-55-16-43afc6e
 
 # revisionHistoryLimit 1 # uncomment to reduce old ReplicaSets, default is 10 https://bit.ly/3hqrzyP
 # maxUnavailable 25

data/lib/templates/new/dsl/backend_config.rb

@@ -0,0 +1,10 @@
+name "<%= app %>"
+spec(
+  timeoutSec: 40,
+  connectionDraining: {
+    drainingTimeoutSec: 60,
+  },
+  sessionAffinity: {
+    affinityType: "CLIENT_IP",
+  }
+)

data/lib/templates/new/dsl/config_map.rb

@@ -0,0 +1,5 @@
+name "<%= app %>"
+data(
+  KEY1: "value1",
+  KEY2: "value2",
+)

data/lib/templates/new/dsl/daemon_set.rb

@@ -0,0 +1,11 @@
+name "<%= app %>"
+labels("app": "<%= app %>")
+updateStrategy(
+  type: "RollingUpdate",
+  rollingUpdate: {
+    maxUnavailable: 1
+  }
+)
+# annotations(
+#   "*scheduler**.alpha.kubernetes.io/critical-pod": '*'
+# )

data/lib/templates/new/dsl/deployment.rb

@@ -0,0 +1,4 @@
+name "<%= role %>"
+labels(role: "<%= role %>")
+replicas 2
+image "<%%= docker_image %>"

data/lib/templates/new/dsl/ingress.rb

@@ -0,0 +1,3 @@
+name "<%= app %>"
+serviceName "<%= app %>"
+servicePort 80

data/lib/templates/new/dsl/job.rb

@@ -0,0 +1,2 @@
+name "<%= app %>"
+image(docker_image)

data/lib/templates/new/dsl/managed_certificate.rb

@@ -0,0 +1,2 @@
+name "cert1"
+domains(["cert1.example.com"])

data/lib/templates/new/dsl/namespace.rb

@@ -0,0 +1,2 @@
+name "<%= app %>"
+labels(app: "<%= app %>") # useful with NetworkPolicy

data/lib/templates/new/dsl/network_policy.rb

@@ -0,0 +1,7 @@
+name "web"
+labels(app: "<%= app %>") # IE: backend
+namespace "<%= app %>"    # IE: backend
+
+matchLabels(app: "<%= app %>", role: "<%= role %>") # IE: app: backend
+fromNamespace(app: "<%= app %>") # IE: app: frontend
+fromPod(app: "<%= app %>")       # IE: backend

data/lib/templates/new/dsl/pod.rb

@@ -0,0 +1,6 @@
+name "<%= app %>"
+containers([
+  image: docker_image,
+  command: ["sleep", "3600"],
+  name: "<%= app %>",
+])

data/lib/templates/new/dsl/role.rb

@@ -0,0 +1,4 @@
+name "<%= app %>"
+apiGroups([""])
+resources(["pods"])
+verbs(["get", "watch", "list"])

data/lib/templates/new/dsl/role_binding.rb

@@ -0,0 +1,7 @@
+name "<%= app %>"
+
+subjects([
+  {kind: "User", name: "me@email.com"},
+])
+
+roleName "<%= app %>"