krates 1.6.0

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

@@ -0,0 +1,93 @@
+require_relative 'common'
+
+module Kontena::Cli::Stacks
+  class MonitorCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Common
+
+    banner "Monitor services in a stack"
+
+    parameter "NAME", "Stack name"
+    parameter "[SERVICES] ...", "Stack services to monitor", attribute_name: 'selected_services'
+
+    requires_current_master
+    requires_current_master_token
+
+    def execute
+      response = client.get("grids/#{current_grid}/services?stack=#{name}")
+      services = response['services']
+      if selected_services.size > 0
+        services.delete_if{ |s| !selected_services.include?(s['name'])}
+      end
+      show_monitor(services)
+    end
+
+    # @param [Array<Hash>]
+    def show_monitor(services)
+      loop do
+        nodes = {}
+        services.each do |service|
+          result = client.get("services/#{service['id']}/containers") rescue nil
+          service['instances'] = 0
+          if result
+            service['instances'] = result['containers'].size
+            result['containers'].each do |container|
+              container['service'] = service['name']
+              nodes[container['node']['name']] ||= []
+              nodes[container['node']['name']] << container
+            end
+          end
+        end
+        clear_terminal
+        puts "grid: #{current_grid}"
+        puts "stack: #{name}"
+        puts "services:"
+        services.each do |service|
+          color = color_for_service(service['name'])
+          puts "  #{pastel.send(color, "■")} #{service['name']} (#{service['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 pastel.send(color, icon)
+          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

data/lib/kontena/cli/stacks/registry/create_command.rb

@@ -0,0 +1,24 @@
+require_relative '../common'
+
+module Kontena::Cli::Stacks::Registry
+  class CreateCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::Stacks::Common
+    include Kontena::Cli::Stacks::Common::RegistryNameParam
+
+    banner "Creates a new Stack to Kontena Stack Registry"
+
+    option '--private', :flag, "Create as private", attribute_name: :is_private
+
+    requires_current_account_token
+
+    def execute
+      exit_with_error "Can't create a stack with a version number" unless stack_name.version.nil?
+      spinner "Creating #{is_private? ? pastel.yellow('private') : 'public'} stack #{pastel.cyan(stack_name)} in Kontena Stack Registry" do
+        stacks_client.create(stack_name, is_private: is_private?)
+      end
+    end
+  end
+end
+
+

data/lib/kontena/cli/stacks/registry/make_private_command.rb

@@ -0,0 +1,24 @@
+require_relative '../common'
+
+module Kontena::Cli::Stacks::Registry
+  class MakePrivateCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::Stacks::Common
+    include Kontena::Cli::Stacks::Common::RegistryNameParam
+
+    banner "Changes Stack visibility private in the Kontena Cloud Stack Registry"
+
+    option '--force', :flag, "Don't ask for confirmation"
+
+    requires_current_account_token
+
+    def execute
+      unless force?
+        confirm("Change stack #{pastel.cyan(stack_name)} visibility to private?")
+      end
+      spinner "Updating Stack #{pastel.cyan(stack_name)} visibility to private" do
+        stacks_client.make_private(stack_name)
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/registry/make_public_command.rb

@@ -0,0 +1,24 @@
+require_relative '../common'
+
+module Kontena::Cli::Stacks::Registry
+  class MakePublicCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::Stacks::Common
+    include Kontena::Cli::Stacks::Common::RegistryNameParam
+
+    banner "Changes Stack visibility private in the Kontena Cloud Stack Registry"
+
+    option '--force', :flag, "Don't ask for confirmation"
+
+    requires_current_account_token
+
+    def execute
+      unless force?
+        confirm("Change stack #{pastel.cyan(stack_name)} visibility to public?")
+      end
+      spinner "Updating Stack #{pastel.cyan(stack_name)} visibility to public" do
+        stacks_client.make_public(stack_name)
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/registry/pull_command.rb

@@ -0,0 +1,28 @@
+require_relative '../common'
+
+module Kontena::Cli::Stacks::Registry
+  class PullCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::Stacks::Common
+    include Kontena::Cli::Stacks::Common::RegistryNameParam
+
+    banner "Pulls / downloads a stack from the stack registry"
+
+    option ['-F', '--file'], '[FILENAME]', "Write to file (default STDOUT)"
+    option '--no-cache', :flag, "Don't use local cache"
+    option '--return', :flag, 'Return the result', hidden: true
+
+    def execute
+      target = no_cache? ? stacks_client : Kontena::StacksCache
+      content = target.pull(stack_name)
+      if return?
+        return content
+      elsif file
+        File.write(file, content)
+        puts pastel.green("Wrote #{content.bytesize} bytes to #{file}")
+      else
+        puts content
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/registry/push_command.rb

@@ -0,0 +1,40 @@
+require_relative '../common'
+
+module Kontena::Cli::Stacks::Registry
+  class PushCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::Stacks::Common
+
+    banner "Pushes (uploads) a stack to the stack registry"
+
+    parameter "[FILE]", "Kontena stack file path", default: "kontena.yml", attribute_name: :source
+
+    requires_current_account_token
+
+    option '--dry-run', :flag, "Do not perform any uploading", hidden: true
+
+    def includes_local_dependencies?(dependencies = loader.dependencies)
+      return false if dependencies.nil?
+      dependencies.any? { |dep| Kontena::Cli::Stacks::YAML::StackFileLoader.for(dep['stack']).origin == 'file' || includes_local_dependencies(dep['depends']) }
+    end
+
+    def includes_local_extends?
+      loader.yaml.fetch('services', {}).any? { |_, svc| svc.key?('extends') && svc['extends'].key?('file') }
+    end
+
+    def execute
+      exit_with_error "Can only perform push from local files" unless loader.origin == "file"
+      exit_with_error "Stack file contains dependencies to local files" if includes_local_dependencies?
+      exit_with_error "Stack file has services that extend from local files" if includes_local_extends?
+
+      spinner("Pushing #{pastel.cyan(source)} to stack registry as #{loader.stack_name}") do
+        unless dry_run?
+          stacks_client.push(
+            loader.stack_name,
+            loader.content
+          )
+        end
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/registry/remove_command.rb

@@ -0,0 +1,30 @@
+require_relative '../common'
+
+module Kontena::Cli::Stacks::Registry
+  class RemoveCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::Stacks::Common
+    include Kontena::Cli::Stacks::Common::RegistryNameParam
+
+    banner "Removes a stack (or version) from the stack registry. Use user/stack_name or user/stack_name:version."
+
+    option ['-f', '--force'], :flag, "Force delete"
+
+    requires_current_account_token
+
+    def execute
+      unless force?
+        if stack_name.version
+          puts "About to delete #{pastel.cyan("#{stack_name}")} from the stacks registry"
+          confirm
+        else
+          puts "About to delete an entire stack and all of its versions from the stacks registry"
+          confirm_command(stack_name)
+        end
+      end
+      spinner "Removing #{pastel.cyan(stack_name)} from the registry" do
+        stacks_client.destroy(stack_name)
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/registry/search_command.rb

@@ -0,0 +1,42 @@
+require_relative '../common'
+
+module Kontena::Cli::Stacks::Registry
+  class SearchCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::Stacks::Common
+    include Kontena::Cli::TableGenerator::Helper
+
+    banner "Search for stacks on the stack registry"
+
+    parameter "[QUERY]", "Query string"
+
+    option ['--[no-]pre'], :flag, "Include pre-release versions", default: true
+    option ['--[no-]private'], :flag, "Include private stacks", default: true, attribute_name: :priv
+
+    option ['--tag', '-t'], '[TAG]', "Search by tags", multivalued: true
+
+    option ['-q', '--quiet'], :flag, "Output the identifying column only"
+
+    def fields
+      quiet? ? ['name'] : %w(name version pulls description)
+    end
+
+    def execute
+      results = stacks_client.search(query.to_s, tags: tag_list, include_prerelease: pre?, include_private: priv?)
+      exit_with_error 'Nothing found' if results.empty?
+      print_table(results.map { |r| r['attributes'] }) do |row|
+        next if quiet?
+        row['name'] = '%s/%s' % [row['organization-id'], row['name']]
+        row['name'] = pastel.yellow(row['name']) if row['is-private']
+        if row['latest-version'] && row['latest-version']['version']
+          row['version'] = row['latest-version']['version']
+          row['description'] = row['latest-version']['description']
+        else
+          row['version'] = '?'
+        end
+
+        row['description'] = '-' if row['description'].to_s.empty?
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/registry/show_command.rb

@@ -0,0 +1,65 @@
+require_relative '../common'
+
+module Kontena::Cli::Stacks::Registry
+  class ShowCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::Stacks::Common
+    include Kontena::Cli::Stacks::Common::RegistryNameParam
+
+    banner "Shows information about a stack on the stacks registry"
+
+    option ['-v', '--versions'], :flag, "Only list available versions"
+
+    requires_current_account_token
+
+    def execute
+      versions = stacks_client.versions(stack_name)
+
+      if versions?
+        stacks_client.versions(stack_name).each do |version|
+          puts version['attributes']['version']
+        end
+      else
+        data = stacks_client.show(stack_name).dig('data', 'attributes')
+        puts "#{data['organization-id']}/#{data['name']}:"
+        puts "  description: #{data.dig('latest-version', 'description') || '-'}"
+        puts "  latest_version: #{data.dig('latest-version', 'version') || '-'}"
+        puts "  created_at: #{data.dig('created-at')}"
+        puts "  pulls: #{data.dig('pulls')}"
+        puts "  private: #{data.dig('is-private')}"
+        meta = data.dig('latest-version', 'meta')
+        if meta
+          puts "  meta:"
+          readme = meta.delete('readme')
+          meta_lines = YAML.dump(meta).split(/[\r\n]/)
+          meta_lines.shift
+          meta_lines.each do |meta_line|
+            puts "    %s" % meta_line
+          end
+          if readme
+            if readme =~ /^http\S+$/
+              puts "    readme: readme"
+            else
+              puts "    readme: |"
+              readme.gsub!(/(\S{#{70}})(?=\S)/, '\1 ')
+              readme.gsub!(/(.{1,#{70}})(?:\s+|$)/, "\\1\n")
+              readme.gsub!(/^/, '      ')
+              puts readme
+            end
+          end
+        else
+          puts "  meta: -"
+        end
+
+        if versions.empty?
+          puts "  versions: -"
+        else
+          puts "  versions:"
+          versions.each do |version|
+            puts "    - #{version['attributes']['version']} (#{version['attributes']['created-at']})"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kontena/cli/stacks/registry_command.rb

@@ -0,0 +1,12 @@
+module Kontena::Cli::Stacks
+  class RegistryCommand < Kontena::Command
+    subcommand "push", "Push a stack into the stacks registry", load_subcommand('stacks/registry/push_command')
+    subcommand "pull", "Pull a stack from the stacks registry", load_subcommand('stacks/registry/pull_command')
+    subcommand ["search"], "Search for stacks in the stacks registry", load_subcommand('stacks/registry/search_command')
+    subcommand "show", "Show info about a stack in the stacks registry", load_subcommand('stacks/registry/show_command')
+    subcommand ["remove", "rm"], "Remove a stack (or version) from the stacks registry", load_subcommand('stacks/registry/remove_command')
+    subcommand "create", "Create a stack in the registry", load_subcommand('stacks/registry/create_command')
+    subcommand "make-private", "Change Stack visibility to private", load_subcommand('stacks/registry/make_private_command')
+    subcommand "make-public", "Change Stack visibility to public", load_subcommand('stacks/registry/make_public_command')
+  end
+end

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

@@ -0,0 +1,80 @@
+require_relative 'common'
+
+module Kontena::Cli::Stacks
+  class RemoveCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Common
+
+    banner "Removes a stack in a grid on Kontena Master"
+
+    parameter "NAME ...", "Stack name", attribute_name: :names
+    option "--force", :flag, "Force remove", default: false, attribute_name: :forced
+    option "--keep-dependencies", :flag, "Do not remove dependencies"
+
+    requires_current_master
+    requires_current_master_token
+
+    def execute
+      names.each do |name|
+        stack = fetch_stack(name)
+        confirm_remove(stack, name) unless forced?
+        unless keep_dependencies?
+          stack.fetch('children', Hash.new).each do |child_stack|
+            caret"Removing dependency #{pastel.cyan(child_stack['name'])}"
+            Kontena.run!(['stack', 'remove', '--force', child_stack['name']])
+          end
+        end
+
+        spinner "Removing stack #{pastel.cyan(name)} " do
+          remove_stack(name)
+          wait_stack_removal(name)
+        end
+      end
+    end
+
+    # @param stack [Hash]
+    # @param name [String]
+    def confirm_remove(stack, name)
+      if stack['parent']
+        puts "#{pastel.yellow('Warning:')} The stack #{pastel.cyan(stack['parent']['name'])} depends on stack #{name}"
+      end
+      if !keep_dependencies? && stack['children'] && !stack['children'].empty?
+        puts "#{pastel.yellow('Warning:')} The stack #{pastel.cyan(name)} has dependencies that will be removed:"
+        stack['children'].each do |child|
+          puts "- #{pastel.yellow(child['name'])}"
+        end
+      end
+      confirm_command(name)
+    end
+
+    # @param name [String]
+    # @return [Hash]
+    def fetch_stack(name)
+      client.get("stacks/#{current_grid}/#{name}")
+    end
+
+    # @param name [String]
+    # @return [Hash]
+    def remove_stack(name)
+      client.delete("stacks/#{current_grid}/#{name}")
+    end
+
+    # @param name [String]
+    def wait_stack_removal(name)
+      removed = false
+      until removed == true
+        begin
+          client.get("stacks/#{current_grid}/#{name}")
+          sleep 1
+        rescue Kontena::Errors::StandardError => exc
+          if exc.status == 404
+            removed = true
+          else
+            raise exc
+          end
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,24 @@
+require_relative 'common'
+
+module Kontena::Cli::Stacks
+  class RestartCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Common
+
+    banner "Restarts all services of a stack that has been installed in a grid on Kontena Master"
+
+    parameter "NAME ...", "Stack name", attribute_name: :names
+
+    requires_current_master
+    requires_current_master_token
+
+    def execute
+      names.each do |name|
+        spinner "Sending restart signal for stack #{name} services" do
+          client.post("stacks/#{current_grid}/#{name}/restart", {})
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,131 @@
+require 'yaml'
+require 'shellwords'
+require_relative '../services/services_helper'
+
+module Kontena::Cli::Stacks
+  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'] || options['env']
+      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['shm_size'] = parse_memory(options['shm_size'].to_s) if options['shm_size']
+      data['cpus'] = options['cpus'] if options['cpus']
+      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 = {
+        'wait_for_port' => deploy_opts['wait_for_port'],
+        'min_health' => deploy_opts['min_health']
+      }
+      if deploy_opts.has_key?('interval')
+        deploy['interval'] = parse_relative_time(deploy_opts['interval'])
+      else
+        deploy['interval'] = nil
+      end
+      data['deploy_opts'] = deploy
+      data['hooks'] = options['hooks'] || {}
+      data['secrets'] = options['secrets'] if options['secrets']
+      data['certificates'] = options['certificates'] if options['certificates']
+      data['build'] = parse_build_options(options) if options['build']
+      data['health_check'] = parse_health_check(options)
+      data['stop_signal'] = options['stop_signal'] if options['stop_signal']
+      data['stop_grace_period'] = options['stop_grace_period'] if options['stop_grace_period']
+      data['read_only'] = options['read_only'] || false
+      data['entrypoint'] = options['entrypoint'] if options['entrypoint']
+      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|
+        if l.kind_of?(String)
+          service_name, alias_name = l.split(':')
+        elsif l.kind_of?(Hash)
+          service_name = l['name']
+          alias_name = l['alias']
+        else
+          raise TypeError, "Invalid link type #{l.class.name}, expecting String or Hash"
+        end
+        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
+
+    # @param [Hash] options
+    # @return [Hash]
+    def parse_health_check(options)
+      health_check = {}
+      %w(port protocol uri timeout interval initial_delay).each do |k|
+        health_check[k] = options.dig('health_check', k)
+      end
+      health_check
+    end
+  end
+end

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

@@ -0,0 +1,27 @@
+require 'yaml'
+require_relative 'service_generator'
+
+module Kontena::Cli::Stacks
+  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