412b7f2727
Split requests made by Danbooru::Http into either internal or external requests. Internal requests are API calls to internal services run by Danbooru. External requests are requests to external websites, for example fetching sources or downloading files. External requests may use a HTTP proxy if one is configured. Internal requests don't. Fixes a few source extractors not using the HTTP proxy for certain API calls.
73 lines
2.7 KiB
Ruby
73 lines
2.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
# An API client for the Reportbooru service. Reportbooru tracks post view
|
|
# counts, post search counts, and missed search counts.
|
|
#
|
|
# @see https://github.com/r888888888/reportbooru
|
|
class ReportbooruService
|
|
attr_reader :http, :reportbooru_server
|
|
|
|
def initialize(http: Danbooru::Http.internal, reportbooru_server: Danbooru.config.reportbooru_server)
|
|
@reportbooru_server = reportbooru_server
|
|
@http = http.timeout(1)
|
|
end
|
|
|
|
# @return [Boolean] true if Reportbooru is configured
|
|
def enabled?
|
|
reportbooru_server.present?
|
|
end
|
|
|
|
# Get the list of today's top missed searches.
|
|
# @param expires_in [Integer] the length of time to cache the results
|
|
# @return [Hash<String, Integer>] a map from searches to search counts
|
|
def missed_search_rankings(expires_in: 1.minute)
|
|
return [] unless enabled?
|
|
|
|
response = http.cache(expires_in).get("#{reportbooru_server}/missed_searches")
|
|
return [] if response.status != 200
|
|
|
|
body = response.to_s.force_encoding("utf-8")
|
|
body.lines.map(&:split).map { [_1, _2.to_i] }
|
|
end
|
|
|
|
# Get the list of the day's top searches.
|
|
# @param date [Date] the date to check (YYYY-MM-DD)
|
|
# @param expires_in [Integer] the length of time to cache the results
|
|
# @return [Array<Array<(String, Float)>>] a map from searches to search counts
|
|
def post_search_rankings(date, expires_in: 1.minute)
|
|
request("#{reportbooru_server}/post_searches/rank?date=#{date}", expires_in)
|
|
end
|
|
|
|
# Get the list of the day's most viewed posts.
|
|
# @param date [Date] the date to check (YYYY-MM-DD)
|
|
# @param expires_in [Integer] the length of time to cache the results
|
|
# @return [Array<Array<(String, Float)>>] a map from post ids to view counts
|
|
def post_view_rankings(date, expires_in: 1.minute)
|
|
request("#{reportbooru_server}/post_views/rank?date=#{date}", expires_in)
|
|
end
|
|
|
|
def popular_searches(date, limit: 100, expires_in: 1.hour)
|
|
ranking = post_search_rankings(date, expires_in: expires_in)
|
|
ranking = post_search_rankings(date.yesterday, expires_in: expires_in) if ranking.blank?
|
|
ranking.take(limit).map(&:first)
|
|
end
|
|
|
|
def popular_posts(date, limit: 100)
|
|
ranking = post_view_rankings(date)
|
|
ranking = post_view_rankings(date.yesterday) if ranking.blank?
|
|
ranking.take(limit).map { |x| Post.find(x[0]) }
|
|
end
|
|
|
|
# Send a request to Reportbooru
|
|
# @param url [String] the full Reportbooru URL
|
|
# @param expires_in [Integer] the length of time to cache the results
|
|
# @return [Object] the parsed JSON response
|
|
def request(url, expires_in)
|
|
return [] unless enabled?
|
|
|
|
response = http.cache(expires_in).get(url)
|
|
return [] if response.status != 200
|
|
JSON.parse(response.to_s.force_encoding("utf-8"))
|
|
end
|
|
end
|