kubes 0.3.5 → 0.4.0

data/lib/kubes/compiler/decorator/pre.rb

@@ -0,0 +1,12 @@
+module Kubes::Compiler::Decorator
+  class Pre < Base
+    def process
+      case @data['kind']
+      when "ConfigMap", "Secret"
+        Hashable.new(@data).store
+      else
+        @data # passthrough
+      end
+    end
+  end
+end

data/lib/kubes/compiler/strategy.rb

@@ -11,7 +11,7 @@ class Kubes::Compiler
 
       strategy = klass.new(@options.merge(path: @path)) # Dsl or Erb
       result = strategy.run
-      result.compile_decorate! # compile phase decoration
+      result.decorate!(:pre) # compile pre phase decoration
       result
     end
 
@@ -19,7 +19,7 @@ class Kubes::Compiler
       ext = File.extname(@path)
       case ext
       when '.rb'   then Dsl
-      when '.yaml' then Erb
+      when '.yaml','.yml' then Erb
       else Pass
       end
     end

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

@@ -7,7 +7,7 @@ class Kubes::Compiler::Strategy
       @path = options[:path]
 
       @filename = @path.sub(%r{.*\.kubes/resources/},'') # IE: web/deployment.rb or web/deployment.yaml
-      @save_file = @filename.sub('.rb','.yaml')
+      @save_file = @filename.sub('.yml','.yaml').sub('.rb','.yaml')
     end
   end
 end

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

@@ -11,12 +11,10 @@ class Kubes::Compiler::Strategy
       @data.respond_to?(:read)
     end
 
-    def compile_decorate!
-      @data = Kubes::Compiler::Decorator::Compile.new(@data).result
-    end
-
-    def write_decorate!
-      @data = Kubes::Compiler::Decorator::Write.new(@data).result
+    # decorate(:pre) or decorate(:post)
+    def decorate!(phase)
+      klass = "Kubes::Compiler::Decorator::#{phase.to_s.camelize}".constantize
+      @data = klass.new(@data).result
     end
 
     def content

data/lib/kubes/config.rb

@@ -11,31 +11,36 @@ module Kubes
     def defaults
       config = ActiveSupport::OrderedOptions.new
 
-      config.state = ActiveSupport::OrderedOptions.new
-      config.state.docker_image_path = "#{Kubes.root}/.kubes/state/docker_image.txt"
+      config.auto_prune = true
 
-      config.logger = Logger.new($stdout)
-      config.logger.level = ENV['KUBES_LOG_LEVEL'] || :info
+      config.builder = "docker" # IE: docker or gcloud
 
       # Auto-switching options
       config.kubectl = ActiveSupport::OrderedOptions.new
       config.kubectl.context = nil
       config.kubectl.context_keep = true # after switching context keep it
-      config.kubectl.exit_on_fail = nil # whether or not continue if the kubectl command fails
 
-      config.kubectl.exit_on_fail_for_apply  = true # whether or not continue if the kubectl apply command fails
-      config.kubectl.exit_on_fail_for_delete = false # whether or not continue if the kubectl delete command fails
-      # Note: not using config.kubectl.delete.exit_on_fail because delete a method internal to ActiveSupport::OrderedOptions
+      # whether or not continue if the kubectl command fails
+      config.kubectl.exit_on_fail = ActiveSupport::OrderedOptions.new
+      config.kubectl.exit_on_fail.apply  = true # whether or not continue if the kubectl apply command fails
+      config.kubectl.exit_on_fail.delete = false # whether or not continue if the kubectl delete command fails
+      # Note: delete is a internal method to ActiveSupport::OrderedOptions so will have to access it with ['...']
 
       config.kubectl.order = ActiveSupport::OrderedOptions.new
       config.kubectl.order.roles = role_order
       config.kubectl.order.kinds = kind_order
 
-      config.suffix_hash = true # append suffix has to ConfigMap and Secret
-
       config.repo = nil # expected to be set by .kubes/config.rb
 
-      config.builder = "docker" # IE: docker or gcloud
+      config.logger = Logger.new($stdout)
+      config.logger.level = ENV['KUBES_LOG_LEVEL'] || :info
+
+      config.skip = []
+
+      config.state = ActiveSupport::OrderedOptions.new
+      config.state.docker_image_path = "#{Kubes.root}/.kubes/state/docker_image.txt"
+
+      config.suffix_hash = true # append suffix hash to ConfigMap and Secret
 
       config
     end

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

@@ -3,7 +3,7 @@ module Kubes::Docker::Strategy::Build
     def perform
       params = args.flatten.join(' ')
       command = "docker build #{params}"
-      run_hooks "build" do
+      run_hooks("docker.rb", name: "build") do
         sh(command)
       end
     end

data/lib/kubes/docker/strategy/build/gcloud.rb

@@ -2,7 +2,7 @@ module Kubes::Docker::Strategy::Build
   class Gcloud < Base
     def perform
       command = "gcloud builds submit --tag #{@@image_name}"
-      run_hooks "build" do
+      run_hooks("docker.rb", name: "build") do
         sh(command)
       end
     end

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

@@ -61,7 +61,7 @@ module Kubes::Docker::Strategy
     end
 
     def custom
-      custom = Kubes::Args::Custom.new(@name, "#{Kubes.root}/.kubes/config/docker/args.rb")
+      custom = Kubes::Args::Custom.new(@name, "#{Kubes.root}/.kubes/config/args/docker.rb")
       custom.build
       custom
     end

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

@@ -17,7 +17,7 @@ module Kubes::Docker::Strategy::Push
     def push
       params = args.flatten.join(' ')
       command = "docker push #{params}"
-      run_hooks "push" do
+      run_hooks("docker.rb", name: "push") do
         sh(command)
       end
     end

data/lib/kubes/docker/strategy/push/gcloud.rb

@@ -1,7 +1,7 @@
 module Kubes::Docker::Strategy::Push
   class Gcloud < Base
     def run
-      run_hooks "push" do
+      run_hooks("docker.rb", name: "push") do
         # noop, gcloud builds submit already pushes the image
       end
     end

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

@@ -4,6 +4,6 @@ module Kubes::Docker::Strategy
     include Kubes::Util::Sh
     include Kubes::Util::Time
     include Kubes::Docker::Strategy::ImageName
-    include Kubes::Docker::Strategy::Hooks
+    include Kubes::Hooks::Concern
   end
 end

data/lib/kubes/hooks/builder.rb

@@ -4,39 +4,53 @@ module Kubes::Hooks
     include Dsl
     include DslEvaluator
     include Kubes::Logging
-    include Kubes::Util::Sh
 
     attr_accessor :name
-    def initialize(name, file)
-      @name = name.to_s
-      @file = file # IE: .kubes/config/kubectl/hooks.rb
+    def initialize(dsl_file, options={})
+      @dsl_file, @options = dsl_file, options # IE: .kubes/config/hooks/kubectl.rb
+      @output_file = options[:file] # IE: .kubes/output/web/service.yaml
+      @name = options[:name].to_s
       @hooks = {before: {}, after: {}}
     end
 
     def build
-      return @hooks unless File.exist?(@file)
-      evaluate_file(@file)
+      return @hooks unless File.exist?(@dsl_file)
+      evaluate_file(@dsl_file)
       @hooks.deep_stringify_keys!
     end
     memoize :build
 
     def run_hooks
       build
-      run_hook("before")
+      run_each_hook("before")
       out = yield if block_given?
-      run_hook("after")
+      run_each_hook("after")
       out
     end
 
-    def run_hook(type)
-      execute = @hooks.dig(type, @name.to_s, "execute")
-      return unless execute
+    def run_each_hook(type)
+      hooks = @hooks.dig(type, @name.to_s) || []
+      hooks.each do |hook|
+        run_hook(type, hook)
+      end
+    end
+
+    def run_hook(type, hook)
+      return unless run?(hook)
 
-      exit_on_fail = @hooks.dig(type, @name.to_s, "exit_on_fail")
-      exit_on_fail = exit_on_fail.nil? ? true : exit_on_fail
+      command = File.basename(@dsl_file).sub('.rb','') # IE: kubes, kubectl, docker
+      id = "#{command} #{type} #{@name}"
+      on = " on: #{hook["on"]}" if hook["on"]
+      label = " label: #{hook["label"]}" if hook["label"]
+      logger.info  "Running #{id} hook.#{on}#{label}"
+      logger.debug "Hook options: #{hook}"
+      Runner.new(hook).run
+    end
 
-      logger.info "Running #{type} hook"
-      sh(execute, exit_on_fail: exit_on_fail)
+    def run?(hook)
+      return false unless hook["execute"]
+      return true  unless hook["on"]
+      @output_file.include?(hook["on"])
     end
   end
 end

data/lib/kubes/hooks/concern.rb

@@ -0,0 +1,10 @@
+module Kubes::Hooks
+  module Concern
+    # options example: {:name=>"apply", :file=>".kubes/output/web/service.yaml"}
+    def run_hooks(file, options={}, &block)
+      hooks = Kubes::Hooks::Builder.new("#{Kubes.root}/.kubes/config/hooks/#{file}", options)
+      hooks.build # build hooks
+      hooks.run_hooks(&block)
+    end
+  end
+end

data/lib/kubes/hooks/dsl.rb

@@ -13,7 +13,8 @@ module Kubes::Hooks
     end
 
     def each_hook(type, name, props={})
-      @hooks[type][name] = props
+      @hooks[type][name] ||= []
+      @hooks[type][name] << props
     end
   end
 end

data/lib/kubes/hooks/runner.rb

@@ -0,0 +1,22 @@
+module Kubes::Hooks
+  class Runner
+    include Kubes::Util::Sh
+    include Kubes::Logging
+
+    def initialize(hook)
+      @hook = hook
+      @execute = @hook["execute"]
+    end
+
+    def run
+      case @execute
+      when String
+        sh(@execute, exit_on_fail: @hook["exit_on_fail"])
+      when -> (e) { e.respond_to?(:public_instance_methods) && e.public_instance_methods.include?(:call) }
+        @execute.new.call
+      else
+        @execute.call
+      end
+    end
+  end
+end

data/lib/kubes/kubectl.rb

@@ -2,6 +2,7 @@ module Kubes
   class Kubectl
     extend Memoist
     include Kubes::Util::Sh
+    include Kubes::Hooks::Concern
 
     def initialize(name, options={})
       @name, @options = name, options
@@ -14,24 +15,19 @@ module Kubes
       options[:exit_on_fail] = exit_on_fail unless exit_on_fail.nil?
 
       params = args.flatten.join(' ')
-      command = "kubectl #{@name} #{params}" # @name: apply or delete
+      args = "#{@name} #{params}" # @name: apply or delete
 
       switch_context do
-        run_hooks(@name) do
+        run_hooks("kubectl.rb", name: @name, file: @options[:file]) do
           if options[:capture]
-            capture(command, options)
+            self.class.capture(args, options) # already includes kubectl
           else
-            sh(command, options)
+            self.class.execute(args, options)
           end
         end
       end
     end
 
-    def execute(args, options={})
-      command = "kubectl #{args}"
-      capture(command)
-    end
-
     # Useful for kustomize mode
     def validate!
       return true unless Kubes.kustomize?
@@ -44,9 +40,8 @@ module Kubes
     end
 
     def exit_on_fail
-      kubectl = Kubes.config.kubectl
-      exit_on_fail = kubectl.send("exit_on_fail_for_#{@name}")
-      exit_on_fail.nil? ? kubectl.exit_on_fail : exit_on_fail
+      return false if ENV['KUBES_EXIT_ON_FAIL'] == '0'
+      Kubes.config.kubectl.exit_on_fail[@name]
     end
 
     def switch_context(&block)
@@ -65,12 +60,6 @@ module Kubes
       end
     end
 
-    def run_hooks(name, &block)
-      hooks = Kubes::Hooks::Builder.new(name, "#{Kubes.root}/.kubes/config/kubectl/hooks.rb")
-      hooks.build # build hooks
-      hooks.run_hooks(&block)
-    end
-
     def args
       # base at end in case of redirection. IE: command > /path
       custom.args + default.args
@@ -90,9 +79,23 @@ module Kubes
     memoize :default
 
     class << self
+      include Kubes::Util::Sh
       def run(name, options={})
         new(name, options).run
       end
+
+      def execute(args, options={})
+        sh("kubectl #{args}", options)
+      end
+
+      def capture(args, options={})
+        resp = sh_capture("kubectl #{args}", options)
+        if args.include?('-o json')
+          JSON.load(resp) # data
+        else
+          resp
+        end
+      end
     end
   end
 end

data/lib/kubes/kubectl/batch.rb

@@ -1,5 +1,6 @@
 class Kubes::Kubectl
   class Batch
+    include Kubes::Hooks::Concern
     include Kubes::Logging
     include Kubes::Util::Consider
     include Ordering
@@ -11,11 +12,13 @@ class Kubes::Kubectl
     def run
       # @options[:preview] is really only used for kubectl delete
       logger.info "Will run:" if @options[:preview]
-      sorted_files.each do |file|
-        if @options[:preview]
-          logger.info "    kubectl #{@name} -f #{file}"
-        else
-          Kubes::Kubectl.run(@name, @options.merge(file: file))
+      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

data/lib/kubes/kubectl/{decider.rb → dispatcher.rb}

@@ -1,5 +1,5 @@
 class Kubes::Kubectl
-  class Decider
+  class Dispatcher
     def initialize(name, options={})
       @name, @options = name.to_s, options
     end

data/lib/kubes/kubectl/fetch/base.rb

@@ -2,6 +2,7 @@ require "json"
 
 module Kubes::Kubectl::Fetch
   class Base
+    extend Memoist
     include Kubes::Logging
     include Kubes::Util::Sh
 
@@ -9,16 +10,18 @@ module Kubes::Kubectl::Fetch
       @options = options
     end
 
-    def fetch_items
-      o = {
-        capture: true,
-        output: "json",
-        show_command: false,
-      }
-      kubectl = Kubes::Kubectl.new(:get, @options.merge(o)) # kubes get -f .kubes/output
-      resp = kubectl.run
-      data = JSON.load(resp)
+    def fetch(kind)
+      return [] unless namespace
+      data = Kubes::Kubectl.capture("get #{kind} -o json -n #{namespace}")
       data['items'] || [] # Note: When fetching only 1 resource, items is not part of structure
     end
+
+    def namespace
+      path = ".kubes/output/shared/namespace.yaml"
+      return unless File.exist?(path)
+      data = Kubes::Kubectl.capture("get -f #{path} -o json")
+      data['metadata']['name']
+    end
+    memoize :namespace
   end
 end

data/lib/kubes/kubectl/fetch/deployment.rb

@@ -3,40 +3,39 @@ module Kubes::Kubectl::Fetch
     extend Memoist
 
     def metadata
-      deployment['metadata']
+      deployment['metadata'] if found
     end
 
     def spec
-      deployment['spec']
+      deployment['spec'] if found
     end
 
     def deployment
-      items = fetch_items
+      items = fetch(:deployment)
       # Not checking if deployment exists because kubes will error on `kubes get` from missing deployments already
       deployments = items.select { |i| i['kind'] == "Deployment" }
 
-      if deployments.size > 1 && !@options[:name]
+      if !@options[:deployment] && !@options[:pod] && deployments.size > 1
         names = deployments.map { |d| d['metadata']['name'] }
         logger.info <<~EOL
           INFO: More than one deployment found.
           Deployment names: #{names.join(', ')}
           Using #{names.first}
-          Note: You can specify the deployment to use with --name or -n
+          Note: You can specify the deployment to use with --deployment or -d
         EOL
       end
 
-      deployment = find_deployment(deployments)
-      unless deployment
-        logger.error "ERROR: No deployment found".color(:red)
-        exit 1
-      end
-      deployment
+      find_deployment(deployments)
     end
     memoize :deployment
 
+    def found
+      !!deployment
+    end
+
     def find_deployment(deployments)
-      if @options[:name]
-        deployments.find { |d| d['metadata']['name'] == @options[:name] }
+      if @options[:deployment]
+        deployments.find { |d| d['metadata']['name'] == @options[:deployment] }
       else
         deployments.first
       end