kontena-plugin-app-command 0.1.0.rc1

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

@@ -0,0 +1,74 @@
+require_relative 'common'
+
+module Kontena::Cli::Apps
+  class RemoveCommand < 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 '--force', :flag, 'Force remove', default: false, attribute_name: :forced
+
+    parameter "[SERVICE] ...", "Remove services"
+
+    attr_reader :services
+
+    def execute
+      require_api_url
+      require_token
+      require_config_file(filename)
+      confirm unless forced?
+
+      @services = services_from_yaml(filename, service_list, service_prefix, true)
+      if services.size > 0
+        remove_services(services)
+      elsif !service_list.empty?
+        puts "No such service: #{service_list.join(', ')}".colorize(:red)
+      end
+    end
+
+    private
+
+    def remove_services(services)
+      services.find_all {|service_name, options| options['links'] && options['links'].size > 0 }.each do |service_name, options|
+        delete(service_name, options, false)
+        services.delete(service_name)
+      end
+      services.each do |service_name, options|
+        delete(service_name, options)
+      end
+    end
+
+    def delete(name, options, async = true)
+      unless deleted_services.include?(name)
+        service = get_service(token, prefixed_name(name)) rescue nil
+        if(service)
+          spinner "removing #{pastel.cyan(name)}" do
+            delete_service(token, prefixed_name(name))
+            unless async
+              wait_for_delete_to_finish(service)
+            end
+          end
+          deleted_services << name
+        else
+          warning "No such service #{name}"
+        end
+      end
+    end
+
+    def wait_for_delete_to_finish(service)
+      until service.nil?
+        service = get_service(token, service['name']) rescue nil
+        sleep 0.5
+      end
+    end
+
+    ##
+    #
+    # @return [Array]
+    def deleted_services
+      @deleted_services ||= []
+    end
+  end
+end

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

@@ -0,0 +1,39 @@
+require_relative 'common'
+
+module Kontena::Cli::Apps
+  class RestartCommand < Kontena::Command
+    include Kontena::Cli::Common
+    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
+        restart_services(services)
+      elsif !service_list.empty?
+        puts "No such service: #{service_list.join(', ')}".colorize(:red)
+      end
+
+    end
+
+    def restart_services(services)
+      services.each do |service_name, opts|
+        if service_exists?(service_name)
+          spinner "Sending restart signal to #{service_name.colorize(:cyan)} " do
+            restart_service(token, prefixed_name(service_name))
+          end
+        else
+          warning "No such service: #{service_name}"
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,33 @@
+require_relative 'common'
+
+module Kontena::Cli::Apps
+  class ScaleCommand < 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", "Service to show"
+    parameter "INSTANCES", "Scales service to given number of instances"
+
+    attr_reader :services
+
+    def execute
+      require_config_file(filename)
+      yml_service = services_from_yaml(filename, [service], service_prefix, true)
+      if yml_service[service]
+        options = yml_service[service]
+        exit_with_error("Service has already instances defined in #{filename}. Please update #{filename} and deploy service instead") if options['instances']
+        spinner "Scaling #{service.colorize(:cyan)} " do
+          deployment = scale_service(require_token, prefixed_name(service), instances)
+          wait_for_deploy_to_finish(token, deployment)
+        end
+
+      else
+        exit_with_error("Service not found")
+      end
+    end
+  end
+end

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

@@ -0,0 +1,114 @@
+require 'yaml'
+require 'shellwords'
+require 'kontena/cli/services/services_helper'
+
+module Kontena::Cli::Apps
+  class ServiceGenerator
+    include Kontena::Cli::Services::ServicesHelper
+
+    attr_reader :service_config
+
+    def initialize(service_config)
+      @service_config = service_config
+    end
+
+    ##
+    # @return [Hash]
+    def generate
+      parse_data(service_config)
+    end
+
+    private
+
+    ##
+    # @param [Hash] options
+    # @return [Hash]
+    def parse_data(options)
+      data = {}
+      data['instances'] = options['instances']
+      data['image'] = parse_image(options['image'])
+      data['env'] = options['environment'] if options['environment']
+      data['links'] = parse_links(options['links'] || [])
+      data['external_links'] = parse_links(options['external_links'] || [])
+      data['ports'] = parse_stringified_ports(options['ports'] || [])
+      data['memory'] = parse_memory(options['mem_limit'].to_s) if options['mem_limit']
+      data['memory_swap'] = parse_memory(options['memswap_limit'].to_s) if options['memswap_limit']
+      data['cpu_shares'] = options['cpu_shares'] if options['cpu_shares']
+      data['volumes'] = options['volumes'] || []
+      data['volumes_from'] = options['volumes_from'] || []
+      data['cmd'] = Shellwords.split(options['command']) if options['command']
+      data['affinity'] = options['affinity'] || []
+      data['user'] = options['user'] if options['user']
+      data['stateful'] = options['stateful'] == true
+      data['privileged'] = options['privileged'] unless options['privileged'].nil?
+      data['cap_add'] = options['cap_add'] if options['cap_add']
+      data['cap_drop'] = options['cap_drop'] if options['cap_drop']
+      data['net'] = options['net'] if options['net']
+      data['pid'] = options['pid'] if options['pid']
+      data['log_driver'] = options['log_driver'] if options['log_driver']
+      data['log_opts'] = options['log_opt'] if options['log_opt'] && !options['log_opt'].empty?
+      deploy_opts = options['deploy'] || {}
+      data['strategy'] = deploy_opts['strategy'] if deploy_opts['strategy']
+      deploy = {}
+      deploy['wait_for_port'] = deploy_opts['wait_for_port'] if deploy_opts.has_key?('wait_for_port')
+      deploy['min_health'] = deploy_opts['min_health'] if deploy_opts.has_key?('min_health')
+      deploy['interval'] = parse_relative_time(deploy_opts['interval']) if deploy_opts.has_key?('interval')
+      unless deploy.empty?
+        data['deploy_opts'] = deploy
+      end
+      data['hooks'] = options['hooks'] || {}
+      data['secrets'] = options['secrets'] if options['secrets']
+      data['build'] = parse_build_options(options) if options['build']
+      health_check = {}
+      health_opts = options['health_check'] || {}
+      health_check['protocol'] = health_opts['protocol'] if health_opts.has_key?('protocol')
+      health_check['uri'] = health_opts['uri'] if health_opts.has_key?('uri')
+      health_check['port'] = health_opts['port'] if health_opts.has_key?('port')
+      health_check['timeout'] = health_opts['timeout'] if health_opts.has_key?('timeout')
+      health_check['interval'] = health_opts['interval'] if health_opts.has_key?('interval')
+      health_check['initial_delay'] = health_opts['initial_delay'] if health_opts.has_key?('initial_delay')
+      unless health_check.empty?
+        data['health_check'] = health_check
+      end
+      data
+    end
+
+    # @param [Array<String>] port_options
+    # @return [Array<Hash>]
+    def parse_stringified_ports(port_options)
+      parse_ports(port_options).map {|p|
+        {
+          'ip' => p[:ip],
+          'container_port' => p[:container_port],
+          'node_port' => p[:node_port],
+          'protocol' => p[:protocol]
+        }
+      }
+    end
+
+    # @param [Array<String>] link_options
+    # @return [Array<Hash>]
+    def parse_links(link_options)
+      link_options.map{|l|
+        service_name, alias_name = l.split(':')
+        if service_name.nil?
+          raise ArgumentError.new("Invalid link value #{l}")
+        end
+        alias_name = service_name if alias_name.nil?
+        {
+            'name' => service_name,
+            'alias' => alias_name
+        }
+      }
+    end
+
+    # @param [Hash] options
+    # @return [Hash]
+    def parse_build_options(options)
+      build = {}
+      build['context'] = options['build'] if options['build']
+      build['dockerfile'] = options['dockerfile'] if options['dockerfile']
+      build
+    end
+  end
+end

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

@@ -0,0 +1,27 @@
+require 'yaml'
+require_relative 'service_generator'
+
+module Kontena::Cli::Apps
+  class ServiceGeneratorV2 < ServiceGenerator
+
+    def parse_data(options)
+      data = super(options)
+      data['net'] = options['network_mode'] if options['network_mode']
+      data['log_driver'] = options.dig('logging', 'driver')
+      data['log_opts'] = options.dig('logging', 'options')
+      if options['depends_on']
+        data['links'] ||= []
+        data['links'] = (data['links'] + parse_links(options['depends_on'])).uniq
+      end
+      data
+    end
+
+    def parse_build_options(options)
+      unless options['build'].is_a?(Hash)
+        options['build'] = { 'context' => options['build']}
+      end
+      options['build']['args'] = parse_build_args(options['build']['args']) if options['build']['args']
+      options['build']
+    end
+  end
+end

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

@@ -0,0 +1,23 @@
+require_relative 'common'
+
+module Kontena::Cli::Apps
+  class ShowCommand < 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", "Service to show"
+
+    attr_reader :services
+
+    def execute
+      require_config_file(filename)
+      token = require_token
+      show_service(token, prefixed_name(service))
+      show_service_instances(token, prefixed_name(service))
+    end
+  end
+end

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

@@ -0,0 +1,40 @@
+require_relative 'common'
+
+module Kontena::Cli::Apps
+  class StartCommand < 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
+        start_services(services)
+      elsif !service_list.empty?
+        puts "No such service: #{service_list.join(', ')}".colorize(:red)
+      end
+
+    end
+
+    def start_services(services)
+      services.each do |service_name, opts|
+        if service_exists?(service_name)
+          spinner "Starting #{service_name.colorize(:cyan)} " do
+            start_service(token, prefixed_name(service_name))
+          end
+        else
+          warning "No such service: #{service_name}"
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,40 @@
+require_relative 'common'
+
+module Kontena::Cli::Apps
+  class StopCommand < 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 stop"
+
+    attr_reader :services
+
+    def execute
+      require_config_file(filename)
+
+      @services = services_from_yaml(filename, service_list, service_prefix, true)
+      if services.size > 0
+        stop_services(services)
+      elsif !service_list.empty?
+        puts "No such service: #{service_list.join(', ')}".colorize(:red)
+      end
+
+    end
+
+    def stop_services(services)
+      services.each do |service_name, opts|
+        if service_exists?(service_name)
+          spinner "Sending stop signal to #{service_name.colorize(:cyan)} " do
+            stop_service(token, prefixed_name(service_name))
+          end
+        else
+          warning "No such service: #{service_name}"
+        end
+      end
+    end
+  end
+end

data/lib/kontena/cli/apps/yaml/custom_validators/affinities_validator.rb

@@ -0,0 +1,19 @@
+module Kontena::Cli::Apps::YAML::Validations::CustomValidators
+  class AffinitiesValidator < HashValidator::Validator::Base
+    def initialize
+      super('valid_affinities')
+    end
+
+    def validate(key, value, validations, errors)
+      unless value.is_a?(Array)
+        errors[key] = 'affinity must be array'
+        return
+      end
+
+      invalid_formats = value.find_all { |a| !a.match(/(?<=\!|\=)=/) }
+      if invalid_formats.count > 0
+        errors[key] = "affinity contains invalid formats: #{invalid_formats.join(', ')}"
+      end
+    end
+  end
+end

data/lib/kontena/cli/apps/yaml/custom_validators/build_validator.rb

@@ -0,0 +1,22 @@
+module Kontena::Cli::Apps::YAML::Validations::CustomValidators
+  class BuildValidator < HashValidator::Validator::Base
+    def initialize
+      super('valid_build')
+    end
+
+    def validate(key, value, validations, errors)
+      unless value.is_a?(String) || value.is_a?(Hash)
+        errors[key] = 'build must be string or hash'
+        return
+      end
+      if value.is_a?(Hash)
+        build_validation = {
+          'context' => 'string',
+          'dockerfile' => HashValidator.optional('string'),
+          'args' => HashValidator.optional(-> (value) { value.is_a?(Array) || value.is_a?(Hash) })
+        }
+        HashValidator.validator_for(build_validation).validate(key, value, build_validation, errors)
+      end
+    end
+  end
+end