demarches-normaliennes/app/services/dossier_search_service.rb

74 lines
2.7 KiB
Ruby
Raw Normal View History

class DossierSearchService
def self.matching_dossiers(dossiers, search_terms, with_annotations = false)
if dossiers.nil?
[]
else
dossier_by_exact_id(dossiers, search_terms)
.presence || dossier_by_full_text(dossiers, search_terms, with_annotations)
end
end
def self.matching_dossiers_for_user(search_terms, user)
dossier_by_exact_id_for_user(search_terms, user)
.presence || dossier_by_full_text_for_user(search_terms, Dossier.includes(:procedure).where(id: user.dossiers.ids + user.dossiers_invites.ids))
end
private
def self.dossier_by_exact_id(dossiers, search_terms)
id = search_terms.to_i
if id != 0 && id_compatible?(id) # Sometimes instructeur is searching dossiers with a big number (ex: SIRET), ActiveRecord can't deal with them and throws ActiveModel::RangeError. id_compatible? prevents this.
dossiers.visible_by_administration.where(id: id).ids
else
[]
end
end
def self.dossier_by_full_text(dossiers, search_terms, with_annotations)
2021-04-29 09:33:04 +02:00
ts_vector = "to_tsvector('french', #{with_annotations ? 'dossiers.search_terms || dossiers.private_search_terms' : 'dossiers.search_terms'})"
ts_query = "to_tsquery('french', #{Dossier.connection.quote(to_tsquery(search_terms))})"
dossiers
.visible_by_administration
2021-04-29 09:33:04 +02:00
.where("#{ts_vector} @@ #{ts_query}")
.order(Arel.sql("COALESCE(ts_rank(#{ts_vector}, #{ts_query}), 0) DESC"))
.pluck('id')
.uniq
end
def self.dossier_by_full_text_for_user(search_terms, dossiers)
ts_vector = "to_tsvector('french', search_terms)"
ts_query = "to_tsquery('french', #{Dossier.includes(:procedure).connection.quote(to_tsquery(search_terms))})"
dossiers
.visible_by_user
.where("#{ts_vector} @@ #{ts_query}")
.order(Arel.sql("COALESCE(ts_rank(#{ts_vector}, #{ts_query}), 0) DESC"))
end
def self.dossier_by_exact_id_for_user(search_terms, user)
id = search_terms.to_i
if id != 0 && id_compatible?(id) # Sometimes user is searching dossiers with a big number (ex: SIRET), ActiveRecord can't deal with them and throws ActiveModel::RangeError. id_compatible? prevents this.
Dossier.includes(:procedure).where(id: user.dossiers.visible_by_user.where(id: id) + user.dossiers_invites.visible_by_user.where(id: id)).distinct
else
Dossier.includes(:procedure).none
end
end
2021-05-06 01:14:57 +02:00
def self.id_compatible?(number)
ActiveRecord::Type::Integer.new.serialize(number)
true
rescue ActiveModel::RangeError
false
2018-07-31 15:08:10 +02:00
end
def self.to_tsquery(search_terms)
(search_terms || "")
2021-02-16 15:05:04 +01:00
.gsub(/['?\\:&|!<>()]/, "") # drop disallowed characters
2020-04-20 14:27:43 +02:00
.strip
.split(/\s+/) # split words
.map { |x| "#{x}:*" } # enable prefix matching
2018-07-31 15:08:10 +02:00
.join(" & ")
end
end