Class: TreasureData::Command::Runner
- Inherits:
-
Object
- Object
- TreasureData::Command::Runner
- Defined in:
- lib/td/command/runner.rb
Instance Attribute Summary collapse
-
#apikey ⇒ Object
Returns the value of attribute apikey.
-
#config_path ⇒ Object
Returns the value of attribute config_path.
-
#prog_name ⇒ Object
Returns the value of attribute prog_name.
-
#secure ⇒ Object
Returns the value of attribute secure.
Instance Method Summary collapse
-
#initialize ⇒ Runner
constructor
A new instance of Runner.
- #run(argv = ARGV) ⇒ Object
Constructor Details
#initialize ⇒ Runner
Returns a new instance of Runner.
7 8 9 10 11 12 |
# File 'lib/td/command/runner.rb', line 7 def initialize @config_path = nil @apikey = nil @prog_name = nil @secure = true end |
Instance Attribute Details
#apikey ⇒ Object
Returns the value of attribute apikey.
14 15 16 |
# File 'lib/td/command/runner.rb', line 14 def apikey @apikey end |
#config_path ⇒ Object
Returns the value of attribute config_path.
14 15 16 |
# File 'lib/td/command/runner.rb', line 14 def config_path @config_path end |
#prog_name ⇒ Object
Returns the value of attribute prog_name.
14 15 16 |
# File 'lib/td/command/runner.rb', line 14 def prog_name @prog_name end |
#secure ⇒ Object
Returns the value of attribute secure.
14 15 16 |
# File 'lib/td/command/runner.rb', line 14 def secure @secure end |
Instance Method Details
#run(argv = ARGV) ⇒ Object
16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 |
# File 'lib/td/command/runner.rb', line 16 def run(argv=ARGV) require 'td/version' require 'td/compat_core' require 'optparse' $prog = @prog_name || File.basename($0) op = OptionParser.new op.version = TreasureData::VERSION op. = <<EOF usage: #{$prog} [options] COMMAND [args] options: EOF op.summary_indent = " " (class << self;self;end).module_eval do define_method(:usage) do |errmsg| require 'td/command/list' puts op.to_s puts "" puts <<EOF Basic commands: db # create/delete/list databases table # create/delete/list/import/export/tail tables query # issue a query job # show/kill/list jobs bulk_import # manage bulk import sessions result # create/delete/list result URLs Additional commands: sched # create/delete/list schedules that run a query periodically schema # create/delete/modify schemas of tables status # show scheds, jobs, tables and results apikey # show/set API key server # show status of the Treasure Data server sample # create a sample log file help # show help messages Type 'td help COMMAND' for more information on a specific command. EOF if errmsg puts "error: #{errmsg}" exit 1 else exit 0 end end end config_path = @config_path apikey = @apikey insecure = nil $verbose = false #$debug = false op.on('-c', '--config PATH', "path to config file (~/.td/td.conf)") {|s| config_path = s } op.on('-k', '--apikey KEY', "use this API key instead of reading the config file") {|s| apikey = s } op.on('--insecure', "Insecure access: disable SSL") { |b| insecure = true } op.on('-v', '--verbose', "verbose mode", TrueClass) {|b| $verbose = b } #op.on('-d', '--debug', "debug mode", TrueClass) {|b| # $debug = b #} op.on('-h', '--help', "show help") { usage nil } begin op.order!(argv) usage nil if argv.empty? cmd = argv.shift require 'td/config' if config_path TreasureData::Config.path = config_path end if apikey TreasureData::Config.apikey = apikey end if insecure TreasureData::Config.secure = false end rescue usage $!.to_s end require 'td/command/list' if defined?(Encoding) #Encoding.default_internal = 'UTF-8' if Encoding.respond_to?(:default_internal) Encoding.default_external = 'UTF-8' if Encoding.respond_to?(:default_external) end method = TreasureData::Command::List.get_method(cmd) unless method $stderr.puts "'#{cmd}' is not a td command. Run '#{$prog}' to show the list." TreasureData::Command::List.show_guess(cmd) exit 1 end begin method.call(argv) rescue TreasureData::ConfigError $stderr.puts "TreasureData account is not configured yet." $stderr.puts "Run '#{$prog} account' first." rescue $stderr.puts "error #{$!.class}: backtrace:" $!.backtrace.each {|b| $stderr.puts " #{b}" } puts "" puts $! end end |