kontena-plugin-app-command 0.1.0.rc1

data/lib/kontena/cli/apps/docker_helper.rb

@@ -0,0 +1,80 @@
+module Kontena::Cli::Apps
+  module DockerHelper
+
+    # @param [Hash] services
+    # @param [Boolean] force_build
+    # @param [Boolean] no_cache
+    def process_docker_images(services, force_build = false, no_cache = false)
+      services.each do |name, service|
+        if service['build'] && (!image_exist?(service['image']) || force_build)
+          dockerfile = service['build']['dockerfile'] || 'Dockerfile'
+          raise ("'#{service['image']}' is not valid Docker image name") unless validate_image_name(service['image'])
+          raise ("'#{service['build']['context']}' does not have #{dockerfile}") unless dockerfile_exist?(service['build']['context'], dockerfile)
+          if service['hooks'] && service['hooks']['pre_build']
+            puts "Running pre_build hook".colorize(:cyan)
+            run_pre_build_hook(service['hooks']['pre_build'])
+          end
+          puts "Building image #{service['image'].colorize(:cyan)}"
+          build_docker_image(service, no_cache)
+          puts "Pushing image #{service['image'].colorize(:cyan)} to registry"
+          push_docker_image(service['image'])
+        end
+      end
+    end
+
+    # @param [String] name
+    # @return [Boolean]
+    def validate_image_name(name)
+      !(/^[\w.\/\-:]+:?+[\w+.]+$/ =~ name).nil?
+    end
+
+    # @param [Hash] service
+    # @param [Boolean] no_cache
+    # @return [Integer]
+    def build_docker_image(service, no_cache = false)
+      dockerfile = dockerfile = service['build']['dockerfile'] || 'Dockerfile'
+      build_context = service['build']['context']
+      cmd = ['docker', 'build', '-t', service['image']]
+      cmd << ['-f', File.join(File.expand_path(build_context), dockerfile)] if dockerfile != "Dockerfile"
+      cmd << '--no-cache' if no_cache
+      args = service['build']['args'] || {}
+      args.each do |k, v|
+        cmd << "--build-arg=#{k}=#{v}"
+      end
+      cmd << build_context
+      ret = system(*cmd.flatten)
+      raise ("Failed to build image #{service['image'].colorize(:cyan)}") unless ret
+      ret
+    end
+
+    # @param [String] image
+    # @return [Integer]
+    def push_docker_image(image)
+      ret = system('docker', 'push', image)
+      raise ("Failed to push image #{image.colorize(:cyan)}") unless ret
+      ret
+    end
+
+    # @param [String] image
+    # @return [Boolean]
+    def image_exist?(image)
+      system("docker history '#{image}' >/dev/null 2>/dev/null")
+    end
+
+    # @param [String] path
+    # @param [String] dockerfile
+    # @return [Boolean]
+    def dockerfile_exist?(path, dockerfile)
+      file = File.join(File.expand_path(path), dockerfile)
+      File.exist?(file)
+    end
+
+    # @param [Hash] hook
+    def run_pre_build_hook(hook)
+      hook.each do |h|
+        ret = system(h['cmd'])
+        raise ("Failed to run pre_build hook: #{h['name']}!") unless ret
+      end
+    end
+  end
+end

data/lib/kontena/cli/apps/dockerfile_generator.rb

@@ -0,0 +1,16 @@
+require 'yaml'
+require_relative 'common'
+
+module Kontena::Cli::Apps
+  class DockerfileGenerator
+    include Common
+
+    def generate(base_image)
+
+      dockerfile = File.new('Dockerfile', 'w')
+      dockerfile.puts "FROM #{base_image}"
+      dockerfile.puts 'CMD ["/start", "web"]'
+      dockerfile.close
+    end
+  end
+end

data/lib/kontena/cli/apps/init_command.rb

@@ -0,0 +1,89 @@
+require 'securerandom'
+require_relative 'common'
+require_relative 'dockerfile_generator'
+require_relative 'docker_compose_generator'
+require_relative 'kontena_yml_generator'
+
+module Kontena::Cli::Apps
+  class InitCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Common
+
+    option ["-f", "--file"], "FILE", "Specify a docker-compose file", attribute_name: :docker_compose_file, default: 'docker-compose.yml'
+    option ["-i", "--image-name"], "IMAGE_NAME", "Specify a docker image name"
+    option ["-b", "--base-image"], "BASE_IMAGE_NAME", "Specify a docker base image name", default: "kontena/buildstep"
+    option ["-p", "--project-name"], "NAME", "Specify an alternate project name (default: directory name)"
+
+
+    def execute
+      if File.exist?('Dockerfile')
+        puts 'Found Dockerfile'
+      elsif create_dockerfile?
+        puts "Creating #{'Dockerfile'.colorize(:cyan)}"
+        DockerfileGenerator.new.generate(base_image)
+      end
+
+      if File.exist?('Procfile')
+        procfile = ::YAML.safe_load(File.read('Procfile'))
+      else
+        procfile = {}
+      end
+
+      app_env = create_env_file(app_json['env']) if app_json['env']
+      addons = app_json['addons'] || []
+
+      if File.exist?(docker_compose_file)
+        puts "Found #{docker_compose_file}."
+      elsif create_docker_compose_yml?
+        puts "Creating #{docker_compose_file.colorize(:cyan)}"
+        docker_compose_generator = DockerComposeGenerator.new(docker_compose_file)
+        docker_compose_generator.generate(procfile, addons, app_env)
+      end
+
+      if File.exist?('kontena.yml')
+        puts "Updating #{'kontena.yml'.colorize(:cyan)}"
+      else
+        puts "Creating #{'kontena.yml'.colorize(:cyan)}"
+      end
+
+      kontena_yml_generator = KontenaYmlGenerator.new(image_name, service_prefix)
+      if File.exist?(docker_compose_file)
+        kontena_yml_generator.generate_from_compose_file(docker_compose_file)
+      else
+        kontena_yml_generator.generate(procfile, addons, app_env)
+      end
+
+      puts "Your app is ready! Deploy with 'kontena app deploy'.".colorize(:green)
+    end
+
+
+    protected
+
+    def service_prefix
+      @service_prefix ||= project_name || current_dir
+    end
+
+    def create_dockerfile?
+      prompt.yes?('Dockerfile not found. Do you want to create it?')
+    end
+
+    def create_env_file(env)
+      app_env = File.new('.env', 'w')
+      app_json['env'].each do |key, env|
+        if env['generator'] == 'secret'
+          value = SecureRandom.hex(64)
+        else
+          value = env['value']
+        end
+        app_env.puts "#{key}=#{value}"
+      end
+      app_env.close
+      '.env'
+    end
+
+    def create_docker_compose_yml?
+      prompt.yes?("#{docker_compose_file} not found. Do you want to create it?")
+    end
+
+  end
+end

data/lib/kontena/cli/apps/kontena_yml_generator.rb

@@ -0,0 +1,105 @@
+require_relative 'common'
+
+module Kontena::Cli::Apps
+  class KontenaYmlGenerator
+    include Common
+
+    attr_reader :image_name, :service_prefix
+
+    def initialize(image_name, service_prefix)
+      @image_name = image_name
+      @service_prefix = service_prefix
+    end
+
+    def generate_from_compose_file(docker_compose_file)
+      services = {}
+      # extend services from docker-compose.yml
+      file = File.read(docker_compose_file)
+
+      yml_services(file).each do |name, options|
+        services[name] = {'extends' => { 'file' => 'docker-compose.yml', 'service' => name }}
+        if options.has_key?('build')
+          image = image_name || "registry.kontena.local/#{File.basename(Dir.getwd)}:latest"
+          services[name]['image'] = image
+        end
+
+        # set Heroku addon service as stateful by default
+        if valid_addons.has_key?(name)
+          services[name]['stateful'] = true
+        end
+
+        # we have to generate Kontena urls to env vars for Heroku addons
+        # redis://openredis:6379 -> redis://project-name-openredis:6379
+        if options['links']
+          options['links'].each do |link|
+            service_link = link.split(':').first
+            if valid_addons.has_key?(service_link)
+              services[name]['environment'] ||= []
+              services[name]['environment'] += valid_addons(service_prefix)[service_link]['environment']
+            end
+          end
+        end
+      end
+      create_yml(services, 'kontena.yml')
+    end
+
+    def yml_services(file)
+      yml = ::YAML.safe_load(file)
+      if yml['version'] == '2'
+        yml['services']
+      else
+        yml
+      end
+    end
+
+    def generate(procfile, addons, env_file)
+      image = image_name || "registry.kontena.local/#{File.basename(Dir.getwd)}:latest"
+      if procfile.keys.size > 0
+        # generate services found in Procfile
+        services = {}
+        procfile.keys.each do |name|
+          services[name] = {'image' => image}
+          services[name]['environment'] = ['PORT=5000'] if app_json && name == 'web' # Heroku generates PORT env variable so should we do too
+          services[name]['command'] = "/start #{name}" if name != 'web'
+          services[name]['env_file'] = env_file if env_file
+
+          # generate addon services
+          addons.each do |addon|
+            addon_service = addon.split(":")[0]
+            addon_service.slice!('heroku-')
+            if valid_addons.has_key?(addon_service)
+              services[name]['links'] ||= []
+              services[name]['links'] << "#{addon_service}:#{addon_service}"
+              services[name]['environment'] ||= []
+              services[name]['environment'] += valid_addons(service_prefix)[addon_service]['environment']
+              services[addon_service] = {'image' => valid_addons[addon_service]['image'], 'stateful' => true}
+            end
+          end
+        end
+      else
+        # no Procfile found, create dummy web service
+        services = {'web' => { 'image' => image}}
+        services['web']['env_file'] = env_file if env_file
+      end
+      # create kontena.yml file
+      create_yml(services, 'kontena.yml')
+    end
+
+    def create_yml(services, filename)
+      if File.exist?(filename) && !File.zero?(filename)
+        kontena_services = yml_services(File.read(filename))
+        services.each do |name, options|
+          if kontena_services[name]
+            services[name].merge!(kontena_services[name])
+          end
+        end
+      end
+      kontena_services = {
+        'version' => '2',
+        'name' => service_prefix,
+        'services' => services
+      }
+      super(kontena_services, filename)
+    end
+  end
+end

data/lib/kontena/cli/apps/list_command.rb

@@ -0,0 +1,59 @@
+require_relative 'common'
+
+module Kontena::Cli::Apps
+  class ListCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Common
+
+    option ['-f', '--file'], 'FILE', 'Specify an alternate Kontena compose file', attribute_name: :filename, default: 'kontena.yml'
+    option ['-p', '--project-name'], 'NAME', 'Specify an alternate project name (default: directory name)'
+
+    option ['-q', '--quiet'], :flag, "Output the identifying column only"
+
+    parameter "[SERVICE] ...", "Services to list"
+
+    attr_reader :services
+
+    def execute
+      require_config_file(filename)
+
+      @services = services_from_yaml(filename, service_list, service_prefix, true)
+
+      if quiet?
+        puts services.map(&:first).join("\n")
+        exit 0
+      end
+
+      if services.size > 0
+        show_services(services)
+      elsif !service_list.empty?
+        puts "No such service: #{service_list.join(', ')}".colorize(:red)
+      end
+
+    end
+
+    def show_services(services)
+      titles = ['NAME', 'IMAGE', 'INSTANCES', 'STATEFUL', 'STATE', 'PORTS']
+      puts "%-30.30s %-50.50s %-15s %-10.10s %-15.20s %-50s" % titles
+
+      services.each do |service_name, opts|
+        service = get_service(token, prefixed_name(service_name)) rescue false
+        if service
+          name = service['name'].sub("#{service_prefix}-", '')
+          state = service['stateful'] ? 'yes' : 'no'
+          ports = service['ports'].map{|p|
+            "#{p['ip']}:#{p['node_port']}->#{p['container_port']}/#{p['protocol']}"
+          }.join(", ")
+          running = service['instance_counts']['running']
+          desired = service['instances']
+          instances = "#{running} / #{desired}"
+          vars = [name, service['image'], instances, state, service['state'], ports]
+        else
+          vars = [service_name, '-', '-', '-', '-', '-']
+        end
+        puts "%-30.30s %-50.50s %-15.10s %-10.10s %-15.20s %-50s" % vars
+      end
+    end
+  end
+end

data/lib/kontena/cli/apps/logs_command.rb

@@ -0,0 +1,37 @@
+require_relative 'common'
+require 'kontena/cli/helpers/log_helper'
+
+module Kontena::Cli::Apps
+  class LogsCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Kontena::Cli::Helpers::LogHelper
+
+    include Common
+
+    option ['-f', '--file'], 'FILE', 'Specify an alternate Kontena compose file', attribute_name: :filename, default: 'kontena.yml'
+    option ['-p', '--project-name'], 'NAME', 'Specify an alternate project name (default: directory name)'
+    parameter "[SERVICE] ...", "Show only specified service logs"
+
+    def execute
+      require_config_file(filename)
+
+      services = services_from_yaml(filename, service_list, service_prefix, true)
+
+      if services.empty? && !service_list.empty?
+        signal_error "No such service: #{service_list.join(', ')}"
+      elsif services.empty?
+        signal_error "No services for application"
+      end
+
+      query_services = services.map{|service_name, opts| prefixed_name(service_name)}.join ','
+      query_params = {
+        services: query_services,
+      }
+
+      show_logs("grids/#{current_grid}/container_logs", query_params) do |log|
+        show_log(log)
+      end
+    end
+  end
+end

data/lib/kontena/cli/apps/monitor_command.rb

@@ -0,0 +1,93 @@
+require_relative 'common'
+
+module Kontena::Cli::Apps
+  class MonitorCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Common
+
+    option ['-f', '--file'], 'FILE', 'Specify an alternate Kontena compose file', attribute_name: :filename, default: 'kontena.yml'
+    option ['-p', '--project-name'], 'NAME', 'Specify an alternate project name (default: directory name)'
+
+    parameter "[SERVICE] ...", "Services to start"
+
+    attr_reader :services
+
+    def execute
+      require_config_file(filename)
+
+      @services = services_from_yaml(filename, service_list, service_prefix, true)
+      if services.size > 0
+        show_monitor(services)
+      elsif !service_list.empty?
+        puts "No such service: #{service_list.join(', ')}".colorize(:red)
+      end
+    end
+
+    def show_monitor(services)
+      require_api_url
+      token = require_token
+      loop do
+        nodes = {}
+        services.each do |name, data|
+          service = prefixed_name(name)
+          result = client(token).get("services/#{current_grid}/#{service}/containers") rescue nil
+          if result
+            services[name]['instances'] = result['containers'].size
+            result['containers'].each do |container|
+              container['service'] = name
+              nodes[container['node']['name']] ||= []
+              nodes[container['node']['name']] << container
+            end
+          end
+        end
+        clear_terminal
+        puts "services:"
+        services.each do |name, data|
+          color = color_for_service(name)
+          puts "  #{"■".colorize(color)} #{name} (#{data['instances']} instances)"
+        end
+        puts "nodes:"
+        node_names = nodes.keys.sort
+        node_names.each do |name|
+          containers = nodes[name]
+          puts "  #{name} (#{containers.size} instances)"
+          print "  "
+          containers.each do |container|
+            icon = "■"
+            if container['status'] != 'running'
+              icon = "□"
+            end
+            color = color_for_service(container['service'])
+            print icon.colorize(color)
+          end
+          puts ''
+        end
+        sleep 1
+      end
+    end
+
+    def color_for_service(service)
+      color_maps[service] = colors.shift unless color_maps[service]
+      color_maps[service].to_sym
+    end
+
+    def color_maps
+      @color_maps ||= {}
+    end
+
+    def colors
+      if(@colors.nil? || @colors.size == 0)
+        @colors = %i(
+          red green yellow blue magenta cyan bright_red bright_green
+          bright_yellow bright_blue bright_magenta bright_cyan
+        )
+      end
+      @colors
+    end
+
+    def clear_terminal
+      print "\e[H\e[2J"
+    end
+  end
+end