require 'fileutils'

namespace :spec do
  def rails_app_path(version)
    File.join(File.dirname(__FILE__), "..", "tmp", "rails_#{version.gsub(".", "_")}_app")
  end

  def gemfile_path(version)
    File.join(File.dirname(__FILE__), "..", "gemfiles", "rails-#{version}")
  end

  def vendor_path(version)
    File.expand_path("vendor/bundle", rails_app_path(version))
  end

  def rails_template_path
    File.join(File.dirname(__FILE__), "..", "spec", "rails_template")
  end

  def version
    ENV['VERSION']
  end

  task :run_with_rails => [:set_gemfile, :generate_rails_app, :initialize_database, :setup_rails_app, :run]

  task :set_gemfile do
    ENV['BUNDLE_PATH']    = vendor_path(version)
    ENV['BUNDLE_GEMFILE'] = gemfile_path(version)

    unless File.exist?(ENV['BUNDLE_PATH'])
      puts "Installing gems for Rails #{version} (this will only be done once)..."
      sh("bundle install #{ENV['BUNDLE_ARGS']}") || exit(1)
    end
  end

  task :generate_rails_app do
    app_path = rails_app_path(version)

    unless File.exist?(File.expand_path("config/environment.rb", app_path))
      puts "Generating Rails #{version} application..."
      sh("bundle exec rails _#{version}_ new \"#{app_path}\" --force --skip-git --skip-javascript --skip-gemfile --skip-sprockets") || exit(1)
    end
  end

  task :initialize_database do
    if ENV['DB'] == 'postgres'
      sh "bundle exec rake db:test:prepare"
    end
  end

  task :setup_rails_app do
    FileUtils.cp_r File.join(rails_template_path, "."), rails_app_path(version)
  end

  task :run do
    ENV['BUNDLE_GEMFILE'] = gemfile_path(version)
    ENV['RAILS_ROOT']     = rails_app_path(version)

    sh "bundle exec rspec #{ENV['SPEC'] || 'spec/*_spec.rb'} --color"
  end
end

def rails_all_versions
  versions = []
  Dir.glob(File.join(File.dirname(__FILE__), "..", "gemfiles", "rails-*")).each do |gemfile|
    if !gemfile.end_with?(".lock") && gemfile =~ /rails-([0-9.]+)/
      versions << $1
    end
  end

  versions
end

def reenable_spec_tasks
  Rake::Task.tasks.each do |task|
    if task.name =~ /spec:/
      task.reenable
    end
  end
end

desc 'Run spec suite in all Rails versions'
task :spec do
  versions = if ENV['RAILS']
               ENV['RAILS'].split(",")
             else
               rails_all_versions
             end

  versions.each do |version|
    puts "Running specs against Rails #{version}..."

    ENV['VERSION'] = version
    reenable_spec_tasks
    Rake::Task['spec:run_with_rails'].invoke
  end
end