module Apartment::Migrator

def migrate(database)
  Tenant.switch(database) do
    version = ENV['VERSION'] ? ENV['VERSION'].to_i : nil
    migration_scope_block = ->(migration) { ENV['SCOPE'].blank? || (ENV['SCOPE'] == migration.scope) }
    if ActiveRecord.version >= Gem::Version.new('7.2.0')
      ActiveRecord::Base.connection_pool.migration_context.migrate(version, &migration_scope_block)
    else
      ActiveRecord::Base.connection.migration_context.migrate(version, &migration_scope_block)
    end
  end
end

def rollback(database, step = 1)
  Tenant.switch(database) do
    if ActiveRecord.version >= Gem::Version.new('7.2.0')
      ActiveRecord::Base.connection_pool.migration_context.rollback(step)
    else
      ActiveRecord::Base.connection.migration_context.rollback(step)
    end
  end
end

def run(direction, database, version)
  Tenant.switch(database) do
    if ActiveRecord.version >= Gem::Version.new('7.2.0')
      ActiveRecord::Base.connection_pool.migration_context.run(direction, version)
    else
      ActiveRecord::Base.connection.migration_context.run(direction, version)
    end
  end
end