stax-helm 0.0.2 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8db16bbf17c66b24cf72a3e807ffeda939a1fda49ced1f7d9b708eae254b66ed
-  data.tar.gz: cd11b9f4eb2d4732a067e1aa51da1613e64903f3668e57e2b9886a084a599e95
+  metadata.gz: ce6add2c9a7276df7ed801b49c7f3a7151ba4d98c34b3690a4fdaca6678fdeb5
+  data.tar.gz: c04af79c0c1fe2a14d7493e175980aeef5864f074f2f405bc67bd7fef35696c8
 SHA512:
-  metadata.gz: c3a6d9400080112015b5dd688939d84f80df7cae7d9f094316f8b51c2cc6c8aa5a1ab127ef0876a148e0761c1e875ebae78ff33c51c12700925e03d72513def2
-  data.tar.gz: be3eb884e62f76be25450face25c860095380ddca05f72b98cb2ba971e538fb874c0ebd7e0e476bab36af2a9dec8d238f239e709844f44dc1e14acbe88bd58cf
+  metadata.gz: fb825149e84265f3f50ea1d17554f4514d2db401d771251063f9f1ce81fb90cfe221053b4a7f9c168797777493679a11f56d850af34fbb3c548cff32237174c3
+  data.tar.gz: 29ff4a39a0ac91b12ea218cff57fd6afa81658ff9fc4ec26d8a90019bab133e8ef3fe5cf7adf3ae895a89a0c3b79f946a8b819e801093ffdf13e8d89418f668f

data/lib/stax/helm.rb

@@ -1,4 +1,8 @@
+require 'stax/helm/base'
 require 'stax/helm/cmd'
 require 'stax/helm/kubectl'
+require 'stax/helm/ingress'
+require 'stax/helm/pod'
+require 'stax/helm/deployment'
 require 'stax/helm/stern'
 Stax.add_command(:helm, Stax::Helm::Cmd)

data/lib/stax/helm/base.rb

@@ -0,0 +1,16 @@
+module Stax
+  class Base < Thor
+
+    no_commands do
+      def helm_release_name
+        @_helm_release_name ||= helm_safe("#{app_name}-#{branch_name}")
+      end
+
+      ## make string safe to use in naming helm stuff
+      def helm_safe(string)
+        string.slice(0, 53).gsub(/[\W_]/, '-').downcase
+      end
+    end
+
+  end
+end

data/lib/stax/helm/cmd.rb

@@ -1,13 +1,10 @@
 module Stax
   module Helm
+
     class Cmd < Base
-      class_option :dry_run, type: :boolean, default: false, desc: 'print command that would be run'
+      class_option :recon, aliases: '--just-print', type: :boolean, default: false, desc: 'print command that would be run'
 
       no_commands do
-        def helm_release_name
-          @_helm_release_name ||= "#{app_name}-#{branch_name}"
-        end
-
         ## location of helm chart
         def helm_dir
           File.join(Stax.root_path, 'helm')
@@ -21,7 +18,12 @@ module Stax
         ## run helm with args
         def helm_run(*args)
           cmd = [helm_bin, *args].join(' ')
-          options[:dry_run] ? puts(cmd) : system(cmd)
+          options[:recon] ? puts(cmd) : system(cmd)
+        end
+
+        ## description added to release
+        def helm_description
+          Git.sha
         end
 
         ## override with full path to a values.yaml file
@@ -37,6 +39,7 @@ module Stax
         ## construct args for install and upgrade commands
         def helm_update_args
           [].tap do |args|
+            args.push("--description #{helm_description}") if helm_description
             args.push("-f #{helm_values_file}") if helm_values_file
             args.push(helm_values&.map { |k,v| "--set #{k}=#{v}" })
           end.flatten

data/lib/stax/helm/deployment.rb

@@ -0,0 +1,56 @@
+## tasks to work on deployments
+module Stax
+  module Helm
+    class Cmd < Base
+
+      no_commands do
+        def helm_deployments
+          jsonpath = '{.items[*].metadata.name}'
+          %x[kubectl get deployments -o=jsonpath='#{jsonpath}' -l #{helm_selector}].split
+        end
+
+        ## prompt user with a list of deployments to choose
+        def helm_ask_deployments(msg)
+          deployments = helm_deployments
+          if deployments.count > 1
+            puts deployments.each_with_index.map { |d, i| "#{i}: #{d}" }
+            resp = ask(msg, default: 'all')
+            if resp != 'all'
+              indices = resp.split.map(&:to_i)
+              deployments = Array(deployments.slice(*indices))
+            end
+          end
+          deployments
+        end
+      end
+
+      desc 'deployments', 'list deployments'
+      def deployments
+        kubectl_run(:get, :deployments, '-l', helm_selector)
+      end
+
+      desc 'restart', 'restart deployments'
+      def restart
+        helm_ask_deployments('choose deployments').each do |deployment|
+          kubectl_run(:rollout, :restart, :deployment, deployment)
+        end
+      end
+
+      desc 'scale', 'show/set scale for deployments'
+      method_option :replicas, aliases: '-r', type: :numeric, default: nil, desc: 'replicas'
+      def scale
+        if options[:replicas]
+          deployments = helm_ask_deployments('choose deployments').join(' ')
+          kubectl_run(:scale, :deployment, deployments, '--replicas', options[:replicas])
+        else
+          debug("Deployment replicas for #{helm_release_name}")
+          deployments = kubectl_json(:get, :deployments, '-l', helm_selector)
+          print_table deployments['items'].map { |i|
+            [ i['metadata']['name'], i['status']['replicas'] || 0 ]
+          }
+        end
+      end
+
+    end
+  end
+end

data/lib/stax/helm/ingress.rb

@@ -0,0 +1,19 @@
+## tasks to get ingress details
+module Stax
+  module Helm
+    class Cmd < Base
+
+      desc 'ingresses', 'list ingresses'
+      def ingresses
+        kubectl_run(:get, :ingresses, '-l', helm_selector)
+      end
+
+      desc 'dns', 'list external-dns hostnames'
+      def dns
+        jsonpath = '{.items[].metadata.annotations.external-dns\.alpha\.kubernetes\.io/hostname}' + "\n"
+        kubectl_run(:get, :ingresses, "-o=jsonpath='#{jsonpath}'", '-l', helm_selector)
+      end
+
+    end
+  end
+end

data/lib/stax/helm/kubectl.rb

@@ -9,54 +9,24 @@ module Stax
 
         def kubectl_run(*args)
           cmd = [kubectl_bin, *args].join(' ')
-          options[:dry_run] ? puts(cmd) : system(cmd)
+          options[:recon] ? puts(cmd) : system(cmd)
         end
 
-        ## build a selector argument from a hash of label and value pairs
-        def selector(hash)
-          '-l ' + hash.compact.map { |k,v| "#{k}=#{v}" }.join(',')
+        def kubectl_json(*args)
+          args.push('-o=json')
+          cmd = [kubectl_bin, *args].join(' ')
+          options[:recon] ? puts(cmd) : JSON.parse(%x(#{cmd}))
+        end
+
+        ## override this to match all objects in your helm release
+        def helm_selector
+          "app.kubernetes.io/instance=#{helm_release_name}"
         end
       end
 
       desc 'services', 'list services'
       def services
-        kubectl_run(:get, :services, selector('app.kubernetes.io/instance': helm_release_name))
-      end
-
-      desc 'ingresses', 'list ingresses'
-      def ingresses
-        kubectl_run(:get, :ingresses, selector('app.kubernetes.io/instance': helm_release_name))
-      end
-
-      desc 'deployments', 'list deployments'
-      def deployments
-        kubectl_run(:get, :deployments, selector('app.kubernetes.io/instance': helm_release_name))
-      end
-
-      desc 'pods', 'list pods'
-      def pods
-        kubectl_run(:get, :pods, selector('app.kubernetes.io/instance': helm_release_name))
-      end
-
-      desc 'containers', 'list containers'
-      def containers
-        columns = 'NAME:.metadata.name,CONTAINERS:.spec.containers[*].name'
-        kubectl_run(:get, :pods, '-o', "custom-columns=#{columns}", selector('app.kubernetes.io/instance': helm_release_name))
-      end
-
-      ## FIXME this is terrible, will be replaced with something better later
-      desc 'logs COMPONENT [CONTAINER]', 'show container logs'
-      method_option :container, aliases: '-c', type: :string, default: nil, desc: 'container from pod'
-      def logs(component)
-        container = options[:container] ? "-c #{options[:container]}" : ''
-        kubectl_run(
-          :logs,
-          container,
-          selector(
-            'app.kubernetes.io/instance': helm_release_name,
-            'app.kubernetes.io/component': component,
-          )
-        )
+        kubectl_run(:get, :services, '-l', helm_selector)
       end
 
     end

data/lib/stax/helm/pod.rb

@@ -0,0 +1,49 @@
+## tasks to work on pods
+module Stax
+  module Helm
+    class Cmd < Base
+
+      no_commands do
+        def helm_pods
+          jsonpath = '{.items[*].metadata.name}'
+          %x[kubectl get pods -o=jsonpath='#{jsonpath}' -l #{helm_selector}].split
+        end
+
+        def helm_ask_pod(msg)
+          pods = helm_pods
+          index = 0
+          if pods.count > 1
+            puts pods.each_with_index.map { |p, i| "#{i}: #{p}" }
+            index = ask(msg, default: index)
+          end
+          pods[index.to_i]
+        end
+      end
+
+      desc 'pods', 'list pods'
+      def pods
+        kubectl_run(:get, :pods, '-l', helm_selector)
+      end
+
+      desc 'containers', 'list containers'
+      def containers
+        columns = 'NAME:.metadata.name,CONTAINERS:.spec.containers[*].name'
+        kubectl_run(:get, :pods, '-o', "custom-columns=#{columns}", '-l', helm_selector)
+      end
+
+      desc 'logs [OPTIONS]', 'run kubectl logs with same options'
+      def logs(*args)
+        trap('SIGINT', 'EXIT')    # clean exit with ctrl-c
+        args = [ '--all-containers', '--prefix', '--follow' ] if args.empty? # helpful default args
+        kubectl_run(:logs, '-l', helm_selector, *args)
+      end
+
+      desc 'exec [CMD]', 'exec command in a web pod'
+      def exec(cmd = 'sh')
+        pod = helm_ask_pod('choose a pod')
+        kubectl_run(:exec, '-it', pod, '--', cmd)
+      end
+
+    end
+  end
+end

data/lib/stax/helm/stern.rb

@@ -16,7 +16,8 @@ module Stax
       ## pass through args to stern
       desc 'stern [STERN_ARGS]', 'use stern to show logs'
       def stern(*args)
-        stern_run(selector('app.kubernetes.io/instance': helm_release_name), *args)
+        trap('SIGINT', 'EXIT')    # clean exit with ctrl-c
+        stern_run('-l', helm_selector, *args)
       end
 
     end

data/lib/stax/helm/version.rb

@@ -1,5 +1,5 @@
 module Stax
   module Helm
-    VERSION = '0.0.2'
+    VERSION = '0.0.7'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: stax-helm
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.7
 platform: ruby
 authors:
 - Richard Lister
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-05-25 00:00:00.000000000 Z
+date: 2020-09-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -67,8 +67,12 @@ files:
 - README.md
 - Rakefile
 - lib/stax/helm.rb
+- lib/stax/helm/base.rb
 - lib/stax/helm/cmd.rb
+- lib/stax/helm/deployment.rb
+- lib/stax/helm/ingress.rb
 - lib/stax/helm/kubectl.rb
+- lib/stax/helm/pod.rb
 - lib/stax/helm/stern.rb
 - lib/stax/helm/version.rb
 - stax-helm.gemspec
@@ -76,7 +80,7 @@ homepage: https://github.com/rlister/stax-helm
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -92,7 +96,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubygems_version: 3.0.3
-signing_key: 
+signing_key:
 specification_version: 4
 summary: Control helm charts with stax.
 test_files: []