module Apartment::Migrator

def migrate(database)
  Tenant.process(database) do
    version = ENV["VERSION"] ? ENV["VERSION"].to_i : nil
    ActiveRecord::Migrator.migrate(ActiveRecord::Migrator.migrations_paths, version) do |migration|
      ENV["SCOPE"].blank? || (ENV["SCOPE"] == migration.scope)
    end
  end
end

def rollback(database, step = 1)
  Tenant.process(database) do
    ActiveRecord::Migrator.rollback(ActiveRecord::Migrator.migrations_paths, step)
  end
end

def run(direction, database, version)
  Tenant.process(database) do
    ActiveRecord::Migrator.run(direction, ActiveRecord::Migrator.migrations_paths, version)
  end
end