kubernetes-cli 0.2.1 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2c8ff82b1830c530f1f26441a1ec99b26a8ae8c1fe0616aff6496e363305ac0b
-  data.tar.gz: 28340df51d8024ac817b65307bcec08ebaa4a839aac55e3e43f50fd9ba7eba8a
+  metadata.gz: 44f3e261b827758d975c3e51c00f9319ac280c25592fd50d5980418de7fc4a46
+  data.tar.gz: ce69cc4e1c0a898e4baad326c14fd15baeebc18f39416157986a992a7e40596f
 SHA512:
-  metadata.gz: c4b46653caac5ff7620678ac1d17dc2fa6c16f7adf68e7c412aee48eb579a072d394fbf1c2d79de6ee56df7c21f6cc9c63eeee49df900c0366928d29c2e91ff8
-  data.tar.gz: '0946d1421aa008222e8c7dc154182188525d36e4745bd591ae7edf64a5e9e8c7e54cdc23f60410c2f03fee693d6f905ef072c0b014e490b9cb35ceee2e6465e8'
+  metadata.gz: 73a7bff727ab4a7f5b42c3b1f1b4d6fd95d2176667978fe454b308c8fc78219eacfeb451ec2bffc50b42f6e8d7d895656ab282c888b48bb9aa0c3ded65fa945d
+  data.tar.gz: 931ebe19d76a237f4e357ec0cac817221a52ba96e53630bfd3f76c669096b50694a52dc27f70bcbc7404ece859249592b433169d795f21736970afa02c2dd3ea

data/CHANGELOG.md

@@ -1,3 +1,6 @@
+## 0.3.0
+* Add ability to redirect kubectl's stdout and stderr streams.
+
 ## 0.2.1
 * Add missing `require` for open3.
 

data/lib/kubernetes-cli.rb

@@ -1,5 +1,6 @@
 require 'kubectl-rb'
 require 'open3'
+require 'stringio'
 
 class KubernetesCLI
   class KubernetesError < StandardError; end
@@ -15,6 +16,8 @@ class KubernetesCLI
   class GetResourceError < KubernetesError; end
 
   STATUS_KEY = :kubernetes_cli_last_status
+  STDOUT_KEY = :kubernetes_cli_stdout
+  STDERR_KEY = :kubernetes_cli_stderr
 
   attr_reader :kubeconfig_path, :executable
 
@@ -49,12 +52,19 @@ class KubernetesCLI
     execc(cmd)
   end
 
+  def system_cmd(container_cmd, namespace, pod, tty = true)
+    cmd = [executable, '--kubeconfig', kubeconfig_path, '-n', namespace, 'exec']
+    cmd += ['-it'] if tty
+    cmd += [pod, '--', *Array(container_cmd)]
+    systemm(cmd)
+  end
+
   def apply(res, dry_run: false)
     cmd = [executable, '--kubeconfig', kubeconfig_path, 'apply', '--validate']
     cmd << '--dry-run=client' if dry_run
     cmd += ['-f', '-']
 
-    open3_w(env, cmd) do |stdin, _wait_thread|
+    open3_w(env, cmd) do |stdin|
       stdin.puts(res.to_resource.to_yaml)
     end
 
@@ -160,6 +170,61 @@ class KubernetesCLI
     backticks(cmd).strip
   end
 
+  def api_resources
+    cmd = [executable, '--kubeconfig', kubeconfig_path, 'api-resources']
+    result = backticks(cmd)
+
+    unless last_status.success?
+      raise KubernetesError, 'could not fetch API resources: kubectl exited with '\
+        "status code #{last_status.exitstatus}. #{result}"
+    end
+
+    result
+  end
+
+  def restart_deployment(namespace, deployment)
+    cmd = [
+      executable,
+      '--kubeconfig', kubeconfig_path,
+      '-n', namespace,
+      'rollout', 'restart', 'deployment', deployment
+    ]
+
+    systemm(cmd)
+
+    unless last_status.success?
+      raise KubernetesError, 'could not restart deployment: kubectl exited with '\
+        "status code #{last_status.exitstatus}"
+    end
+  end
+
+  def with_pipes(out = STDOUT, err = STDERR)
+    previous_stdout = self.stdout
+    previous_stderr = self.stderr
+    self.stdout = out
+    self.stderr = err
+    yield
+  ensure
+    self.stdout = previous_stdout
+    self.stderr = previous_stderr
+  end
+
+  def stdout
+    Thread.current[STDOUT_KEY] || STDOUT
+  end
+
+  def stdout=(new_stdout)
+    Thread.current[STDOUT_KEY] = new_stdout
+  end
+
+  def stderr
+    Thread.current[STDERR_KEY] || STDERR
+  end
+
+  def stderr=(new_stderr)
+    Thread.current[STDERR_KEY] = new_stderr
+  end
+
   private
 
   def env
@@ -179,30 +244,82 @@ class KubernetesCLI
   def systemm(cmd)
     run_before_callbacks(cmd)
     cmd_s = cmd.join(' ')
-    system(cmd_s).tap do
-      self.last_status = $?
+
+    Open3.popen3(cmd_s) do |p_stdin, p_stdout, p_stderr, wait_thread|
+      Thread.new(stdout) do |t_stdout|
+        begin
+          p_stdout.each { |line| t_stdout.puts(line) }
+        rescue IOError
+        end
+      end
+
+      Thread.new(stderr) do |t_stderr|
+        begin
+          p_stderr.each { |line| t_stderr.puts(line) }
+        rescue IOError
+        end
+      end
+
+      p_stdin.close
+      self.last_status = wait_thread.value
       run_after_callbacks(cmd)
+      wait_thread.join
     end
   end
 
   def backticks(cmd)
     run_before_callbacks(cmd)
     cmd_s = cmd.join(' ')
-    `#{cmd_s}`.tap do
-      self.last_status = $?
+    result = StringIO.new
+
+    Open3.popen3(cmd_s) do |p_stdin, p_stdout, p_stderr, wait_thread|
+      Thread.new do
+        begin
+          p_stdout.each { |line| result.puts(line) }
+        rescue IOError
+        end
+      end
+
+      Thread.new(stderr) do |t_stderr|
+        begin
+          p_stderr.each { |line| t_stderr.puts(line) }
+        rescue IOError
+        end
+      end
+
+      p_stdin.close
+      self.last_status = wait_thread.value
       run_after_callbacks(cmd)
+      wait_thread.join
     end
+
+    result.string
   end
 
   def open3_w(env, cmd, opts = {}, &block)
     run_before_callbacks(cmd)
     cmd_s = cmd.join(' ')
 
-    Open3.pipeline_w([env, cmd_s], opts) do |stdin, wait_threads|
-      yield(stdin, wait_threads).tap do
-        stdin.close
-        self.last_status = wait_threads.last.value
+    Open3.popen3(env, cmd_s, opts) do |p_stdin, p_stdout, p_stderr, wait_thread|
+      Thread.new(stdout) do |t_stdout|
+        begin
+          p_stdout.each { |line| t_stdout.puts(line) }
+        rescue IOError
+        end
+      end
+
+      Thread.new(stderr) do |t_stderr|
+        begin
+          p_stderr.each { |line| t_stderr.puts(line) }
+        rescue IOError
+        end
+      end
+
+      yield(p_stdin).tap do
+        p_stdin.close
+        self.last_status = wait_thread.value
         run_after_callbacks(cmd)
+        wait_thread.join
       end
     end
   end

data/lib/kubernetes-cli/version.rb

@@ -1,3 +1,3 @@
 class KubernetesCLI
-  VERSION = '0.2.1'
+  VERSION = '0.3.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: kubernetes-cli
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.0
 platform: ruby
 authors:
 - Cameron Dutro
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-08-02 00:00:00.000000000 Z
+date: 2020-09-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: kubectl-rb