optimus-prime 0.0.4

data/lib/optimus_prime.rb

@@ -0,0 +1,65 @@
+require 'optparse'
+
+$LOAD_PATH.unshift File.dirname(__FILE__)
+
+require 'optimus_prime/core_ext/object'
+require 'optimus_prime/command'
+require 'optimus_prime/optor'
+
+module OptimusPrime
+  def self.included(klass)
+    klass.class_eval do
+      extend ClassMethods
+
+      command :help do |cmd|
+        ##
+        # Show this help message
+        if cmd and self.class.commands.include?(cmd)
+          puts help(cmd)
+        else
+          puts "Commands:"
+          puts self.class.commands.map { |name| "- #{name}" }
+        end
+      end
+
+      def initialize
+        self.class.init(self)
+      end
+
+      def help(name)
+        self.class.help(name)
+      end
+    end
+  end
+
+  module ClassMethods
+    def __optor__
+      @__optor__ ||= Optor.new(self, ARGV.dup)
+    end
+
+    def init(instance)
+      __optor__.init(instance)
+    end
+
+    def help(name)
+      __optor__.help(name)
+    end
+
+    def command(name, &block)
+      __optor__.command(name, block)
+    end
+
+    def commands
+      __optor__.commands.keys
+    end
+
+    def flag(*flags)
+      flags.each { |name| __optor__.flag(name) }
+    end
+
+    def option(*names)
+      options = names.last.is_a?(Hash) ? names.pop : {}
+      names.each { |name| __optor__.option(name, options) }
+    end
+  end
+end

data/lib/optimus_prime/command.rb

@@ -0,0 +1,41 @@
+module OptimusPrime
+  class Command
+    def initialize(handler, caller=nil)
+      @handler, @caller = handler, (caller && caller[1])
+    end
+
+    def help
+      @help ||= begin
+        lines = file.split(/\n/)
+        result = []
+        i = 0
+        while (string = lines[line+i]) =~ /\s*##?/
+          result << string.gsub(/^\s*#*\s?/, '')
+          i += 1
+        end
+        result.shift if result.first.empty?
+        result.pop   if result.last.empty?
+        result.join("\n")
+      end
+    end
+
+    def file
+      @file ||= File.read(@caller.split(':').first)
+    end
+
+    def line
+      @caller.split(':').last.to_i
+    end
+
+    def arity
+      @handler.arity
+    end
+
+    def to_proc
+      case handler = @handler
+      when Proc then handler
+      else @handler.to_proc
+      end
+    end
+  end
+end

data/lib/optimus_prime/core_ext/object.rb

@@ -0,0 +1,22 @@
+class Object
+  unless respond_to?(:instance_exec)
+    module InstanceExecHelper; end
+    include InstanceExecHelper
+    def instance_exec(*args, &block)
+      begin
+        old_critical, Thread.critical = Thread.critical, true
+        n = 0
+        n += 1 while respond_to?(mname="__instance_exec#{n}")
+        InstanceExecHelper.module_eval{ define_method(mname, &block) }
+      ensure
+        Thread.critical = old_critical
+      end
+      begin
+        ret = send(mname, *args)
+      ensure
+        InstanceExecHelper.module_eval{ remove_method(mname) } rescue nil
+      end
+      ret
+    end
+  end
+end

data/lib/optimus_prime/optor.rb

@@ -0,0 +1,74 @@
+module OptimusPrime
+  class Optor
+    def initialize(klass, args)
+      @klass, @args = klass, args
+    end
+
+    def options
+      @options ||= begin
+        list = [''] # need single options for getopts. psh.
+        list += @option_list || []
+        OptionParser.getopts(@args, *list)
+      end
+    end
+
+    def commands
+      @commands ||= {}
+    end
+
+    def command(name, handler)
+      commands[name.to_s] = handler && Command.new(handler, caller)
+    end
+
+    def help(name)
+      commands[name].help
+    end
+
+    def flag(name)
+      @option_list ||= []
+      @option_list << name.to_s
+    end
+
+    def option(name, options={})
+      @option_list ||= []
+      @option_list << name.to_s + ':'
+
+      if options[:prompt]
+        @prompt_list ||= {}
+        @prompt_list[name.to_s] = options[:prompt]
+      end
+    end
+
+    def init(instance)
+      options.each do |key,val|
+        instance.instance_variable_set("@#{key}", val)
+      end
+
+      Array(@prompt_list).each do |key, prompt|
+        next if options[key]
+        $stdout.print prompt + ' '
+        options[key] = $stdin.gets.chomp
+        instance.instance_variable_set("@#{key}", options[key])
+      end
+
+      args = @args.dup
+      args.each do |val|
+        args.delete(val)
+        run_command(instance, val, args)
+      end
+    end
+
+    private
+
+    def get_command(instance, name)
+      Command.new(@klass.instance_method(name).bind(instance))
+    end
+
+    def run_command(instance, name, args)
+      command = commands[name] || get_command(instance, name)
+      block_args = []
+      command.arity.times { block_args << args.shift }
+      instance.instance_exec(*block_args, &command)
+    end
+  end
+end

metadata

@@ -0,0 +1,58 @@
+--- !ruby/object:Gem::Specification 
+name: optimus-prime
+version: !ruby/object:Gem::Version 
+  version: 0.0.4
+platform: ruby
+authors: 
+- Pat Nakajima
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-07-10 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: 
+email: patnakajima@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/optimus_prime/command.rb
+- lib/optimus_prime/optor.rb
+- lib/optimus_prime.rb
+- lib/optimus_prime/core_ext/object.rb
+has_rdoc: true
+homepage: http://github.com/nakajima/optimus-prime
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Easy command line options.
+test_files: []
+