central-cli 0.6.3

data/lib/central/cli/stacks/create_command.rb

@@ -0,0 +1,26 @@
+require_relative 'common'
+
+module Central::Cli::Stacks
+  class CreateCommand < Clamp::Command
+    include Central::Cli::Common
+    include Common
+
+    parameter 'NAME', 'Stack name'
+    option '--initial-size', 'INITIAL_SIZE', 'Initial stack size (number of nodes)', default: 1
+
+    def execute
+      require_api_url
+
+      token = require_token
+      payload = {
+        name: name
+      }
+      payload[:initial_size] = initial_size if initial_size
+      stack = client(token).post('stacks', payload)
+      if stack
+        self.current_stack = stack
+        puts "Using stack: #{stack['name'].cyan}"
+      end
+    end
+  end
+end

data/lib/central/cli/stacks/current_command.rb

@@ -0,0 +1,25 @@
+require_relative 'common'
+
+module Central::Cli::Stacks
+  class CurrentCommand < Clamp::Command
+    include Central::Cli::Common
+    include Common
+
+    option ['--name'], :flag, 'Show name only', default: false
+
+    def execute
+      require_api_url
+      if current_stack.nil?
+        abort 'No stack selected. To select stack, please run: cm stack use <stack name>'
+      else
+
+        stack = client(require_token).get("stacks/#{current_stack}")
+        if name?
+          puts (stack['name']).to_s
+        else
+          print_stack(stack)
+        end
+      end
+    end
+  end
+end

data/lib/central/cli/stacks/env_command.rb

@@ -0,0 +1,32 @@
+require_relative 'common'
+
+module Central::Cli::Stacks
+  class EnvCommand < Clamp::Command
+    include Central::Cli::Common
+    include Common
+
+    parameter '[NAME]', 'Stack name'
+    option ['-e', '--export'], :flag, 'Add export', default: false
+
+    def execute
+      require_api_url
+
+      name_or_current = name.nil? ? current_stack : name
+
+      if name_or_current.nil?
+        abort 'No stack selected. Use: cm stack env <name>, or select a stack with: cm stack use <name>'
+      else
+        stack = find_stack_by_name(name_or_current)
+        abort('Stack not found'.colorize(:red)) unless stack
+
+        prefix = export? ? 'export ' : ''
+
+        server = settings['servers'].find { |s| s['name'] == settings['current_server'] }
+        if server
+          puts "#{prefix}CENTRAL_URI=#{server['url'].sub('http', 'ws')}"
+          puts "#{prefix}CENTRAL_TOKEN=#{server['token']}"
+        end
+      end
+    end
+  end
+end

data/lib/central/cli/stacks/list_command.rb

@@ -0,0 +1,35 @@
+require_relative 'common'
+require 'tty'
+
+module Central::Cli::Stacks
+  class ListCommand < Clamp::Command
+    include Central::Cli::Common
+    include Common
+
+    def execute
+      require_api_url
+
+      if stacks['stacks'].size == 0
+        puts "You don't have any stacks yet. Create first one with 'cm stack create' command".colorize(:yellow)
+      end
+
+      header = %w(Name Nodes Services Users)
+      rows = []
+
+      stacks['stacks'].each do |stack|
+        name = if stack['id'] == current_stack
+                 "#{stack['name']} *"
+               else
+                 stack['name']
+               end
+        rows << [name, stack['node_count'], stack['service_count'], stack['user_count']]
+      end
+
+      ttable = TTY::Table.new header: header, rows: Array.new(rows)
+      renderer = TTY::Table::Renderer::Unicode.new(ttable)
+      renderer.border.style = :cyan
+      renderer.padding = [0, 1, 0, 1]
+      puts renderer.render
+    end
+  end
+end

data/lib/central/cli/stacks/list_users_command.rb

@@ -0,0 +1,26 @@
+require_relative 'common'
+require 'tty'
+
+module Central::Cli::Stacks
+  class ListUsersCommand < Clamp::Command
+    include Central::Cli::Common
+    include Common
+
+    def execute
+      require_api_url
+      token = require_token
+      result = client(token).get("stacks/#{current_stack}/users")
+      header = %w(Email Name)
+      rows = []
+      result['users'].each do |user|
+        rows << [user['email'], user['name']]
+      end
+
+      ttable = TTY::Table.new header: header, rows: Array.new(rows)
+      renderer = TTY::Table::Renderer::Unicode.new(ttable)
+      renderer.border.style = :cyan
+      renderer.padding = [0, 1, 0, 1]
+      puts renderer.render
+    end
+  end
+end

data/lib/central/cli/stacks/logs_command.rb

@@ -0,0 +1,81 @@
+module Central::Cli::Stacks
+  class LogsCommand < Clamp::Command
+    include Central::Cli::Common
+
+    option ['-t', '--tail'], :flag, 'Tail (follow) logs', default: false
+    option '--lines', 'LINES', 'Number of lines to show from the end of the logs'
+    option '--since', 'SINCE', 'Show logs since given timestamp'
+    option '--node', 'NODE', 'Filter by node name', multivalued: true
+    option '--service', 'SERVICE', 'Filter by service name', multivalued: true
+    option ['-c', '--container'], 'CONTAINER', 'Filter by container', multivalued: true
+
+    def execute
+      require_api_url
+      token = require_token
+
+      query_params = {}
+      query_params[:nodes] = node_list.join(',') unless node_list.empty?
+      query_params[:services] = service_list.join(',') unless service_list.empty?
+      query_params[:containers] = container_list.join(',') unless container_list.empty?
+      query_params[:limit] = lines if lines
+      query_params[:since] = since if since
+
+      if tail?
+        @buffer = ''
+        query_params[:follow] = 1
+        stream_logs(token, query_params)
+      else
+        list_logs(token, query_params)
+      end
+    end
+
+    def list_logs(token, query_params)
+      result = client(token).get("stacks/#{current_stack}/container_logs", query_params)
+      result['logs'].each do |log|
+        color = color_for_container(log['name'])
+        prefix = ''
+        prefix << "#{log['created_at']} "
+        prefix << "#{log['name']}:"
+        prefix = prefix.colorize(color)
+        puts "#{prefix} #{log['data']}"
+      end
+    end
+
+    def stream_logs(token, query_params)
+      streamer = lambda do |chunk, _remaining_bytes, _total_bytes|
+        begin
+          chunk = @buffer + chunk unless @buffer.empty?
+          log = JSON.parse(chunk) unless chunk.empty?
+          @buffer = ''
+        rescue => exc
+          @buffer << chunk
+        end
+        if log
+          color = color_for_container(log['name'])
+          puts "#{log['name'].colorize(color)} | #{log['data']}"
+        end
+      end
+
+      result = client(token).get_stream(
+        "stacks/#{current_stack}/container_logs", streamer, query_params
+      )
+    end
+
+    def color_for_container(container_id)
+      color_maps[container_id] = colors.shift unless color_maps[container_id]
+      color_maps[container_id].to_sym
+    end
+
+    def color_maps
+      @color_maps ||= {}
+    end
+
+    def colors
+      if @colors.nil? || @colors.size == 0
+        @colors = [:green, :yellow, :magenta, :cyan, :red,
+                   :light_green, :light_yellow, :ligh_magenta, :light_cyan, :light_red]
+      end
+      @colors
+    end
+  end
+end

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

@@ -0,0 +1,26 @@
+require_relative 'common'
+
+module Central::Cli::Stacks
+  class RemoveCommand < Clamp::Command
+    include Central::Cli::Common
+    include Common
+
+    parameter 'NAME', 'Stack name'
+
+    def execute
+      require_api_url
+      token = require_token
+      stack = find_stack_by_name(name)
+
+      if !stack.nil?
+        response = client(token).delete("stacks/#{stack['id']}")
+        if response
+          clear_current_stack if stack['id'] == current_stack
+          puts "removed #{stack['name'].cyan}"
+        end
+      else
+        abort "Could not resolve stack by name [#{name}]. For a list of existing stacks please run: cm stack list".colorize(:red)
+      end
+    end
+  end
+end

data/lib/central/cli/stacks/remove_user_command.rb

@@ -0,0 +1,17 @@
+require_relative 'common'
+
+module Central::Cli::Stacks
+  class RemoveUserCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+    include Common
+
+    parameter 'EMAIL', 'Email address'
+
+    def execute
+      require_api_url
+      token = require_token
+      result = client(token).delete("stacks/#{current_stack}/users/#{email}")
+    end
+  end
+end

data/lib/central/cli/stacks/show_command.rb

@@ -0,0 +1,19 @@
+require_relative 'common'
+
+module Central::Cli::Stacks
+  class ShowCommand < Clamp::Command
+    include Central::Cli::Common
+    include Common
+
+    parameter 'NAME', 'Stack name'
+
+    def execute
+      require_api_url
+
+      stack = find_stack_by_name(name)
+      abort('Stack not found'.colorize(:red)) unless stack
+
+      print_stack(stack)
+    end
+  end
+end

data/lib/central/cli/stacks/trusted_subnets/add_command.rb

@@ -0,0 +1,16 @@
+module Central::Cli::Stacks::TrustedSubnets
+  class AddCommand < Clamp::Command
+    include Central::Cli::Common
+
+    parameter 'NAME', 'Stack name'
+    parameter 'SUBNET', 'Trusted subnet'
+
+    def execute
+      require_api_url
+      token = require_token
+      stack = client(token).get("stacks/#{name}")
+      data = { trusted_subnets: stack['trusted_subnets'] + [subnet] }
+      client(token).put("stacks/#{name}", data)
+    end
+  end
+end

data/lib/central/cli/stacks/trusted_subnets/list_command.rb

@@ -0,0 +1,17 @@
+module Central::Cli::Stacks::TrustedSubnets
+  class ListCommand < Clamp::Command
+    include Central::Cli::Common
+
+    parameter 'NAME', 'Stack name'
+
+    def execute
+      require_api_url
+      token = require_token
+      stack = client(token).get("stacks/#{current_stack}")
+      trusted_subnets = stack['trusted_subnets'] || []
+      trusted_subnets.each do |subnet|
+        puts subnet
+      end
+    end
+  end
+end

data/lib/central/cli/stacks/trusted_subnets/remove_command.rb

@@ -0,0 +1,20 @@
+module Central::Cli::Stacks::TrustedSubnets
+  class RemoveCommand < Clamp::Command
+    include Central::Cli::Common
+
+    parameter 'NAME', 'Stack name'
+    parameter 'SUBNET', 'Trusted subnet'
+
+    def execute
+      require_api_url
+      token = require_token
+      stack = client(token).get("stacks/#{name}")
+      trusted_subnets = stack['trusted_subnets'] || []
+      unless trusted_subnets.delete(subnet)
+        abort("Stack does not have trusted subnet #{subnet}")
+      end
+      data = { trusted_subnets: trusted_subnets }
+      client(token).put("stacks/#{name}", data)
+    end
+  end
+end

data/lib/central/cli/stacks/update_command.rb

@@ -0,0 +1,27 @@
+require_relative 'common'
+
+module Central::Cli::Stacks
+  class UpdateCommand < Clamp::Command
+    include Central::Cli::Common
+    include Common
+
+    parameter 'NAME', 'Stack name'
+    option '--statsd-server', 'STATSD_SERVER', 'Statsd server address (host:port)'
+
+    def execute
+      require_api_url
+      token = require_token
+      payload = {}
+      if statsd_server
+        server, port = statsd_server.split(':')
+        payload[:stats] = {
+          statsd: {
+            server: server,
+            port: port || 8125
+          }
+        }
+      end
+      client(token).put("stacks/#{name}", payload)
+    end
+  end
+end

data/lib/central/cli/stacks/use_command.rb

@@ -0,0 +1,21 @@
+require_relative 'common'
+
+module Central::Cli::Stacks
+  class UseCommand < Clamp::Command
+    include Central::Cli::Common
+    include Common
+
+    parameter 'NAME', 'Stack name to use'
+
+    def execute
+      require_api_url
+      stack = find_stack_by_name(name)
+      if !stack.nil?
+        self.current_stack = stack
+        puts "Using stack: #{stack['name'].cyan}"
+      else
+        abort "Could not resolve stack by name [#{name}]. For a list of existing stacks please run: cm stack list".colorize(:red)
+      end
+    end
+  end
+end

data/lib/central/cli/stacks/user_command.rb

@@ -0,0 +1,11 @@
+module Central::Cli::Stacks
+  require_relative 'users/list_command'
+  require_relative 'users/add_command'
+  require_relative 'users/remove_command'
+
+  class UserCommand < Clamp::Command
+    subcommand %w(list ls), 'List stack users', Users::ListCommand
+    subcommand 'add', 'Add user to stack', Users::AddCommand
+    subcommand %w(remove rm), 'Remove user from stack', Users::RemoveCommand
+  end
+end

data/lib/central/cli/stacks/users/add_command.rb

@@ -0,0 +1,18 @@
+require_relative '../common'
+
+module Central::Cli::Stacks::Users
+  class AddCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+    include Central::Cli::Stacks::Common
+
+    parameter 'EMAIL', 'Email address'
+
+    def execute
+      require_api_url
+      token = require_token
+      data = { email: email }
+      client(token).post("stacks/#{current_stack}/users", data)
+    end
+  end
+end

data/lib/central/cli/stacks/users/list_command.rb

@@ -0,0 +1,18 @@
+require_relative '../common'
+
+module Central::Cli::Stacks::Users
+  class ListCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::Stacks::Common
+
+    def execute
+      require_api_url
+      token = require_token
+      result = client(token).get("stacks/#{current_stack}/users")
+      puts '%-40s %-40s' % %w(Email Name)
+      result['users'].each do |user|
+        puts '%-40.40s %-40.40s' % [user['email'], user['name']]
+      end
+    end
+  end
+end