krates 1.6.0

data/lib/kontena/cli/services/create_command.rb

@@ -0,0 +1,105 @@
+require_relative 'services_helper'
+require 'shellwords'
+
+module Kontena::Cli::Services
+  class CreateCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include ServicesHelper
+
+    parameter "NAME", "Service name"
+    parameter "IMAGE", "Service image (example: redis:3.0)"
+
+    option ["-p", "--ports"], "PORTS", "Publish a service's port to the host", multivalued: true
+    option ["-e", "--env"], "ENV", "Set environment variables", multivalued: true
+    option ["-l", "--link"], "LINK", "Add link to another service in the form of name:alias", multivalued: true
+    option ["-v", "--volume"], "VOLUME", "Add a volume or bind mount it from the host", multivalued: true
+    option "--volumes-from", "VOLUMES_FROM", "Mount volumes from another container", multivalued: true
+    option ["-a", "--affinity"], "AFFINITY", "Set service affinity", multivalued: true
+    option "--cpus", "CPUS", "Number of CPUs" do |cpus|
+      Float(cpus)
+    end
+    option ["-c", "--cpu-shares"], "CPU_SHARES", "CPU shares (relative weight)"
+    option ["-m", "--memory"], "MEMORY", "Memory limit (format: <number><optional unit>, where unit = b, k, m or g)"
+    option ["--memory-swap"], "MEMORY_SWAP", "Total memory usage (memory + swap), set \'-1\' to disable swap (format: <number><optional unit>, where unit = b, k, m or g)"
+    option ["--shm-size"], "SHM_SIZE", "Size of /dev/shm (format: <number><optional unit>, where unit = b, k, m or g)"
+    option "--cmd", "CMD", "Command to execute"
+    option "--instances", "INSTANCES", "How many instances should be deployed"
+    option ["-u", "--user"], "USER", "Username who executes first process inside container"
+    option "--stateful", :flag, "Set service as stateful", default: false
+    option "--privileged", :flag, "Give extended privileges to this service", default: false
+    option "--cap-add", "CAP_ADD", "Add capabitilies", multivalued: true
+    option "--cap-drop", "CAP_DROP", "Drop capabitilies", multivalued: true
+    option "--net", "NET", "Network mode"
+    option "--log-driver", "LOG_DRIVER", "Set logging driver"
+    option "--log-opt", "LOG_OPT", "Add logging options", multivalued: true
+    option "--deploy-strategy", "STRATEGY", "Deploy strategy to use (ha, daemon, random)"
+    option "--deploy-wait-for-port", "PORT", "Wait for port to respond when deploying"
+    option "--deploy-min-health", "FLOAT", "The minimum percentage (0.0 - 1.0) of healthy instances that do not sacrifice overall service availability while deploying"
+    option "--deploy-interval", "TIME", "Auto-deploy with given interval (format: <number><unit>, where unit = min, h, d)"
+    option "--pid", "PID", "Pid namespace to use"
+    option "--secret", "SECRET", "Import secret from Vault (format: <secret>:<name>:<env>)", multivalued: true
+    option "--health-check-uri", "URI", "URI path for HTTP health check"
+    option "--health-check-timeout", "TIMEOUT", "Timeout for health check"
+    option "--health-check-interval", "INTERVAL", "Interval for health check"
+    option "--health-check-initial-delay", "DELAY", "Initial delay for health check"
+    option "--health-check-port", "PORT", "Port for health check"
+    option "--health-check-protocol", "PROTOCOL", "Protocol of health check"
+    option "--stop-signal", "STOP_SIGNAL", "Alternative signal to stop container"
+    option "--stop-timeout", "STOP_TIMEOUT", "Timeout (duration) to stop a container"
+    option "--read-only", :flag, "Read-only root fs for the container", default: false
+
+    def execute
+      require_api_url
+      token = require_token
+      data = {
+        name: name,
+        image: image,
+        stateful: stateful?
+      }
+      data.merge!(parse_service_data_from_options)
+      spinner "Creating #{pastel.cyan(name)} service " do
+        create_service(token, current_grid, data)
+      end
+    end
+
+    ##
+    # parse given options to hash
+    # @return [Hash]
+    def parse_service_data_from_options
+      data = {}
+      data[:ports] = parse_ports(ports_list) unless ports_list.empty?
+      data[:links] = parse_links(link_list) unless link_list.empty?
+      data[:volumes] = volume_list unless volume_list.empty?
+      data[:volumes_from] = volumes_from_list unless volumes_from_list.empty?
+      data[:memory] = parse_memory(memory) if memory
+      data[:memory_swap] = parse_memory(memory_swap) if memory_swap
+      data[:shm_size] = parse_memory(shm_size) if shm_size
+      data[:cpus] = cpus if cpus
+      data[:cpu_shares] = cpu_shares if cpu_shares
+      data[:affinity] = affinity_list unless affinity_list.empty?
+      data[:env] = env_list unless env_list.empty?
+      data[:secrets] = parse_secrets(secret_list)
+      data[:container_count] = instances if instances
+      data[:cmd] = Shellwords.split(cmd) if cmd
+      data[:user] = user if user
+      data[:image] = parse_image(image) if image
+      data[:privileged] = privileged?
+      data[:cap_add] = cap_add_list unless cap_add_list.empty?
+      data[:cap_drop] = cap_drop_list unless cap_drop_list.empty?
+      data[:net] = net if net
+      data[:log_driver] = log_driver if log_driver
+      data[:log_opts] = parse_log_opts(log_opt_list)
+      data[:strategy] = deploy_strategy if deploy_strategy
+      deploy_opts = parse_deploy_opts
+      data[:deploy_opts] = deploy_opts unless deploy_opts.empty?
+      health_check = parse_health_check
+      data[:health_check] = health_check unless health_check.empty?
+      data[:pid] = pid if pid
+      data[:stop_signal] = stop_signal if stop_signal
+      data[:stop_grace_period] = stop_timeout if stop_timeout
+      data[:read_only] = read_only?
+      data
+    end
+  end
+end

data/lib/kontena/cli/services/deploy_command.rb

@@ -0,0 +1,25 @@
+require_relative 'services_helper'
+
+module Kontena::Cli::Services
+  class DeployCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include ServicesHelper
+
+    parameter "NAME", "Service name"
+    option '--[no-]wait', :flag, 'Wait for service deployment to finish', default: true
+    option '--force', :flag, 'Force deploy even if service does not have any changes'
+
+    def execute
+      require_api_url
+      token = require_token
+      service_id = name
+      data = {}
+      data[:force] = true if force?
+      spinner "Deploying service #{pastel.cyan(name)} " do
+        deployment = deploy_service(token, name, data)
+        wait_for_deploy_to_finish(token, deployment) if wait?
+      end
+    end
+  end
+end

data/lib/kontena/cli/services/env_command.rb

@@ -0,0 +1,9 @@
+module Kontena::Cli::Services
+
+
+  class EnvCommand < Kontena::Command
+    subcommand ["list", "ls"], "List service environment variables", load_subcommand('services/envs/list_command')
+    subcommand "add", "Add environment variable", load_subcommand('services/envs/add_command')
+    subcommand ["remove", "rm"], "Remove environment variable", load_subcommand('services/envs/remove_command')
+  end
+end

data/lib/kontena/cli/services/envs/add_command.rb

@@ -0,0 +1,21 @@
+require_relative '../services_helper'
+
+module Kontena::Cli::Services::Envs
+  class AddCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Kontena::Cli::Services::ServicesHelper
+
+    parameter "NAME", "Service name"
+    parameter "ENV", "Environment variable"
+
+    def execute
+      require_api_url
+      token = require_token
+      data = {env: env}
+      spinner "Adding env variable to #{pastel.cyan(name)} service " do
+        client(token).post("services/#{parse_service_id(name)}/envs", data)
+      end
+    end
+  end
+end

data/lib/kontena/cli/services/envs/list_command.rb

@@ -0,0 +1,22 @@
+require_relative '../services_helper'
+
+module Kontena::Cli::Services::Envs
+  class ListCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Kontena::Cli::Services::ServicesHelper
+    include Kontena::Cli::TableGenerator::Helper
+
+    parameter "NAME", "Service name"
+
+    requires_current_master
+    requires_current_master_token
+
+    def execute
+      service = client.get("services/#{parse_service_id(name)}")
+      service["env"].sort.each do |env|
+        puts quiet? ? env.split('=', 2).first : env
+      end
+    end
+  end
+end

data/lib/kontena/cli/services/envs/remove_command.rb

@@ -0,0 +1,22 @@
+require_relative '../services_helper'
+
+module Kontena::Cli::Services::Envs
+  class RemoveCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Kontena::Cli::Services::ServicesHelper
+
+    parameter "NAME", "Service name"
+    parameter "ENV ...", "Environment variable name", attribute_name: :envs
+
+    def execute
+      require_api_url
+      token = require_token
+      envs.each do |env|
+        spinner "Removing env variable #{pastel.cyan(env)} from #{pastel.cyan(name)} service " do
+          client(token).delete("services/#{parse_service_id(name)}/envs/#{env}")
+        end
+      end
+    end
+  end
+end

data/lib/kontena/cli/services/events_command.rb

@@ -0,0 +1,36 @@
+require_relative 'services_helper'
+require_relative '../helpers/log_helper'
+
+module Kontena::Cli::Services
+  class EventsCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Kontena::Cli::Helpers::LogHelper
+    include ServicesHelper
+
+    parameter "NAME", "Service name"
+
+    def execute
+      require_api_url
+
+      query_params = {}
+
+      titles = ['TIME', 'TYPE', 'MESSAGE']
+      puts "%-25s %-20s %s" % titles
+      show_logs("services/#{parse_service_id(name)}/event_logs", query_params) do |log|
+        show_log(log)
+      end
+    end
+
+    def show_log(log)
+      msg = log['message']
+      node = log['relationships'].find { |r| r['type'] == 'node' }
+      if node
+        msg = "#{msg} (#{node['id'].split('/')[-1]})"
+      end
+      puts '%-25s %-20s %s' % [
+        log['created_at'], log['type'].sub('service:'.freeze, ''.freeze), msg
+      ]
+    end
+  end
+end

data/lib/kontena/cli/services/exec_command.rb

@@ -0,0 +1,107 @@
+require 'shellwords'
+require 'json'
+require_relative 'services_helper'
+require_relative '../helpers/exec_helper'
+
+module Kontena::Cli::Services
+  class ExecCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Kontena::Cli::Helpers::ExecHelper
+    include ServicesHelper
+
+    class ExecExit < StandardError
+      attr_reader :exit_status
+
+      def initialize(exit_status, message = nil)
+        super(message)
+        @exit_status = exit_status
+      end
+    end
+
+    usage "[OPTIONS] [NAME] -- [CMD] ..."
+
+    parameter "NAME", "Service name"
+    parameter "CMD ...", "Command"
+
+    option ["--instance"], "INSTANCE", "Exec on given numbered instance, default first running" do |value| Integer(value) end
+    option ["-a", "--all"], :flag, "Exec on all running instances"
+    option ["--shell"], :flag, "Execute as a shell command", default: false
+    option ["-i", "--interactive"], :flag, "Keep stdin open", default: false
+    option ["-t", "--tty"], :flag, "Allocate a pseudo-TTY", default: false
+    option ["--skip"], :flag, "Skip failed instances when executing --all"
+    option ["--silent"], :flag, "Do not show exec status"
+    option ["--verbose"], :flag, "Show exec status"
+
+    requires_current_master
+    requires_current_grid
+
+    def execute
+      exit_with_error "the input device is not a TTY" if tty? && !STDIN.tty?
+      exit_with_error "--interactive cannot be used with --all" if all? && interactive?
+
+      service_containers = client.get("services/#{parse_service_id(name)}/containers")['containers']
+      service_containers.sort_by! { |container| container['instance_number'] }
+      running_containers = service_containers.select{|container| container['status'] == 'running' }
+
+      exit_with_error "Service #{name} does not have any running containers" if running_containers.empty?
+
+      if all?
+        ret = true
+        service_containers.each do |container|
+          if container['status'] == 'running'
+            if !execute_container(container)
+              ret = false
+            end
+          else
+            warning "Service #{name} container #{container['name']} is #{container['status']}, skipping"
+          end
+        end
+        return ret
+      elsif instance
+        if !(container = service_containers.find{|c| c['instance_number'] == instance})
+          exit_with_error "Service #{name} does not have container instance #{instance}"
+        elsif container['status'] != 'running'
+          exit_with_error "Service #{name} container #{container['name']} is not running, it is #{container['status']}"
+        else
+          execute_container(container)
+        end
+      else
+        execute_container(running_containers.first)
+      end
+    end
+
+    # Run block with spinner by default if --all, or when using --verbose.
+    # Do not use spinner if --silent.
+    def maybe_spinner(msg, &block)
+      if (all? || verbose?) && !silent?
+        spinner(msg, &block)
+      else
+        yield
+      end
+    end
+
+    # @param [Hash] container
+    # @raise [SystemExit] if exec exits with non-zero status, and not --skip
+    # @return [true] exit exit status zero
+    # @return [false] exit exit status non-zero and --skip
+    def execute_container(container)
+      maybe_spinner "Executing command on #{container['name']}" do
+        exit_status = container_exec(container['id'], self.cmd_list,
+          interactive: interactive?,
+          shell: shell?,
+          tty: tty?,
+        )
+        raise ExecExit.new(exit_status) unless exit_status.zero?
+      end
+    rescue ExecExit => exc
+      if skip?
+        return false
+      else
+        exit exc.exit_status
+      end
+    else
+      return true
+    end
+  end
+end

data/lib/kontena/cli/services/link_command.rb

@@ -0,0 +1,35 @@
+require_relative 'services_helper'
+
+module Kontena::Cli::Services
+  class LinkCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include ServicesHelper
+
+    parameter "NAME", "Service name"
+    parameter "TARGET", "Link target service name"
+
+    def execute
+      require_api_url
+      token = require_token
+      target_service = target
+
+      target_service = "null/#{target_service}" unless target_service.include?('/')
+
+      service = client(token).get("services/#{parse_service_id(name)}")
+      existing_targets = service['links'].map{ |l| l['id'].split('/', 2)[1] }
+      if existing_targets.include?(target_service.to_s)
+        exit_with_error("Service is already linked to #{target.to_s}")
+      end
+      links = service['links'].map{ |l|
+        { name: l['id'].split('/', 2)[1], alias: l['alias'] }
+      }
+      links << {name: target_service.to_s, alias: target.to_s}
+      links.compact!
+      data = {links: links}
+      spinner "Linking #{pastel.cyan(name)} to #{pastel.cyan(target)} " do
+        update_service(token, name, data)
+      end
+    end
+  end
+end

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

@@ -0,0 +1,66 @@
+require_relative 'services_helper'
+
+module Kontena::Cli::Services
+  class ListCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Kontena::Cli::TableGenerator::Helper
+    include ServicesHelper
+
+    option '--stack', 'STACK', 'Stack name'
+
+    requires_current_master
+    requires_current_master_token
+
+    def services
+      client.get("grids/#{current_grid}/services#{"?stack=#{stack}" if stack}")['services'].sort_by{|s| s['updated_at'] }.reverse
+    end
+
+    def fields
+      quiet? ? ['name'] : {name: 'name', instances: 'instances', stateful: 'stateful', state: 'state', "exposed ports" => 'ports' }
+    end
+
+    def service_port(port)
+      "#{port['ip']}:#{port['node_port']}->#{port['container_port']}/#{port['protocol']}"
+    end
+
+    def stack_id(service)
+      if quiet?
+        service.fetch('stack', {}).fetch('id', 'null')
+      else
+        service.fetch('stack', {}).fetch('name', nil)
+      end
+    end
+
+    def service_name(service)
+      stack_id = stack_id(service)
+      return service['name'] if stack_id == 'null'
+      [ stack_id(service), service['name'] ].compact.join('/')
+    end
+
+    def state_color(state)
+      case state
+      when 'running' then :green
+      when 'initialized' then :cyan
+      when 'stopped' then :red
+      else :blue
+      end
+    end
+
+    def execute
+      print_table(services) do |row|
+        row['name'] = quiet? ? service_name(row) :  health_status_icon(health_status(row)) + " " + service_name(row)
+        next if quiet?
+        row['stateful'] = row['stateful'] ? pastel.green('yes') : 'no'
+        row['ports'] = row['ports'].map(&method(:service_port)).join(',')
+        row['state'] = pastel.send(state_color(row['state']), row['state'])
+
+        instances = [row['instance_counts']['running'], row['instances']]
+        if instances.first < instances.last
+          instances[0] = pastel.cyan(instances[0].to_s)
+        end
+        row['instances'] = instances.join(' / ')
+      end
+    end
+  end
+end

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

@@ -0,0 +1,33 @@
+require_relative 'services_helper'
+require_relative '../helpers/log_helper'
+
+module Kontena::Cli::Services
+  class LogsCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Kontena::Cli::Helpers::LogHelper
+    include ServicesHelper
+
+    parameter "NAME", "Service name"
+    option ["-i", "--instance"], "INSTANCE", "Show only given instance specific logs"
+
+    def execute
+      require_api_url
+
+      query_params = {}
+      query_params[:instance] = instance if instance
+
+      show_logs("services/#{parse_service_id(name)}/container_logs", query_params) do |log|
+        show_log(log)
+      end
+    end
+
+    def show_log(log)
+      color = color_for_container(log['name'])
+      instance_number = log['name'].match(/^.+-(\d+)$/)[1]
+      name = instance_number.nil? ? log['name'] : instance_number
+      prefix = pastel.send(color, "#{log['created_at']} [#{name}]:")
+      puts "#{prefix} #{log['data']}"
+    end
+  end
+end

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

@@ -0,0 +1,58 @@
+require_relative 'services_helper'
+
+module Kontena::Cli::Services
+  class MonitorCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include ServicesHelper
+
+    parameter "NAME", "Service name"
+    option "--interval", "SECONDS", "How often view is refreshed", default: 2
+
+    def execute
+      require_api_url
+      token = require_token
+
+      loop do
+        nodes = {}
+        service = client(token).get("services/#{parse_service_id(name)}")
+        result = client(token).get("services/#{parse_service_id(name)}/containers")
+        result['containers'].each do |container|
+          nodes[container['node']['name']] ||= []
+          nodes[container['node']['name']] << container
+        end
+        clear_terminal
+        puts "service: #{name} (#{result['containers'].size}/#{service['instances']} instances)"
+        puts "strategy: #{service['strategy']}"
+        puts "status: #{service['state']}"
+        puts "stateful: #{service['stateful'] == true ? 'yes' : 'no' }"
+        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|
+            color = container['status']
+            if container['status'] == 'running'
+              color = :green
+            elsif container['status'] == 'killed'
+              color = :red
+            elsif container['status'] == 'stopped'
+              color = :bright_black
+            else
+              color = :yellow
+            end
+            print pastel.send(color, "■")
+          end
+          puts ''
+        end
+        sleep interval.to_f
+      end
+    end
+
+    def clear_terminal
+      print "\e[H\e[2J"
+    end
+  end
+end

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

@@ -0,0 +1,60 @@
+require_relative 'services_helper'
+
+module Kontena::Cli::Services
+  class RemoveCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include ServicesHelper
+
+    parameter "NAME ...", "Service name", attribute_name: :names
+    option "--instance", "INSTANCE", "Remove only given instance"
+    option "--force", :flag, "Force remove", default: false, attribute_name: :forced
+
+    banner "Remove a service"
+
+    requires_current_master
+    requires_current_master_token
+
+    def execute
+      names.each do |name|
+        if instance
+          remove_instance(name)
+        else
+          remove(name)
+        end
+      end
+    end
+
+    def remove(name)
+      confirm_command(name) unless forced?
+
+      spinner "Removing service #{pastel.cyan(name)} " do
+        client.delete("services/#{parse_service_id(name)}")
+        removed = false
+        until removed == true
+          sleep 1
+          begin
+            client.get("services/#{parse_service_id(name)}")
+          rescue Kontena::Errors::StandardError => exc
+            if exc.status == 404
+              removed = true
+            else
+              raise exc
+            end
+          end
+        end
+      end
+    end
+
+    def remove_instance(name)
+      instance_name = "#{name}/#{instance}"
+      confirm_command("#{name}/#{instance}") unless forced?
+      service_instance = client.get("services/#{parse_service_id(name)}/instances")['instances'].find{ |i|
+        i['instance_number'] == instance.to_i
+      }
+      exit_with_error("Instance not found") unless service_instance
+      spinner "Removing service instance #{pastel.cyan(instance_name)} " do
+        client.delete("services/#{parse_service_id(name)}/instances/#{service_instance['id']}")
+      end
+    end
+  end
+end

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

@@ -0,0 +1,19 @@
+require_relative 'services_helper'
+
+module Kontena::Cli::Services
+  class RestartCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include ServicesHelper
+
+    parameter "NAME", "Service name"
+
+    def execute
+      require_api_url
+      token = require_token
+      spinner "Sending restart signal to service #{pastel.cyan(name)} " do
+        restart_service(token, name)
+      end
+    end
+  end
+end

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

@@ -0,0 +1,21 @@
+require_relative 'services_helper'
+
+module Kontena::Cli::Services
+  class ScaleCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include ServicesHelper
+
+    parameter "NAME", "Service name"
+    parameter "INSTANCES", "Scales service to given number of instances"
+    option '--[no-]wait', :flag, 'Wait for service deployment', default: true
+
+    def execute
+      token = require_token
+      spinner "Scaling #{pastel.cyan(name)} to #{instances} instances " do
+        deployment = scale_service(token, name, instances)
+        wait_for_deploy_to_finish(token, deployment) if wait?
+      end
+    end
+  end
+end

data/lib/kontena/cli/services/secret_command.rb

@@ -0,0 +1,8 @@
+module Kontena::Cli::Services
+
+
+  class SecretCommand < Kontena::Command
+    subcommand "link", "Link secret from Vault", load_subcommand('services/secrets/link_command')
+    subcommand "unlink", "Unlink secret from Vault", load_subcommand('services/secrets/unlink_command')
+  end
+end