kubec 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e29547dc53808d55661f20349483c90c709d8ae4
-  data.tar.gz: c2de23b6dbc3444b4ec3d5017ad1e31f227416e2
+  metadata.gz: 62c55e240f35b19478f090bcb66fd0b9a32d9180
+  data.tar.gz: c484d7863b58f11ef1c35067d3c51ef72732898c
 SHA512:
-  metadata.gz: 27ac3b15f987988bc7ec555a98824c3e7377aa386eae3aab00b36b166fad84f8c6ec7ba8acaea572ca295e52acc4dc733dce214e53045fe961c58c94e37f1c6b
-  data.tar.gz: 2ce1d12f2f50f085e1121ade09320841308545add1eaa5921e843740d553090e6316fe5754276543e421302edced92a21b96c1fec7b35f2d7ce3d76bd5556385
+  metadata.gz: 7e95a5579e35ed2dac74a919c2206a95f32832577db1fed071c0b7ef7697a2498b8ae63f85d4d662bd6412abae62a7bf76ede7416e0411be76a396879c295ca6
+  data.tar.gz: b16b95cbe52efc8408f534599b3fdfe62cf32b6ede61d442aa708cd4363bf2f4248c6aae90eb27aadc90338ce6ee5ea0e4e6c59352cfc19405fd9826bb0ca7f8

data/lib/kubec/config/base.rb

@@ -0,0 +1,6 @@
+module Kubec
+  # :nodoc:
+  module Config
+    autoload :Downloader, 'kubec/config/downloader'
+  end
+end

data/lib/kubec/config/downloader.rb

@@ -0,0 +1,51 @@
+module Kubec
+  module Config
+    # :nodoc:
+    class Downloader
+      def initialize
+        @result = `kubectl -n #{fetch(:stage, :staging)} get configmap -o json`
+        @success = $CHILD_STATUS.success?
+        @items = {}
+
+        prepare
+      end
+
+      def save
+        Kubernetes.config.each do |config|
+          downloaded = @items[config.name]
+          config.files.each do |(key, path)|
+            write path, downloaded.dig('data', key)
+          end
+        end
+      end
+
+      private
+
+      def prepare
+        return unless @success
+        items = JSON.parse(@result).dig('items') || []
+        items.each do |item|
+          name = item.dig('metadata', 'name').to_sym
+          @items[name] = item
+        end
+      end
+
+      def write(path, body)
+        puts "=> #{path} saved"
+        File.write(
+          path_with_stage(path),
+          body
+        )
+      end
+
+      # TODO: Refactor
+      def path_with_stage(path)
+        path.split('.').tap do |ary|
+          ext = ary.pop
+          ary.push(fetch(:stage, :staging))
+          ary.push(ext)
+        end.join('.')
+      end
+    end
+  end
+end

data/lib/kubec/config.rb

@@ -0,0 +1 @@
+load File.expand_path('../tasks/config.rake', __FILE__)

data/lib/kubec/deploy.rb

@@ -1 +1,2 @@
 load File.expand_path('../tasks/deploy.rake', __FILE__)
+load File.expand_path('../tasks/restart.rake', __FILE__)

data/lib/kubec/dsl/kubernetes.rb

@@ -6,7 +6,7 @@ module Kubec
     module Kubernetes
       extend Forwardable
 
-      def_delegators :kubernetes, :service, :deployment
+      def_delegators :kubernetes, :service, :deployment, :config, :cronjob
 
       def kubernetes
         Kubec::Kubernetes.instance

data/lib/kubec/kubernetes/config_map.rb

@@ -0,0 +1,49 @@
+module Kubec
+  class Kubernetes
+    # :nodoc:
+    class ConfigMap < Hash
+      attr_reader :name, :files
+
+      def initialize(name, &block)
+        @name = name.to_sym
+        @files = {}
+
+        prepare
+        instance_eval(&block)
+      end
+
+      def set(key, value)
+        self[:data][key] = value
+      end
+
+      # TODO: Refactor
+      def file(path)
+        key = path.split('/').last
+        @files[key] = path
+        path = path_with_stage(path) if stage_config_exist?(path)
+        set key, File.read(path)
+      end
+
+      private
+
+      def prepare
+        self[:apiVersion] = 'v1'
+        self[:metadata] = Metadata.new(@name)
+        self[:kind] = 'ConfigMap'
+        self[:data] = {}
+      end
+
+      def stage_config_exist?(path)
+        File.exist?(path_with_stage(path))
+      end
+
+      def path_with_stage(path)
+        path.split('.').tap do |ary|
+          ext = ary.pop
+          ary.push(fetch(:stage, :staging))
+          ary.push(ext)
+        end.join('.')
+      end
+    end
+  end
+end

data/lib/kubec/kubernetes/container.rb

@@ -3,7 +3,6 @@ module Kubec
     # :nodoc:
     class Container < Hash
       include HasAttribute
-
       attribute :image
       attribute :name
       attribute :ports
@@ -13,6 +12,26 @@ module Kubec
         instance_eval(&block)
       end
 
+      def env(key, value)
+        self[:env] ||= []
+        self[:env].push name: key,
+                        value: value.to_s
+      end
+
+      def mount(name, at:)
+        self[:volumeMounts] ||= []
+        self[:volumeMounts].push name: name,
+                                 mountPath: at
+      end
+
+      # TODO: Auto setup config map volume
+      def config_file(name, path:, from:)
+        self[:volumeMounts] ||= []
+        self[:volumeMounts].push name: from,
+                                 mountPath: [path, name].join('/'),
+                                 subPath: name
+      end
+
       # TODO: Add object to check fields
       def port(container_port, host_port = nil,
                ip: nil, name: nil, protocol: nil)
@@ -26,6 +45,16 @@ module Kubec
         }.compact
         self[:ports].push port
       end
+
+      def command(*args)
+        args = args.flatten
+        self[:command] = args.take(1)
+        self[:args] = args.drop(1)
+      end
+
+      def args(*args)
+        self[:args] = args.flatten
+      end
     end
   end
 end

data/lib/kubec/kubernetes/cron_job.rb

@@ -0,0 +1,22 @@
+module Kubec
+  class Kubernetes
+    # :nodoc:
+    class CronJob < Config
+      api_version 'batch/v1beta1'
+
+      def schedule(at)
+        spec[:schedule] = at
+      end
+
+      def template(&block)
+        inst = Template.new
+        spec[:jobTemplate] = {
+          spec: {
+            template: inst
+          }
+        }
+        inst.instance_eval(&block)
+      end
+    end
+  end
+end

data/lib/kubec/kubernetes/deployment.rb

@@ -4,7 +4,8 @@ module Kubec
     class Deployment < Config
       api_version 'extensions/v1beta1'
 
-      def replicas(size)
+      def replicas(size = nil)
+        return spec[:replicas] || 1 if size.nil?
         spec[:replicas] = size.to_i
       end
 
@@ -12,6 +13,19 @@ module Kubec
         spec[:template] ||= Template.new
         spec[:template].instance_eval(&block)
       end
+
+      # TODO: Move into module
+      def select(key, value)
+        spec[:selector] ||= { matchLabels: {} }
+        spec[:selector][:matchLabels][key] = value
+      end
+
+      def selector=(labels)
+        # TODO: Check labels type
+        spec[:selector] = labels
+      end
+
+      alias selector selector=
     end
   end
 end

data/lib/kubec/kubernetes/template.rb

@@ -11,6 +11,11 @@ module Kubec
         self[:spec] = {}
       end
 
+      def volume(name, &block)
+        spec[:volumes] ||= []
+        spec[:volumes].push Volume.new(name, &block)
+      end
+
       def metadata(&block)
         return self[:metadata] unless block_given?
         self[:metadata].instance_eval(&block)
@@ -24,6 +29,10 @@ module Kubec
         spec[:containers] ||= []
         spec[:containers].push Container.new(name, &block)
       end
+
+      def restart(policy)
+        spec[:restartPolicy] = policy.to_s.capitalize
+      end
     end
   end
 end

data/lib/kubec/kubernetes/volume.rb

@@ -0,0 +1,42 @@
+module Kubec
+  class Kubernetes
+    # :nodoc:
+    class Volume < Hash
+      def initialize(name, &block)
+        self[:name] = name
+        instance_eval(&block)
+      end
+
+      def empty(memory = false)
+        self[:emptyDir] = if memory
+                            { medium: true }
+                          else
+                            {}
+                          end
+      end
+
+      def host_path(path, type = nil)
+        self[:hostPath] = { path: path }
+        self[:hostPath][:type] = type unless type.nil?
+      end
+
+      def gce(name, type: 'ext4', ro: false)
+        self[:gcePersistentDisk] = {
+          pdName: name,
+          fsType: type,
+          readOnly: ro
+        }
+      end
+
+      # TODO: Check for should create a new class to handle
+      def config(name, items)
+        self[:configMap] = {
+          name: name,
+          items: items.map do |key, path|
+            { key: key, path: path }
+          end
+        }
+      end
+    end
+  end
+end

data/lib/kubec/kubernetes.rb

@@ -1,21 +1,22 @@
-# frozen_string_literal: true
-
 module Kubec
   # Kubernetes
   class Kubernetes
-    autoload :Config,       'kubec/kubernetes/config'
-    autoload :Template,     'kubec/kubernetes/template'
-    autoload :Service,      'kubec/kubernetes/service'
-    autoload :Deployment,   'kubec/kubernetes/deployment'
-    autoload :Metadata,     'kubec/kubernetes/metadata'
-    autoload :Spec,         'kubec/kubernetes/spec'
-    autoload :Container,    'kubec/kubernetes/container'
+    autoload :Config,     'kubec/kubernetes/config'
+    autoload :ConfigMap,  'kubec/kubernetes/config_map'
+    autoload :CronJob,    'kubec/kubernetes/cron_job'
+    autoload :Template,   'kubec/kubernetes/template'
+    autoload :Service,    'kubec/kubernetes/service'
+    autoload :Deployment, 'kubec/kubernetes/deployment'
+    autoload :Metadata,   'kubec/kubernetes/metadata'
+    autoload :Spec,       'kubec/kubernetes/spec'
+    autoload :Container,  'kubec/kubernetes/container'
+    autoload :Volume,     'kubec/kubernetes/volume'
 
     autoload :HasAttribute, 'kubec/kubernetes/has_attribute'
 
     include Singleton
 
-    APPLYABLE_TYPES = %i[service deployment].freeze
+    APPLYABLE_TYPES = %i[service deployment config cronjob].freeze
 
     class << self
       def apply(type)
@@ -56,6 +57,8 @@ module Kubec
     def initialize
       @services = []
       @deployments = []
+      @configs = []
+      @cronjobs = []
     end
 
     def service(name = nil, &block)
@@ -67,5 +70,15 @@ module Kubec
       return @deployments if name.nil?
       @deployments << Kubernetes::Deployment.new(name, &block)
     end
+
+    def config(name = nil, &block)
+      return @configs if name.nil?
+      @configs << Kubernetes::ConfigMap.new(name, &block)
+    end
+
+    def cronjob(name = nil, &block)
+      return @cronjobs if name.nil?
+      @cronjobs << Kubernetes::CronJob.new(name, &block)
+    end
   end
 end

data/lib/kubec/tasks/config.rake

@@ -0,0 +1,13 @@
+namespace :config do
+  desc 'Upload config files using ConfigMap'
+  task push: ['deploy:namespace'] do
+    Kubec::Utils::Helper.header 'Starting apply config maps'
+    Kubec::Kubernetes.apply(:config)
+  end
+
+  desc 'Download config files from ConfigMap'
+  task pull: ['deploy:namespace'] do
+    Kubec::Utils::Helper.header 'Starting save config maps'
+    Kubec::Config::Downloader.new.save
+  end
+end

data/lib/kubec/tasks/deploy.rake

@@ -15,11 +15,18 @@ namespace :deploy do
     Kubec::Utils::Helper.header 'Starting apply services'
     Kubec::Kubernetes.apply(:service)
   end
+
+  desc 'Apply Service to Kubernetes'
+  task :cronjobs do
+    Kubec::Utils::Helper.header 'Starting apply cron jobs'
+    Kubec::Kubernetes.apply(:cronjob)
+  end
 end
 
 desc 'Deploy to Kubernetes'
 task deploy: [
   'deploy:namespace',
   'deploy:deployments',
-  'deploy:services'
+  'deploy:services',
+  'deploy:cronjobs'
 ]

data/lib/kubec/tasks/restart.rake

@@ -0,0 +1,17 @@
+desc 'Restart all services'
+task :restart do
+  # TODO: Refactor this section
+  stage = fetch(:stage, :staging)
+  def scale(stage, name, size)
+    `kubectl -n #{stage} scale deployment/#{name} --replicas=#{size}`
+  end
+
+  Kubec::Utils::Helper.header 'Restarting deployments'
+  Kubec::Kubernetes.deployment.each do |deploy|
+    Thread.new do
+      puts "=> Restart #{deploy.name}"
+      scale(stage, deploy.name, 0)
+      scale(stage, deploy.name, deploy.replicas)
+    end.join
+  end
+end

data/lib/kubec/templates/Kubeconfig

@@ -3,4 +3,5 @@ require 'kubec/setup'
 
 # Add deploy scripts
 require 'kubec/deploy'
+require 'kubec/config'
 require 'kubec/status'

data/lib/kubec/utils/humanize_time.rb

@@ -10,7 +10,7 @@ module Kubec
       ].freeze
 
       SHORT = {
-        seocnds: 's',
+        seconds: 's',
         minutes: 'm',
         hours:   'h',
         days:    'd'

data/lib/kubec/version.rb

@@ -1,3 +1,3 @@
 module Kubec
-  VERSION = '0.2.0'.freeze
+  VERSION = '0.3.0'.freeze
 end

data/lib/kubec.rb

@@ -17,4 +17,5 @@ module Kubec
   autoload :DSL,         'kubec/dsl'
   autoload :Status,      'kubec/status/base'
   autoload :Utils,       'kubec/utils'
+  autoload :Config,      'kubec/config/base'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: kubec
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - 5xRuby
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-01-12 00:00:00.000000000 Z
+date: 2018-01-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: colorize
@@ -147,6 +147,9 @@ files:
 - lib/Kubeconfig
 - lib/kubec.rb
 - lib/kubec/application.rb
+- lib/kubec/config.rb
+- lib/kubec/config/base.rb
+- lib/kubec/config/downloader.rb
 - lib/kubec/deploy.rb
 - lib/kubec/dsl.rb
 - lib/kubec/dsl/env.rb
@@ -157,20 +160,25 @@ files:
 - lib/kubec/install.rb
 - lib/kubec/kubernetes.rb
 - lib/kubec/kubernetes/config.rb
+- lib/kubec/kubernetes/config_map.rb
 - lib/kubec/kubernetes/container.rb
+- lib/kubec/kubernetes/cron_job.rb
 - lib/kubec/kubernetes/deployment.rb
 - lib/kubec/kubernetes/has_attribute.rb
 - lib/kubec/kubernetes/metadata.rb
 - lib/kubec/kubernetes/service.rb
 - lib/kubec/kubernetes/template.rb
+- lib/kubec/kubernetes/volume.rb
 - lib/kubec/setup.rb
 - lib/kubec/status.rb
 - lib/kubec/status/base.rb
 - lib/kubec/status/deployment.rb
 - lib/kubec/status/pod.rb
 - lib/kubec/status/service.rb
+- lib/kubec/tasks/config.rake
 - lib/kubec/tasks/deploy.rake
 - lib/kubec/tasks/install.rake
+- lib/kubec/tasks/restart.rake
 - lib/kubec/tasks/status.rake
 - lib/kubec/templates/Kubeconfig
 - lib/kubec/templates/config/kubec.rb