require 'rails/generators/base'
require 'rails/generators/named_base'

module RSpec
  module Generators
    class Base < Rails::Generators::Base #:nodoc:
      def self.source_root
        @_rspec_source_root ||= File.expand_path(File.join(File.dirname(__FILE__), 'rspec', generator_name, 'templates'))
      end

      def self.banner
        "#{$0} rspec:#{generator_name} #{self.arguments.map{ |a| a.usage }.join(' ')} [options]"
      end
    end

    class NamedBase < Rails::Generators::NamedBase #:nodoc:
      def self.source_root
        @_rspec_source_root ||= File.expand_path(File.join(File.dirname(__FILE__), 'rspec', generator_name, 'templates'))
      end

      def self.banner
        "#{$0} rspec:#{generator_name} #{self.arguments.map{ |a| a.usage }.join(' ')} [options]"
      end
    end

  end
end