central-cli 0.6.3

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

@@ -0,0 +1,80 @@
+require_relative 'common'
+
+module Central::Cli::Apps
+  class RemoveCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+    include Common
+
+    option ['-f', '--file'], 'FILE', 'Specify an alternate Central compose file', attribute_name: :filename, default: 'central.yml'
+    option ['-p', '--project-name'], 'NAME', 'Specify an alternate project name (default: directory name)'
+
+    parameter '[SERVICE] ...', 'Remove services'
+
+    attr_reader :services, :service_prefix
+
+    def execute
+      require_api_url
+      require_token
+      require_config_file(filename)
+
+      @service_prefix = project_name || current_dir
+      @services = load_services(filename, service_list, service_prefix)
+      if services.size > 0
+        remove_services(services)
+      elsif !service_list.empty?
+        puts "No such service: #{service_list.join(', ')}".colorize(:red)
+      end
+    end
+
+    private
+
+    def remove_services(services)
+      services.find_all do |_service_name, options|
+        options['links'] && options['links'].size > 0
+      end.each do |service_name, options|
+        delete(service_name, options, false)
+        services.delete(service_name)
+      end
+      services.each do |service_name, options|
+        delete(service_name, options)
+      end
+    end
+
+    def delete(name, _options, async = true)
+      unless deleted_services.include?(name)
+        print "deleting #{name.colorize(:cyan)}"
+        ShellSpinner ' ' do
+          service = begin
+                      get_service(token, prefixed_name(name))
+                    rescue
+                      nil
+                    end
+          if service
+            delete_service(token, prefixed_name(name))
+            wait_for_delete_to_finish(service) unless async
+            deleted_services << name
+          else
+            puts "No such service: #{name}".colorize(:red)
+          end
+        end
+      end
+    end
+
+    def wait_for_delete_to_finish(service)
+      until service.nil?
+        service = begin
+                    get_service(token, service['name'])
+                  rescue
+                    nil
+                  end
+        sleep 0.5
+      end
+    end
+
+    # @return [Array]
+    def deleted_services
+      @deleted_services ||= []
+    end
+  end
+end

data/lib/central/cli/apps/restart_command.rb

@@ -0,0 +1,38 @@
+require_relative 'common'
+
+module Central::Cli::Apps
+  class RestartCommand < Clamp::Command
+    include Central::Cli::Common
+    include Common
+
+    option ['-f', '--file'], 'FILE', 'Specify an alternate Central compose file', attribute_name: :filename, default: 'central.yml'
+    option ['-p', '--project-name'], 'NAME', 'Specify an alternate project name (default: directory name)'
+
+    parameter '[SERVICE] ...', 'Services to start'
+
+    attr_reader :services, :service_prefix
+
+    def execute
+      require_config_file(filename)
+
+      @service_prefix = project_name || current_dir
+      @services = load_services(filename, service_list, service_prefix)
+      if services.size > 0
+        restart_services(services)
+      elsif !service_list.empty?
+        puts "No such service: #{service_list.join(', ')}".colorize(:red)
+      end
+    end
+
+    def restart_services(services)
+      services.each do |service_name, _opts|
+        if service_exists?(service_name)
+          puts "restarting #{prefixed_name(service_name)}"
+          restart_service(token, prefixed_name(service_name))
+        else
+          puts "No such service: #{service_name}".colorize(:red)
+        end
+      end
+    end
+  end
+end

data/lib/central/cli/apps/scale_command.rb

@@ -0,0 +1,31 @@
+require_relative 'common'
+
+module Central::Cli::Apps
+  class ScaleCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+    include Common
+
+    option ['-f', '--file'], 'FILE', 'Specify an alternate Central compose file', attribute_name: :filename, default: 'central.yml'
+    option ['-p', '--project-name'], 'NAME', 'Specify an alternate project name (default: directory name)'
+
+    parameter 'SERVICE', 'Service to show'
+    parameter 'INSTANCES', 'Scales service to given number of instances'
+
+    attr_reader :services, :service_prefix
+
+    def execute
+      require_config_file(filename)
+      @service_prefix = project_name || current_dir
+      yml_service = load_services(filename, [service], service_prefix)
+      if yml_service[service]
+        options = yml_service[service]
+        abort("Service has already instances defined in #{filename}. Please update #{filename} and deploy service instead") if options['instances']
+        @service_prefix = project_name || current_dir
+        scale_service(require_token, prefixed_name(service), instances)
+      else
+        abort('Service not found')
+      end
+    end
+  end
+end

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

@@ -0,0 +1,23 @@
+require_relative 'common'
+
+module Central::Cli::Apps
+  class ShowCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+    include Common
+
+    option ['-f', '--file'], 'FILE', 'Specify an alternate Central compose file', attribute_name: :filename, default: 'central.yml'
+    option ['-p', '--project-name'], 'NAME', 'Specify an alternate project name (default: directory name)'
+
+    parameter 'SERVICE', 'Service to show'
+
+    attr_reader :services, :service_prefix
+
+    def execute
+      require_config_file(filename)
+
+      @service_prefix = project_name || current_dir
+      show_service(require_token, prefixed_name(service))
+    end
+  end
+end

data/lib/central/cli/apps/start_command.rb

@@ -0,0 +1,39 @@
+require_relative 'common'
+
+module Central::Cli::Apps
+  class StartCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+    include Common
+
+    option ['-f', '--file'], 'FILE', 'Specify an alternate Central compose file', attribute_name: :filename, default: 'central.yml'
+    option ['-p', '--project-name'], 'NAME', 'Specify an alternate project name (default: directory name)'
+
+    parameter '[SERVICE] ...', 'Services to start'
+
+    attr_reader :services, :service_prefix
+
+    def execute
+      require_config_file(filename)
+
+      @service_prefix = project_name || current_dir
+      @services = load_services(filename, service_list, service_prefix)
+      if services.size > 0
+        start_services(services)
+      elsif !service_list.empty?
+        puts "No such service: #{service_list.join(', ')}".colorize(:red)
+      end
+    end
+
+    def start_services(services)
+      services.each do |service_name, _opts|
+        if service_exists?(service_name)
+          puts "starting #{prefixed_name(service_name)}"
+          start_service(token, prefixed_name(service_name))
+        else
+          puts "No such service: #{service_name}".colorize(:red)
+        end
+      end
+    end
+  end
+end

data/lib/central/cli/apps/stop_command.rb

@@ -0,0 +1,39 @@
+require_relative 'common'
+
+module Central::Cli::Apps
+  class StopCommand < Clamp::Command
+    include Central::Cli::Common
+    include Central::Cli::StackOptions
+    include Common
+
+    option ['-f', '--file'], 'FILE', 'Specify an alternate Central compose file', attribute_name: :filename, default: 'central.yml'
+    option ['-p', '--project-name'], 'NAME', 'Specify an alternate project name (default: directory name)'
+
+    parameter '[SERVICE] ...', 'Services to stop'
+
+    attr_reader :services, :service_prefix
+
+    def execute
+      require_config_file(filename)
+
+      @service_prefix = project_name || current_dir
+      @services = load_services(filename, service_list, service_prefix)
+      if services.size > 0
+        stop_services(services)
+      elsif !service_list.empty?
+        puts "No such service: #{service_list.join(', ')}".colorize(:red)
+      end
+    end
+
+    def stop_services(services)
+      services.each do |service_name, _opts|
+        if service_exists?(service_name)
+          puts "stopping #{prefixed_name(service_name)}"
+          stop_service(token, prefixed_name(service_name))
+        else
+          puts "No such service: #{service_name}".colorize(:red)
+        end
+      end
+    end
+  end
+end

data/lib/central/cli/auth/aws/create_command.rb

@@ -0,0 +1,34 @@
+
+module Central::Cli::Auth::Aws
+  class CreateCommand < Clamp::Command
+    include Central::Cli::Common
+
+    option '--access-key', 'ACCESS_KEY', 'AWS access key ID', environment_variable: 'AWS_ACCESS_KEY_ID', required: true
+    option '--secret-key', 'SECRET_KEY', 'AWS secret key', environment_variable: 'AWS_SECRET_ACCESS_KEY', required: true
+    option '--key-pair', 'KEY_PAIR', 'EC2 key pair name', required: true
+    option '--ssl-cert', 'SSL CERT', 'SSL certificate file (default: generate self-signed cert)'
+    option '--region', 'REGION', 'EC2 Region', default: 'us-west-2'
+    option '--zone', 'ZONE', 'EC2 Availability Zone', default: 'a'
+    option '--vpc-id', 'VPC ID', 'Virtual Private Cloud (VPC) ID (default: default vpc)'
+    option '--subnet-id', 'SUBNET ID', 'VPC option to specify subnet to launch instance into (default: first subnet from vpc/az)'
+    option '--type', 'SIZE', 'Instance type', default: 't2.small'
+    option '--storage', 'STORAGE', 'Storage size (GiB)', default: '30'
+    option '--version', 'VERSION', 'Define installed Central version', default: 'latest'
+
+    def execute
+      require 'central/machine/aws'
+
+      provisioner = Central::Machine::Aws::AuthProvisioner.new(access_key, secret_key, region)
+      provisioner.run!(
+        type: type,
+        vpc: vpc_id,
+        zone: zone,
+        subnet: subnet_id,
+        ssl_cert: ssl_cert,
+        storage: storage,
+        version: version,
+        key_pair: key_pair
+      )
+    end
+  end
+end

data/lib/central/cli/auth/aws_command.rb

@@ -0,0 +1,7 @@
+module Central::Cli::Auth
+  require_relative 'aws/create_command'
+
+  class AwsCommand < Clamp::Command
+    subcommand 'create', 'Create a new AWS oAuth Server', Aws::CreateCommand
+  end
+end

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

@@ -0,0 +1,28 @@
+require 'tty'
+
+module Central::Cli::Auth
+  class ListCommand < Clamp::Command
+    include Central::Cli::Common
+
+    def execute
+      header = %w(Name Url)
+      rows = []
+
+      current_server = settings['current_server']
+      settings['servers'].each do |server|
+        name = if server['name'] == current_server
+                 "* #{server['name']}"
+               else
+                 server['name']
+               end
+        rows << [name, server['url']]
+      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/auth_command.rb

@@ -0,0 +1,9 @@
+require_relative 'auth/aws_command'
+require_relative 'auth/list_command'
+
+class Central::Cli::AuthCommand < Clamp::Command
+  subcommand 'aws', 'AWS specific commands', Central::Cli::Auth::AwsCommand
+  subcommand %w(list ls), 'List oAuth servers', Central::Cli::Auth::ListCommand
+  def execute
+  end
+end

data/lib/central/cli/bytes_helper.rb

@@ -0,0 +1,38 @@
+
+module Central
+  module Cli
+    module BytesHelper
+      KILOBYTE = 1024
+      MEGABYTE = KILOBYTE * 1024
+      GIGABYTE = MEGABYTE * 1024
+      TERABYTE = GIGABYTE * 1024
+
+      def to_kilobytes(bytes, ndigits = nil)
+        return 0.0 if bytes.nil?
+        round(bytes.to_f / KILOBYTE, ndigits)
+      end
+
+      def to_megabytes(bytes, ndigits = nil)
+        return 0.0 if bytes.nil?
+        round(bytes.to_f / MEGABYTE, ndigits)
+      end
+
+      def to_gigabytes(bytes, ndigits = nil)
+        return 0.0 if bytes.nil?
+        round(bytes.to_f / GIGABYTE, ndigits)
+      end
+
+      def to_terabytes(bytes, ndigits)
+        return 0.0 if bytes.nil?
+        round(bytes.to_f / TERABYTE, ndigits)
+      end
+
+      private
+
+      def round(value, ndigits = nil)
+        return value if ndigits.nil?
+        value.round(ndigits)
+      end
+    end
+  end
+end

data/lib/central/cli/common.rb

@@ -0,0 +1,148 @@
+require 'colorize'
+require 'uri'
+
+module Central
+  module Cli
+    module Common
+      def require_api_url
+        api_url
+      end
+
+      def require_token
+        token = ENV['CENTRAL_TOKEN'] || current_master['token']
+        unless token
+          fail ArgumentError.new('Please login first using: cm login')
+        end
+        token
+      end
+
+      def client(token = nil)
+        if @client.nil?
+          headers = {}
+          headers['Authorization'] = "Bearer #{token}" unless token.nil?
+
+          @client = Central::Client.new(api_url, headers)
+        end
+        @client
+      end
+
+      def reset_client
+        @client = nil
+      end
+
+      def settings_filename
+        File.join(Dir.home, '/.central_client.json')
+      end
+
+      def settings
+        if @settings.nil?
+          if File.exist?(settings_filename)
+            @settings = JSON.parse(File.read(settings_filename))
+            unless @settings['current_server']
+              # Let's migrate the old settings model to new
+              @settings['server']['name'] = 'default'
+              @settings = {
+                'current_server' => 'default',
+                'servers' => [@settings['server']]
+              }
+              save_settings
+            end
+          else
+            @settings = {
+              'current_server' => 'default',
+              'servers' => [{}]
+            }
+          end
+        end
+        @settings
+      end
+
+      def api_url
+        url = ENV['CENTRAL_URL'] || current_master['url']
+        unless url
+          fail ArgumentError.new("It seem's that you are not logged into Central master, please login with: cm login")
+        end
+        ensure_custom_ssl_ca(url)
+        url
+      end
+
+      def ensure_custom_ssl_ca(url)
+        return if Excon.defaults[:ssl_ca_file]
+
+        uri = URI.parse(url)
+        cert_file = File.join(Dir.home, "/.central/certs/#{uri.host}.pem")
+        Excon.defaults[:ssl_ca_file] = cert_file if File.exist?(cert_file)
+      end
+
+      def current_stack=(stack)
+        settings['servers'][current_master_index]['stack'] = stack['id']
+        save_settings
+      end
+
+      def require_current_stack
+        if current_stack.nil?
+          fail ArgumentError.new('Please select stack first using: cm stack use <stack name>')
+        end
+      end
+
+      def clear_current_stack
+        settings['servers'][current_master_index].delete('stack')
+        save_settings
+      end
+
+      def current_stack
+        if respond_to?(:stack)
+          ENV['CENTRAL_STACK'] || stack || current_master['stack']
+        else
+          ENV['CENTRAL_STACK'] || current_master['stack']
+        end
+      rescue ArgumentError => e
+        nil
+      end
+
+      def current_master_index
+        current_server = settings['current_server'] || 'default'
+        settings['servers'].find_index { |m| m['name'] == current_server }
+      end
+
+      def current_master
+        index = current_master_index
+        unless index
+          fail ArgumentError.new("It seem's that you are not logged into ANY Central master, please login with: cm login")
+        end
+        settings['servers'][index]
+      end
+
+      def current_master=(master_alias)
+        settings['current_server'] = master_alias
+        save_settings
+      end
+
+      def api_url=(api_url)
+        settings['servers'][current_master_index]['url'] = api_url
+        save_settings
+      end
+
+      def access_token=(token)
+        settings['servers'][current_master_index]['token'] = token
+        save_settings
+      end
+
+      def add_master(server_name, master_info)
+        server_name ||= 'default'
+        index = settings['servers'].find_index { |m| m['name'] == server_name }
+        if index
+          settings['servers'][index] = master_info
+        else
+          settings['servers'] << master_info
+        end
+        settings['current_server'] = server_name
+        save_settings
+      end
+
+      def save_settings
+        File.write(settings_filename, JSON.pretty_generate(settings))
+      end
+    end
+  end
+end