krates 1.6.0

data/lib/kontena/cli/master/token/show_command.rb

@@ -0,0 +1,34 @@
+require_relative 'common'
+
+module Kontena::Cli::Master::Token
+  class ShowCommand < Kontena::Command
+
+    parameter "TOKEN_OR_ID", "Access token or access token id"
+
+    include Kontena::Cli::Common
+    include Common
+
+    requires_current_master
+    requires_current_master_token
+
+    option '--id', :flag, "Only output access token id", hidden: true
+
+    def execute
+      data = client.get("/oauth2/tokens/#{token_or_id}")
+      output = token_data_to_hash(data)
+
+      if id?
+        puts output[:id]
+        return
+      end
+
+      id = output.delete(:id)
+      puts '%s:' % id
+      output.each do |key, value|
+        puts "  %s: %s" % [key, value.nil? ? '-' : value]
+      end
+    end
+  end
+end
+
+

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

@@ -0,0 +1,13 @@
+
+module Kontena::Cli::Master
+  class TokenCommand < Kontena::Command
+    subcommand ["list", "ls"], "List access tokens", load_subcommand('master/token/list_command')
+    subcommand ["rm", "remove"], "Remove / revoke an access token", load_subcommand('master/token/remove_command')
+    subcommand "show", "Display access token", load_subcommand('master/token/show_command')
+    subcommand "current", "Display current access token", load_subcommand('master/token/current_command')
+    subcommand "create", "Generate an access token", load_subcommand('master/token/create_command')
+
+    def execute
+    end
+  end
+end

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

@@ -0,0 +1,31 @@
+module Kontena::Cli::Master
+  class UseCommand < Kontena::Command
+    include Kontena::Cli::Common
+
+    parameter "[NAME]", "Master name to use"
+
+    option '--clear', :flag, "Clear current master setting"
+
+    def execute
+      if clear?
+        config.current_master = nil
+        config.write
+        exit 0
+      elsif name.nil?
+        signal_usage_error Clamp.message(:parameter_argument_error, :param => 'NAME', :message => "missing")
+        exit 1
+      end
+
+      master = config.find_server(name)
+      if master.nil?
+        exit_with_error "Could not resolve master by name '#{name}'. For a list of known masters please run: kontena master list"
+      else
+        config.current_master = master['name']
+        config.write
+        puts "Using master: #{pastel.cyan(master['name'])} (#{master['url']})"
+        puts "Using grid: #{current_grid ? pastel.cyan(current_grid) : "<none>"}"
+      end
+    end
+  end
+
+end

data/lib/kontena/cli/master/user/invite_command.rb

@@ -0,0 +1,51 @@
+require_relative '../../common'
+require_relative 'role/add_command'
+
+module Kontena::Cli::Master::User
+  class InviteCommand < Kontena::Command
+    include Kontena::Cli::Common
+
+    parameter "EMAIL ...", "List of emails"
+
+    option ['-r', '--roles'], '[ROLES]', 'Comma separated list of roles to assign to the invited users'
+    option ['-c', '--code'], :flag, 'Only output the invite code'
+    option '--external-id', '[EXTERNAL ID]', 'Assign external id to user', hidden: true
+    option '--return', :flag, 'Return the code', hidden: true
+
+    requires_current_master
+    requires_current_master_token
+
+    def execute
+      if self.roles
+        roles = self.roles.split(',')
+      else
+        roles = []
+      end
+      external_id = nil
+      if email_list.size == 1 && self.external_id
+        external_id = self.external_id
+      end
+      email_list.each do |email|
+        begin
+          data = { email: email, external_id: external_id, response_type: 'invite' }
+          response = client.post('/oauth2/authorize', data)
+          if self.code?
+            puts response['invite_code']
+          elsif self.return?
+            return response
+          else
+            puts pastel.green("Invitation created for #{response['email']}")
+            puts "  * code:    #{response['invite_code']}"
+            puts "  * command: kontena master join #{current_master.url} #{response['invite_code']}"
+          end
+          roles.each do |role|
+            raise "Failed to add role" unless Kontena.run(["master", "user", "role", "add", role, email])
+          end
+        rescue => ex
+          logger.error(ex)
+          exit_with_error "Failed to invite #{email} : #{ex.message}"
+        end
+      end
+    end
+  end
+end

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

@@ -0,0 +1,29 @@
+require_relative '../../common'
+
+module Kontena::Cli::Master::User
+  class ListCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::TableGenerator::Helper
+
+    requires_current_master
+    requires_current_master_token
+
+    def fields
+      quiet? ? ['id'] : { email: 'email', roles: 'role_list' }
+    end
+
+    def execute
+      response = spin_if(!quiet? && $stdout.tty?, "Retrieving user list from Kontena Master") do
+        client.get('users')['users']
+      end
+
+      print_table(response) do |row|
+        if row['roles'].empty?
+          row['role_list'] = ''
+        else
+          row['role_list'] = row['roles'].map { |r| r['name'] }.join(',')
+        end
+      end
+    end
+  end
+end

data/lib/kontena/cli/master/user/remove_command.rb

@@ -0,0 +1,24 @@
+require_relative '../../common'
+
+module Kontena::Cli::Master::User
+  class RemoveCommand < Kontena::Command
+    include Kontena::Cli::Common
+
+    parameter "EMAIL ...", "List of emails"
+    option "--force", :flag, "Force remove", default: false, attribute_name: :forced
+
+    def execute
+      require_api_url
+      token = require_token
+      confirm unless forced?
+
+      email_list.each do |email|
+        begin
+          client(token).delete("users/#{email}")
+        rescue => ex
+          $stderr.puts pastel.red("Failed to remove user #{email} : #{ex.message}")
+        end
+      end
+    end
+  end
+end

data/lib/kontena/cli/master/user/role/add_command.rb

@@ -0,0 +1,29 @@
+require_relative '../../../common'
+
+module Kontena::Cli::Master::User
+  module Role
+    class AddCommand < Kontena::Command
+      include Kontena::Cli::Common
+
+      parameter "ROLE", "Role name"
+      parameter "USER ...", "List of users"
+
+      option '--silent', :flag, 'Reduce output verbosity'
+
+      def execute
+        require_api_url
+        token = require_token
+        data = { role: role }
+
+        user_list.each do |email|
+          begin
+            response = client(token).post("users/#{email}/roles", data)
+            puts "Added role #{role} to #{email}" unless running_silent?
+          rescue => ex
+            abort pastel.red("Failed to add role #{role} to #{email} : #{ex.message}")
+          end
+        end
+      end
+    end
+  end
+end

data/lib/kontena/cli/master/user/role/remove_command.rb

@@ -0,0 +1,27 @@
+require_relative '../../../common'
+
+module Kontena::Cli::Master::User::Role
+  class RemoveCommand < Kontena::Command
+    include Kontena::Cli::Common
+
+    parameter "ROLE", "Role name"
+    parameter "USER ...", "List of users"
+    option "--force", :flag, "Force remove", default: false, attribute_name: :forced
+
+
+    def execute
+      require_api_url
+      token = require_token
+      confirm unless forced?
+
+      user_list.each do |email|
+        begin
+          response = client(token).delete("users/#{email}/roles/#{role}")
+          puts "Removed role #{role} from #{email}" if response
+        rescue => ex
+          $stderr.puts pastel.red("Failed to remove role #{role} from #{email} : #{ex.message}")
+        end
+      end
+    end
+  end
+end

data/lib/kontena/cli/master/user/role_command.rb

@@ -0,0 +1,6 @@
+module Kontena::Cli::Master::User
+  class RoleCommand < Kontena::Command
+    subcommand "add", "Add role to user", load_subcommand('master/user/role/add_command')
+    subcommand ["remove", "rm"], "Remove role from user", load_subcommand('master/user/role/remove_command')
+  end
+end

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

@@ -0,0 +1,9 @@
+module Kontena::Cli::Master
+  class UserCommand < Kontena::Command
+    subcommand "invite", "Invite user to Kontena Master", load_subcommand('master/user/invite_command')
+    subcommand ["remove", "rm"], "Remove user from Kontena Master", load_subcommand('master/user/remove_command')
+    subcommand ["list", "ls"], "List users", load_subcommand('master/user/list_command')
+    subcommand "role", "User role specific commands", load_subcommand('master/user/role_command')
+  end
+end
+

data/lib/kontena/cli/master_command.rb

@@ -0,0 +1,21 @@
+class Kontena::Cli::MasterCommand < Kontena::Command
+  include Kontena::Util
+
+  subcommand ["list", "ls"], "List masters where client has logged in", load_subcommand('master/list_command')
+  subcommand ["remove", "rm"], "Remove a master from configuration", load_subcommand('master/remove_command')
+  subcommand ["config", "cfg"], "Configure master settings", load_subcommand('master/config_command')
+  subcommand "use", "Switch to use selected master", load_subcommand('master/use_command')
+  subcommand "user", "User specific commands", load_subcommand('master/user_command')
+  subcommand "current", "Show current master details", load_subcommand('master/current_command')
+  subcommand "login", "Authenticate to Kontena Master", load_subcommand('master/login_command')
+  subcommand "logout", "Log out of Kontena Master", load_subcommand('master/logout_command')
+  subcommand "token", "Manage Kontena Master access tokens", load_subcommand('master/token_command')
+  subcommand "join", "Join Kontena Master using an invitation code", load_subcommand('master/join_command')
+  subcommand "audit-log", "Show master audit logs", load_subcommand('master/audit_log_command')
+  subcommand "create", "Install a new Kontena Master", load_subcommand('master/create_command') if experimental?
+  subcommand "init-cloud", "Configure current master to use Kontena Cloud services", load_subcommand('master/init_cloud_command')
+  subcommand "ssh", "Connect to the master via SSH", load_subcommand('master/ssh_command')
+
+  def execute
+  end
+end

data/lib/kontena/cli/node_command.rb

@@ -0,0 +1,17 @@
+class Kontena::Cli::NodeCommand < Kontena::Command
+
+  subcommand ["list","ls"], "List grid nodes", load_subcommand('nodes/list_command')
+  subcommand "show", "Show node", load_subcommand('nodes/show_command')
+  subcommand "ssh", "Ssh into node", load_subcommand('nodes/ssh_command')
+  subcommand "create", "Create node", load_subcommand('nodes/create_command')
+  subcommand "update", "Update node", load_subcommand('nodes/update_command')
+  subcommand "reset-token", "Reset node token for agent websocket connection", load_subcommand('nodes/reset_token_command')
+  subcommand ["remove","rm"], "Remove node", load_subcommand('nodes/remove_command')
+  subcommand "label", "Node label specific commands", load_subcommand('nodes/label_command')
+  subcommand "health", "Check node health", load_subcommand('nodes/health_command')
+  subcommand "env", "Generate kontena-agent.env configuration", load_subcommand('nodes/env_command')
+
+
+  def execute
+  end
+end

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

@@ -0,0 +1,25 @@
+module Kontena::Cli::Nodes
+  class CreateCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+
+    requires_current_master
+    requires_current_master_token
+    requires_current_grid
+
+    parameter "NAME", "Node name"
+    option ["--token"], "TOKEN", "Node token"
+    option ["-l", "--label"], "LABEL", "Node label", multivalued: true
+
+    def execute
+      data = { name: name }
+
+      data[:token] = token if token
+      data[:labels] = label_list
+
+      spinner "Creating #{pastel.cyan(name)} node " do
+        client.post("grids/#{current_grid}/nodes", data)
+      end
+    end
+  end
+end

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

@@ -0,0 +1,37 @@
+module Kontena::Cli::Nodes
+  class EnvCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+
+    requires_current_master
+    requires_current_master_token
+    requires_current_grid
+
+    parameter "NAME", "Node name"
+    option ['--token'], :flag, 'Only show token', default: false
+
+    def grid_uri
+      grid_uri = self.current_master['url'].sub('http', 'ws')
+    end
+
+    def get_node_token
+      return client.get("nodes/#{current_grid}/#{name}/token")
+    rescue Kontena::Errors::StandardError => exc
+      raise unless exc.status == 404
+      return nil
+    end
+
+    def execute
+      unless token_node = get_node_token()
+        exit_with_error "Node #{name} was not created with a node token. Use `kontena grid env` instead"
+      end
+
+      if self.token?
+        puts token_node['token']
+      else
+        puts "KONTENA_URI=#{grid_uri}"
+        puts "KONTENA_NODE_TOKEN=#{token_node['token']}"
+      end
+    end
+  end
+end

data/lib/kontena/cli/nodes/health_command.rb

@@ -0,0 +1,47 @@
+require 'kontena/cli/helpers/health_helper'
+require 'kontena/cli/helpers/time_helper'
+
+module Kontena::Cli::Nodes
+  class HealthCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Kontena::Cli::Helpers::HealthHelper
+    include Kontena::Cli::Helpers::TimeHelper
+
+    parameter "NODE", "Node name"
+
+    requires_current_master
+    requires_current_grid
+
+    def execute
+      return show_node_health("#{current_grid}/#{self.node}")
+    end
+
+    # @param id [String] :grid/:node
+    # @return [Boolean] true if healthy
+    def show_node_health(id)
+      node_health = client.get("nodes/#{id}/health")
+
+      if node_health['status'] == 'online'
+        puts "#{health_icon(:ok)} Node is online for #{time_since(node_health['connected_at'])}"
+      else
+        puts "#{health_icon(:warning)} Node is #{node_health['status']}"
+      end
+
+      etcd_health, etcd_status = node_etcd_health(node_health['etcd_health'])
+
+      puts "#{health_icon etcd_health} Node #{node_health['name']} etcd is #{etcd_status}"
+
+      return etcd_health == :ok
+
+    rescue Kontena::Errors::StandardErrorHash => exc
+      raise unless exc.status == 422
+
+      exc.errors.each do |what, error|
+        puts "#{health_icon :offline} Node #{id} #{what} error: #{error}"
+      end
+
+      return false
+    end
+  end
+end

data/lib/kontena/cli/nodes/label_command.rb

@@ -0,0 +1,10 @@
+module Kontena::Cli::Nodes
+  class LabelCommand < Kontena::Command
+    subcommand ["list", "ls"], "List node labels", load_subcommand('nodes/labels/list_command')
+    subcommand "add", "Add label to node", load_subcommand('nodes/labels/add_command')
+    subcommand ["remove", "rm"], "Remove label from node", load_subcommand('nodes/labels/remove_command')
+
+    def execute
+    end
+  end
+end

data/lib/kontena/cli/nodes/labels/add_command.rb

@@ -0,0 +1,18 @@
+module Kontena::Cli::Nodes::Labels
+  class AddCommand < Kontena::Command
+    include Kontena::Cli::Common
+
+    parameter "NODE", "Node name"
+    parameter "LABEL ...", "Labels"
+
+    requires_current_master
+    requires_current_master_token
+    requires_current_grid
+
+    def execute
+      node = client.get("nodes/#{current_grid}/#{self.node}")
+      data = { labels: (Array(node['labels']) + label_list).uniq }
+      client.put("nodes/#{node['id']}", data)
+    end
+  end
+end

data/lib/kontena/cli/nodes/labels/list_command.rb

@@ -0,0 +1,19 @@
+module Kontena::Cli::Nodes::Labels
+  class ListCommand < Kontena::Command
+    include Kontena::Cli::Common
+
+    parameter "NODE", "Node name"
+
+    # the command outputs id info only anyway, this is here strictly for ignoring purposes
+    option ['-q', '--quiet'], :flag, "Output the identifying column only", hidden: true
+
+    requires_current_master
+    requires_current_master_token
+    requires_current_grid
+
+    def execute
+      node = client.get("nodes/#{current_grid}/#{self.node}")
+      puts Array(node['labels']).join("\n")
+    end
+  end
+end

data/lib/kontena/cli/nodes/labels/remove_command.rb

@@ -0,0 +1,32 @@
+module Kontena::Cli::Nodes::Labels
+  class RemoveCommand < Kontena::Command
+    include Kontena::Cli::Common
+
+    parameter "NODE", "Node name"
+    parameter "LABEL ...", "Labels"
+
+    option '--force', :flag, "Do not abort if items in label list are not found"
+
+    requires_current_master
+    requires_current_master_token
+    requires_current_grid
+
+    def execute
+      node_data = client.get("nodes/#{current_grid}/#{self.node}")
+
+      node_data['labels'] ||= []
+
+      found_labels = label_list.uniq & node_data['labels']
+
+      if !force? && found_labels.size != label_list.uniq.size
+        missing = label_list - found_labels
+        exit_with_error "Label#{'s' if missing.size > 1} #{pastel.cyan(missing.join(', '))} not found on node #{pastel.cyan(node)}"
+      end
+
+      return nil if found_labels.empty?
+
+      data = { labels: node_data['labels'] - found_labels }
+      client.put("nodes/#{node_data['id']}", data)
+    end
+  end
+end

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

@@ -0,0 +1,97 @@
+require_relative '../helpers/health_helper'
+require_relative '../helpers/time_helper'
+
+module Kontena::Cli::Nodes
+  class ListCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+    include Kontena::Cli::Helpers::HealthHelper
+    include Kontena::Cli::Helpers::TimeHelper
+    include Kontena::Cli::TableGenerator::Helper
+
+    option ['-a', '--all'], :flag, 'List nodes for all grids', default: false
+
+    requires_current_master
+    requires_current_master_token
+    requires_current_grid
+
+    def node_name(node, grid)
+      return node['name'] unless all?
+      "#{grid['name']}/#{node['name']}"
+    end
+
+    def node_status(node)
+      case node_status = node['status']
+      when 'created'
+        "#{pastel.dark('created')} #{time_since(node['created_at'], terse: true)}"
+      when 'connecting'
+        "#{pastel.cyan('connecting')} #{time_since(node['connected_at'], terse: true)}"
+      when 'online'
+        "#{pastel.green('online')} #{time_since(node['connected_at'], terse: true)}"
+      when 'drain'
+        "#{pastel.yellow('drain')}"
+      when 'offline'
+        "#{pastel.red('offline')} #{time_since(node['disconnected_at'], terse: true)}"
+      else
+        pastel.white(node_status.to_s)
+      end
+    end
+
+    def node_initial(node, grid)
+      return '-' unless node['initial_member']
+      "#{node['node_number']} / #{grid['initial_size']}"
+    end
+
+    def node_labels(node)
+      (node['labels'] || ['-']).join(',')
+    end
+
+    def fields
+      return ['name'] if quiet?
+      {
+        name:    'name',
+        version: 'agent_version',
+        status:  'status',
+        initial: 'initial',
+        labels:  'labels',
+      }
+    end
+
+    def grids
+      all? ? client.get("grids")['grids'] : [client.get("grids/#{current_grid}")]
+    end
+
+    def grid_nodes(grid_name)
+      client.get("grids/#{grid_name}/nodes")['nodes']
+    end
+
+    def node_data
+      grids.flat_map do |grid|
+        grid_nodes = []
+
+        grid_nodes(grid['id']).each do |node|
+          node['name'] = node_name(node, grid)
+          grid_nodes << node
+          next if quiet?
+          node['agent_version'] ||= '-'
+          node['initial'] = node_initial(node, grid)
+          node['status'] = node_status(node)
+          node['labels'] = node_labels(node)
+        end
+
+        unless quiet?
+          grid_health = grid_health(grid, grid_nodes)
+          grid_nodes.each do |node|
+            node['name'] = health_icon(node_health(node, grid_health)) + " " + (node['name'] || node['node_id'])
+          end
+        end
+
+        grid_nodes.sort_by { |n| n['node_number'] }
+      end
+    end
+
+    def execute
+      print_table(node_data)
+    end
+  end
+end

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

@@ -0,0 +1,34 @@
+module Kontena::Cli::Nodes
+  class RemoveCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+
+    parameter "NODE ...", "Node name", attribute_name: :nodes
+    option "--force", :flag, "Force remove", default: false, attribute_name: :forced
+
+    def execute
+      require_api_url
+      require_current_grid
+      token = require_token
+
+      nodes.each do |node_name|
+        node = client(token).get("nodes/#{current_grid}/#{node_name}")
+        provider = Array(node["labels"]).find{ |l| l.start_with?('provider=')}.to_s.split('=').last
+        if node['connected'] && provider
+          plugin = provider == 'kontena' ? 'cloud' : provider
+          exit_with_error "Node #{node['name']} is still connected. You should terminate the node instead: kontena #{plugin} node terminate #{node['name']}"
+        elsif node['connected']
+          exit_with_error "Node #{node['name']} is still connected. You must terminate the node before removing it"
+        else
+          warning "Removing a node from the grid does not terminate the host machine from any cloud provider, so your cloud provider may continue to bill you for the machine until you terminate it"
+        end
+
+        confirm_command(node_name) unless forced?
+
+        spinner "Removing #{pastel.cyan(node_name)} node from #{pastel.cyan(current_grid)} grid " do
+          client(token).delete("nodes/#{node['id']}")
+        end
+      end
+    end
+  end
+end

data/lib/kontena/cli/nodes/reset_token_command.rb

@@ -0,0 +1,34 @@
+module Kontena::Cli::Nodes
+  class ResetTokenCommand < Kontena::Command
+    include Kontena::Cli::Common
+    include Kontena::Cli::GridOptions
+
+    requires_current_master
+    requires_current_master_token
+    requires_current_grid
+
+    parameter "NODE", "Node name"
+
+    option ["--token"], "TOKEN", "Use given node token instead of generating a random token"
+    option ["--clear-token"], :flag, "Clear node token, reverting to grid token"
+    option "--[no-]reset-connection", :flag, "Reset agent websocket connection", default: true
+    option "--force", :flag, "Force token update"
+
+    def execute
+      confirm("Resetting the node token will disconnect the agent (unless using --no-reset-connection), and require you to reconfigure the kontena-agent using the new `kontena node env` values before it will be able to reconnect. Are you sure?")
+
+      spinner "Resetting node #{pastel.cyan(self.node)} websocket connection token" do
+        if self.clear_token?
+          client.delete("nodes/#{current_grid}/#{self.node}/token",
+            reset_connection: self.reset_connection?,
+          )
+        else
+          client.put("nodes/#{current_grid}/#{self.node}/token",
+            token: self.token,
+            reset_connection: self.reset_connection?,
+          )
+        end
+      end
+    end
+  end
+end