# File lib/main/program/class_methods.rb, line 196 def argument(*a, &b) (parameters << Parameter.create(:argument, main=self, *a, &b)).last end
# File lib/main/program/class_methods.rb, line 224 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
# File lib/main/program/class_methods.rb, line 301 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.is_a?(Hash) ? args.last : {} lines = config.to_yaml.split(/\n/) dash = lines.shift open(config_path, 'w') do |fd| fd.puts '### you may need to edit this config!' 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
# File lib/main/program/class_methods.rb, line 323 def config_path(*config_path) @config_path = File.join(dotdir, 'config.yml') unless defined?(@config_path) @config_path = File.join(*config_path) unless config_path.empty? @config_path end
# File lib/main/program/class_methods.rb, line 285 def db(*args, &block) unless defined?(@db) require 'sequel' unless defined?(Sequel) require 'amalgalite' unless defined?(Amalgalite) @db = dotdir{ Sequel.amalgalite(db_path) } @db.instance_eval(&block) if block end @db end
# File lib/main/program/class_methods.rb, line 295 def db_path(*db_path) @db_path = File.join(dotdir, 'db.sqlite') unless defined?(@db_path) @db_path = File.join(*db_path) unless db_path.empty? @db_path end
# File lib/main/program/class_methods.rb, line 212 def default_options! option 'help', 'h' unless parameters.has_option?('help', 'h') end
# File lib/main/program/class_methods.rb, line 271 def dotdir(*dotdir, &block) @dotdir = File.join(Util.home, ".#{ name }") unless defined?(@dotdir) @dotdir = dotdir.join('/') unless dotdir.empty? if block require 'fileutils' unless defined?(FileUtils) FileUtils.mkdir_p(@dotdir) unless test(d, @dotdir) Dir.chdir(&block) else @dotdir end end
extend the class based on modules given in argv
# File lib/main/program/class_methods.rb, line 107 def dynamically_extend_via_commandline_modes! self.breadth_first_modes = modes.dup size = modes.size loop do modes.each do |mode| arg = argv.first && /^#{ 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 && /^#{ 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
# File lib/main/program/class_methods.rb, line 208 def environment(*a, &b) (parameters << Parameter.create(:environment, main=self, *a, &b)).last end
# File lib/main/program/class_methods.rb, line 95 def evaluate(&block) module_eval(&block) end
# File lib/main/program/class_methods.rb, line 33 def factory(&block) Factory.new(&block) end
# File lib/main/program/class_methods.rb, line 38 def factory=(factory) @factory = factory end
TODO
# File lib/main/program/class_methods.rb, line 174 def fully_qualified_mode modes.map{|mode| mode.name} end
# File lib/main/program/class_methods.rb, line 230 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
# File lib/main/program/class_methods.rb, line 329 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
# File lib/main/program/class_methods.rb, line 345 def io(*args) i = input(*[args.shift].compact).default('-') o = output(*[args.shift].compact).default('-') [i, o] end
# File lib/main/program/class_methods.rb, line 204 def keyword(*a, &b) (parameters << Parameter.create(:keyword, main=self, *a, &b)).last end
# File lib/main/program/class_methods.rb, line 240 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
# File lib/main/program/class_methods.rb, line 216 def mode(name, &block) name = name.to_s block.fattr(:name => name) modes[name] = block breadth_first_modes[name] = block block end
# File lib/main/program/class_methods.rb, line 178 def mode_name return 'main' if modes.empty? fully_qualified_mode.join(' ') end
# File lib/main/program/class_methods.rb, line 78 def new() instance = allocate instance.instance_eval do pre_initialize() before_initialize() main_initialize() initialize() after_initialize() post_initialize() end instance end
# File lib/main/program/class_methods.rb, line 200 def option(*a, &b) (parameters << Parameter.create(:option, main=self, *a, &b)).last end
# File lib/main/program/class_methods.rb, line 337 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
# File lib/main/program/class_methods.rb, line 192 def parameter(*a, &b) (parameters << Parameter.create(:parameter, main=self, *a, &b)).last end
# File lib/main/program/class_methods.rb, line 91 def params parameters end
# File lib/main/program/class_methods.rb, line 142 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
# File lib/main/program/class_methods.rb, line 99 def set_default_options! option('help', 'h') unless parameters.has_option?('help', 'h') end
# File lib/main/program/class_methods.rb, line 184 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 up users run method to handle errors, etc
# File lib/main/program/class_methods.rb, line 138 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