namespace :db do

  unless Rake::Task.task_defined?("db:drop")
    desc "Drops all the collections for the database for the current Rails.env"
    task :drop => "mongoid:drop"
  end

  unless Rake::Task.task_defined?("db:seed")
    # if another ORM has defined db:seed, don"t run it twice.
    desc "Load the seed data from db/seeds.rb"
    task :seed => :environment do
      seed_file = File.join(Rails.root, "db", "seeds.rb")
      load(seed_file) if File.exist?(seed_file)
    end
  end

  unless Rake::Task.task_defined?("db:setup")
    desc "Create the database, and initialize with the seed data"
    task :setup => [ "db:create", "db:mongoid:create_indexes", "db:seed" ]
  end

  unless Rake::Task.task_defined?("db:reseed")
    desc "Delete data and seed"
    task :reseed => [ "db:drop", "db:seed" ]
  end

  unless Rake::Task.task_defined?("db:create")
    task :create => :environment do
      # noop
    end
  end

  unless Rake::Task.task_defined?("db:migrate")
    task :migrate => :environment do
      # noop
    end
  end

  unless Rake::Task.task_defined?("db:schema:load")
    namespace :schema do
      task :load do
        # noop
      end
    end
  end

  unless Rake::Task.task_defined?("db:test:prepare")
    namespace :test do
      task :prepare do
        # noop
      end
    end
  end

  unless Rake::Task.task_defined?("db:create_indexes")
    task :create_indexes => "mongoid:create_indexes"
  end

  unless Rake::Task.task_defined?("db:remove_indexes")
    task :remove_indexes => "mongoid:remove_indexes"
  end

  namespace :mongoid do

    desc "Create the indexes defined on your mongoid models"
    task :create_indexes => :environment do
      engines_models_paths = Rails.application.railties.engines.map do |engine|
        engine.paths["app/models"].expanded
      end
      root_models_paths = Rails.application.paths["app/models"]
      models_paths = engines_models_paths.push(root_models_paths).flatten

      models_paths.each do |path|
        ::Rails::Mongoid.create_indexes("#{path}/**/*.rb")
      end
    end

    desc "Remove the indexes defined on your mongoid models without questions!"
    task :remove_indexes => :environment do
      engines_models_paths = Rails.application.railties.engines.map do |engine|
        engine.paths["app/models"].expanded
      end
      root_models_paths = Rails.application.paths["app/models"]
      models_paths = engines_models_paths.push(root_models_paths).flatten

      models_paths.each do |path|
        ::Rails::Mongoid.remove_indexes("#{path}/**/*.rb")
      end
    end

    desc "Drops the database for the current Rails.env"
    task :drop => :environment do
      ::Mongoid::Sessions.default.drop
    end
  end
end