module Thor::RakeCompat

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

def self.rake_classes
  @rake_classes ||= []
end