Rails db:refresh task

db.rake

namespace :db do
  desc "Do a clean reset of your DB from a backup file"
  task refresh: [:environment, :check_protected_environments] do
    filename = ARGV.last
    jobs     = ENV.fetch('JOBS', 2)
    database = ActiveRecord::Base.connection.current_database

    abort "Usage: rails db:refresh filename" if filename == 'db:refresh'
    abort "Can't find: #{filename}" unless File.exists? filename
    abort "Can't find pg_restore, make sure it is in your $PATH" unless system "which", "pg_restore"

    Rake::Task['db:drop'].invoke
    Rake::Task['db:create'].invoke

    system "pg_restore --verbose --clean --no-owner --dbname=#{database} --jobs=#{jobs} #{filename}"

    # This re-populates ar_internal_metadata with the correct environment so that db:seed_fu doesn't fail
    # see db:environment:set task: https://github.com/rails/rails/blob/master/activerecord/lib/active_record/railties/databases.rake
    ActiveRecord::InternalMetadata.create_table
    ActiveRecord::InternalMetadata[:environment] = ActiveRecord::Migrator.current_environment

    if ENV.fetch('MIGRATE', 'true') == 'true'
      Rake::Task['db:migrate'].invoke
    end
  end
end