Module: Main::Program::ClassMethods
- Included in:
- Main::Program
- Defined in:
- lib/main/program/class_methods.rb
Defined Under Namespace
Classes: Factory
Instance Method Summary collapse
- #argument(*a, &b) ⇒ Object
- #can_has(ptype, *a, &b) ⇒ Object
-
#chunkname ⇒ Object
TODO - for some reason these hork the usage!.
- #config(*args, &block) ⇒ Object (also: #edit_config_file!)
- #config_path(*config_path) ⇒ Object
- #daemonizes!(*args) ⇒ Object
- #db(*args, &block) ⇒ Object
- #db_path(*db_path) ⇒ Object
- #default_options! ⇒ Object
-
#dynamically_extend_via_commandline_modes! ⇒ Object
extend the class based on modules given in argv.
- #environment(*a, &b) ⇒ Object
- #evaluate(&block) ⇒ Object
- #factory(&block) ⇒ Object (also: #create)
- #factory=(factory) ⇒ Object
-
#fully_qualified_mode ⇒ Object
TODO.
- #has(key, *keys) ⇒ Object
- #input(*args, &block) ⇒ Object
- #io(*args) ⇒ Object
- #keyword(*a, &b) ⇒ Object
- #mixin(name, *names, &block) ⇒ Object
- #mode(name, &block) ⇒ Object
- #mode_name ⇒ Object
- #new ⇒ Object
- #option(*a, &b) ⇒ Object
- #output(*args, &block) ⇒ Object
- #parameter(*a, &b) ⇒ Object
- #params ⇒ Object
- #run(&block) ⇒ Object
- #set_default_options! ⇒ Object
- #state_basename(*args) ⇒ Object
- #state_dirname(*args) ⇒ Object
- #state_path(*state_path, &block) ⇒ Object (also: #dotdir)
- #usage(*args, &block) ⇒ Object
-
#wrap_run! ⇒ Object
wrap up users run method to handle errors, etc.
Instance Method Details
#argument(*a, &b) ⇒ Object
197 198 199 |
# File 'lib/main/program/class_methods.rb', line 197 def argument(*a, &b) (parameters << Parameter.create(:argument, main=self, *a, &b)).last end |
#can_has(ptype, *a, &b) ⇒ Object
225 226 227 228 229 |
# File 'lib/main/program/class_methods.rb', line 225 def can_has(ptype, *a, &b) key = a.map{|s| s.to_s}.sort_by{|s| -s.size }.first can_has_hash.update key => [ptype, a, b] key end |
#chunkname ⇒ Object
TODO - for some reason these hork the usage!
256 257 258 259 260 261 262 263 |
# File 'lib/main/program/class_methods.rb', line 256 %w[ examples samples api ].each do |chunkname| module_eval <<-code def #{ chunkname } *a, &b txt = b ? b.call : a.join("\\n") usage['#{ chunkname }'] = txt end code end |
#config(*args, &block) ⇒ Object Also known as: edit_config_file!
317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 |
# File 'lib/main/program/class_methods.rb', line 317 def config(*args, &block) unless defined?(@config) require 'yaml' unless defined?(YAML) if test(?s, config_path) @config = Map.for(YAML.load(IO.read(config_path))) else config = args.last lines = case config when Hash config.to_yaml.split(/\n/) when String Util.unindent(config).split(/\n/) else [] end dash = lines.shift if lines.first.to_s =~ /^---/ require 'fileutils' unless defined?(FileUtils) FileUtils.mkdir_p(File.dirname(config_path)) open(config_path, 'w') do |fd| fd.puts "## file: #{ config_path }" fd.puts "#" fd.puts fd.puts lines end editor = ENV['EDITOR'] || ENV['EDIT'] || 'vi' system("#{ editor.inspect } #{ config_path }") @config = Map.for(YAML.load(IO.read(config_path))) end end @config end |
#config_path(*config_path) ⇒ Object
351 352 353 354 355 |
# File 'lib/main/program/class_methods.rb', line 351 def config_path(*config_path) @config_path = File.join(state_path, 'config.yml') unless defined?(@config_path) @config_path = File.join(*config_path) unless config_path.empty? @config_path end |
#daemonizes!(*args) ⇒ Object
379 380 381 382 383 384 385 386 387 |
# File 'lib/main/program/class_methods.rb', line 379 def daemonizes!(*args) mode(:daemon){ run { cmd = argv.shift || :usage daemon.cmd(cmd) } } end |
#db(*args, &block) ⇒ Object
301 302 303 304 305 306 307 308 309 |
# File 'lib/main/program/class_methods.rb', line 301 def db(*args, &block) unless defined?(@db) require 'sequel' unless defined?(Sequel) require 'amalgalite' unless defined?(Amalgalite) @db = state_path{ Sequel.amalgalite(db_path) } @db.instance_eval(&block) if block end @db end |
#db_path(*db_path) ⇒ Object
311 312 313 314 315 |
# File 'lib/main/program/class_methods.rb', line 311 def db_path(*db_path) @db_path = File.join(state_path, 'db.sqlite') unless defined?(@db_path) @db_path = File.join(*db_path) unless db_path.empty? @db_path end |
#default_options! ⇒ Object
213 214 215 |
# File 'lib/main/program/class_methods.rb', line 213 def option 'help', 'h' unless parameters.has_option?('help', 'h') end |
#dynamically_extend_via_commandline_modes! ⇒ Object
extend the class based on modules given in argv
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 |
# File 'lib/main/program/class_methods.rb', line 108 def dynamically_extend_via_commandline_modes! self.breadth_first_modes = modes.dup size = modes.size loop do modes.each do |mode| arg = argv.first && %r/^#{ argv.first }/ if arg and mode.name =~ arg argv.shift modes.clear() breadth_first_modes.clear() evaluate(&mode) self.breadth_first_modes = modes.dup depth_first_modes[mode.name] = mode break end end arg = argv.first && %r/^#{ argv.first }/ more_modes = ( !modes.empty? and modes.any?{|mode| arg && mode.name =~ arg} ) break unless more_modes end self.modes = depth_first_modes.dup end |
#environment(*a, &b) ⇒ Object
209 210 211 |
# File 'lib/main/program/class_methods.rb', line 209 def environment(*a, &b) (parameters << Parameter.create(:environment, main=self, *a, &b)).last end |
#evaluate(&block) ⇒ Object
96 97 98 |
# File 'lib/main/program/class_methods.rb', line 96 def evaluate(&block) module_eval(&block) end |
#factory(&block) ⇒ Object Also known as: create
34 35 36 |
# File 'lib/main/program/class_methods.rb', line 34 def factory(&block) Factory.new(&block) end |
#factory=(factory) ⇒ Object
39 40 41 |
# File 'lib/main/program/class_methods.rb', line 39 def factory=(factory) @factory = factory end |
#fully_qualified_mode ⇒ Object
TODO
175 176 177 |
# File 'lib/main/program/class_methods.rb', line 175 def fully_qualified_mode modes.map{|mode| mode.name} end |
#has(key, *keys) ⇒ Object
231 232 233 234 235 236 237 238 239 |
# File 'lib/main/program/class_methods.rb', line 231 def has(key, *keys) keys = [key, *keys].flatten.compact.map{|k| k.to_s} keys.map do |key| ptype, a, b = can_has_hash[key] abort "yo - can *not* has #{ key.inspect }!?" unless(ptype and a and b) send ptype, *a, &b key end end |
#input(*args, &block) ⇒ Object
357 358 359 360 361 362 363 |
# File 'lib/main/program/class_methods.rb', line 357 def input(*args, &block) first = args.first args.push(:input) unless(first.is_a?(Symbol) or first.is_a?(String)) param = argument(*args, &block) param.cast(:input) param end |
#io(*args) ⇒ Object
373 374 375 376 377 |
# File 'lib/main/program/class_methods.rb', line 373 def io(*args) i = input(*[args.shift].compact).default('-') o = output(*[args.shift].compact).default('-') [i, o] end |
#keyword(*a, &b) ⇒ Object
205 206 207 |
# File 'lib/main/program/class_methods.rb', line 205 def keyword(*a, &b) (parameters << Parameter.create(:keyword, main=self, *a, &b)).last end |
#mixin(name, *names, &block) ⇒ Object
241 242 243 244 245 246 247 248 249 250 251 252 |
# File 'lib/main/program/class_methods.rb', line 241 def mixin(name, *names, &block) names = [name, *names].flatten.compact.map{|name| name.to_s} if block names.each do |name| mixin_table[name] = block end else names.each do |name| module_eval(&mixin_table[name]) end end end |
#mode(name, &block) ⇒ Object
217 218 219 220 221 222 223 |
# File 'lib/main/program/class_methods.rb', line 217 def mode(name, &block) name = name.to_s block.fattr(:name => name) modes[name] = block breadth_first_modes[name] = block block end |
#mode_name ⇒ Object
179 180 181 182 |
# File 'lib/main/program/class_methods.rb', line 179 def mode_name return 'main' if modes.empty? fully_qualified_mode.join(' ') end |
#new ⇒ Object
79 80 81 82 83 84 85 86 87 88 89 90 |
# File 'lib/main/program/class_methods.rb', line 79 def new() instance = allocate instance.instance_eval do pre_initialize() before_initialize() main_initialize() initialize() after_initialize() post_initialize() end instance end |
#option(*a, &b) ⇒ Object
201 202 203 |
# File 'lib/main/program/class_methods.rb', line 201 def option(*a, &b) (parameters << Parameter.create(:option, main=self, *a, &b)).last end |
#output(*args, &block) ⇒ Object
365 366 367 368 369 370 371 |
# File 'lib/main/program/class_methods.rb', line 365 def output(*args, &block) first = args.first args.push(:output) unless(first.is_a?(Symbol) or first.is_a?(String)) param = argument(*args, &block) param.cast(:output) param end |
#parameter(*a, &b) ⇒ Object
193 194 195 |
# File 'lib/main/program/class_methods.rb', line 193 def parameter(*a, &b) (parameters << Parameter.create(:parameter, main=self, *a, &b)).last end |
#params ⇒ Object
92 93 94 |
# File 'lib/main/program/class_methods.rb', line 92 def params parameters end |
#run(&block) ⇒ Object
267 268 269 270 |
# File 'lib/main/program/class_methods.rb', line 267 def run(&block) block ||= lambda{} define_method(:run, &block) if block end |
#set_default_options! ⇒ Object
100 101 102 |
# File 'lib/main/program/class_methods.rb', line 100 def option('help', 'h') unless parameters.has_option?('help', 'h') end |
#state_basename(*args) ⇒ Object
279 280 281 282 283 284 |
# File 'lib/main/program/class_methods.rb', line 279 def state_basename(*args) @state_basename = File.join(*args) unless args.empty? @state_basename ||= ENV['STATE_BASENAME'] @state_basename ||= ".#{ name }" @state_basename end |
#state_dirname(*args) ⇒ Object
272 273 274 275 276 277 |
# File 'lib/main/program/class_methods.rb', line 272 def state_dirname(*args) @state_dirname = File.join(*args) unless args.empty? @state_dirname ||= ENV['STATE_DIRNAME'] @state_dirname ||= Util.home @state_dirname end |
#state_path(*state_path, &block) ⇒ Object Also known as: dotdir
286 287 288 289 290 291 292 293 294 295 296 297 298 |
# File 'lib/main/program/class_methods.rb', line 286 def state_path(*state_path, &block) @state_path = File.join(state_dirname, state_basename) unless defined?(@state_path) @state_path = state_path.join('/') unless state_path.empty? if block require 'fileutils' unless defined?(FileUtils) FileUtils.mkdir_p(@state_path) unless test(?d, @state_path) Dir.chdir(&block) else @state_path end end |
#usage(*args, &block) ⇒ Object
185 186 187 188 189 190 191 |
# File 'lib/main/program/class_methods.rb', line 185 def usage(*args, &block) usage! unless defined? @usage return @usage if args.empty? and block.nil? key, value, *ignored = args value = block.call if block @usage[key.to_s] = value.to_s end |
#wrap_run! ⇒ Object
wrap up users run method to handle errors, etc
139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 |
# File 'lib/main/program/class_methods.rb', line 139 def wrap_run! evaluate do alias_method 'run!', 'run' def run() exit_status = catch :exit do begin parse_parameters if help? puts(usage.to_s) exit end pre_run before_run run! after_run post_run finalize rescue Object => exception self.exit_status ||= exception.status if exception.respond_to?(:status) handle_exception(exception) end nil end self.exit_status ||= (exit_status || exit_success) handle_exit(self.exit_status) end end end |