class Rspec::Generators::InstallGenerator

def self.source_root
  @source_root ||= File.expand_path(File.join(File.dirname(__FILE__), 'templates'))
end

def copy_rails_files
  template 'spec/rails_helper.rb'
end

def copy_spec_files
  Dir.mktmpdir do |dir|
    generate_rspec_init dir
    template File.join(dir, '.rspec'), '.rspec'
    directory File.join(dir, 'spec'), 'spec'
  end
end

def generate_rspec_init(tmpdir)
  initializer = ::RSpec::Core::ProjectInitializer.new(
    :destination => tmpdir,
    :report_stream => StringIO.new
  )
  initializer.run
  spec_helper_path = File.join(tmpdir, 'spec', 'spec_helper.rb')
  replace_generator_command(spec_helper_path)
  remove_warnings_configuration(spec_helper_path)
end

def remove_warnings_configuration(spec_helper_path)
  empty_line = '^\n'
  comment_line = '^\s*#.+\n'
  gsub_file spec_helper_path,
            /#{empty_line}(#{comment_line})+\s+config\.warnings = true\n/,
            '',
            :verbose => false
end

def replace_generator_command(spec_helper_path)
  gsub_file spec_helper_path,
            'rspec --init',
            'rails generate rspec:install',
            :verbose => false
end