k8s_kit 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: d6536da6860c0b0bcfa14bae28fbee59110c028b589bab71deb2437f1cd28d2d
+  data.tar.gz: 4317a20be3364cfb36851c81105d9f59df3aa0ad718350e33ed12a1396e3e27a
+SHA512:
+  metadata.gz: 8ecfb4eba9a6ee5832aea993271160d052fd404c9eef4bef58ebb455eb661a1c627fe021dfed1921d5dc39eb33a20091894b6a8db5142ed673a865567c2924bf
+  data.tar.gz: b9e70468631580c5000831ee0d382580ef9d59a403b3fc78b2eea616bd2096f552b1289b6a127967a9a0ba5c28e5ceb3a9aba95dc70b7f8b0406b01250bbadfc

data/Gemfile

@@ -0,0 +1,2 @@
+source 'https://rubygems.org'
+gemspec

data/Gemfile.lock

@@ -0,0 +1,62 @@
+PATH
+  remote: .
+  specs:
+    k8s_kit (0.0.1)
+      gli (= 2.18.1)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    aruba (0.14.11)
+      childprocess (>= 0.6.3, < 3.0.0)
+      contracts (~> 0.9)
+      cucumber (>= 1.3.19)
+      ffi (~> 1.9)
+      rspec-expectations (>= 2.99)
+      thor (~> 0.19)
+    backports (3.15.0)
+    builder (3.2.3)
+    childprocess (2.0.0)
+      rake (< 13.0)
+    contracts (0.16.0)
+    cucumber (3.1.2)
+      builder (>= 2.1.2)
+      cucumber-core (~> 3.2.0)
+      cucumber-expressions (~> 6.0.1)
+      cucumber-wire (~> 0.0.1)
+      diff-lcs (~> 1.3)
+      gherkin (~> 5.1.0)
+      multi_json (>= 1.7.5, < 2.0)
+      multi_test (>= 0.1.2)
+    cucumber-core (3.2.1)
+      backports (>= 3.8.0)
+      cucumber-tag_expressions (~> 1.1.0)
+      gherkin (~> 5.0)
+    cucumber-expressions (6.0.1)
+    cucumber-tag_expressions (1.1.1)
+    cucumber-wire (0.0.1)
+    diff-lcs (1.3)
+    ffi (1.11.1)
+    gherkin (5.1.0)
+    gli (2.18.1)
+    multi_json (1.13.1)
+    multi_test (0.1.2)
+    rake (12.3.3)
+    rdoc (6.1.1)
+    rspec-expectations (3.8.4)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.8.0)
+    rspec-support (3.8.2)
+    thor (0.20.3)
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  aruba
+  k8s_kit!
+  rake
+  rdoc
+
+BUNDLED WITH
+   2.0.2

data/bin/k8s_kit

@@ -0,0 +1,73 @@
+#!/usr/bin/env ruby
+require 'gli'
+begin # XXX: Remove this begin/rescue before distributing your app
+require 'k8s_kit'
+rescue LoadError
+  STDERR.puts "In development, you need to use `bundle exec bin/k8s_kit` to run your app"
+  STDERR.puts "At install-time, RubyGems will make sure lib, etc. are in the load path"
+  STDERR.puts "Feel free to remove this message from bin/k8s_kit now"
+  exit 64
+end
+
+class App
+  extend GLI::App
+
+  program_desc 'Describe your application here'
+
+  version K8sKit::VERSION
+
+  subcommand_option_handling :normal
+  arguments :strict
+
+  desc 'Choose the namespace (current namespace is used if not set)'
+  flag [:n, :namespace]
+
+  desc 'Describe attach-job here'
+  arg_name '<job name>'
+  command 'attach-job' do |c|
+    c.desc 'Container name (needed if more than one container is running in the pod)'
+    c.flag [:c, :container]
+
+    c.action do |global_options, options, args|
+      K8sKit::Context.new(namespace: global_options[:namespace]).tap do |ctx|
+        ctx.job(args[0]).attach(container: options[:container])
+      end
+    end
+  end
+
+  desc 'Waits until all pods in the namespace are "Ready"'
+  command 'wait-all-pods-ready' do |c|
+    c.desc 'Timeout in seconds'
+    c.default_value '300'
+    c.flag [:t, :timeout]
+
+    c.action do |global_options, options, args|
+      K8sKit::Context.new(namespace: global_options[:namespace]).tap do |ctx|
+        ctx.wait_for_all_pods_ready(timeout: options[:timeout].to_i > 0 ? options[:timeout].to_i : 300)
+      end
+    end
+  end
+
+  pre do |global,command,options,args|
+    # Pre logic here
+    # Return true to proceed; false to abort and not call the
+    # chosen command
+    # Use skips_pre before a command to skip this block
+    # on that command only
+    true
+  end
+
+  post do |global,command,options,args|
+    # Post logic here
+    # Use skips_post before a command to skip this
+    # block on that command only
+  end
+
+  on_error do |exception|
+    # Error logic here
+    # return false to skip default error handling
+    true
+  end
+end
+
+exit App.run(ARGV)

data/k8s_kit.gemspec

@@ -0,0 +1,21 @@
+# Ensure we require the local version and not one we might have installed already
+require File.join([File.dirname(__FILE__),'lib','k8s_kit','version.rb'])
+spec = Gem::Specification.new do |s|
+  s.name = 'k8s_kit'
+  s.version = K8sKit::VERSION
+  s.author = 'Your Name Here'
+  s.email = 'your@email.address.com'
+  s.homepage = 'http://your.website.com'
+  s.platform = Gem::Platform::RUBY
+  s.summary = 'A description of your project'
+  s.files = `git ls-files`.split("
+")
+  s.require_paths << 'lib'
+  s.rdoc_options << '--title' << 'k8s_kit' << '--main' << 'README.rdoc' << '-ri'
+  s.bindir = 'bin'
+  s.executables << 'k8s_kit'
+  s.add_development_dependency('rake')
+  s.add_development_dependency('rdoc')
+  s.add_development_dependency('aruba')
+  s.add_runtime_dependency('gli','2.18.1')
+end

data/lib/k8s_kit/context.rb

@@ -0,0 +1,57 @@
+require 'open3'
+
+module K8sKit
+  class Context
+    attr_reader :namespace
+
+    def initialize(namespace: nil)
+      @namespace = namespace
+    end
+
+    def kubectl
+      ns = namespace ? "-n #{namespace}" : ''
+      "kubectl #{ns}"
+    end
+
+    def run(cmd, silent: true)
+      cmd = "#{kubectl} #{cmd}"
+
+      out = []
+      Open3.popen2e(cmd) do |stdin, stdout_err, wait_thr|
+        while line = stdout_err.gets
+          puts line unless silent
+          out << line
+        end
+
+        exit_status = wait_thr.value
+        unless exit_status.success?
+          abort "Failed: #{cmd}"
+        end
+      end
+
+      out.join("\n").chomp
+    end
+
+    def pod(name)
+      Pod.new(self, name)
+    end
+
+    def job(name)
+      Job.new(self, name)
+    end
+
+    def wait_for_all_pods_ready(timeout: 300)
+      t0 = Time.now
+
+      loop do
+        statuses = run('get pods -o jsonpath="{.items.*.status.containerStatuses[*].ready}"')
+        return unless statuses.include?('false')
+
+        raise StandardError, 'Timeout while waiting for pods to become ready' if t0 + timeout < Time.now
+
+        puts 'Some pods are not ready yet, retrying...'
+        sleep 5
+      end
+    end
+  end
+end

data/lib/k8s_kit/job.rb

@@ -0,0 +1,32 @@
+module K8sKit
+  class Job
+    attr_reader :context, :name
+
+    def initialize(context, name)
+      @context = context
+      @name = name
+    end
+
+    def attach(container:, delete_on_completion: true, exit_on_completion: true)
+      pod.wait_until_ready
+      pod.logs(container: container, stream: true)
+      exit_code = pod.exit_code(container: container)
+
+      delete if delete_on_completion
+      exit exit_code if exit_on_completion
+
+      exit_code
+    end
+
+    def delete
+      context.run("delete job #{name}")
+    end
+
+    def pod
+      @pod ||= begin
+        pod_name = context.run("get pod -l job-name=#{name} -o jsonpath='{.items[0].metadata.name}'")
+        Pod.new(context, pod_name)
+      end
+    end
+  end
+end

data/lib/k8s_kit/pod.rb

@@ -0,0 +1,29 @@
+module K8sKit
+  class Pod
+    attr_reader :context, :name
+
+    def initialize(context, name)
+      @context = context
+      @name = name
+    end
+
+    def logs(container: nil, stream: false)
+      command = "logs #{name}"
+      command += " -c #{container}" if container
+      command += " -f" if stream
+
+      context.run(command, silent: false)
+    end
+
+    def wait_until_ready(timeout: '5m')
+      context.run("wait pod/#{name} --for=condition=Ready --timeout=#{timeout}")
+    end
+
+    def exit_code(container:)
+      json_path = "{ .status.containerStatuses[?(@.name == '#{container}')].state.terminated.exitCode }"
+      context.run("get pod #{name} -o jsonpath=\"#{json_path}\"").to_i
+    rescue StandardError
+      nil
+    end
+  end
+end

data/lib/k8s_kit/version.rb

@@ -0,0 +1,3 @@
+module K8sKit
+  VERSION = '0.0.1'
+end

data/lib/k8s_kit.rb

@@ -0,0 +1,4 @@
+require 'k8s_kit/context'
+require 'k8s_kit/job'
+require 'k8s_kit/pod'
+require 'k8s_kit/version.rb'

metadata

@@ -0,0 +1,113 @@
+--- !ruby/object:Gem::Specification
+name: k8s_kit
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Your Name Here
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-08-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rdoc
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: aruba
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: gli
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 2.18.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 2.18.1
+description: 
+email: your@email.address.com
+executables:
+- k8s_kit
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- Gemfile.lock
+- bin/k8s_kit
+- k8s_kit.gemspec
+- lib/k8s_kit.rb
+- lib/k8s_kit/context.rb
+- lib/k8s_kit/job.rb
+- lib/k8s_kit/pod.rb
+- lib/k8s_kit/version.rb
+homepage: http://your.website.com
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options:
+- "--title"
+- k8s_kit
+- "--main"
+- README.rdoc
+- "-ri"
+require_paths:
+- lib
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.0.3
+signing_key: 
+specification_version: 4
+summary: A description of your project
+test_files: []