lib/osdn/cli/runner.rb

   1 require 'getoptlong'
   2 require 'logger'
   3 require 'pp'
   4
   5 module OSDN
   6   module CLI
   7     class Runner
   8       def initialize
   9         @logger = Logger.new(STDERR)
  10         @logger.level = Logger::INFO
  11         @logger.formatter = proc { |severity, time, progname, msg|
  12           "[%s] %s\n" % [severity, msg]
  13         }
  14         OSDNClient.configure do |config|
  15           ENV['OSDN_API_OVERRIDE_HOST'] and
  16             config.host = ENV['OSDN_API_OVERRIDE_HOST']
  17           ENV['OSDN_API_SKIP_SSL_VERIFY'].to_s =~ /^(1|t(rue)?|y(es)?)$/i and
  18             config.verify_ssl = false
  19             config.verify_ssl_host = false
  20         end
  21       end
  22       attr_reader :logger
  23
  24       def parse_opt
  25         opts = GetoptLong.new(
  26           [ '--help', '-h', GetoptLong::NO_ARGUMENT ],
  27           [ '--verbose', '-v', GetoptLong::NO_ARGUMENT ],
  28           [ '--quiet', '-q', GetoptLong::NO_ARGUMENT ],
  29         )
  30         opts.ordering = GetoptLong::REQUIRE_ORDER
  31         opts.each do |opt, arg|
  32           case opt
  33           when '--help'
  34             help
  35             exit 0
  36           when '--verbose'
  37             if logger.level == Logger::DEBUG
  38               OSDNClient.configure do |config|
  39                 config.debugging = true
  40               end
  41             end
  42             logger.level > Logger::DEBUG and
  43               logger.level -= 1
  44           when '--quiet'
  45             logger.level < Logger::UNKNOWN and
  46               logger.level += 1
  47           when '--help'
  48             help
  49             exit
  50           end
  51         end
  52         logger.debug "Loglevel is #{logger.level}"
  53       end
  54
  55       def get_command_class(command_name)
  56         class_name = command_name.to_s.split('_').map(&:capitalize).join
  57         begin
  58           return self.class.const_get("OSDN::CLI::Command::#{class_name}")
  59         rescue NameError => e
  60           logger.fatal "Invalid command name '#{command_name}'. Use 'help' to list commands."
  61           exit
  62         end
  63         false
  64       end
  65
  66       def run
  67         parse_opt
  68
  69         command_name = ARGV.shift
  70         unless command_name
  71           help
  72           exit 1
  73         end
  74
  75         if command_name == 'help'
  76           help
  77           exit
  78         end
  79
  80         command = get_command_class(command_name).new(logger)
  81         logger.debug "Run command #{command_name}"
  82         begin
  83           command.run
  84         rescue OSDNClient::ApiError => e
  85           begin
  86             err = JSON.parse(e.response_body)
  87             if err["message"]
  88               logger.fatal "#{err["status"]}: #{err["message"]}"
  89             elsif err["error_description"]
  90               logger.fatal err["error_description"]
  91             else
  92               logger.fatal "Command failed by ApiError: #{e.response_body}"
  93             end
  94           rescue
  95             logger.fatal "Command failed: #{e.inspect}"
  96           end
  97         end
  98       end
  99
 100       def help
 101         command_name = ARGV.shift
 102         if command_name
 103           get_command_class(command_name).new(logger).help
 104         else
 105           puts "#{$0} [global-options] <command> [command-options] [args]"
 106           puts "#{$0} help <command>"
 107           puts "Global Options:"
 108           puts "  -h --help      Show help message. use 'help <command>' for specific command. "
 109           puts "  -v --verbose   Increase log level (multiple)"
 110           puts "  -q --quiet     Decrease log level (multiple)"
 111           puts "Avaiable Commands:"
 112           puts "  help"
 113           OSDN::CLI::Command.constants.each do |c|
 114             c = c.to_s.split(/(?=[A-Z])/).join('_').downcase
 115             c == 'base' and next
 116             puts "  %-14s %s" % [c, get_command_class(c).description]
 117           end
 118         end
 119       end
 120     end
 121   end
 122 end