aboutsummaryrefslogtreecommitdiffstats
path: root/railties/lib/rails/vendor/thor-0.11.6/lib/thor/rake_compat.rb
blob: 3ab6bb21f59db43de7fff93dbd976cd437fea0dd (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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
require 'rake'

class Thor
  # Adds a compatibility layer to your Thor classes which allows you to use
  # rake package tasks. For example, to use rspec rake tasks, one can do:
  #
  #   require 'thor/rake_compat'
  #
  #   class Default < Thor
  #     include Thor::RakeCompat
  #
  #     Spec::Rake::SpecTask.new(:spec) do |t|
  #       t.spec_opts = ['--options', "spec/spec.opts"]
  #       t.spec_files = FileList['spec/**/*_spec.rb']
  #     end
  #   end
  #
  module RakeCompat
    def self.rake_classes
      @rake_classes ||= []
    end

    def self.included(base)
      # Hack. Make rakefile point to invoker, so rdoc task is generated properly.
      Rake.application.instance_variable_set(:@rakefile, caller[0].match(/(.*):\d+/)[1])
      self.rake_classes << base
    end
  end
end

class Object #:nodoc:
  alias :rake_task :task
  alias :rake_namespace :namespace

  def task(*args, &block)
    task = rake_task(*args, &block)

    if klass = Thor::RakeCompat.rake_classes.last
      non_namespaced_name = task.name.split(':').last

      description = non_namespaced_name
      description << task.arg_names.map{ |n| n.to_s.upcase }.join(' ')
      description.strip!

      klass.desc description, task.comment || non_namespaced_name
      klass.class_eval <<-METHOD
        def #{non_namespaced_name}(#{task.arg_names.join(', ')})
          Rake::Task[#{task.name.to_sym.inspect}].invoke(#{task.arg_names.join(', ')})
        end
      METHOD
    end

    task
  end

  def namespace(name, &block)
    if klass = Thor::RakeCompat.rake_classes.last
      const_name = Thor::Util.camel_case(name.to_s).to_sym
      klass.const_set(const_name, Class.new(Thor))
      new_klass = klass.const_get(const_name)
      Thor::RakeCompat.rake_classes << new_klass
    end

    rake_namespace(name, &block)
    Thor::RakeCompat.rake_classes.pop
  end
end