kubes 0.1.0 → 0.2.0

data/exe/kubes

@@ -0,0 +1,14 @@
+#!/usr/bin/env ruby
+
+# Trap ^C
+Signal.trap("INT") {
+  puts "\nCtrl-C detected. Exiting..."
+  sleep 0.1
+  exit
+}
+
+$:.unshift(File.expand_path("../../lib", __FILE__))
+require "kubes"
+require "kubes/cli"
+
+Kubes::CLI.start(ARGV)

data/kubes.gemspec

@@ -7,28 +7,38 @@ Gem::Specification.new do |spec|
   spec.name          = "kubes"
   spec.version       = Kubes::VERSION
   spec.authors       = ["Tung Nguyen"]
-  spec.email         = ["tongueroo@gmail.com"]
+  spec.email         = ["tung@boltops.com"]
+  spec.summary       = "Kubernetes Deployment Tool: build docker image, compile Kubernetes YAML files, and apply them"
+  spec.homepage      = "https://github.com/boltops-tools/kubes"
+  spec.license       = "Apache2.0"
 
-  spec.summary       = %q{Kubes tool}
-  spec.description   = %q{Kubes tool}
-  spec.homepage      = "https://github.com/tongueroo/kubes"
-
-  # Prevent pushing this gem to RubyGems.org. To allow pushes either set the 'allowed_push_host'
-  # to allow pushing to a single host or delete this section to allow pushing to any host.
-  # if spec.respond_to?(:metadata)
-  #   spec.metadata["allowed_push_host"] = "TODO: Set to 'http://mygemserver.com'"
-  # else
-  #   raise "RubyGems 2.0 or newer is required to protect against " \
-  #     "public gem pushes."
-  # end
-
-  spec.files         = `git ls-files -z`.split("\x0").reject do |f|
-    f.match(%r{^(test|spec|features)/})
-  end
+  spec.files         = File.directory?('.git') ? `git ls-files`.split($/) : Dir.glob("**/*")
   spec.bindir        = "exe"
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
   spec.require_paths = ["lib"]
 
-  spec.add_development_dependency "bundler", "~> 1.15"
-  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_dependency "activesupport"
+  spec.add_dependency "deep_merge"
+  spec.add_dependency "dsl_evaluator"
+  spec.add_dependency "hash_squeezer"
+  spec.add_dependency "memoist"
+  spec.add_dependency "rainbow"
+  spec.add_dependency "render_me_pretty"
+  spec.add_dependency "thor"
+  spec.add_dependency "zeitwerk"
+
+  spec.add_development_dependency "bundler"
+  spec.add_development_dependency "byebug"
+  spec.add_development_dependency "cli_markdown"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec"
+
+  # Prevent pushing this gem to RubyGems.org. To allow pushes either set the 'allowed_push_host'
+  # to allow pushing to a single host or delete this section to allow pushing to any host.
+  if spec.respond_to?(:metadata)
+    spec.metadata["homepage_uri"] = spec.homepage
+  else
+    raise "RubyGems 2.0 or newer is required to protect against public gem pushes."
+  end
 end

data/lib/kubes.rb

@@ -1,5 +1,25 @@
+$stdout.sync = true unless ENV["KUBES_STDOUT_SYNC"] == "0"
+
+$:.unshift(File.expand_path("../", __FILE__))
+require "active_support/core_ext/class"
+require "active_support/core_ext/hash"
+require "active_support/core_ext/string"
+require "active_support/ordered_options"
+require "deep_merge/rails_compat"
+require "dsl_evaluator"
+require "fileutils"
+require "hash_squeezer"
 require "kubes/version"
+require "memoist"
+require "rainbow/ext/string"
+require "yaml"
+
+DslEvaluator.backtrace_reject = ".kubes"
+
+require "kubes/autoloader"
+Kubes::Autoloader.setup
 
 module Kubes
-  # Your code goes here...
+  class Error < StandardError; end
+  extend Core
 end

data/lib/kubes/args/custom.rb

@@ -0,0 +1,30 @@
+module Kubes::Args
+  class Custom
+    extend Memoist
+    include Dsl
+    include DslEvaluator
+
+    attr_accessor :name
+    def initialize(name, file)
+      @name = name.to_s
+      @file = file # IE: .kubes/config/kubectl/args.rb
+      @commands = {}
+    end
+
+    def build
+      return @commands unless File.exist?(@file)
+      evaluate_file(@file)
+      @commands.deep_stringify_keys!
+    end
+    memoize :build
+
+    def args
+      args = dig("args")
+      args.compact.flatten
+    end
+
+    def dig(prop, default=[])
+      @commands.dig(@name, prop) || default
+    end
+  end
+end

data/lib/kubes/args/dsl.rb

@@ -0,0 +1,14 @@
+module Kubes::Args
+  module Dsl
+    def command(*commands, **props)
+      commands.each do |name|
+        each_command(name, props)
+      end
+    end
+    alias_method :commands, :command
+
+    def each_command(name, props={})
+      @commands[name.to_s] = props
+    end
+  end
+end

data/lib/kubes/auth.rb

@@ -0,0 +1,21 @@
+module Kubes
+  class Auth
+    def initialize(image)
+      @image = image
+    end
+
+    def run
+      klass = strategy_class
+      return unless klass
+      klass.new(@image).run
+    end
+
+    # Currently only support ECR
+    def strategy_class
+      case @image
+      when /\.amazonaws\.com/ # IE: 112233445566.dkr.ecr.us-west-2.amazonaws.com/demo/sinatra
+        Ecr
+      end
+    end
+  end
+end

data/lib/kubes/auth/ecr.rb

@@ -0,0 +1,55 @@
+# Normally, you must authorized to AWS ECR to push to their registry with:
+#
+#   eval $(aws ecr get-login --no-include-email)
+#
+# If you haven't ever ran the ecr get-login command before then you'll get this error:
+#
+#   no basic auth credentials
+#
+# If you have ran it before but the auto token has expired you'll get this message:
+#
+#   denied: Your Authorization Token has expired. Please run 'aws ecr get-login' to fetch a new one.
+#
+# This class updates the ~/.docker/config.json file which is an internal docker file to automatically update the auto token for you.
+# If that format changes, the update will need to be updated.
+#
+class Kubes::Auth
+  class Ecr
+    include Kubes::AwsServices
+
+    def initialize(image)
+      @image = image
+      @repo_domain = "#{image.split('/').first}"
+    end
+
+    def run
+      auth_token = fetch_auth_token
+      if File.exist?(docker_config)
+        data = JSON.load(IO.read(docker_config))
+        data["auths"][@repo_domain] = {auth: auth_token}
+      else
+        data = {"auths" => {@repo_domain => {auth: auth_token}}}
+      end
+
+      # Handle legacy docker clients that still have old format with https://
+      legacy_entry = "https://#{@repo_domain}"
+      data["auths"][legacy_entry] = {auth: auth_token}
+
+      ensure_dotdocker_exists
+      IO.write(docker_config, JSON.pretty_generate(data))
+    end
+
+    def fetch_auth_token
+      ecr.get_authorization_token.authorization_data.first.authorization_token
+    end
+
+    def docker_config
+      "#{ENV['HOME']}/.docker/config.json"
+    end
+
+    def ensure_dotdocker_exists
+      dirname = File.dirname(docker_config)
+      FileUtils.mkdir_p(dirname) unless File.exist?(dirname)
+    end
+  end
+end

data/lib/kubes/autoloader.rb

@@ -0,0 +1,21 @@
+require "zeitwerk"
+
+module Kubes
+  class Autoloader
+    class Inflector < Zeitwerk::Inflector
+      def camelize(basename, _abspath)
+        map = { cli: "CLI", version: "VERSION" }
+        map[basename.to_sym] || super
+      end
+    end
+
+    class << self
+      def setup
+        loader = Zeitwerk::Loader.new
+        loader.inflector = Inflector.new
+        loader.push_dir(File.dirname(__dir__)) # lib
+        loader.setup
+      end
+    end
+  end
+end

data/lib/kubes/aws_services.rb

@@ -0,0 +1,14 @@
+require "aws-sdk-ecr"
+
+# require "aws_mfa_secure/ext/aws" # add MFA support
+
+module Kubes
+  module AwsServices
+    extend Memoist
+
+    def ecr
+      Aws::ECR::Client.new
+    end
+    memoize :ecr
+  end
+end

data/lib/kubes/cli.rb

@@ -0,0 +1,93 @@
+module Kubes
+  class CLI < Command
+    class_option :verbose, type: :boolean
+    class_option :noop, type: :boolean
+
+    image_option = Proc.new {
+      option :image, desc: "override image"
+    }
+    compile_option = Proc.new {
+      option :compile, type: :boolean, default: true, desc: "whether or not to compile the .kube/resources"
+    }
+
+    desc "docker SUBCOMMAND", "Docker subcommands"
+    long_desc Help.text(:docker)
+    subcommand "docker", Docker
+
+    desc "apply [ROLE] [RESOURCE]", "Apply the Kubernetes YAML files without building docker image"
+    long_desc Help.text(:apply)
+    image_option.call
+    compile_option.call
+    def apply(role=nil, resource=nil)
+      Apply.new(options.merge(role: role, resource: resource)).run
+    end
+
+    desc "clean", "Remove .kubes/output files"
+    long_desc Help.text(:clean)
+    image_option.call
+    def clean
+      Clean.new(options).run
+    end
+
+    desc "compile", "Compile Kubernetes YAML files from DSL"
+    long_desc Help.text(:compile)
+    image_option.call
+    def compile
+      Compile.new(options).run
+    end
+
+    desc "delete [ROLE] [RESOURCE]", "Delete Kubernetes resources within the app folder"
+    long_desc Help.text(:delete)
+    image_option.call
+    option :yes, aliases: %w[y], type: :boolean, desc: "Skip are you sure prompt"
+    def delete(role=nil, resource=nil)
+      Delete.new(options.merge(role: role, resource: resource)).run
+    end
+
+    desc "deploy [ROLE] [RESOURCE]", "Deploy to Kubernetes: docker build/push, kubes compile, and kubectl apply"
+    long_desc Help.text(:deploy)
+    image_option.call
+    option :build, type: :boolean, default: true, desc: "whether or not to build docker image"
+    def deploy(role=nil, resource=nil)
+      Deploy.new(options.merge(role: role, resource: resource)).run
+    end
+
+    desc "describe [ROLE] [RESOURCE]", "Describe Kubernetes resource using the compiled YAML files"
+    long_desc Help.text(:describe)
+    image_option.call
+    compile_option.call
+    def describe(role=nil, resource=nil)
+      Describe.new(options.merge(role: role, resource: resource)).run
+    end
+
+    desc "get [ROLE] [RESOURCE]", "Get Kubernetes resource using the compiled YAML files"
+    long_desc Help.text(:get)
+    image_option.call
+    compile_option.call
+    option :output, aliases: %w[o], desc: "Output format: json|yaml|wide|name"
+    def get(role=nil, resource=nil)
+      Get.new(options.merge(role: role, resource: resource)).run
+    end
+
+    long_desc Help.text(:init)
+    Init.options.each { |args| option(*args) }
+    register(Init, "init", "init", "Init project")
+
+    desc "completion *PARAMS", "Prints words for auto-completion."
+    long_desc Help.text(:completion)
+    def completion(*params)
+      Completer.new(CLI, *params).run
+    end
+
+    desc "completion_script", "Generates a script that can be eval to setup auto-completion."
+    long_desc Help.text(:completion_script)
+    def completion_script
+      Completer::Script.generate
+    end
+
+    desc "version", "prints version"
+    def version
+      puts VERSION
+    end
+  end
+end

data/lib/kubes/cli/apply.rb

@@ -0,0 +1,9 @@
+class Kubes::CLI
+  class Apply < Base
+    def run
+      Compile.new(@options).run unless @options[:compile] == false
+      logger.info "Deploying kubes resources"
+      Kubes::Kubectl::Decider.new(:apply, @options).run
+    end
+  end
+end

data/lib/kubes/cli/base.rb

@@ -0,0 +1,9 @@
+class Kubes::CLI
+  class Base
+    include Kubes::Logging
+
+    def initialize(options={})
+      @options = options
+    end
+  end
+end

data/lib/kubes/cli/build.rb

@@ -0,0 +1,8 @@
+class Kubes::CLI
+  class Build < Base
+    def run
+      Kubes::Docker::Build.new(@options).run
+      Kubes::Docker::Push.new(@options).run
+    end
+  end
+end

data/lib/kubes/cli/clean.rb

@@ -0,0 +1,9 @@
+class Kubes::CLI
+  class Clean < Base
+    def run
+      path = ".kubes/output"
+      FileUtils.rm_rf("#{Kubes.root}/#{path}")
+      logger.info "Removed #{path}" unless @options[:mute]
+    end
+  end
+end

data/lib/kubes/cli/compile.rb

@@ -0,0 +1,8 @@
+class Kubes::CLI
+  class Compile < Base
+    def run
+      Clean.new(@options.merge(mute: true)).run
+      Kubes::Compiler.new(@options).run
+    end
+  end
+end

data/lib/kubes/cli/delete.rb

@@ -0,0 +1,11 @@
+class Kubes::CLI
+  class Delete < Base
+    include Kubes::Util::Sure
+
+    def run
+      sure?("This will delete resources. Are you sure?")
+      Compile.new(@options).run
+      Kubes::Kubectl::Decider.new(:delete, @options).run
+    end
+  end
+end

data/lib/kubes/cli/deploy.rb

@@ -0,0 +1,13 @@
+class Kubes::CLI
+  class Deploy < Base
+    def run
+      Build.new(@options).run if build?
+      Apply.new(@options).run # also calls Compile
+    end
+
+    def build?
+      return false if @options[:build] == false || @options[:image]
+      @options[:resource].nil? || @options[:resource] == "deployment"
+    end
+  end
+end

data/lib/kubes/cli/describe.rb

@@ -0,0 +1,8 @@
+class Kubes::CLI
+  class Describe < Base
+    def run
+      Compile.new(@options).run unless @options[:compile] == false
+      Kubes::Kubectl.run(:describe, @options)
+    end
+  end
+end