require 'earth'
require 'rake'
require 'active_record/connection_adapters/abstract/connection_specification'

module Earth
  class Tasks
    include Rake::DSL

    def initialize
      init_earth_tasks
      init_bare unless rails?

      namespace :db do
        unless rails?
          task :create => 'earth:db:create'
          task :drop => 'earth:db:drop'
        end
        task :migrate do
          Rake::Task['earth:db:migrate'].invoke
        end
        task :seed do
          Rake::Task['earth:db:seed'].invoke
        end
      end
    end

    def rails?
      return @rails unless @rails.nil?
      @rails = Object.const_defined?('Rails')
    end

    def init_bare
      Object.const_set 'Rails', Earth

      require 'active_record'
      load 'active_record/railties/databases.rake'

      Rake::Task['db:load_config'].clear
      Rake::Task['db:create'].clear
      Rake::Task['db:drop'].clear
      Rake::Task['db:migrate'].clear
      Rake::Task['db:seed'].clear
    end

    def config
      spec = ENV['DATABASE_URL']
      resolver = ActiveRecord::Base::ConnectionSpecification::Resolver.new spec, {}
      resolver.spec.config.stringify_keys
    end
        
    def init_earth_tasks
      namespace :earth do
        namespace :db do
          task :create do
            create_database(config)
          end
          task :drop do
            drop_database_and_rescue(config)
          end
          task :load_config do
            if rails?
              Rake::Task['db:load_config'].invoke
            else
              Earth.connect
            end
          end
          task :migrate => :load_config do
            DataMiner::Run.auto_upgrade!
            Earth.reset_schemas!
          end
          task :seed => :load_config do
            require 'falls_back_on'

            FallsBackOn.clear
            Earth.run_data_miner!
          end
        end
      end
    end
  end
end