Files
danbooru/app/presenters/post_set_presenters/post.rb
2013-06-24 15:08:42 -07:00

70 lines
1.8 KiB
Ruby

module PostSetPresenters
class Post < Base
attr_accessor :post_set, :tag_set_presenter
delegate :posts, :to => :post_set
def initialize(post_set)
@post_set = post_set
@tag_set_presenter = TagSetPresenter.new(related_tags)
end
def related_tags
if post_set.is_pattern_search?
pattern_tags
elsif post_set.is_tag_subscription?
post_set.tag_subscription_tags
elsif post_set.is_single_tag?
related_tags_for_single
elsif post_set.is_empty_tag?
popular_tags
else
related_tags_for_group
end
end
def popular_tags
Cache.get("popular-tags", 1.hour) do
CurrentUser.scoped(User.admins.first, "127.0.0.1") do
n = 1
results = []
while results.empty? && n < 256
query = n.days.ago.strftime("date:>%Y-%m-%d")
results = RelatedTagCalculator.calculate_from_sample_to_array(query)
n *= 2
end
results.map! do |tag_name, recent_count|
tag = Tag.find_or_create_by_name(tag_name)
[tag_name, recent_count.to_f / tag.post_count.to_f]
end
results.sort_by! {|x| -x[1]}.map(&:first)
end
end
end
def pattern_tags
Tag.name_matches(post_set.tag_string).all(:select => "name", :limit => Danbooru.config.tag_query_limit, :order => "post_count DESC").map(&:name)
end
def related_tags_for_group
RelatedTagCalculator.calculate_from_sample_to_array(post_set.tag_string).map(&:first)
end
def related_tags_for_single
tag = Tag.find_by_name(post_set.tag_string.downcase)
if tag
tag.related_tag_array.map(&:first)
else
[]
end
end
def tag_list_html(template)
tag_set_presenter.tag_list_html(template)
end
end
end