jherdman-thor 0.9.5

data/CHANGELOG.rdoc

@@ -0,0 +1,35 @@
+== TODO
+
+* Change Thor.start to parse ARGV in a single pass
+* Improve spec coverage for Thor::Runner
+* Improve help output to list shorthand switches, too
+* Investigate and fix deep namespacing ("foo:bar:baz") issues
+
+== 0.9.5, released 2008-08-27
+
+* Improve Windows compatibility
+* Update (incorrect) README and task.thor sample file
+* Options hash is now frozen (once returned)
+* Allow magic predicates on options object. For instance: `options.force?`
+* Add support for :numeric type
+* BACKWARDS INCOMPATIBLE: Refactor Thor::Options. You cannot access shorthand forms in options hash anymore (for instance, options[:f])
+* Allow specifying optional args with default values: method_options(:user => "mislav")
+* Don't write options for nil or false values. This allows, for example, turning color off when running specs.
+* Exit with the status of the spec command to help CI stuff out some.
+
+== 0.9.4, released 2008-08-13
+
+* Try to add Windows compatibility.
+* BACKWARDS INCOMPATIBLE: options hash is now accessed as a property in your class and is not passed as last argument anymore
+* Allow options at the beginning of the argument list as well as the end.
+* Make options available with symbol keys in addition to string keys.
+* Allow true to be passed to Thor#method_options to denote a boolean option.
+* If loading a thor file fails, don't give up, just print a warning and keep going.
+* Make sure that we re-raise errors if they happened further down the pipe than we care about.
+* Only delete the old file on updating when the installation of the new one is a success
+* Make it Ruby 1.8.5 compatible.
+* Don't raise an error if a boolean switch is defined multiple times.
+* Thor::Options now doesn't parse through things that look like options but aren't.
+* Add URI detection to install task, and make sure we don't append ".thor" to URIs
+* Add rake2thor to the gem binfiles.
+* Make sure local Thorfiles override system-wide ones.

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2008 Yehuda Katz
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.markdown

@@ -0,0 +1,76 @@
+thor
+====
+
+Map options to a class. Simply create a class with the appropriate annotations, and have options automatically map
+to functions and parameters.
+
+Example:
+
+    class MyApp < Thor                                                # [1]
+      map "-L" => :list                                               # [2]
+                                                                    
+      desc "install APP_NAME", "install one of the available apps"    # [3]
+      method_options :force => :boolean, :alias => :optional          # [4]
+      def install(name)
+        user_alias = options[:alias]
+        if options.force?
+          # do something
+        end
+        # ... other code ...
+      end
+      
+      desc "list [SEARCH]", "list all of the available apps, limited by SEARCH"
+      def list(search = "")
+        # list everything
+      end
+    end
+    
+Thor automatically maps commands as such:
+
+    app install myname --force
+    
+That gets converted to:
+
+    MyApp.new.install("myname")
+    # with {'force' => true} as options hash
+
+1.  Inherit from Thor to turn a class into an option mapper
+2.  Map additional non-valid identifiers to specific methods. In this case,
+    convert -L to :list
+3.  Describe the method immediately below. The first parameter is the usage information,
+    and the second parameter is the description.
+4.  Provide any additional options. These will be marshaled from `--` and `-` params.
+    In this case, a `--force` and a `-f` option is added.
+    
+Types for `method_options`
+--------------------------
+
+<dl>
+  <dt><code>:boolean</code></dt>
+    <dd>true if the option is passed</dd>
+  <dt><code>true</code></dt>
+    <dd>same as <code>:boolean</code></dd>
+  <dt><code>:required</code></dt>
+    <dd>the value for this option MUST be provided</dd>
+  <dt><code>:optional</code></dt>
+    <dd>the value for this option MAY be provided</dd>
+  <dt><code>:numeric</code></dt>
+    <dd>the value MAY be provided, but MUST be in numeric form</dd>
+  <dt>a String or Numeric</dt>
+    <dd>same as <code>:optional</code>, but fall back to the given object as default value</dd>
+</dl>
+
+In case of unsatisfied requirements, `Thor::Options::Error` is raised.
+
+Examples of option parsing:
+
+    # let's say this is how we defined options for a method:
+    method_options(:force => :boolean, :retries => :numeric)
+    
+    # here is how the following command-line invocations would be parsed:
+    
+    command -f --retries 5    # => {'force' => true, 'retries' => 5}
+    command --force -r=5      # => {'force' => true, 'retries' => 5}
+    command -fr 5             # => {'force' => true, 'retries' => 5}
+    command --retries=5       # => {'retries' => 5}
+    command -r5               # => {'retries' => 5}

data/Rakefile

@@ -0,0 +1,6 @@
+task :default => :install
+
+desc "install the gem locally"
+task :install do
+  sh %{ruby "#{File.dirname(__FILE__)}/bin/thor" :install}
+end

data/bin/rake2thor

@@ -0,0 +1,83 @@
+#!/usr/bin/env ruby
+
+require 'rubygems'
+require 'ruby2ruby'
+require 'rake'
+
+input  = ARGV[0] || 'Rakefile'
+output = ARGV[1] || 'Thorfile'
+
+$requires = []
+
+module Kernel
+  def require_with_record(file)
+    $requires << file if caller[1] =~ /rake2thor:/
+    require_without_record file
+  end
+  alias_method :require_without_record, :require
+  alias_method :require, :require_with_record
+end
+
+load input
+
+@private_methods = []
+
+def file_task_name(name)
+  "compile_" + name.gsub('/', '_slash_').gsub('.', '_dot_').gsub(/\W/, '_')
+end
+
+def method_for_task(task)
+  file_task = task.is_a?(Rake::FileTask)
+  comment = task.instance_variable_get('@comment')
+  prereqs = task.instance_variable_get('@prerequisites').select(&Rake::Task.method(:task_defined?))
+  actions = task.instance_variable_get('@actions')
+  name = task.name.gsub(/^([^:]+:)+/, '')
+  name = file_task_name(name) if file_task
+  meth = ''
+
+  meth << "desc #{name.inspect}, #{comment.inspect}\n" if comment
+  meth << "def #{name}\n"
+
+  meth << prereqs.map do |pre|
+    pre = pre.to_s
+    pre = file_task_name(pre) if Rake::Task[pre].is_a?(Rake::FileTask)
+    '  ' + pre
+  end.join("\n")
+
+  meth << "\n\n" unless prereqs.empty? || actions.empty?
+
+  meth << actions.map do |act|
+    act = act.to_ruby
+    unless act.gsub!(/^proc \{ \|(\w+)\|\n/,
+                     "  \\1 = Struct.new(:name).new(#{name.inspect}) # A crude mock Rake::Task object\n")
+      act.gsub!(/^proc \{\n/, '')
+    end
+    act.gsub(/\n\}$/, '')
+  end.join("\n")
+
+  meth << "\nend"
+
+  if file_task
+    @private_methods << meth
+    return
+  end
+
+  meth
+end
+
+body = Rake::Task.tasks.map(&method(:method_for_task)).compact.map { |meth| meth.gsub(/^/, '  ') }.join("\n\n")
+
+unless @private_methods.empty?
+  body << "\n\n  private\n\n"
+  body << @private_methods.map { |meth| meth.gsub(/^/, '  ') }.join("\n\n")
+end
+
+requires = $requires.map { |r| "require #{r.inspect}" }.join("\n")
+
+File.open(output, 'w') { |f| f.write(<<END.lstrip) }
+#{requires}
+
+class Default < Thor
+#{body}
+end
+END

data/bin/thor

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+# -*- mode: ruby -*-
+
+require File.dirname(__FILE__) + "/../lib/thor"
+require 'thor/runner'
+
+Thor::Runner.start

data/lib/thor.rb

@@ -0,0 +1,138 @@
+$:.unshift File.expand_path(File.dirname(__FILE__))
+require "thor/options"
+require "thor/util"
+require "thor/task"
+require "thor/task_hash"
+
+class Thor
+  attr_accessor :options
+  
+  def self.map(map)
+    @map ||= superclass.instance_variable_get("@map") || {}
+    map.each do |key, value|
+      if key.respond_to?(:each)
+        key.each {|subkey| @map[subkey] = value}
+      else
+        @map[key] = value
+      end
+    end
+  end
+
+  def self.desc(usage, description)
+    @usage, @desc = usage, description
+  end
+  
+  def self.method_options(opts)
+    @method_options = opts
+  end
+
+  def self.subclass_files
+    @subclass_files ||= Hash.new {|h,k| h[k] = []}
+  end
+  
+  def self.subclasses
+    @subclasses ||= []
+  end
+  
+  def self.tasks
+    @tasks ||= TaskHash.new(self)
+  end
+
+  def self.opts
+    (@opts || {}).merge(self == Thor ? {} : superclass.opts)
+  end
+
+  def self.[](task)
+    namespaces = task.split(":")
+    klass = Thor::Util.constant_from_thor_path(namespaces[0...-1].join(":"))
+    raise Error, "`#{klass}' is not a Thor class" unless klass <= Thor
+    klass.tasks[namespaces.last]
+  end
+
+  def self.maxima
+    @maxima ||= begin
+      max_usage = tasks.map {|_, t| t.usage}.max {|x,y| x.to_s.size <=> y.to_s.size}.size
+      max_desc  = tasks.map {|_, t| t.description}.max {|x,y| x.to_s.size <=> y.to_s.size}.size
+      max_opts  = tasks.map {|_, t| t.opts ? t.opts.formatted_usage : ""}.max {|x,y| x.to_s.size <=> y.to_s.size}.size
+      Struct.new(:description, :usage, :opt).new(max_desc, max_usage, max_opts)
+    end
+  end
+  
+  def self.start(args = ARGV)
+    options = Thor::Options.new(self.opts)
+    opts = options.parse(args, false)
+    args = options.trailing_non_opts
+
+    meth = args.first
+    meth = @map[meth].to_s if @map && @map[meth]
+    meth ||= "help"
+    
+    tasks[meth].parse new(opts, *args), args[1..-1]
+  rescue Thor::Error => e
+    $stderr.puts e.message
+  end
+
+  class << self
+    protected
+    def inherited(klass)
+      register_klass_file klass
+    end
+
+    def method_added(meth)
+      meth = meth.to_s
+
+      if meth == "initialize"
+        @opts = @method_options
+        @method_options = nil
+        return
+      end
+
+      return if !public_instance_methods.include?(meth) || !@usage
+      register_klass_file self
+
+      tasks[meth] = Task.new(meth, @desc, @usage, @method_options)
+
+      @usage, @desc, @method_options = nil
+    end
+
+    def register_klass_file(klass, file = caller[1].split(":")[0])
+      unless self == Thor
+        superclass.register_klass_file(klass, file)
+        return
+      end
+
+      file_subclasses = subclass_files[File.expand_path(file)]
+      file_subclasses << klass unless file_subclasses.include?(klass)
+      subclasses << klass unless subclasses.include?(klass)
+    end
+  end
+
+  def initialize(opts = {}, *args)
+  end
+  
+  map ["-h", "-?", "--help", "-D"] => :help
+  
+  desc "help [TASK]", "describe available tasks or one specific task"
+  def help(task = nil)
+    if task
+      if task.include? ?:
+        task = self.class[task]
+        namespace = true
+      else
+        task = self.class.tasks[task]
+      end
+
+      puts task.formatted_usage(namespace)
+      puts task.description
+    else
+      puts "Options"
+      puts "-------"
+      self.class.tasks.each do |_, task|
+        format = "%-" + (self.class.maxima.usage + self.class.maxima.opt + 4).to_s + "s"
+        print format % ("#{task.formatted_usage}")      
+        puts  task.description.split("\n").first
+      end
+    end
+  end
+  
+end

data/lib/thor/error.rb

@@ -0,0 +1,3 @@
+class Thor
+  class Error < Exception; end
+end

data/lib/thor/options.rb

@@ -0,0 +1,238 @@
+# This is a modified version of Daniel Berger's Getopt::Long class,
+# licensed under Ruby's license.
+
+class Thor
+  class Options
+    class Error < StandardError; end
+    
+    # simple Hash with indifferent access
+    class Hash < ::Hash
+      def initialize(hash)
+        super()
+        update hash
+      end
+      
+      def [](key)
+        super convert_key(key)
+      end
+      
+      def values_at(*indices)
+        indices.collect { |key| self[convert_key(key)] }
+      end
+      
+      protected
+        def convert_key(key)
+          key.kind_of?(Symbol) ? key.to_s : key
+        end
+        
+        # Magic predicates. For instance:
+        #   options.force? # => !!options['force']
+        def method_missing(method, *args, &block)
+          method.to_s =~ /^(\w+)\?$/ ? !!self[$1] : super
+        end
+    end
+
+    NUMERIC     = /(\d*\.\d+|\d+)/
+    LONG_RE     = /^(--\w+[-\w+]*)$/
+    SHORT_RE    = /^(-[a-z])$/i
+    EQ_RE       = /^(--\w+[-\w+]*|-[a-z])=(.*)$/i
+    SHORT_SQ_RE = /^-([a-z]{2,})$/i # Allow either -x -v or -xv style for single char args
+    SHORT_NUM   = /^(-[a-z])#{NUMERIC}$/i
+    
+    attr_reader :leading_non_opts, :trailing_non_opts
+    
+    def non_opts
+      leading_non_opts + trailing_non_opts
+    end
+
+    # Takes an array of switches. Each array consists of up to three
+    # elements that indicate the name and type of switch. Returns a hash
+    # containing each switch name, minus the '-', as a key. The value
+    # for each key depends on the type of switch and/or the value provided
+    # by the user.
+    #
+    # The long switch _must_ be provided. The short switch defaults to the
+    # first letter of the short switch. The default type is :boolean.
+    #
+    # Example:
+    #
+    #   opts = Thor::Options.new(
+    #      "--debug" => true,
+    #      ["--verbose", "-v"] => true,
+    #      ["--level", "-l"] => :numeric
+    #   ).parse(args)
+    #
+    def initialize(switches)
+      @defaults = {}
+      @shorts = {}
+
+      @switches = switches.inject({}) do |mem, (name, type)|
+        if name.is_a?(Array)
+          name, *shorts = name
+        else
+          name = name.to_s
+          shorts = []
+        end
+        # we need both nice and dasherized form of switch name
+        if name.index('-') == 0
+          nice_name = undasherize name
+        else
+          nice_name = name
+          name = dasherize name
+        end
+        # if there are no shortcuts specified, generate one using the first character
+        shorts << "-" + nice_name[0,1] if shorts.empty? and nice_name.length > 1
+        shorts.each { |short| @shorts[short] = name }
+        
+        # normalize type
+        case type
+        when TrueClass then type = :boolean
+        when String
+          @defaults[nice_name] = type
+          type = :optional
+        when Numeric
+          @defaults[nice_name] = type
+          type = :numeric
+        end
+        
+        mem[name] = type
+        mem
+      end
+      
+      # remove shortcuts that happen to coincide with any of the main switches
+      @shorts.keys.each do |short|
+        @shorts.delete(short) if @switches.key?(short)
+      end
+    end
+
+    def parse(args, skip_leading_non_opts = true)
+      @args = args
+      # start with Thor::Options::Hash pre-filled with defaults
+      hash = Hash.new @defaults
+      
+      @leading_non_opts = []
+      if skip_leading_non_opts
+        @leading_non_opts << shift until current_is_option? || @args.empty?
+      end
+
+      while current_is_option?
+        case shift
+        when SHORT_SQ_RE
+          unshift $1.split('').map { |f| "-#{f}" }
+          next
+        when EQ_RE, SHORT_NUM
+          unshift $2
+          switch = $1
+        when LONG_RE, SHORT_RE
+          switch = $1
+        end
+        
+        switch    = normalize_switch(switch)
+        nice_name = undasherize(switch)
+        type      = switch_type(switch)
+        
+        case type
+        when :required
+          assert_value!(switch)
+          raise Error, "cannot pass switch '#{peek}' as an argument" if valid?(peek)
+          hash[nice_name] = shift
+        when :optional
+          hash[nice_name] = peek.nil? || valid?(peek) || shift
+        when :boolean
+          hash[nice_name] = true
+        when :numeric
+          assert_value!(switch)
+          unless peek =~ NUMERIC and $& == peek
+            raise Error, "expected numeric value for '#{switch}'; got #{peek.inspect}"
+          end
+          hash[nice_name] = $&.index('.') ? shift.to_f : shift.to_i
+        end
+      end
+      
+      @trailing_non_opts = @args
+
+      check_required! hash
+      hash.freeze
+      hash
+    end
+    
+    def formatted_usage
+      return "" if @switches.empty?
+      @switches.map do |opt, type|
+        case type
+        when :boolean
+          "[#{opt}]"
+        when :required
+          opt + "=" + opt.gsub(/\-/, "").upcase
+        else
+          sample = @defaults[undasherize(opt)]
+          sample ||= case type
+            when :optional then opt.gsub(/\-/, "").upcase
+            when :numeric  then "N"
+            end
+          "[" + opt + "=" + sample.to_s + "]"
+        end
+      end.join(" ")
+    end
+
+    private
+    
+    def assert_value!(switch)
+      raise Error, "no value provided for argument '#{switch}'" if peek.nil?
+    end
+    
+    def undasherize(str)
+      str.sub(/^-{1,2}/, '')
+    end
+    
+    def dasherize(str)
+      (str.length > 1 ? "--" : "-") + str
+    end
+    
+    def peek
+      @args.first
+    end
+
+    def shift
+      @args.shift
+    end
+
+    def unshift(arg)
+      unless arg.kind_of?(Array)
+        @args.unshift(arg)
+      else
+        @args = arg + @args
+      end
+    end
+    
+    def valid?(arg)
+      @switches.key?(arg) or @shorts.key?(arg)
+    end
+
+    def current_is_option?
+      case peek
+      when LONG_RE, SHORT_RE, EQ_RE, SHORT_NUM
+        valid?($1)
+      when SHORT_SQ_RE
+        $1.split('').any? { |f| valid?("-#{f}") }
+      end
+    end
+    
+    def normalize_switch(switch)
+      @shorts.key?(switch) ? @shorts[switch] : switch
+    end
+    
+    def switch_type(switch)
+      @switches[switch]
+    end
+    
+    def check_required!(hash)
+      for name, type in @switches
+        if type == :required and !hash[undasherize(name)]
+          raise Error, "no value provided for required argument '#{name}'"
+        end
+      end
+    end
+    
+  end
+end

data/lib/thor/ordered_hash.rb

@@ -0,0 +1,64 @@
+class Thor
+  # This class is based on the Ruby 1.9 ordered hashes.
+  # It keeps the semantics and most of the efficiency of normal hashes
+  # while also keeping track of the order in which elements were set.
+  class OrderedHash
+    Node = Struct.new(:key, :value, :next, :prev)
+    include Enumerable
+
+    def initialize
+      @hash = {}
+    end
+
+    def initialize_copy(other)
+      @hash = other.instance_variable_get('@hash').clone
+    end
+
+    def [](key)
+      @hash[key] && @hash[key].value
+    end
+
+    def []=(key, value)
+      node = Node.new(key, value)
+
+      if old = @hash[key]
+        if old.prev
+          old.prev.next = old.next
+        else # old is @first and @last
+          @first = @last = nil
+        end
+      end
+
+      if @first.nil?
+        @first = @last = node
+      else
+        node.prev = @last
+        @last.next = node
+        @last = node
+      end
+
+      @hash[key] = node
+      value
+    end
+
+    def each
+      return unless @first
+      yield [@first.key, @first.value]
+      node = @first
+      yield [node.key, node.value] while node = node.next
+      self
+    end
+
+    def values
+      self.map { |k, v| v }
+    end
+
+    def +(other)
+      new = clone
+      other.each do |key, value|
+        new[key] = value unless self[key]
+      end
+      new
+    end
+  end
+end

data/lib/thor/runner.rb

@@ -0,0 +1,260 @@
+require 'thor'
+require "thor/util"
+require "open-uri"
+require "fileutils"
+require "yaml"
+require "digest/md5"
+require "readline"
+
+class Thor::Runner < Thor
+  
+  def self.globs_for(path)
+    ["#{path}/Thorfile", "#{path}/*.thor", "#{path}/tasks/*.thor", "#{path}/lib/tasks/*.thor"]
+  end
+
+  map "-T" => :list, "-i" => :install, "-u" => :update
+  
+  desc "install NAME", "install a Thor file into your system tasks, optionally named for future updates"
+  method_options :as => :optional, :relative => :boolean
+  def install(name)
+    initialize_thorfiles
+    begin
+      contents = open(name).read
+    rescue OpenURI::HTTPError
+      raise Error, "Error opening URI `#{name}'"
+    rescue Errno::ENOENT
+      raise Error, "Error opening file `#{name}'"
+    end
+    
+    is_uri = File.exist?(name) ? false : true
+    
+    puts "Your Thorfile contains: "
+    puts contents
+    print "Do you wish to continue [y/N]? "
+    response = Readline.readline
+    
+    return false unless response =~ /^\s*y/i
+    
+    constants = Thor::Util.constants_in_contents(contents)
+    
+    # name = name =~ /\.thor$/ || is_uri ? name : "#{name}.thor"
+    
+    as = options["as"] || begin
+      first_line = contents.split("\n")[0]
+      (match = first_line.match(/\s*#\s*module:\s*([^\n]*)/)) ? match[1].strip : nil
+    end
+        
+    if !as
+      print "Please specify a name for #{name} in the system repository [#{name}]: "
+      as = Readline.readline
+      as = name if as.empty?
+    end
+    
+    FileUtils.mkdir_p thor_root
+    
+    yaml_file = File.join(thor_root, "thor.yml")
+    FileUtils.touch(yaml_file)
+    yaml = thor_yaml
+    
+    location = (options[:relative] || is_uri) ? name : File.expand_path(name)
+    yaml[as] = {:filename => Digest::MD5.hexdigest(name + as), :location => location, :constants => constants}
+    
+    save_yaml(yaml)
+    
+    puts "Storing thor file in your system repository"
+    
+    File.open(File.join(thor_root, yaml[as][:filename]), "w") do |file|
+      file.puts contents
+    end
+    
+    yaml[as][:filename] # Indicate sucess
+  end
+  
+  desc "uninstall NAME", "uninstall a named Thor module"
+  def uninstall(name)
+    yaml = thor_yaml
+    raise Error, "Can't find module `#{name}'" unless yaml[name]
+    
+    puts "Uninstalling #{name}."
+    
+    file = File.join(thor_root, "#{yaml[name][:filename]}")
+    File.delete(file)
+    yaml.delete(name)
+    save_yaml(yaml)
+    
+    puts "Done."
+  end
+  
+  desc "update NAME", "update a Thor file from its original location"
+  def update(name)
+    yaml = thor_yaml
+    raise Error, "Can't find module `#{name}'" if !yaml[name] || !yaml[name][:location]
+
+    puts "Updating `#{name}' from #{yaml[name][:location]}"
+    old_filename = yaml[name][:filename]
+    options["as"] = name
+    filename = install(yaml[name][:location])
+    unless filename == old_filename
+      File.delete(File.join(thor_root, old_filename))
+    end
+  end
+  
+  desc "installed", "list the installed Thor modules and tasks (--internal means list the built-in tasks as well)"
+  method_options :internal => :boolean
+  def installed
+    thor_root_glob.each do |f|
+      next if f =~ /thor\.yml$/
+      load_thorfile f unless Thor.subclass_files.keys.include?(File.expand_path(f))
+    end
+
+    klasses = Thor.subclasses
+    klasses -= [Thor, Thor::Runner] unless options['internal']
+    display_klasses(true, klasses)
+  end
+  
+  desc "list [SEARCH]", "list the available thor tasks (--substring means SEARCH can be anywhere in the module)"
+  method_options :substring => :boolean
+  def list(search = "")
+    initialize_thorfiles
+    search = ".*#{search}" if options["substring"]
+    search = /^#{search}.*/i
+    
+    display_klasses(false, Thor.subclasses.select {|k| 
+      Thor::Util.constant_to_thor_path(k.name) =~ search})
+  end
+
+  # Override Thor#help so we can give info about not-yet-loaded tasks
+  def help(task = nil)
+    initialize_thorfiles(task) if task && task.include?(?:)
+    super
+  end
+    
+  def method_missing(meth, *args)
+    meth = meth.to_s
+    super(meth.to_sym, *args) unless meth.include? ?:
+
+    initialize_thorfiles(meth)
+    task = Thor[meth]
+    task.parse task.klass.new, ARGV[1..-1]
+  end
+
+  def self.thor_root
+    File.join(ENV["HOME"] || ENV["APPDATA"], ".thor")
+  end
+
+  def self.thor_root_glob
+    # On Windows thor_root will be something like this:
+    #
+    #   C:\Documents and Settings\james\.thor
+    #
+    # If we don't #gsub the \ character, Dir.glob will fail.
+    Dir["#{thor_root.gsub(/\\/, '/')}/**/*"]
+  end
+  
+  private
+  def thor_root
+    self.class.thor_root
+  end
+
+  def thor_root_glob
+    self.class.thor_root_glob
+  end
+  
+  def thor_yaml
+    yaml_file = File.join(thor_root, "thor.yml")
+    yaml = YAML.load_file(yaml_file) if File.exists?(yaml_file)
+    yaml || {}
+  end
+  
+  def save_yaml(yaml)
+    yaml_file = File.join(thor_root, "thor.yml")
+    File.open(yaml_file, "w") {|f| f.puts yaml.to_yaml }
+  end
+  
+  def display_klasses(with_modules = false, klasses = Thor.subclasses)
+    klasses -= [Thor, Thor::Runner] unless with_modules
+    raise Error, "No Thor tasks available" if klasses.empty?
+    
+    if with_modules && !(yaml = thor_yaml).empty?
+      max_name = yaml.max {|(xk,xv),(yk,yv)| xk.size <=> yk.size }.first.size
+      modules_label = "Modules"
+      namespaces_label = "Namespaces"
+      column_width = [max_name + 4, modules_label.size + 1].max
+      
+      print "%-#{column_width}s" % modules_label
+      puts namespaces_label
+      print "%-#{column_width}s" % ("-" * modules_label.size)
+      puts "-" * namespaces_label.size
+      
+      yaml.each do |name, info|
+        print "%-#{column_width}s" % name
+        puts info[:constants].map {|c| Thor::Util.constant_to_thor_path(c)}.join(", ")
+      end
+    
+      puts
+    end
+    
+    puts "Tasks"
+    puts "-----"
+    
+    # Calculate the largest base class name
+    max_base = klasses.max do |x,y| 
+      Thor::Util.constant_to_thor_path(x.name).size <=> Thor::Util.constant_to_thor_path(y.name).size
+    end.name.size
+    
+    # Calculate the size of the largest option description
+    max_left_item = klasses.max do |x,y| 
+      (x.maxima.usage + x.maxima.opt).to_i <=> (y.maxima.usage + y.maxima.opt).to_i
+    end
+    
+    max_left = max_left_item.maxima.usage + max_left_item.maxima.opt
+        
+    klasses.each {|k| display_tasks(k, max_base, max_left)}
+  end  
+  
+  def display_tasks(klass, max_base, max_left)
+    base = Thor::Util.constant_to_thor_path(klass.name)
+    klass.tasks.each true do |name, task|
+      format_string = "%-#{max_left + max_base + 5}s"
+      print format_string % task.formatted_usage(true)
+      puts task.description
+    end
+  end
+
+  def initialize_thorfiles(relevant_to = nil)
+    thorfiles(relevant_to).each {|f| load_thorfile f unless Thor.subclass_files.keys.include?(File.expand_path(f))}
+  end
+  
+  def load_thorfile(path)
+    begin
+      load path
+    rescue Object => e
+      $stderr.puts "WARNING: unable to load thorfile #{path.inspect}: #{e.message}"
+    end
+  end
+  
+  def thorfiles(relevant_to = nil)
+    path = Dir.pwd
+    thorfiles = []
+    
+    # Look for Thorfile or *.thor in the current directory or a parent directory, until the root
+    while thorfiles.empty?
+      thorfiles = Thor::Runner.globs_for(path).map {|g| Dir[g]}.flatten
+      path = File.dirname(path)
+      break if path == "/"
+    end
+
+    # We want to load system-wide Thorfiles first
+    # so the local Thorfiles will override them.
+    (relevant_to ? thorfiles_relevant_to(relevant_to) :
+     thor_root_glob) + thorfiles - ["#{thor_root}/thor.yml"]
+  end
+
+  def thorfiles_relevant_to(meth)
+    klass_str = Thor::Util.to_constant(meth.split(":")[0...-1].join(":"))
+    thor_yaml.select do |k, v|
+      v[:constants] && v[:constants].include?(klass_str)
+    end.map { |k, v| File.join(thor_root, "#{v[:filename]}") }
+  end
+
+end

data/lib/thor/task.rb

@@ -0,0 +1,68 @@
+require 'thor/error'
+require 'thor/util'
+
+class Thor
+  class Task < Struct.new(:meth, :description, :usage, :opts, :klass)
+    def self.dynamic(meth, klass)
+      new(meth, "A dynamically-generated task", meth.to_s, nil, klass)
+    end
+
+    def parse(obj, args)
+      list, hash = parse_args(args)
+      obj.options = hash
+      run(obj, *list)
+    end
+
+    def run(obj, *params)
+      raise NoMethodError, "the `#{meth}' task of #{obj.class} is private" if
+        (obj.private_methods + obj.protected_methods).include?(meth)
+      
+      obj.send(meth, *params)
+    rescue ArgumentError => e
+      # backtrace sans anything in this file
+      backtrace = e.backtrace.reject {|frame| frame =~ /^#{Regexp.escape(__FILE__)}/}
+      # and sans anything that got us here
+      backtrace -= caller
+      raise e unless backtrace.empty?
+    
+      # okay, they really did call it wrong
+      raise Error, "`#{meth}' was called incorrectly. Call as `#{formatted_usage}'"
+    rescue NoMethodError => e
+      begin
+        raise e unless e.message =~ /^undefined method `#{meth}' for #{Regexp.escape(obj.inspect)}$/
+      rescue
+        raise e
+      end
+      raise Error, "The #{namespace false} namespace doesn't have a `#{meth}' task"
+    end
+
+    def namespace(remove_default = true)
+      Thor::Util.constant_to_thor_path(klass, remove_default)
+    end
+
+    def with_klass(klass)
+      new = self.dup
+      new.klass = klass
+      new
+    end
+    
+    def opts
+      return super unless super.kind_of? Hash
+      self.opts = Options.new(super)
+    end
+
+    def formatted_usage(namespace = false)
+      (namespace ? self.namespace + ':' : '') + usage +
+        (opts ? " " + opts.formatted_usage : "")
+    end
+
+    protected
+
+    def parse_args(args)
+      return [args, {}] unless opts
+      hash = opts.parse(args)
+      list = opts.non_opts
+      [list, hash]
+    end
+  end
+end

data/lib/thor/task_hash.rb

@@ -0,0 +1,22 @@
+require 'thor/ordered_hash'
+require 'thor/task'
+
+class Thor::TaskHash < Thor::OrderedHash
+  def initialize(klass)
+    super()
+    @klass = klass
+  end
+
+  def each(local = false, &block)
+    super() { |k, t| yield k, t.with_klass(@klass) }
+    @klass.superclass.tasks.each { |k, t| yield k, t.with_klass(@klass) } unless local || @klass == Thor
+  end
+
+  def [](name)
+    if task = super(name) || (@klass == Thor && @klass.superclass.tasks[name])
+      return task.with_klass(@klass)
+    end
+
+    Thor::Task.dynamic(name, @klass)
+  end
+end

data/lib/thor/tasks.rb

@@ -0,0 +1,77 @@
+require "thor"
+require "fileutils"
+
+class Thor
+  def self.package_task(spec)
+    desc "package", "package up the gem"
+    define_method :package do
+      FileUtils.mkdir_p(File.join(Dir.pwd, "pkg"))
+      Gem::Builder.new(spec).build
+      FileUtils.mv(spec.file_name, File.join(Dir.pwd, "pkg", spec.file_name))
+    end
+  end
+  
+  def self.install_task(spec)
+    package_task spec
+
+    null, sudo, gem = RUBY_PLATFORM =~ /w(in)?32$/ ? ['NUL', '', 'gem.bat'] :
+                                                     ['/dev/null', 'sudo', 'gem']
+    
+    desc "install", "install the gem"
+    define_method :install do
+      old_stderr, $stderr = $stderr.dup, File.open(null, "w")
+      package
+      $stderr = old_stderr
+      system %{#{sudo} #{gem} install pkg/#{spec.name}-#{spec.version} --no-rdoc --no-ri --no-update-sources}
+    end
+  end
+  
+  def self.spec_task(file_list, opts = {})
+    name = opts.delete(:name) || "spec"
+    rcov_dir = opts.delete(:rcov_dir) || "coverage"
+    file_list = file_list.map {|f| %["#{f}"]}.join(" ")
+    verbose = opts.delete(:verbose)
+    opts = {:format => "specdoc", :color => true}.merge(opts)
+    
+    rcov_opts = convert_task_options(opts.delete(:rcov) || {})
+    rcov = !rcov_opts.empty?
+    options = convert_task_options(opts)
+    
+    if rcov
+      FileUtils.rm_rf(File.join(Dir.pwd, rcov_dir))
+    end
+    
+    desc(name, "spec task")
+    define_method(name) do
+      cmd = "ruby "
+      if rcov
+        cmd << "-S rcov -o #{rcov_dir} #{rcov_opts} "
+      end
+      cmd << `which spec`.chomp
+      cmd << " -- " if rcov
+      cmd << " "
+      cmd << file_list
+      cmd << " "
+      cmd << options
+      puts cmd if verbose
+      system(cmd)
+      exit($?.exitstatus)
+    end
+  end
+  
+  private
+  def self.convert_task_options(opts)
+    opts.map do |key, value|
+      case value
+      when true
+        "--#{key}"
+      when Array
+        value.map {|v| "--#{key} #{v.inspect}"}.join(" ")
+      when nil, false
+        ""
+      else
+        "--#{key} #{value.inspect}"
+      end
+    end.join(" ")    
+  end  
+end

data/lib/thor/tasks/package.rb

@@ -0,0 +1,18 @@
+require "thor/task"
+
+class Thor::PackageTask < Thor::Task
+  attr_accessor :spec
+  attr_accessor :opts
+
+  def initialize(gemspec, opts = {})
+    super(:package, "build a gem package")
+    @spec = gemspec
+    @opts = {:dir => File.join(Dir.pwd, "pkg")}.merge(opts)
+  end
+
+  def run
+    FileUtils.mkdir_p(@opts[:dir])
+    Gem::Builder.new(spec).build
+    FileUtils.mv(spec.file_name, File.join(@opts[:dir], spec.file_name))
+  end
+end

data/lib/thor/util.rb

@@ -0,0 +1,43 @@
+require 'thor/error'
+
+class Thor
+  module Util
+    
+    def self.constant_to_thor_path(str, remove_default = true)
+      str = snake_case(str.to_s).squeeze(":")
+      str.gsub!(/^default/, '') if remove_default
+      str
+    end
+
+    def self.constant_from_thor_path(str)
+      make_constant(to_constant(str))
+    rescue NameError => e
+      raise e unless e.message =~ /^uninitialized constant (.*)$/
+      raise Error, "There was no available namespace `#{str}'."
+    end
+
+    def self.to_constant(str)
+      str = 'default' if str.empty?
+      str.gsub(/:(.?)/) { "::#{$1.upcase}" }.gsub(/(?:^|_)(.)/) { $1.upcase }
+    end
+
+    def self.constants_in_contents(str)
+      klasses = self.constants.dup
+      eval(str)
+      ret = self.constants - klasses
+      ret.each {|k| self.send(:remove_const, k)}
+      ret
+    end
+
+    def self.make_constant(str)
+      list = str.split("::").inject(Object) {|obj, x| obj.const_get(x)}
+    end
+    
+    def self.snake_case(str)
+      return str.downcase if str =~ /^[A-Z_]+$/
+      str.gsub(/\B[A-Z]/, '_\&').squeeze('_') =~ /_*(.*)/
+      return $+.downcase
+    end  
+    
+  end
+end

metadata

@@ -0,0 +1,73 @@
+--- !ruby/object:Gem::Specification 
+name: jherdman-thor
+version: !ruby/object:Gem::Version 
+  version: 0.9.5
+platform: ruby
+authors: 
+- Yehuda Katz
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2008-08-27 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: A gem that maps options to a class
+email: wycats@gmail.com
+executables: 
+- thor
+- rake2thor
+extensions: []
+
+extra_rdoc_files: 
+- README.markdown
+- CHANGELOG.rdoc
+- LICENSE
+files: 
+- README.markdown
+- LICENSE
+- CHANGELOG.rdoc
+- Rakefile
+- bin/rake2thor
+- bin/thor
+- lib/thor
+- lib/thor/error.rb
+- lib/thor/options.rb
+- lib/thor/ordered_hash.rb
+- lib/thor/runner.rb
+- lib/thor/task.rb
+- lib/thor/task_hash.rb
+- lib/thor/tasks
+- lib/thor/tasks/package.rb
+- lib/thor/tasks.rb
+- lib/thor/util.rb
+- lib/thor.rb
+has_rdoc: true
+homepage: http://yehudakatz.com
+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: thor
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: A gem that maps options to a class
+test_files: []
+