class ActiveRecord::PredicateBuilder::PolymorphicArrayValue

def convert_to_id(value)
  if value.is_a?(Base)
    primary_key = primary_key(value)
    if primary_key.is_a?(Array)
      primary_key.map { |column| value._read_attribute(column) }
    else
      value._read_attribute(primary_key)
    end
  elsif value.is_a?(Relation)
    value.select(primary_key(value))
  else
    value
  end
end

def initialize(associated_table, values)
  @associated_table = associated_table
  @values = values
end

def klass(value)
  if value.is_a?(Base)
    value.class
  elsif value.is_a?(Relation)
    value.model
  end
end

def primary_key(value)
  associated_table.join_primary_key(klass(value))
end

def queries
  return [ associated_table.join_foreign_key => values ] if values.empty?
  type_to_ids_mapping.map do |type, ids|
    query = {}
    query[associated_table.join_foreign_type] = type if type
    query[associated_table.join_foreign_key] = ids
    query
  end
end

def type_to_ids_mapping
  default_hash = Hash.new { |hsh, key| hsh[key] = [] }
  values.each_with_object(default_hash) do |value, hash|
    hash[klass(value)&.polymorphic_name] << convert_to_id(value)
  end
end