vmc 0.0.8 → 0.2.4

data/lib/cli/runner.rb

@@ -0,0 +1,437 @@
+
+require 'optparse'
+
+require File.dirname(__FILE__) + '/usage'
+
+class VMC::Cli::Runner
+
+  attr_reader   :namespace
+  attr_reader   :action
+  attr_reader   :args
+  attr_reader   :options
+
+  def self.run(args)
+    new(args).run
+  end
+
+  def initialize(args=[])
+    @args = args
+    @options = { :colorize => true }
+    @exit_status = true
+  end
+
+  # Collect all the available options for all commands
+  # Some duplicates exists to capture all scenarios
+  def parse_options!
+    opts_parser = OptionParser.new do |opts|
+      opts.banner = "\nAvailable options:\n\n"
+
+      opts.on('--email EMAIL')     { |email| @options[:email] = email }
+      opts.on('--user EMAIL')      { |email| @options[:email] = email }
+      opts.on('--passwd PASS')     { |pass|  @options[:password] = pass }
+      opts.on('--pass PASS')       { |pass|  @options[:password] = pass }
+      opts.on('--password PASS')   { |pass|  @options[:password] = pass }
+      opts.on('--app NAME')        { |name|  @options[:name] = name }
+      opts.on('--name NAME')       { |name|  @options[:name] = name }
+      opts.on('--bind BIND')       { |bind|  @options[:bind] = bind }
+      opts.on('--instance INST')   { |inst|  @options[:instance] = inst }
+      opts.on('--instances INST')  { |inst|  @options[:instances] = inst }
+      opts.on('--url URL')         { |url|   @options[:url] = url }
+      opts.on('--mem MEM')         { |mem|   @options[:mem] = mem }
+      opts.on('--path PATH')       { |path|  @options[:path] = path }
+      opts.on('--no-start')        {         @options[:nostart] = true }
+      opts.on('--nostart')         {         @options[:nostart] = true }
+      opts.on('--force')           {         @options[:force] = true }
+      opts.on('--all')             {         @options[:all] = true }
+
+      # generic tracing and debugging
+      opts.on('-t', '--trace')     {         @options[:trace] = true }
+      opts.on('-q', '--quiet')     {         @options[:quiet] = true }
+
+      # Don't use builtin zip
+      opts.on('--no-zip')          {         @options[:nozip] = true }
+      opts.on('--nozip')           {         @options[:nozip] = true }
+
+      opts.on('--no-resources')    {         @options[:noresources] = true }
+      opts.on('--noresources')     {         @options[:noresources] = true }
+
+      opts.on('--no-color')        {         @options[:colorize] = false }
+      opts.on('--verbose')         {         @options[:verbose] = true }
+
+      opts.on('-n','--no-prompt')  {         @options[:noprompts] = true }
+      opts.on('--noprompt')        {         @options[:noprompts] = true }
+      opts.on('--non-interactive') {         @options[:noprompts] = true }
+
+      opts.on('--prefix')          {         @options[:prefixlogs] = true }
+      opts.on('--prefix-logs')     {         @options[:prefixlogs] = true }
+      opts.on('--prefixlogs')      {         @options[:prefixlogs] = true }
+
+      opts.on('--json')            {         @options[:json] = true }
+
+      opts.on('-v', '--version')   {         set_cmd(:misc, :version) }
+      opts.on('-h', '--help')      {         puts "#{command_usage}\n"; exit }
+
+      # deprecated
+      opts.on('--exec EXEC')       { |exec|  @options[:exec] = exec }
+      opts.on('--noframework')     {         @options[:noframework] = true }
+      opts.on('--canary')          {         @options[:canary] = true }
+
+      # Proxying for another user, requires admin privileges
+      opts.on('-u PROXY')          { |proxy| @options[:proxy] = proxy }
+
+      opts.on_tail('--options')    {          puts "#{opts}\n"; exit }
+    end
+    @args = opts_parser.parse!(@args)
+    convert_options!
+    self
+  end
+
+  def display_help
+    puts command_usage
+    exit
+  end
+
+  def convert_options!
+    # make sure certain options are valid and in correct form.
+    @options[:instances] = Integer(@options[:instances]) if @options[:instances]
+    @options[:instance] = Integer(@options[:instance]) if @options[:instance]
+  end
+
+  def set_cmd(namespace, action, args_range=0)
+    return if @help_only
+    unless args_range == "*" || args_range.is_a?(Range)
+      args_range = (args_range.to_i..args_range.to_i)
+    end
+
+    if args_range == "*" || args_range.include?(@args.size)
+      @namespace = namespace
+      @action    = action
+    else
+      @exit_status = false
+      if @args.size > args_range.last
+        usage_error("Too many arguments for [#{action}]: %s" % [ @args[args_range.last..-1].map{|a| "'#{a}'"}.join(', ') ])
+      else
+        usage_error("Not enough arguments for [#{action}]")
+      end
+    end
+  end
+
+  def parse_command!
+    # just return if already set, happends with -v, -h
+    return if @namespace && @action
+
+    verb = @args.shift
+    case verb
+
+    when 'version'
+      usage('vmc version')
+      set_cmd(:misc, :version)
+
+    when 'target'
+      usage('vmc target [url] [--url]')
+      if @args.size == 1
+        set_cmd(:misc, :set_target, 1)
+      else
+        set_cmd(:misc, :target)
+      end
+
+    when 'targets'
+      usage('vmc targets')
+      set_cmd(:misc, :targets)
+
+    when 'tokens'
+      usage('vmc tokens')
+      set_cmd(:misc, :tokens)
+
+    when 'info'
+      usage('vmc info')
+      set_cmd(:misc, :info)
+
+    when 'user'
+      usage('vmc user')
+      set_cmd(:user, :info)
+
+    when 'login'
+      usage('vmc login [email] [--email EMAIL] [--passwd PASS]')
+      if @args.size == 1
+        set_cmd(:user, :login, 1)
+      else
+        set_cmd(:user, :login)
+      end
+
+    when 'logout'
+      usage('vmc logout')
+      set_cmd(:user, :logout)
+
+    when 'passwd'
+      usage('vmc passwd')
+      if @args.size == 1
+        set_cmd(:user, :change_password, 1)
+      else
+        set_cmd(:user, :change_password)
+      end
+
+    when 'add-user', 'add_user', 'create_user', 'create-user', 'register'
+      usage('vmc add-user [user] [--email EMAIL] [--passwd PASS]')
+      if @args.size == 1
+        set_cmd(:admin, :add_user, 1)
+      else
+        set_cmd(:admin, :add_user)
+      end
+
+    when 'delete-user', 'delete_user', 'unregister'
+      usage('vmc delete-user <user>')
+      set_cmd(:admin, :delete_user, 1)
+
+    when 'apps'
+      usage('vmc apps')
+      set_cmd(:apps, :apps)
+
+    when 'list'
+      usage('vmc list')
+      set_cmd(:apps, :list)
+
+    when 'start'
+      usage('vmc start <appname>')
+      set_cmd(:apps, :start, 1)
+
+    when 'stop'
+      usage('vmc stop <appname>')
+      set_cmd(:apps, :stop, 1)
+
+    when 'restart'
+      usage('vmc restart <appname>')
+      set_cmd(:apps, :restart, 1)
+
+    when 'rename'
+      usage('vmc rename <appname> <newname>')
+      set_cmd(:apps, :rename, 2)
+
+    when 'mem'
+      usage('vmc mem <appname> [memsize]')
+      if @args.size == 2
+        set_cmd(:apps, :mem, 2)
+      else
+        set_cmd(:apps, :mem, 1)
+      end
+
+    when 'stats'
+      usage('vmc stats <appname>')
+      set_cmd(:apps, :stats, 1)
+
+    when 'map'
+      usage('vmc map <appname> <url>')
+      set_cmd(:apps, :map, 2)
+
+    when 'unmap'
+      usage('vmc unmap <appname> <url>')
+      set_cmd(:apps, :unmap, 2)
+
+    when 'delete'
+      usage('vmc delete <appname>')
+      if @options[:all] && @args.size == 0
+        set_cmd(:apps, :delete)
+      else
+        set_cmd(:apps, :delete, 1)
+      end
+
+    when 'files'
+      usage('vmc files <appname> [path] [--instance N] [--all] [--prefix]')
+      if @args.size == 1
+        set_cmd(:apps, :files, 1)
+      else
+        set_cmd(:apps, :files, 2)
+      end
+
+    when 'logs'
+      usage('vmc logs <appname> [--instance N] [--all] [--prefix]')
+      set_cmd(:apps, :logs, 1)
+
+    when 'instances', 'scale'
+      if @args.size == 1
+        usage('vmc instances <appname>')
+        set_cmd(:apps, :instances, 1)
+      else
+        usage('vmc instances <appname> <num|delta>')
+        set_cmd(:apps, :instances, 2)
+      end
+
+    when 'crashes'
+      usage('vmc crashes <appname>')
+      set_cmd(:apps, :crashes, 1)
+
+    when 'crashlogs'
+      usage('vmc crashlogs <appname>')
+      set_cmd(:apps, :crashlogs, 1)
+
+    when 'push'
+      usage('vmc push [appname] [--path PATH] [--url URL] [--instances N] [--mem] [--no-start]')
+      if @args.size == 1
+        set_cmd(:apps, :push, 1)
+      else
+        set_cmd(:apps, :push, 0)
+      end
+
+    when 'update'
+      usage('vmc update <appname> [--path PATH]')
+      set_cmd(:apps, :update, 1)
+
+    when 'services'
+      usage('vmc services')
+      set_cmd(:services, :services)
+
+    when 'create-service', 'create_service'
+      usage('vmc create-service [service] [servicename] [appname] [--name servicename] [--bind appname]')
+      set_cmd(:services, :create_service) if @args.size == 0
+      set_cmd(:services, :create_service, 1) if @args.size == 1
+      set_cmd(:services, :create_service, 2) if @args.size == 2
+      set_cmd(:services, :create_service, 3) if @args.size == 3
+
+    when 'delete-service', 'delete_service'
+      usage('vmc delete-service <service>')
+      if @args.size == 1
+        set_cmd(:services, :delete_service, 1)
+      else
+        set_cmd(:services, :delete_service)
+      end
+
+    when 'bind-service', 'bind_service'
+      usage('vmc bind-service <servicename> <appname>')
+      set_cmd(:services, :bind_service, 2)
+
+    when 'unbind-service', 'unbind_service'
+      usage('vmc unbind-service <servicename> <appname>')
+      set_cmd(:services, :unbind_service, 2)
+
+    when 'clone-services'
+      usage('vmc clone-services <src-app> <dest-app>')
+      set_cmd(:services, :clone_services, 2)
+
+    when 'aliases'
+      usage('vmc aliases')
+      set_cmd(:misc, :aliases)
+
+    when 'alias'
+      usage('vmc alias <alias[=]command>')
+      if @args.size == 1
+        set_cmd(:misc, :alias, 1)
+      elsif @args.size == 2
+        set_cmd(:misc, :alias, 2)
+      end
+
+    when 'unalias'
+      usage('vmc unalias <alias>')
+      set_cmd(:misc, :unalias, 1)
+
+    when 'help'
+      display_help if @args.size == 0
+      @help_only = true
+      parse_command!
+
+    when 'usage'
+      display basic_usage
+      exit(true)
+
+    when 'options'
+      # Simulate --options
+      @args = @args.unshift('--options')
+      parse_options!
+
+    else
+      if verb
+        display "vmc: Unknown command [#{verb}]"
+        display basic_usage
+        exit(false)
+      end
+    end
+  end
+
+  def process_aliases!
+    return if @args.empty?
+    aliases = VMC::Cli::Config.aliases
+    if key = aliases.values.grep(@args[0]).first
+      display "[#{@args[0]} aliased to #{aliases.invert[key]}]" if @options[:verbose]
+      @args[0] = aliases.invert[key]
+    end
+  end
+
+  def usage(msg = nil)
+    @usage = msg if msg
+    @usage
+  end
+
+  def usage_error(msg = nil)
+    @usage_error = msg if msg
+    @usage_error
+  end
+
+  def run
+    trap('TERM') { print "\nInterupted\n"; exit(false)}
+    trap("INT")  { print "\nInterupted\n"; exit(false)}
+
+    parse_options!
+
+    @options[:colorize] = false unless STDOUT.tty?
+
+    VMC::Cli::Config.colorize   = @options.delete(:colorize)
+    VMC::Cli::Config.trace      = @options.delete(:trace)
+    VMC::Cli::Config.nozip      = @options.delete(:nozip)
+    VMC::Cli::Config.output   ||= STDOUT unless @options[:quiet]
+
+    process_aliases!
+    parse_command!
+
+    if @namespace && @action
+      eval("VMC::Cli::Command::#{@namespace.to_s.capitalize}").new(@options).send(@action.to_sym, *@args)
+    elsif @help_only || @usage
+      display_usage
+    else
+      display basic_usage
+      exit(false)
+    end
+
+  rescue OptionParser::InvalidOption => e
+  rescue OptionParser::AmbiguousOption => e
+    puts(e.message.red)
+    puts("\n")
+    puts(basic_usage)
+    @exit_status = false
+  rescue VMC::Client::AuthError => e
+    if VMC::Cli::Config.auth_token.nil?
+      puts "Login Required".red
+    else
+      puts "Not Authorized".red
+    end
+    @exit_status = false
+  rescue VMC::Client::TargetError, VMC::Client::NotFound, VMC::Client::BadTarget  => e
+    puts e.message.red
+    @exit_status = false
+  rescue VMC::Cli::GracefulExit => e
+    # Redirected commands end up generating this exception (kind of goto)
+  rescue VMC::Cli::CliExit => e
+    puts e.message.red
+    @exit_status = false
+  rescue VMC::Cli::CliError => e
+    say("Error #{e.error_code}: #{e.message}".red)
+    @exit_status = false
+  rescue SystemExit => e
+    @exit_status = e.success?
+  rescue => e
+    puts e.message.red
+    puts e.backtrace
+    @exit_status = false
+  ensure
+    say("\n")
+    @exit_status == true if @exit_status.nil?
+    if @options[:verbose]
+      if @exit_status
+        puts "[#{@namespace}:#{@action}] SUCCEEDED".green
+      else
+        puts "[#{@namespace}:#{@action}] FAILED".red
+      end
+      say("\n")
+    end
+    exit(@exit_status)
+  end
+
+end

data/lib/cli/services_helper.rb

@@ -0,0 +1,74 @@
+
+module VMC::Cli
+  module ServicesHelper
+    def display_system_services(services=nil)
+      services ||= client.services_info
+
+      display "\n============== System Services ==============\n\n"
+
+      return display "No system services available" if services.empty?
+
+      services_table = table do |t|
+        t.headings = 'Service', 'Version', 'Description'
+        services.each do |service_type, value|
+          value.each do |vendor, version|
+            version.each do |version_str, service|
+              t << [ vendor, version_str, service[:description] ]
+            end
+          end
+        end
+      end
+      display services_table
+    end
+
+    def display_provisioned_services(services=nil)
+      services ||= client.services
+      display "\n=========== Provisioned Services ============\n\n"
+      display_provisioned_services_table(services)
+    end
+
+    def display_provisioned_services_table(services)
+      return unless services && !services.empty?
+      services_table = table do |t|
+        t.headings = 'Name', 'Service'
+        services.each do |service|
+          t << [ service[:name], service[:vendor] ]
+        end
+      end
+      display services_table
+    end
+
+    def create_service_banner(service, name, display_name=false)
+      sn = " [#{name}]" if display_name
+      display "Creating Service#{sn}: ", false
+      client.create_service(service, name)
+      display 'OK'.green
+    end
+
+    def bind_service_banner(service, appname, check_restart=true)
+      display "Binding Service: ", false
+      client.bind_service(service, appname)
+      display 'OK'.green
+      check_app_for_restart(appname) if check_restart
+    end
+
+    def unbind_service_banner(service, appname, check_restart=true)
+      display "Unbinding Service: ", false
+      client.unbind_service(service, appname)
+      display 'OK'.green
+      check_app_for_restart(appname) if check_restart
+    end
+
+    def random_service_name(service)
+      r = "%04x" % [rand(0x0100000)]
+      "#{service.to_s}-#{r}"
+    end
+
+    def check_app_for_restart(appname)
+      app = client.app_info(appname)
+      cmd = VMC::Cli::Command::Apps.new(@options)
+      cmd.restart(appname) if app[:state] == 'STARTED'
+    end
+
+  end
+end

data/lib/cli/usage.rb

@@ -0,0 +1,94 @@
+class VMC::Cli::Runner
+
+  def basic_usage
+    "Usage: vmc [options] command [<args>] [command_options]\n" +
+    "Try 'vmc help [command]' or 'vmc help options' for more information."
+  end
+
+  def display_usage
+    if @usage
+      say @usage_error if @usage_error
+      say "Usage: #{@usage}"
+      return
+    elsif @verb_usage
+      say @verb_usage
+      return
+    end
+    say command_usage
+  end
+
+    def command_usage
+    <<-USAGE
+
+#{basic_usage}
+
+Currently available vmc commands are:
+
+  Getting Started
+    target [url]                                 Reports current target or sets a new target
+    login  [email] [--email, --passwd]           Login
+    info                                         System and account information
+
+  Applications
+    apps                                         List deployed applications
+
+  Application Creation
+    push [appname]                               Create, push, map, and start a new application
+    push [appname] --path                        Push application from specified path
+    push [appname] --url                         Set the url for the application
+    push [appname] --instances <N>               Set the expected number <N> of instances
+    push [appname] --mem M                       Set the memory reservation for the application
+    push [appname] --no-start                    Do not auto-start the application
+
+  Application Operations
+    start <appname>                              Start the application
+    stop  <appname>                              Stop the application
+    restart <appname>                            Restart the application
+    delete <appname>                             Delete the application
+    rename <appname> <newname>                   Rename the application
+
+  Application Updates
+    update <appname> [--path]                    Update the application bits
+    mem <appname> [memsize]                      Update the memory reservation for an application
+    map <appname> <url>                          Register the application to the url
+    unmap <appname> <url>                        Unregister the application from the url
+    instances <appname> <num|delta>              Scale the application instances up or down
+
+  Application Information
+    crashes <appname>                            List recent application crashes
+    crashlogs <appname>                          Display log information for crashed applications
+    logs <appname> [--all]                       Display log information for the application
+    files <appname> [path] [--all]               Display directory listing or file download for [path]
+    stats <appname>                              Display resource usage for the application
+    instances <appname>                          List application instances
+
+  Services
+    services                                     Lists of services available and provisioned
+    create-service <service> [--name,--bind]     Create a provisioned service
+    create-service <service> <name>              Create a provisioned service and assign it <name>
+    create-service <service> <name> <app>        Create a provisioned service and assign it <name>, and bind to <app>
+    delete-service [servicename]                 Delete a provisioned service
+    bind-service <servicename> <appname>         Bind a service to an application
+    unbind-service <servicename> <appname>       Unbind service from the application
+    clone-services <src-app> <dest-app>          Clone service bindings from <src-app> application to <dest-app>
+
+  Administration
+    user                                         Display user account information
+    passwd                                       Change the password for the current user
+    logout                                       Logs current user out of the target system
+    add-user [--email, --passwd]                 Register a new user (requires admin privileges)
+    delete-user <user>                           Delete a user and all apps and services (requires admin privileges)
+
+  Misc
+    aliases                                      List aliases
+    alias <alias[=]command>                      Create an alias for a command
+    unalias <alias>                              Remove an alias
+    targets                                      List known targets and associated authorization tokens
+
+  Help
+    help [command]                               Get general help or help on a specific command
+    help options                                 Get help on available options
+USAGE
+
+  end
+end

data/lib/cli/version.rb

@@ -0,0 +1,5 @@
+module VMC
+  module Cli
+    VERSION = '0.2.4'
+  end
+end

data/lib/cli/zip_util.rb

@@ -0,0 +1,61 @@
+
+require 'zip/zipfilesystem'
+
+module VMC::Cli
+
+  class ZipUtil
+
+    class << self
+      def entry_lines(file)
+        contents = nil
+        unless VMC::Cli::Config.nozip
+          contents = `unzip -l #{file} 2> /dev/null`
+          contents = nil if $? != 0
+        end
+        # Do Ruby version if told to or native version failed
+        unless contents
+          entries = []
+          Zip::ZipFile.foreach(file) { |zentry| entries << zentry }
+          contents = entries.join("\n")
+        end
+        contents
+      end
+
+      def unpack(file, dest)
+        unless VMC::Cli::Config.nozip
+          FileUtils.mkdir(dest)
+          `unzip -q #{file} -d #{dest} 2> /dev/null`
+          return unless $? != 0
+        end
+        # Do Ruby version if told to or native version failed
+        Zip::ZipFile.foreach(file) do |zentry|
+          epath = "#{dest}/#{zentry}"
+          dirname = File.dirname(epath)
+          FileUtils.mkdir_p(dirname) unless File.exists?(dirname)
+          zentry.extract(epath) unless File.exists?(epath)
+        end
+      end
+
+      def pack(dir, zipfile)
+        exclude = ['..', '.', '*~', '#*#', '*.log']
+        unless VMC::Cli::Config.nozip
+          excludes = exclude.map { |e| "\\#{e}" }
+          excludes = excludes.join(' ')
+          Dir.chdir(dir) do
+            `zip -q -r -x #{excludes} #{zipfile} . 2> /dev/null`
+            return unless $? != 0
+          end
+        end
+        # Do Ruby version if told to or native version failed
+        Zip::ZipFile::open(zipfile, true) do |zf|
+          Dir.glob("#{dir}/**/*", File::FNM_DOTMATCH).each do |f|
+            process = true
+            exclude.each { |e| process = false if File.fnmatch(e, File.basename(f)) }
+            zf.add(f.sub("#{dir}/",''), f) if (process && File.exists?(f))
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/cli.rb

@@ -0,0 +1,30 @@
+
+ROOT = File.expand_path(File.dirname(__FILE__))
+
+module VMC
+
+  autoload :Client,           "#{ROOT}/vmc/client"
+
+  module Cli
+
+    autoload :Config,         "#{ROOT}/cli/config"
+    autoload :Framework,      "#{ROOT}/cli/frameworks"
+    autoload :Runner,         "#{ROOT}/cli/runner"
+    autoload :ZipUtil,        "#{ROOT}/cli/zip_util"
+    autoload :ServicesHelper, "#{ROOT}/cli/services_helper"
+
+    module Command
+      autoload :Base,         "#{ROOT}/cli/commands/base"
+      autoload :Admin,        "#{ROOT}/cli/commands/admin"
+      autoload :Apps,         "#{ROOT}/cli/commands/apps"
+      autoload :Misc,         "#{ROOT}/cli/commands/misc"
+      autoload :Services,     "#{ROOT}/cli/commands/services"
+      autoload :User,         "#{ROOT}/cli/commands/user"
+    end
+
+  end
+end
+
+require "#{ROOT}/cli/version"
+require "#{ROOT}/cli/core_ext"
+require "#{ROOT}/cli/errors"