mirror of
https://github.com/scinote-eln/scinote-web.git
synced 2024-11-14 05:05:55 +08:00
124 lines
4 KiB
Ruby
124 lines
4 KiB
Ruby
module SearchableModel
|
|
extend ActiveSupport::Concern
|
|
|
|
included do
|
|
# Helper function for relations that
|
|
# adds OR ILIKE where clause for all specified attributes
|
|
# for the given search query
|
|
scope :where_attributes_like, lambda { |attributes, query, options = {}|
|
|
return unless query
|
|
attrs = []
|
|
if attributes.blank?
|
|
# Do nothing in this case
|
|
elsif attributes.is_a? Symbol
|
|
attrs = [attributes.to_s]
|
|
elsif attributes.is_a? String
|
|
attrs = [attributes]
|
|
elsif attributes.is_a? Array
|
|
attrs = attributes.collect(&:to_s)
|
|
else
|
|
raise ArgumentError, ':attributes must be an array, symbol or string'
|
|
end
|
|
|
|
if options[:whole_word].to_s == 'true' ||
|
|
options[:whole_phrase].to_s == 'true' ||
|
|
options[:at_search].to_s == 'true'
|
|
unless attrs.empty?
|
|
like = options[:match_case].to_s == 'true' ? '~' : '~*'
|
|
like = 'SIMILAR TO' if options[:at_search].to_s == 'true'
|
|
|
|
if options[:whole_word].to_s == 'true'
|
|
a_query = query.split
|
|
.map { |a| Regexp.escape(a) }
|
|
.join('|')
|
|
elsif options[:at_search].to_s == 'true'
|
|
a_query = "%#{Regexp.escape(query).downcase}%"
|
|
else
|
|
a_query = Regexp.escape(query)
|
|
end
|
|
|
|
# quick fix to enable searching by repositoy_row id
|
|
id_index = { present: false }
|
|
where_str =
|
|
(attrs.map.with_index do |a, i|
|
|
if a == 'repository_rows.id'
|
|
id_index = { present: true, val: i }
|
|
"(#{a}) = :t#{i} OR "
|
|
else
|
|
col = options[:at_search].to_s == 'true' ? "lower(#{a})": a
|
|
"(trim_html_tags(#{col})) #{like} :t#{i} OR "
|
|
end
|
|
end
|
|
).join[0..-5]
|
|
vals = (
|
|
attrs.map.with_index do |_, i|
|
|
if id_index[:present] && id_index[:val] == i
|
|
["t#{i}".to_sym, a_query.to_i]
|
|
else
|
|
["t#{i}".to_sym, '\\y(' + a_query + ')\\y']
|
|
end
|
|
end
|
|
).to_h
|
|
return where(where_str, vals)
|
|
end
|
|
end
|
|
|
|
like = options[:match_case].to_s == 'true' ? 'LIKE' : 'ILIKE'
|
|
|
|
if query.count(' ') > 0
|
|
unless attrs.empty?
|
|
a_query = query.split.map { |a| "%#{sanitize_sql_like(a)}%" }
|
|
# quick fix to enable searching by repositoy_row id
|
|
id_index = { present: false }
|
|
where_str =
|
|
(attrs.map.with_index do |a, i|
|
|
if a == 'repository_rows.id'
|
|
id_index = { present: true, val: i }
|
|
"(#{a}) IN (:t#{i}) OR "
|
|
else
|
|
"(trim_html_tags(#{a})) #{like} ANY (array[:t#{i}]) OR "
|
|
end
|
|
end
|
|
).join[0..-5]
|
|
vals = (
|
|
attrs.map.with_index do |_, i|
|
|
if id_index[:present] && id_index[:val] == i
|
|
["t#{i}".to_sym, a_query.map(&:to_i)]
|
|
else
|
|
["t#{i}".to_sym, a_query]
|
|
end
|
|
end
|
|
).to_h
|
|
|
|
return where(where_str, vals)
|
|
end
|
|
else
|
|
unless attrs.empty?
|
|
# quick fix to enable searching by repositoy_row id
|
|
id_index = { present: false }
|
|
where_str =
|
|
(attrs.map.with_index do |a, i|
|
|
if a == 'repository_rows.id'
|
|
id_index = { present: true, val: i }
|
|
"(#{a}) = :t#{i} OR "
|
|
else
|
|
"(trim_html_tags(#{a})) #{like} :t#{i} OR "
|
|
end
|
|
end
|
|
).join[0..-5]
|
|
vals = (
|
|
attrs.map.with_index do |_, i|
|
|
if id_index[:present] && id_index[:val] == i
|
|
["t#{i}".to_sym, sanitize_sql_like(query).to_i]
|
|
else
|
|
["t#{i}".to_sym, "%#{sanitize_sql_like(query.to_s)}%"]
|
|
end
|
|
end
|
|
).to_h
|
|
|
|
return where(where_str, vals)
|
|
end
|
|
end
|
|
}
|
|
end
|
|
end
|