docker_rails_proxy 0.0.10 → 0.0.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 482f5b31d3f9c713c16a356a03b922cc8af3c101
-  data.tar.gz: d0a88b59450098afa2c61bef41c5b70941d51976
+  metadata.gz: a3a12d0b5a75064d9d2536c11ccd90898c8d88f6
+  data.tar.gz: 279c5c537357f88c68cad0dd49ba48d7b2b224d8
 SHA512:
-  metadata.gz: bf4f656786e8e4b7cfbac98bc66a7f2afa1deb8cef534df99f5c79aa1c7c338f4f9b63a3485ad123076d8fceab0726ba298b74e41401ce411b212c8de4023c83
-  data.tar.gz: 402d732079a1723be4ccbdcd199dc6f23c71a234b38b97e963b0aa6d98f9b1e1e9a46f2e0bd93ba1c820888874a7ba8c51c5a7ef6964114ab7da7877fb496f52
+  metadata.gz: 8d9e1bbfb5714a48d34b7b42846677db5b5278876b066a4289fa465b1d2a3baa465e6743700078f43a39641fea4943aca27bb6c8ccc6d9fc0b55a004929c04be
+  data.tar.gz: d59f6de86cc746002b3f59a9bfcf3d4d4ddaf0aebae75f8e8bbc4018f81c0c6de70b2da1a864853f5de1702c33092a4cdcaf25e4e4f8e0009eec161e769e386c

data/lib/docker_rails_proxy.rb

@@ -98,6 +98,17 @@ module DockerRailsProxy
       end
     end
 
+    def until_get_option(values, message, options = {})
+      value = nil
+
+      while value.nil? do
+        print_options(values, message)
+        value = get_option(values, options)
+      end
+
+      value
+    end
+
     def flush_stdin
       loop do
         Timeout::timeout(0.1) { $stdin.gets.chomp } rescue break

data/lib/docker_rails_proxy/commands/kubectl.rb

@@ -0,0 +1,45 @@
+require 'optparse'
+
+module DockerRailsProxy
+  class Kubectl < Base
+    autoload :Bash,          'docker_rails_proxy/commands/kubectl/bash'
+    autoload :SetKubeconfig, 'docker_rails_proxy/commands/kubectl/set_kubeconfig'
+
+    KUBECONFIG_PATH = build_path('tmp/.kube/kubeconfig')
+
+    before_initialize do
+      'kubectl is required, `brew install kubectl`' unless system 'type kubectl &> /dev/null'
+    end
+
+    validates do
+      unless File.exist? KUBECONFIG_PATH
+        <<-EOS
+
+        kubeconfig is required, run this command to set it
+        `bin/#{APP_NAME} kubectl set-kubeconfig --path /path/kubeconfig`
+        EOS
+      end unless arguments.first == 'set-kubeconfig'
+    end
+
+    builds -> (params:) do
+      case params[:arguments].first
+      when 'bash'           then Bash
+      when 'set-kubeconfig' then SetKubeconfig
+      end
+    end
+
+    def process
+      kubectl arguments.join(' ')
+    end
+
+  private
+
+    def kubectl(command)
+      exec "kubectl --kubeconfig='#{KUBECONFIG_PATH}' #{command}"
+    end
+
+    def kubectl_output(command)
+      %x(kubectl --kubeconfig='#{KUBECONFIG_PATH}' #{command}).strip
+    end
+  end
+end

data/lib/docker_rails_proxy/commands/kubectl/bash.rb

@@ -0,0 +1,77 @@
+require 'optparse'
+require 'json'
+
+module DockerRailsProxy
+  class Kubectl < Base
+    class Bash < self
+      attr_accessor :data
+
+      before_process do
+        self.data = JSON.parse kubectl_output("get #{pod} -o json")
+      end
+
+      def process
+        %w[livenessProbe readinessProbe command].each do |attr|
+          container.delete attr
+        end
+
+
+        overrides = {
+          metadata: {
+            annotations: {
+              'iam.amazonaws.com/role' => data.dig('metadata', 'annotations', 'iam.amazonaws.com/role')
+            }
+          },
+          spec: {
+            volumes: data.dig('spec', 'volumes'),
+            imagePullSecrets: data.dig('spec', 'imagePullSecrets'),
+            securityContext: data.dig('spec', 'securityContext'),
+            serviceAccount: data.dig('spec', 'serviceAccount'),
+            serviceAccountName: data.dig('spec', 'serviceAccountName'),
+            containers: [
+              container.merge!({
+                args: %w[bash],
+                stdin: true,
+                stdinOnce: true,
+                tty: true
+              })
+            ]
+          }
+        }
+
+        pod_name = "#{container['name']}-bash-#{Time.now.to_i}"
+        puts "Starting #{pod_name} pod ..."
+
+        kubectl <<-EOS
+          run #{pod_name} --rm -i --tty \
+            --image='#{container['image']}' \
+            --generator='run-pod/v1' \
+            --overrides='#{overrides.to_json}'
+        EOS
+      end
+
+    private
+
+      def pod
+        @pod_name ||= begin
+          pods = kubectl_output('get pods -o name').split(' ')
+          until_get_option pods, "Choose a pod and press [ENTER]"
+        end
+      end
+
+      def container
+        @container ||= begin
+          containers = data.dig('spec', 'containers')
+
+          if containers.size > 1
+            names = containers.map{|c| c['name'] }
+            name  = until_get_option names, "Choose a container and press [ENTER]"
+            containers.detect{|c| c['name'] == name }
+          else
+            containers.first
+          end
+        end
+      end
+    end
+  end
+end

data/lib/docker_rails_proxy/commands/kubectl/set_kubeconfig.rb

@@ -0,0 +1,70 @@
+require 'optparse'
+require 'yaml'
+
+module DockerRailsProxy
+  class Kubectl < Base
+    class SetKubeconfig < self
+      attr_accessor :options, :data
+
+      after_initialize { self.options = {} }
+      after_initialize { opt_parser.parse!(arguments) }
+
+      validates do
+        if options[:path].nil?
+          '--path is required'
+        elsif !File.exist?(options[:path])
+          "#{options[:path]} kubeconfig does not exist."
+        end
+      end
+
+      validates do
+        self.data = YAML.load_file(options[:path])
+
+        unless (data['kind'] == 'Config')
+          "#{options[:path]} is not a valid kubeconfig file"
+        end
+      end
+
+      before_process do
+        Dir.mkdir('tmp/.kube') unless Dir.exist?('tmp/.kube')
+
+        absolute_dir = File.dirname(options[:path])
+
+        data['clusters'].each do |hash|
+          cluster = hash['cluster']
+          cluster['certificate-authority'] = File.absolute_path(cluster['certificate-authority'], absolute_dir)
+        end
+
+        data['users'].each do |hash|
+          user = hash['user']
+          user['client-certificate'] = File.absolute_path(user['client-certificate'], absolute_dir)
+          user['client-key']         = File.absolute_path(user['client-key'], absolute_dir)
+        end
+      end
+
+      def process
+        File.write(KUBECONFIG_PATH, data.to_yaml)
+        puts 'kubeconfig was set correctly'
+      end
+
+    private
+
+      def opt_parser
+        @opt_parser ||= OptionParser.new do |opts|
+          opts.banner = "Usage: bin/#{APP_NAME} kubectl #{self.class.name.demodulize.parameterize} [options]"
+
+          opts.on('--path PATH', 'kubeconfig file path') do |path|
+            options[:path] = path
+          end
+
+          yield opts if block_given?
+
+          opts.on('-h', '--help', 'Display this screen') do
+            puts opts
+            exit
+          end
+        end
+      end
+    end
+  end
+end

data/lib/docker_rails_proxy/version.rb

@@ -1,3 +1,3 @@
 module DockerRailsProxy
-  VERSION = '0.0.10'
+  VERSION = '0.0.12'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: docker_rails_proxy
 version: !ruby/object:Gem::Version
-  version: 0.0.10
+  version: 0.0.12
 platform: ruby
 authors:
 - Jairo
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-11-20 00:00:00.000000000 Z
+date: 2017-12-02 00:00:00.000000000 Z
 dependencies: []
 description: Configures docker-compose and provides rails command helpers
 email:
@@ -33,6 +33,9 @@ files:
 - lib/docker_rails_proxy/commands/data_bags.rb
 - lib/docker_rails_proxy/commands/data_bags/pull.rb
 - lib/docker_rails_proxy/commands/data_bags/push.rb
+- lib/docker_rails_proxy/commands/kubectl.rb
+- lib/docker_rails_proxy/commands/kubectl/bash.rb
+- lib/docker_rails_proxy/commands/kubectl/set_kubeconfig.rb
 - lib/docker_rails_proxy/commands/rails.rb
 - lib/docker_rails_proxy/commands/rake.rb
 - lib/docker_rails_proxy/commands/rspec.rb