* 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.
1259 lines
35 KiB
Ruby
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
|