class RSpec::Core::CommandLineConfiguration::Autotest

def create_autotest_directory
  Dir.mkdir('autotest') unless File.exist?('autotest')
end

def create_discover_file
  optionally_remove_discover_file if discover_file_exists?
  File.open(discover_file_path, 'w') do |file|
    file << 'Autotest.add_discovery { "rspec2" }'
  end
end

def discover_file_exists?
  File.exist?(discover_file_path)
end

def discover_file_path
  File.join('autotest', 'discover.rb')
end

def generate
  create_autotest_directory
  create_discover_file
  puts "autotest/discover.rb has been added"
end

def optionally_remove_discover_file
  print "Discover file already exists, overwrite [y/N]? "
  exit if gets !~ /y/i
  FileUtils.rm_rf(discover_file_path)
end