Files
danbooru/app/logical/post_query_builder.rb
evazion 3dab648d0e search: refactor PostQueryBuilder class methods into instance methods.
* Make scan_query, parse_query, normalize_query into instance methods
  instead of class methods. This is to a) clean up the API and b)
  prepare for moving certain tag utility methods into PostQueryBuilder.

* Fix a few cases where a caller used scan_query when they should have
  used split_query or parse_tag_edit.
2020-04-22 19:38:17 -05:00

1259 lines
35 KiB
Ruby

require "strscan"
class PostQueryBuilder
extend Memoist
COUNT_METATAGS = %w[
comment_count deleted_comment_count active_comment_count
note_count deleted_note_count active_note_count
flag_count resolved_flag_count unresolved_flag_count
child_count deleted_child_count active_child_count
pool_count deleted_pool_count active_pool_count series_pool_count collection_pool_count
appeal_count approval_count replacement_count
]
# allow e.g. `deleted_comments` as a synonym for `deleted_comment_count`
COUNT_METATAG_SYNONYMS = COUNT_METATAGS.map { |str| str.delete_suffix("_count").pluralize }
METATAGS = %w[
-user user
-approver approver
-commenter commenter comm
-noter noter
-noteupdater noteupdater
-artcomm artcomm
-commentaryupdater commentaryupdater
-flagger flagger
-appealer appealer
-upvote upvote
-downvote downvote
-fav fav
-ordfav ordfav
-favgroup favgroup
-pool pool ordpool
-commentary commentary
-id id
-rating rating
-locked locked
-source source
-status status
-filetype filetype
-disapproved disapproved
-parent parent
-child child
-search search
-embedded embedded
md5
width
height
mpixels
ratio
score
favcount
filesize
date
age
order
limit
tagcount
pixiv_id pixiv
] + TagCategory.short_name_list.map {|x| "#{x}tags"} + COUNT_METATAGS + COUNT_METATAG_SYNONYMS
ORDER_METATAGS = %w[
id id_desc
score score_asc
favcount favcount_asc
created_at created_at_asc
change change_asc
comment comment_asc
comment_bumped comment_bumped_asc
note note_asc
artcomm artcomm_asc
mpixels mpixels_asc
portrait landscape
filesize filesize_asc
tagcount tagcount_asc
rank
curated
modqueue
random
custom
] +
COUNT_METATAGS +
COUNT_METATAG_SYNONYMS.flat_map { |str| [str, "#{str}_asc"] } +
TagCategory.short_name_list.flat_map { |str| ["#{str}tags", "#{str}tags_asc"] }
attr_accessor :query_string
def initialize(query_string)
@query_string = query_string
end
def escape_string_for_tsquery(array)
array.map(&:to_escaped_for_tsquery)
end
def attribute_matches(values, field, type = :integer)
values.to_a.reduce(Post.all) do |relation, value|
operator, *args = parse_metatag_value(value, type)
relation.where_operator(field, operator, *args)
end
end
def user_matches(field, username)
case username.downcase
when "any"
Post.where.not(field => nil)
when "none"
Post.where(field => nil)
else
Post.where(field => User.name_matches(username))
end
end
def user_subquery_matches(subquery, username, field: :creator, &block)
subquery = subquery.where("post_id = posts.id").select(1)
if username == "any"
Post.where("EXISTS (#{subquery.to_sql})")
elsif username == "none"
Post.where("NOT EXISTS (#{subquery.to_sql})")
elsif block.nil?
subquery = subquery.where(field => User.name_matches(username))
Post.where("EXISTS (#{subquery.to_sql})")
else
subquery = subquery.merge(block.call(username))
return Post.none if subquery.to_sql.blank?
Post.where("EXISTS (#{subquery.to_sql})")
end
end
def flagger_matches(username)
flags = PostFlag.unscoped.category_matches("normal")
user_subquery_matches(flags, username) do |username|
flagger = User.find_by_name(username)
PostFlag.unscoped.creator_matches(flagger, CurrentUser.user)
end
end
def add_tag_string_search_relation(tags, relation)
tag_query_sql = []
if tags[:include].any?
tag_query_sql << "(" + escape_string_for_tsquery(tags[:include]).join(" | ") + ")"
end
if tags[:related].any?
tag_query_sql << "(" + escape_string_for_tsquery(tags[:related]).join(" & ") + ")"
end
if tags[:exclude].any?
tag_query_sql << "!(" + escape_string_for_tsquery(tags[:exclude]).join(" | ") + ")"
end
if tag_query_sql.any?
relation = relation.where("posts.tag_index @@ to_tsquery('danbooru', E?)", tag_query_sql.join(" & "))
end
relation
end
def saved_search_matches(label)
case label.downcase
when "all"
Post.where(id: SavedSearch.post_ids_for(CurrentUser.id))
else
Post.where(id: SavedSearch.post_ids_for(CurrentUser.id, label: label))
end
end
def status_matches(status)
case status.downcase
when "pending"
Post.pending
when "flagged"
Post.flagged
when "modqueue"
Post.pending_or_flagged
when "deleted"
Post.deleted
when "banned"
Post.banned
when "active"
Post.active
when "unmoderated"
Post.pending_or_flagged.available_for_moderation
when "all", "any"
Post.all
else
Post.none
end
end
def disapproved_matches(query)
if query.downcase.in?(PostDisapproval::REASONS)
Post.where(disapprovals: PostDisapproval.where(reason: query.downcase))
elsif User.normalize_name(query) == CurrentUser.user.name
Post.where(disapprovals: PostDisapproval.where(user: CurrentUser.user))
else
Post.none
end
end
def parent_matches(parent)
case parent.downcase
when "none"
Post.where(parent: nil)
when "any"
Post.where.not(parent: nil)
when /\A\d+\z/
Post.where(id: parent).or(Post.where(parent: parent))
else
Post.none
end
end
def child_matches(child)
case child.downcase
when "none"
Post.where(has_children: false)
when "any"
Post.where(has_children: true)
else
Post.none
end
end
def source_matches(source)
case source.downcase
when "none"
Post.where_like(:source, "")
else
Post.where_ilike(:source, source + "*")
end
end
def embedded_matches(embedded)
if embedded.truthy?
Post.bit_flags_match(:has_embedded_notes, true)
elsif embedded.falsy?
Post.bit_flags_match(:has_embedded_notes, false)
else
Post.none
end
end
def pool_matches(pool_name)
case pool_name.downcase
when "none"
Post.where.not(id: Pool.select("unnest(post_ids)"))
when "any"
Post.where(id: Pool.select("unnest(post_ids)"))
when "series"
Post.where(id: Pool.series.select("unnest(post_ids)"))
when "collection"
Post.where(id: Pool.collection.select("unnest(post_ids)"))
when /\*/
Post.where(id: Pool.name_matches(pool_name).select("unnest(post_ids)"))
else
Post.where(id: Pool.named(pool_name).select("unnest(post_ids)"))
end
end
def ordpool_matches(pool_name)
# XXX unify with Pool#posts
pool_posts = Pool.named(pool_name).joins("CROSS JOIN unnest(pools.post_ids) WITH ORDINALITY AS row(post_id, pool_index)").select(:post_id, :pool_index)
Post.joins("JOIN (#{pool_posts.to_sql}) pool_posts ON pool_posts.post_id = posts.id").order("pool_posts.pool_index ASC")
end
def favgroup_matches(query)
favgroup = FavoriteGroup.visible(CurrentUser.user).name_or_id_matches(query, CurrentUser.user)
Post.where(id: favgroup.select("unnest(post_ids)"))
end
def commentary_matches(query)
case query.downcase
when "none", "false"
Post.where.not(artist_commentary: ArtistCommentary.all).or(Post.where(artist_commentary: ArtistCommentary.deleted))
when "any", "true"
Post.where(artist_commentary: ArtistCommentary.undeleted)
when "translated"
Post.where(artist_commentary: ArtistCommentary.translated)
when "untranslated"
Post.where(artist_commentary: ArtistCommentary.untranslated)
else
Post.where(artist_commentary: ArtistCommentary.text_matches(query))
end
end
def locked_matches(query)
case query.downcase
when "rating"
Post.where(is_rating_locked: true)
when "note", "notes"
Post.where(is_note_locked: true)
when "status"
Post.where(is_status_locked: true)
else
Post.none
end
end
def table_for_metatag(metatag)
if metatag.in?(COUNT_METATAGS)
metatag[/(?<table>[a-z]+)_count\z/i, :table]
else
nil
end
end
def tables_for_query(q)
metatags = q.keys
metatags << q[:order].remove(/_(asc|desc)\z/i) if q[:order].present?
tables = metatags.map { |metatag| table_for_metatag(metatag.to_s) }
tables.compact.uniq
end
def add_joins(q, relation)
tables = tables_for_query(q)
relation = relation.with_stats(tables)
relation
end
def hide_deleted_posts?(q)
return false if CurrentUser.admin_mode?
return false if q[:status].to_a.any?(%w[deleted active any all])
return false if q[:status_neg].to_a.any?(%w[deleted active any all])
return CurrentUser.user.hide_deleted_posts?
end
def build
q = parse_query
relation = Post.all
if q[:tag_count].to_i > Danbooru.config.tag_query_limit
raise ::Post::SearchError
end
if CurrentUser.safe_mode?
relation = relation.where("posts.rating = 's'")
end
if hide_deleted_posts?(q)
relation = relation.undeleted
end
relation = add_joins(q, relation)
relation = relation.merge(attribute_matches(q[:id], :id))
relation = relation.merge(attribute_matches(q[:md5], :md5, :md5))
relation = relation.merge(attribute_matches(q[:mpixels], "posts.image_width * posts.image_height / 1000000.0", :float))
relation = relation.merge(attribute_matches(q[:ratio], "ROUND(1.0 * posts.image_width / GREATEST(1, posts.image_height), 2)", :ratio))
relation = relation.merge(attribute_matches(q[:width], :image_width))
relation = relation.merge(attribute_matches(q[:height], :image_height))
relation = relation.merge(attribute_matches(q[:score], :score))
relation = relation.merge(attribute_matches(q[:fav_count], :fav_count))
relation = relation.merge(attribute_matches(q[:file_size], :file_size, :filesize))
relation = relation.merge(attribute_matches(q[:date], :created_at, :date))
relation = relation.merge(attribute_matches(q[:age], :created_at, :age))
relation = relation.merge(attribute_matches(q[:pixiv_id], :pixiv_id))
relation = relation.merge(attribute_matches(q[:post_tag_count], :tag_count))
relation = relation.merge(attribute_matches(q[:filetype], :file_ext, :enum))
relation = relation.merge(attribute_matches(q[:filetype_neg], :file_ext, :enum).negate(:nor)) if q[:filetype_neg].present?
TagCategory.categories.each do |category|
relation = relation.merge(attribute_matches(q["#{category}_tag_count".to_sym], "tag_count_#{category}".to_sym))
end
COUNT_METATAGS.each do |column|
relation = relation.merge(attribute_matches(q[column.to_sym], column.to_sym))
end
q[:status].to_a.each do |query|
relation = relation.merge(status_matches(query))
end
q[:status_neg].to_a.each do |query|
relation = relation.merge(status_matches(query).negate)
end
q[:source].to_a.each do |query|
relation = relation.merge(source_matches(query))
end
q[:source_neg].to_a.each do |query|
relation = relation.merge(source_matches(query).negate)
end
q[:pool_neg].to_a.each do |pool_name|
relation = relation.merge(pool_matches(pool_name).negate)
end
q[:pool].to_a.each do |pool_name|
relation = relation.merge(pool_matches(pool_name))
end
q[:commentary_neg].to_a.each do |query|
relation = relation.merge(commentary_matches(query).negate)
end
q[:commentary].to_a.each do |query|
relation = relation.merge(commentary_matches(query))
end
q[:saved_searches_neg].to_a.each do |query|
relation = relation.merge(saved_search_matches(query).negate)
end
q[:saved_searches].to_a.each do |query|
relation = relation.merge(saved_search_matches(query))
end
q[:user_neg].to_a.each do |username|
relation = relation.merge(user_matches(:uploader, username).negate)
end
q[:user].to_a.each do |username|
relation = relation.merge(user_matches(:uploader, username))
end
q[:approver_neg].to_a.each do |username|
relation = relation.merge(user_matches(:approver, username).negate)
end
q[:approver].to_a.each do |username|
relation = relation.merge(user_matches(:approver, username))
end
q[:disapproved_neg].to_a.each do |query|
relation = relation.merge(disapproved_matches(query).negate)
end
q[:disapproved].to_a.each do |query|
relation = relation.merge(disapproved_matches(query))
end
q[:flagger_neg].to_a.each do |username|
relation = relation.merge(flagger_matches(username).negate)
end
q[:flagger].to_a.each do |username|
relation = relation.merge(flagger_matches(username))
end
q[:appealer_neg].to_a.each do |username|
relation = relation.merge(user_subquery_matches(PostAppeal.unscoped, username).negate)
end
q[:appealer].to_a.each do |username|
relation = relation.merge(user_subquery_matches(PostAppeal.unscoped, username))
end
q[:commenter_neg].to_a.each do |username|
relation = relation.merge(user_subquery_matches(Comment.unscoped, username).negate)
end
q[:commenter].to_a.each do |username|
relation = relation.merge(user_subquery_matches(Comment.unscoped, username))
end
q[:noter_neg].to_a.each do |username|
relation = relation.merge(user_subquery_matches(NoteVersion.unscoped.where(version: 1), username, field: :updater).negate)
end
q[:noter].to_a.each do |username|
relation = relation.merge(user_subquery_matches(NoteVersion.unscoped.where(version: 1), username, field: :updater))
end
q[:note_updater_neg].to_a.each do |username|
relation = relation.merge(user_subquery_matches(NoteVersion.unscoped, username, field: :updater).negate)
end
q[:note_updater].to_a.each do |username|
relation = relation.merge(user_subquery_matches(NoteVersion.unscoped, username, field: :updater))
end
q[:commentary_updater_neg].to_a.each do |username|
relation = relation.merge(user_subquery_matches(ArtistCommentaryVersion.unscoped, username, field: :updater).negate)
end
q[:commentary_updater].to_a.each do |username|
relation = relation.merge(user_subquery_matches(ArtistCommentaryVersion.unscoped, username, field: :updater))
end
q[:id_neg].to_a.each do |id|
relation = relation.where.not(id: id)
end
q[:parent].to_a.each do |parent|
relation = relation.merge(parent_matches(parent))
end
q[:parent_neg].to_a.each do |parent_neg|
relation = relation.merge(parent_matches(parent_neg).negate)
end
q[:child].to_a.each do |child|
relation = relation.merge(child_matches(child))
end
q[:child_neg].to_a.each do |child|
relation = relation.merge(child_matches(child).negate)
end
q[:rating].to_a.each do |rating|
relation = relation.where(rating: rating.first.downcase)
end
q[:rating_neg].to_a.each do |rating|
relation = relation.where.not(rating: rating.first.downcase)
end
q[:locked].to_a.each do |lock|
relation = relation.merge(locked_matches(lock))
end
q[:locked_neg].to_a.each do |lock|
relation = relation.merge(locked_matches(lock).negate)
end
q[:embedded].to_a.each do |lock|
relation = relation.merge(embedded_matches(lock))
end
q[:embedded_neg].to_a.each do |lock|
relation = relation.merge(embedded_matches(lock).negate)
end
q[:ordpool].to_a.each do |pool_name|
relation = relation.merge(ordpool_matches(pool_name))
end
q[:favgroup_neg].to_a.each do |favgroup_name|
relation = relation.merge(favgroup_matches(favgroup_name).negate)
end
q[:favgroup].to_a.each do |favgroup_name|
relation = relation.merge(favgroup_matches(favgroup_name))
end
q[:upvoter].to_a.each do |username|
relation = relation.merge(user_subquery_matches(PostVote.positive.visible(CurrentUser.user), username, field: :user))
end
q[:upvoter_neg].to_a.each do |username|
relation = relation.merge(user_subquery_matches(PostVote.positive.visible(CurrentUser.user), username, field: :user).negate)
end
q[:downvoter].to_a.each do |username|
relation = relation.merge(user_subquery_matches(PostVote.negative.visible(CurrentUser.user), username, field: :user))
end
q[:downvoter_neg].to_a.each do |username|
relation = relation.merge(user_subquery_matches(PostVote.negative.visible(CurrentUser.user), username, field: :user).negate)
end
q[:fav_neg].to_a.each do |username|
favuser = User.find_by_name(username)
if favuser.present? && Pundit.policy!([CurrentUser.user, nil], favuser).can_see_favorites?
q[:tags][:exclude] << "fav:#{favuser.id}"
else
relation = relation.all # no-op; excluding a nonexistent user returns everything
end
end
q[:fav].to_a.each do |username|
favuser = User.find_by_name(username)
if favuser.present? && Pundit.policy!([CurrentUser.user, nil], favuser).can_see_favorites?
q[:tags][:related] << "fav:#{favuser.id}"
else
relation = relation.none
end
end
q[:ordfav].to_a.each do |username|
favuser = User.find_by_name(username)
if favuser.present? && Pundit.policy!([CurrentUser.user, nil], favuser).can_see_favorites?
q[:tags][:related] << "fav:#{favuser.id}"
relation = relation.joins("INNER JOIN favorites ON favorites.post_id = posts.id")
relation = relation.where("favorites.user_id % 100 = ? and favorites.user_id = ?", favuser.id % 100, favuser.id).order("favorites.id DESC")
else
relation = relation.none
end
end
relation = add_tag_string_search_relation(q[:tags], relation)
# HACK: if we're using a date: or age: metatag, default to ordering by
# created_at instead of id so that the query will use the created_at index.
if q[:date].present? || q[:age].present?
case q[:order]
when "id", "id_asc"
q[:order] = "created_at_asc"
when "id_desc", nil
q[:order] = "created_at_desc"
end
end
if q[:order] == "rank"
relation = relation.where("posts.score > 0 and posts.created_at >= ?", 2.days.ago)
elsif q[:order] == "landscape" || q[:order] == "portrait"
relation = relation.where("posts.image_width IS NOT NULL and posts.image_height IS NOT NULL")
end
if q[:order] == "custom" && q[:post_id].present? && q[:post_id][0] == :in
relation = relation.find_ordered(q[:post_id][1])
else
relation = search_order(relation, q[:order])
end
relation
end
def search_order(relation, order)
case order.to_s
when "id", "id_asc"
relation = relation.order("posts.id ASC")
when "id_desc"
relation = relation.order("posts.id DESC")
when "score", "score_desc"
relation = relation.order("posts.score DESC, posts.id DESC")
when "score_asc"
relation = relation.order("posts.score ASC, posts.id ASC")
when "favcount"
relation = relation.order("posts.fav_count DESC, posts.id DESC")
when "favcount_asc"
relation = relation.order("posts.fav_count ASC, posts.id ASC")
when "created_at", "created_at_desc"
relation = relation.order("posts.created_at DESC")
when "created_at_asc"
relation = relation.order("posts.created_at ASC")
when "change", "change_desc"
relation = relation.order("posts.updated_at DESC, posts.id DESC")
when "change_asc"
relation = relation.order("posts.updated_at ASC, posts.id ASC")
when "comment", "comm"
relation = relation.order("posts.last_commented_at DESC NULLS LAST, posts.id DESC")
when "comment_asc", "comm_asc"
relation = relation.order("posts.last_commented_at ASC NULLS LAST, posts.id ASC")
when "comment_bumped"
relation = relation.order("posts.last_comment_bumped_at DESC NULLS LAST")
when "comment_bumped_asc"
relation = relation.order("posts.last_comment_bumped_at ASC NULLS FIRST")
when "note"
relation = relation.order("posts.last_noted_at DESC NULLS LAST")
when "note_asc"
relation = relation.order("posts.last_noted_at ASC NULLS FIRST")
when "artcomm"
relation = relation.joins("INNER JOIN artist_commentaries ON artist_commentaries.post_id = posts.id")
relation = relation.order("artist_commentaries.updated_at DESC")
when "artcomm_asc"
relation = relation.joins("INNER JOIN artist_commentaries ON artist_commentaries.post_id = posts.id")
relation = relation.order("artist_commentaries.updated_at ASC")
when "mpixels", "mpixels_desc"
relation = relation.where(Arel.sql("posts.image_width is not null and posts.image_height is not null"))
# Use "w*h/1000000", even though "w*h" would give the same result, so this can use
# the posts_mpixels index.
relation = relation.order(Arel.sql("posts.image_width * posts.image_height / 1000000.0 DESC"))
when "mpixels_asc"
relation = relation.where("posts.image_width is not null and posts.image_height is not null")
relation = relation.order(Arel.sql("posts.image_width * posts.image_height / 1000000.0 ASC"))
when "portrait"
relation = relation.order(Arel.sql("1.0 * posts.image_width / GREATEST(1, posts.image_height) ASC"))
when "landscape"
relation = relation.order(Arel.sql("1.0 * posts.image_width / GREATEST(1, posts.image_height) DESC"))
when "filesize", "filesize_desc"
relation = relation.order("posts.file_size DESC")
when "filesize_asc"
relation = relation.order("posts.file_size ASC")
when /\A(?<column>#{COUNT_METATAGS.join("|")})(_(?<direction>asc|desc))?\z/i
column = $~[:column]
direction = $~[:direction] || "desc"
relation = relation.order(column => direction, :id => direction)
when "tagcount", "tagcount_desc"
relation = relation.order("posts.tag_count DESC")
when "tagcount_asc"
relation = relation.order("posts.tag_count ASC")
when /(#{TagCategory.short_name_regex})tags(?:\Z|_desc)/
relation = relation.order("posts.tag_count_#{TagCategory.short_name_mapping[$1]} DESC")
when /(#{TagCategory.short_name_regex})tags_asc/
relation = relation.order("posts.tag_count_#{TagCategory.short_name_mapping[$1]} ASC")
when "rank"
relation = relation.order(Arel.sql("log(3, posts.score) + (extract(epoch from posts.created_at) - extract(epoch from timestamp '2005-05-24')) / 35000 DESC"))
when "curated"
contributors = User.bit_prefs_match(:can_upload_free, true)
relation = relation
.joins(:favorites)
.where(favorites: { user: contributors })
.group("posts.id")
.select("posts.*, COUNT(*) AS contributor_fav_count")
.order("contributor_fav_count DESC, posts.fav_count DESC, posts.id DESC")
when "modqueue", "modqueue_desc"
relation = relation.left_outer_joins(:flags).order(Arel.sql("GREATEST(posts.created_at, post_flags.created_at) DESC, posts.id DESC"))
when "modqueue_asc"
relation = relation.left_outer_joins(:flags).order(Arel.sql("GREATEST(posts.created_at, post_flags.created_at) ASC, posts.id ASC"))
else
relation = relation.order("posts.id DESC")
end
relation
end
concerning :ParseMethods do
def scan_query
terms = []
query = query_string.to_s.gsub(/[[:space:]]/, " ")
scanner = StringScanner.new(query)
until scanner.eos?
scanner.skip(/ +/)
if scanner.scan(/(#{METATAGS.join("|")}):/io)
metatag = scanner.captures.first
if scanner.scan(/"(.+)"/)
value = scanner.captures.first
elsif scanner.scan(/'(.+)'/)
value = scanner.captures.first
else
value = scanner.scan(/[^ ]*/)
end
terms << OpenStruct.new({ type: :metatag, name: metatag.downcase, value: value })
elsif scanner.scan(/[^ ]+/)
terms << OpenStruct.new({ type: :tag, value: scanner.matched.downcase })
end
end
terms
end
def split_query
scan_query.map do |term|
if term.type == :metatag && term.value.include?(" ")
"#{term.name}:\"#{term.value}\""
elsif term.type == :metatag
"#{term.name}:#{term.value}"
elsif term.type == :tag
term.value
end
end
end
def normalize_query(normalize_aliases: true, sort: true)
tags = split_query
tags = tags.map { |t| Tag.normalize_name(t) }
tags = TagAlias.to_aliased(tags) if normalize_aliases
tags = tags.sort if sort
tags = tags.uniq
tags.join(" ")
end
def parse_tag_edit
split_query
end
def parse_query
q = {}
q[:tag_count] = 0
q[:tags] = {
:related => [],
:include => [],
:exclude => []
}
scan_query.each do |term|
q[:tag_count] += 1 unless Danbooru.config.is_unlimited_tag?(term)
if term.type == :metatag
g1 = term.name
g2 = term.value
case g1
when "-user"
q[:user_neg] ||= []
q[:user_neg] << g2
when "user"
q[:user] ||= []
q[:user] << g2
when "-approver"
q[:approver_neg] ||= []
q[:approver_neg] << g2
when "approver"
q[:approver] ||= []
q[:approver] << g2
when "flagger"
q[:flagger] ||= []
q[:flagger] << g2
when "-flagger"
q[:flagger_neg] ||= []
q[:flagger_neg] << g2
when "appealer"
q[:appealer] ||= []
q[:appealer] << g2
when "-appealer"
q[:appealer_neg] ||= []
q[:appealer_neg] << g2
when "commenter", "comm"
q[:commenter] ||= []
q[:commenter] << g2
when "-commenter", "-comm"
q[:commenter_neg] ||= []
q[:commenter_neg] << g2
when "noter"
q[:noter] ||= []
q[:noter] << g2
when "-noter"
q[:noter_neg] ||= []
q[:noter_neg] << g2
when "noteupdater"
q[:note_updater] ||= []
q[:note_updater] << g2
when "-noteupdater"
q[:note_updater_neg] ||= []
q[:note_updater_neg] << g2
when "-commentaryupdater", "-artcomm"
q[:commentary_updater_neg] ||= []
q[:commentary_updater_neg] << g2
when "commentaryupdater", "artcomm"
q[:commentary_updater] ||= []
q[:commentary_updater] << g2
when "disapproved"
q[:disapproved] ||= []
q[:disapproved] << g2
when "-disapproved"
q[:disapproved_neg] ||= []
q[:disapproved_neg] << g2
when "-pool"
q[:pool_neg] ||= []
q[:pool_neg] << g2
when "pool"
q[:pool] ||= []
q[:pool] << g2
when "ordpool"
q[:ordpool] ||= []
q[:ordpool] << g2
when "-favgroup"
q[:favgroup_neg] ||= []
q[:favgroup_neg] << g2
when "favgroup"
q[:favgroup] ||= []
q[:favgroup] << g2
when "-fav", "-ordfav"
q[:fav_neg] ||= []
q[:fav_neg] << g2
when "fav"
q[:fav] ||= []
q[:fav] << g2
when "ordfav"
q[:ordfav] ||= []
q[:ordfav] << g2
when "-commentary"
q[:commentary_neg] ||= []
q[:commentary_neg] << g2
when "commentary"
q[:commentary] ||= []
q[:commentary] << g2
when "-search"
q[:saved_searches_neg] ||= []
q[:saved_searches_neg] << g2
when "search"
q[:saved_searches] ||= []
q[:saved_searches] << g2
when "md5"
q[:md5] ||= []
q[:md5] << g2
when "-rating"
q[:rating_neg] ||= []
q[:rating_neg] << g2
when "rating"
q[:rating] ||= []
q[:rating] << g2
when "-locked"
q[:locked_neg] ||= []
q[:locked_neg] << g2
when "locked"
q[:locked] ||= []
q[:locked] << g2
when "id"
q[:id] ||= []
q[:id] << g2
when "-id"
q[:id_neg] ||= []
q[:id_neg] << g2
when "width"
q[:width] ||= []
q[:width] << g2
when "height"
q[:height] ||= []
q[:height] << g2
when "mpixels"
q[:mpixels] ||= []
q[:mpixels] << g2
when "ratio"
q[:ratio] ||= []
q[:ratio] << g2
when "score"
q[:score] ||= []
q[:score] << g2
when "favcount"
q[:fav_count] ||= []
q[:fav_count] << g2
when "filesize"
q[:file_size] ||= []
q[:file_size] << g2
when "source"
q[:source] ||= []
q[:source] << g2
when "-source"
q[:source_neg] ||= []
q[:source_neg] << g2
when "date"
q[:date] ||= []
q[:date] << g2
when "age"
q[:age] ||= []
q[:age] << g2
when "tagcount"
q[:post_tag_count] ||= []
q[:post_tag_count] << g2
when /(#{TagCategory.short_name_regex})tags/
q["#{TagCategory.short_name_mapping[$1]}_tag_count".to_sym] ||= []
q["#{TagCategory.short_name_mapping[$1]}_tag_count".to_sym] << g2
when "parent"
q[:parent] ||= []
q[:parent] << g2
when "-parent"
q[:parent_neg] ||= []
q[:parent_neg] << g2
when "child"
q[:child] ||= []
q[:child] << g2
when "-child"
q[:child_neg] ||= []
q[:child_neg] << g2
when "order"
g2 = g2.downcase
order, suffix, _tail = g2.partition(/_(asc|desc)\z/i)
if order.in?(COUNT_METATAG_SYNONYMS)
g2 = order.singularize + "_count" + suffix
end
q[:order] = g2
when "limit"
# Do nothing. The controller takes care of it.
when "-status"
q[:status_neg] ||= []
q[:status_neg] << g2
when "status"
q[:status] ||= []
q[:status] << g2
when "embedded"
q[:embedded] ||= []
q[:embedded] << g2
when "-embedded"
q[:embedded_neg] ||= []
q[:embedded_neg] << g2
when "filetype"
q[:filetype] ||= []
q[:filetype] << g2
when "-filetype"
q[:filetype_neg] ||= []
q[:filetype_neg] << g2
when "pixiv_id", "pixiv"
q[:pixiv_id] ||= []
q[:pixiv_id] << g2
when "-upvote"
q[:upvoter_neg] ||= []
q[:upvoter_neg] << g2
when "upvote"
q[:upvoter] ||= []
q[:upvoter] << g2
when "-downvote"
q[:downvoter_neg] ||= []
q[:downvoter_neg] << g2
when "downvote"
q[:downvoter] ||= []
q[:downvoter] << g2
when *COUNT_METATAGS
q[g1.to_sym] ||= []
q[g1.to_sym] << g2
when *COUNT_METATAG_SYNONYMS
g1 = "#{g1.singularize}_count"
q[g1.to_sym] ||= []
q[g1.to_sym] << g2
end
else
parse_tag(term.value, q[:tags])
end
end
q[:tags][:exclude] = TagAlias.to_aliased(q[:tags][:exclude])
q[:tags][:include] = TagAlias.to_aliased(q[:tags][:include])
q[:tags][:related] = TagAlias.to_aliased(q[:tags][:related])
return q
end
def parse_tag_operator(tag)
tag = Tag.normalize_name(tag)
if tag.starts_with?("-")
["-", tag.delete_prefix("-")]
elsif tag.starts_with?("~")
["~", tag.delete_prefix("~")]
else
[nil, tag]
end
end
def parse_tag(tag, output)
operator, tag = parse_tag_operator(tag)
if tag.blank?
# XXX ignore "-", "~" operators without a tag.
elsif tag.include?("*")
tags = Tag.wildcard_matches(tag)
if operator == "-"
output[:exclude] += tags
else
tags = ["~no_matches~"] if tags.empty? # force empty results if wildcard found no matches.
output[:include] += tags
end
elsif operator == "-"
output[:exclude] << tag
elsif operator == "~"
output[:include] << tag
else
output[:related] << tag
end
end
def parse_cast(object, type)
case type
when :enum
object.to_s.downcase
when :integer
object.to_i
when :float
object.to_f
when :md5
object.to_s.downcase
when :date, :datetime
Time.zone.parse(object) rescue nil
when :age
DurationParser.parse(object).ago
when :ratio
object =~ /\A(\d+(?:\.\d+)?):(\d+(?:\.\d+)?)\Z/i
if $1 && $2.to_f != 0.0
($1.to_f / $2.to_f).round(2)
else
object.to_f.round(2)
end
when :filesize
object =~ /\A(\d+(?:\.\d*)?|\d*\.\d+)([kKmM]?)[bB]?\Z/
size = $1.to_f
unit = $2
conversion_factor = case unit
when /m/i
1024 * 1024
when /k/i
1024
else
1
end
(size * conversion_factor).to_i
end
end
def parse_metatag_value(string, type)
if type == :enum
[:in, string.split(/[, ]+/).map { |x| parse_cast(x, type) }]
else
parse_range(string, type)
end
end
def parse_range(string, type)
range = case string
when /\A(.+?)\.\.\.(.+)/ # A...B
lo, hi = [parse_cast($1, type), parse_cast($2, type)].sort
[:between, (lo...hi)]
when /\A(.+?)\.\.(.+)/
lo, hi = [parse_cast($1, type), parse_cast($2, type)].sort
[:between, (lo..hi)]
when /\A<=(.+)/, /\A\.\.(.+)/
[:lteq, parse_cast($1, type)]
when /\A<(.+)/
[:lt, parse_cast($1, type)]
when /\A>=(.+)/, /\A(.+)\.\.\Z/
[:gteq, parse_cast($1, type)]
when /\A>(.+)/
[:gt, parse_cast($1, type)]
when /[, ]/
[:in, string.split(/[, ]+/).map {|x| parse_cast(x, type)}]
when "any"
[:not_eq, nil]
when "none"
[:eq, nil]
else
# add a 5% tolerance for float and filesize values
if type == :float || (type == :filesize && string =~ /[km]b?\z/i)
value = parse_cast(string, type)
[:between, (value * 0.95..value * 1.05)]
elsif type.in?([:date, :age])
value = parse_cast(string, type)
[:between, (value.beginning_of_day..value.end_of_day)]
else
[:eq, parse_cast(string, type)]
end
end
range = reverse_range(range) if type == :age
range
end
def reverse_range(range)
case range
in [:lteq, value]
[:gteq, value]
in [:lt, value]
[:gt, value]
in [:gteq, value]
[:lteq, value]
in [:gt, value]
[:lt, value]
else
range
end
end
end
memoize :scan_query, :split_query, :parse_query
end