module PaperTrailAssociationTracking::Reifiers::HasAndBelongsToMany

def load_version(assoc, id, transaction_id, version_at)
  assoc.klass.paper_trail.version_class.
    where("item_type = ?", assoc.klass.base_class.name).
    where("item_id = ?", id).
    where("created_at >= ? OR transaction_id = ?", version_at, transaction_id).
    order("id").
    limit(1).
    first
end

def reify(pt_enabled, assoc, model, options, transaction_id)
  version_ids = model.class.paper_trail.version_association_class.
    where("foreign_key_name = ?", assoc.name).
    where("version_id = ?", transaction_id).
    pluck(:foreign_key_id)
  model.send(assoc.name).proxy_association.target =
    version_ids.map do |id|
      if pt_enabled
        version = load_version(assoc, id, transaction_id, options[:version_at])
        if version
          next version.reify(
            options.merge(
              has_many: false,
              has_one: false,
              belongs_to: false,
              has_and_belongs_to_many: false
            )
          )
        end
      end
      assoc.klass.where(assoc.klass.primary_key => id).first
    end
end