krates 1.6.0

data/lib/kontena/cli/log_formatters/strip_color.rb

@@ -0,0 +1,13 @@
+require 'logger'
+
+module Kontena
+  module Cli
+    module LogFormatter
+      class StripColor < Logger::Formatter
+        def msg2str(msg)
+          super(msg.kind_of?(String) ? msg.gsub(/\e+\[{1,2}[0-9;:?]+m/m, '') : msg)
+        end
+      end
+    end
+  end
+end

data/lib/kontena/cli/logout_command.rb

@@ -0,0 +1,10 @@
+class Kontena::Cli::LogoutCommand < Kontena::Command
+  include Kontena::Cli::Common
+
+  banner "Command removed, use 'kontena master logout' to log out of the Kontena Master"
+  banner "or 'kontena cloud logout' to log out of the Kontena Cloud", false
+
+  def execute
+    exit_with_error("Command removed. Use #{pastel.yellow("kontena master logout")} to log out of the Kontena Master")
+  end
+end

data/lib/kontena/cli/master/audit_log_command.rb

@@ -0,0 +1,19 @@
+module Kontena::Cli::Master
+  class AuditLogCommand < Kontena::Command
+    include Kontena::Cli::Common
+
+    option ["-l", "--lines"], "LINES", "Number of lines"
+
+    requires_current_master
+    requires_current_master_token
+
+    def execute
+      audit_logs = client.get("audit_logs", {limit: lines})
+      puts '%-30.30s %-10s %-15s %-25s %-15s %-25s %-15s %-15s' % ['Time', 'Grid', 'Resource Type', 'Resource Name', 'Event Name', 'User', 'Source IP', 'User-Agent']
+      audit_logs['logs'].each do |log|
+        puts '%-30.30s %-10s %-15s %-25s %-15s %-25s %-15s %-15s' % [ log['time'], log['grid'], log['resource_type'], log['resource_name'], log['event_name'], log['user_identity']['email'], log['source_ip'], log['user_agent']]
+      end
+    end
+  end
+end
+

data/lib/kontena/cli/master/config/export_command.rb

@@ -0,0 +1,46 @@
+module Kontena::Cli::Master::Config
+  class ExportCommand < Kontena::Command
+
+    include Kontena::Cli::Common
+
+    requires_current_master
+    requires_current_master_token
+
+    banner "Reads configuration from master"
+
+    parameter '[PATH]', "Output to file in PATH, default: STDOUT", required: false
+    option ['-f', '--format'], '[FORMAT]', "Specify output format (json, yaml) (default: guess from PATH or json)"
+
+    option ['--filter'], "[FILTER]", "Filter keys, example: oauth2.*"
+
+    def decorate(data)
+      case self.format.downcase
+      when 'json'
+        require 'json'
+        JSON.pretty_generate(data)
+      when 'yaml', 'yml'
+        require 'yaml'
+        YAML.dump(data)
+      else
+        exit_with_error "Unknown output format '#{self.format}'"
+      end
+    end
+
+    def output(content)
+      self.path ? File.write(self.path, content) : puts(content)
+    end
+
+    def data
+      client.get("config", self.filter ? { filter: self.filter } : nil)
+    end
+
+    def set_default_format
+      self.format ||= self.path.to_s.end_with?('.yml') ? 'yaml' : 'json'
+    end
+
+    def execute
+      set_default_format
+      output(decorate(data))
+    end
+  end
+end

data/lib/kontena/cli/master/config/get_command.rb

@@ -0,0 +1,26 @@
+module Kontena::Cli::Master::Config
+  class GetCommand < Kontena::Command
+
+    include Kontena::Cli::Common
+
+    requires_current_master
+    requires_current_master_token
+
+    banner "Reads a configuration value from master"
+
+    parameter "KEY", "Configuration key to read from master", required: true
+
+    option ['-p', '--pair'], :flag, "Print key=value instead of only value"
+
+    def response
+      client.get("config/#{key}")
+    end
+
+    def execute
+      value = response[key]
+      print(key + '=') if pair?
+      puts value
+    end
+  end
+end
+

data/lib/kontena/cli/master/config/import_command.rb

@@ -0,0 +1,67 @@
+module Kontena::Cli::Master::Config
+  class ImportCommand < Kontena::Command
+
+    include Kontena::Cli::Common
+
+    requires_current_master
+    requires_current_master_token
+
+    banner "Updates configuration from a file into Master"
+
+    parameter '[PATH]', "Input from file in PATH (default: STDIN)", required: false
+
+    option ['--preset'], '[NAME]', 'Load preset', hidden: true
+
+    option ['--format'], '[FORMAT]', "Specify input format (json, yaml) (default: guess from PATH or json)"
+    option ['--full'], :flag, "Perform full update, keys that are not present in the input are cleared"
+    option ['-f', '--force'], :flag, "Don't ask for confirmation"
+
+
+    def input_as_hash
+      if self.path && self.preset
+        exit_with_error "Options --preset and PATH can not be used together"
+      elsif self.path
+        unless File.exist?(self.path) && File.readable?(self.path)
+          exit_with_error "Can not read '#{self.path}'"
+        end
+        File.read(self.path)
+      elsif self.preset
+        self.format = 'yaml'
+        path = File.join(Kontena.root, 'lib/kontena/presets', "#{self.preset}.yml")
+        File.read(path)
+      else
+        stdin_input("Enter master configuration as #{format.upcase}", :multiline)
+      end
+    end
+
+    def convert(data)
+      case self.format.downcase
+      when 'json'
+        require 'json'
+        JSON.parse(data)
+      when 'yaml', 'yml'
+        ::YAML.safe_load(data, [], [], true)
+      else
+        exit_with_error "Unknown input format '#{self.format}'"
+      end
+    end
+
+    def http_method
+      self.full? ? :patch : :put
+    end
+
+    def upload(data)
+      confirm unless self.force?
+      client.send(http_method, "config", data)
+    end
+
+    def set_default_format
+      self.format ||= self.path.to_s.end_with?('.yml') ? 'yaml' : 'json'
+    end
+
+    def execute
+      set_default_format
+      upload(convert(input_as_hash))
+    end
+  end
+end

data/lib/kontena/cli/master/config/set_command.rb

@@ -0,0 +1,19 @@
+module Kontena::Cli::Master::Config
+  class SetCommand < Kontena::Command
+
+    include Kontena::Cli::Common
+
+    requires_current_master
+    requires_current_master_token
+
+    banner "Sets a configuration value to Master"
+
+    parameter "KEY_VALUE_PAIR ...", "Key/value pair, for example server.root_url=http://example.com", required: true
+
+    def execute
+      data = self.key_value_pair_list.map{ |p| p.split('=') }.to_h
+      client.patch('config', data)
+    end
+  end
+end
+

data/lib/kontena/cli/master/config/unset_command.rb

@@ -0,0 +1,20 @@
+module Kontena::Cli::Master::Config
+  class UnsetCommand < Kontena::Command
+
+    include Kontena::Cli::Common
+
+    requires_current_master
+    requires_current_master_token
+
+    parameter "KEY ...", "Key(s) to unset", required: true
+
+    banner "Clears a configuration value from Master"
+
+    def execute
+      self.key_list.each do |key|
+        client.delete("config/#{key}")
+      end
+    end
+  end
+end
+

data/lib/kontena/cli/master/config_command.rb

@@ -0,0 +1,17 @@
+
+module Kontena
+  module Cli
+    module Master
+      class ConfigCommand < Kontena::Command
+        subcommand "set", "Set a config value", load_subcommand('master/config/set_command')
+        subcommand "get", "Get a config value", load_subcommand('master/config/get_command')
+        subcommand "unset", "Clear a config value", load_subcommand('master/config/unset_command')
+        subcommand ["load", "import"], "Upload config to Master", load_subcommand('master/config/import_command')
+        subcommand ["dump", "export"], "Download config from Master", load_subcommand('master/config/export_command')
+
+        def execute
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,74 @@
+module Kontena::Cli::Master
+  class CreateCommand < Kontena::Command
+    include Kontena::Cli::Common
+
+    callback_matcher :master, :create_with_plugin_select
+
+    def subcommand_tree(command = nil)
+      command ||= Kontena::MainCommand
+
+      real_command = command.respond_to?(:subcommand_class) ? command.subcommand_class : command
+
+      tree = {}
+      real_command.recognised_subcommands.each do |sub_command|
+        sub_command.names.each do |command_name|
+          if sub_command.subcommand_class.has_subcommands?
+            tree[command_name] = subcommand_tree(sub_command)
+          else
+            tree[command_name] = sub_command.subcommand_class
+          end
+        end
+      end
+      tree
+    end
+
+    def master_create_subcommands(tree)
+      creators = []
+      tree.each do |k,cmd|
+        if cmd.kind_of?(Hash)
+          creators += master_create_subcommands(cmd)
+        elsif cmd.respond_to?(:callback_matcher) && cmd.callback_matcher == [:master, :create]
+          creators << cmd
+        end
+      end
+      creators
+    end
+
+    def execute
+
+      require 'shellwords'
+
+      tree = master_create_subcommands(subcommand_tree)
+
+      if tree.empty?
+        exit_with_error "Install platform plugins first, use: kontena plugin"
+      end
+      cmd_class = prompt.select("Choose target platform") do |menu|
+        tree.each do |cmd_class|
+          plugin_name = cmd_class.to_s[/Plugin::(.+?)::/, 1]
+          next unless plugin_name
+          menu.choice plugin_name, cmd_class
+        end
+      end
+      skip_options = ['--no-prompt', '--silent', '--help', '--version']
+      options = []
+      cmd_class.recognised_options.each do |option|
+        next if option.switches.any?{ |sw| skip_options.include?(sw) }
+        if option.type == :flag
+          answer = prompt.yes?(option.description)
+          if answer
+            options << option.switches.first
+          end
+        else
+          answer = prompt.ask("#{option.description}: ", required: option.required?, default: option.default_value)
+          if answer
+            options << "#{option.switches.first} #{answer.shellescape}"
+          end
+        end
+      end
+      cmd = [cmd_class.to_s[/Plugin::(.+?)::/, 1].downcase, 'master', 'create'] + options
+      Kontena.run!(cmd)
+    end
+  end
+
+end

data/lib/kontena/cli/master/current_command.rb

@@ -0,0 +1,25 @@
+require 'uri'
+
+module Kontena::Cli::Master
+  class CurrentCommand < Kontena::Command
+    include Kontena::Cli::Common
+
+    option ["--name"], :flag, "Show name only", default: false
+    option ["--address"], :flag, "Show IP address or FQDN only", default: false
+    option ["--url"], :flag, "Show URL only", default: false
+
+    def execute
+      master = require_current_master
+
+      if name?
+        puts master['name']
+      elsif address?
+        puts URI.parse(master['url']).host
+      elsif url?
+        puts master['url']
+      else
+        puts "#{master['name']} #{master['url']}"
+      end
+    end
+  end
+end

data/lib/kontena/cli/master/init_cloud_command.rb

@@ -0,0 +1,45 @@
+module Kontena::Cli::Master
+  class InitCloudCommand < Kontena::Command
+
+    include Kontena::Cli::Common
+
+    banner "Configures the current Kontena Master to use Kontena Cloud services and authentication"
+
+    option '--force',           :flag,       "Don't ask questions"
+    option '--cloud-master-id', '[ID]',      "Use existing cloud master ID"
+    option '--provider',        '[NAME]',    "Set master provider name"
+    option '--version',         '[VERSION]', "Set master version"
+
+    requires_current_master
+    requires_current_master_token
+    requires_current_account_token
+
+    def master_config
+      @master_config ||= client.get('config')
+    end
+
+    def current_master_cloud_config
+      return @cloud_config if @cloud_config
+      master_client_id = master_config['oauth2.client_id']
+      @cloud_config = cloud_client.get('user/masters')['data'].find { |cm| cm['attributes']['client-id'] == master_client_id }
+    end
+
+    def current_master_cloud_name
+      @cloud_name ||= current_master_cloud_config.nil? ? nil : current_master_cloud_config['attributes']['name']
+    end
+
+    def already_cloud_enabled?
+      !current_master_cloud_config.nil?
+    end
+
+    def execute
+      exit_with_error "Current master is already registered to use Kontena Cloud as #{pastel.cyan(current_master_cloud_name)}" if already_cloud_enabled?
+      args = ["--current"]
+      args << "--force" if self.force?
+      args += ["--cloud-master-id", self.cloud_master_id.shellescape] if self.cloud_master_id
+      args += ["--provider", self.provider.shellescape] if self.provider
+      args += ["--version", self.version.shellescape] if self.version
+      Kontena.run!(['cloud', 'master', 'add'] + args)
+    end
+  end
+end

data/lib/kontena/cli/master/join_command.rb

@@ -0,0 +1,22 @@
+module Kontena::Cli::Master
+  class JoinCommand < Kontena::Command
+    parameter "URL", "Kontena Master URL or name"
+    parameter "INVITE_CODE", "Invitation code"
+
+    option ['-n', '--name'], '[NAME]', 'Set server name'
+    option ['-r', '--remote'], :flag, 'Do not try to open a browser'
+    option ['-v', '--verbose'], :flag, 'Increase output verbosity'
+
+    def execute
+      params = []
+      params += ["--join", self.invite_code]
+      params << "--remote" if self.remote?
+      params += ["--name", self.name] if self.name
+      params << "--verbose" if self.verbose?
+
+      cmd = ['master', 'login'] + params
+      cmd << url
+      Kontena.run!(cmd)
+    end
+  end
+end

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

@@ -0,0 +1,24 @@
+module Kontena::Cli::Master
+  class ListCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::TableGenerator::Helper
+
+    def fields
+      @fields ||= quiet? ? %w(name) : %w(name url)
+    end
+
+    def current_master_name
+      @current_master_name ||= current_master.nil? ? nil : current_master.name
+    end
+
+    def mark_if_current(row)
+      unless quiet?
+        row.name.to_s.insert(0, pastel.yellow('* ')) if row.name == current_master_name
+      end
+    end
+
+    def execute
+      print_table(config.servers, fields, &method(:mark_if_current))
+    end
+  end
+end