require 'rake'
require File.dirname(__FILE__) + '/../lib/exodus'
require File.dirname(__FILE__) + '/../lib/exodus/helpers/rake_helper'

task :require_env do
  include Exodus::RakeHelper
  Exodus.load_migrations
end

namespace Exodus.configuration.rake_namespace + 'db' do
  desc "Migrate the database"
  task :migrate => :require_env do 
    time_it "db:migrate#{" step #{step}" if step}" do 
      migrations = Exodus::Migration.load_all(Exodus.migrations_info.migrate)
      Exodus::sort_and_run_migrations('up', migrations, step)
    end
  end

  desc "Rolls the database back to the previous version"
  task :rollback => :require_env do 
    time_it "db:rollback#{" step #{step}" if step}" do 
      migrations = Exodus::Migration.load_all(Exodus.migrations_info.rollback)
      Exodus::sort_and_run_migrations('down', migrations, step)
    end
  end

  desc "Reset the database back by rolling back all migrations and then migrating"
  task :reset => :require_env do 
    time_it "db:reset" do 
      migrations_to_revert = Exodus::Migration.load_all(Exodus.migrations_info.rollback)
      migrations_to_migrate = Exodus::Migration.load_all(Exodus.migrations_info.migrate)
      Exodus::sort_and_run_migrations('down', migrations_to_revert)

      puts "rolling back is done. Database is now in a virgin state. Migrating in progress..."
      Exodus::sort_and_run_migrations('up', migrations_to_migrate)
    end
  end

  desc "Shows informations about the current mongo connection"
  task :mongo_info => :require_env do 
    p MongoMapper.database
  end

  namespace :migrate do
    desc "Show which migrations will be run when calling 'rake db:migrate'"
    task :show => :require_env do 
      migrations = Exodus::Migration.load_all(Exodus.migrations_info.migrate)
      puts "List of all the migrations that will be executed by running 'rake db:migrate#{" STEP=#{step}" if step}': \n\n"
      
      migrations_info = Exodus::sort_and_run_migrations('up', migrations, step, true)
      puts migrations_info.empty? ? "There is no new migrations to migrate." : migrations_info
    end

    desc "Manually migrates specified migrations (specify migrations or use config/migration.yml)"
    task :custom, [:migrations_info] => :require_env do |t, args|
      time_it "db:migrate_custom#{" step #{step}" if step}" do 
         migrations = if args[:migrations_info]
          YAML.load(args[:migrations_info])
        else
          Exodus::Migration.load_custom(Exodus.migrations_info.migrate_custom)
        end

        puts "No migrations to run have been found" if migrations.empty?
        Exodus::sort_and_run_migrations('up', migrations, step)
      end
    end

    desc "Lists all the migrations"
    task :list => :require_env do 
      Exodus::Migration.list
    end

    desc "Loads migration.yml and displays it"
    task :yml_status => :require_env do 
      pp Exodus.migrations_info.to_s
    end

    desc "Displays the current status of migrations"
    task :status => :require_env do 
      Exodus::Migration.db_status
    end
  end

  namespace :rollback do
    desc "Show which migrations will be run when calling 'rake db:rollback'"
    task :show => :require_env do 
      migrations = Exodus::Migration.load_all(Exodus.migrations_info.migrate)
      puts "List of all the migrations that will be executed by running 'rake db:rollback#{" STEP=#{step}" if step}': \n\n"

      migrations_info = Exodus::sort_and_run_migrations('down', migrations, step, true)
      puts migrations_info.empty? ? "There is no new migrations to rollback." : migrations_info
    end

    desc "Manually rolls the database back using specified migrations (specify migrations or use config/migration.yml)"
    task :custom, [:migrations_info] => :require_env do |t, args|
      time_it "db:rollback_custom#{" step #{step}" if step}" do 
         migrations = if args[:migrations_info]
          YAML.load(args[:migrations_info])
        else
          Exodus::Migration.load_custom(Exodus.migrations_info.rollback_custom)
        end

        puts "No migrations to run have been found" if migrations.empty?
        Exodus::sort_and_run_migrations('down', migrations, step)
      end
    end
  end
end