class Apartment::RakeTaskEnhancer

def enhance!
  return unless should_enhance?
  # insert task before
  TASKS::ENHANCE_BEFORE.each do |name|
    task = Rake::Task[name]
    enhance_before_task(task)
  end
  # insert task after
  TASKS::ENHANCE_AFTER.each do |name|
    task = Rake::Task[name]
    enhance_after_task(task)
  end
end

def enhance_after_task(task)
  task.enhance do
    Rake::Task[inserted_task_name(task)].invoke
  end
end

def enhance_before_task(task)
  task.enhance([inserted_task_name(task)])
end

def inserted_task_name(task)
  task.name.sub(/db:/, 'apartment:')
end

def should_enhance?
  Apartment.db_migrate_tenants
end