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.
111 lines
2.7 KiB
Ruby
111 lines
2.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
# An API client for the NicoSeiga XML API.
|
|
class NicoSeigaApiClient
|
|
extend Memoist
|
|
XML_API = "https://seiga.nicovideo.jp/api"
|
|
|
|
attr_reader :http, :user_session
|
|
|
|
def initialize(work_id:, type:, user_session: Danbooru.config.nico_seiga_user_session, http:)
|
|
@work_id = work_id
|
|
@work_type = type
|
|
@user_session = user_session
|
|
@http = http
|
|
end
|
|
|
|
def image_ids
|
|
case @work_type
|
|
when "illust"
|
|
[api_response["id"]]
|
|
when "manga"
|
|
manga_api_response.map { |x| Source::URL.parse(x.dig("meta", "source_url"))&.image_id }.compact
|
|
end
|
|
end
|
|
|
|
def title
|
|
api_response["title"]
|
|
end
|
|
|
|
def description
|
|
api_response["description"]
|
|
end
|
|
|
|
def tags
|
|
tags = api_response.dig("tag_list", "tag")
|
|
if tags.instance_of? Hash
|
|
# When a manga post has a single tag, the XML parser thinks it's a hash instead of an array of hashes,
|
|
# so we need to manually turn it into the latter. Example: https://seiga.nicovideo.jp/watch/mg302561
|
|
tags = [tags]
|
|
end
|
|
tags.to_a.map { |t| t["name"] }.compact
|
|
end
|
|
|
|
def user_id
|
|
if api_response["user_id"].to_i == 0 # anonymous: https://nico.ms/mg310193
|
|
nil
|
|
else
|
|
api_response["user_id"]
|
|
end
|
|
end
|
|
|
|
def user_name
|
|
case @work_type
|
|
when "illust"
|
|
api_response["nickname"]
|
|
when "manga"
|
|
user_api_response(user_id)["nickname"]
|
|
end
|
|
end
|
|
|
|
def api_response
|
|
case @work_type
|
|
when "illust"
|
|
resp = get("https://sp.seiga.nicovideo.jp/ajax/seiga/im#{@work_id}")
|
|
return {} if resp.blank? || resp.code.to_i == 404
|
|
api_response = JSON.parse(resp)["target_image"]
|
|
|
|
when "manga"
|
|
resp = get("#{XML_API}/theme/info?id=#{@work_id}")
|
|
return {} if resp.blank? || resp.code.to_i == 404
|
|
api_response = Hash.from_xml(resp.to_s)["response"]["theme"]
|
|
end
|
|
|
|
api_response || {}
|
|
rescue JSON::ParserError
|
|
{}
|
|
end
|
|
|
|
def manga_api_response
|
|
resp = get("https://ssl.seiga.nicovideo.jp/api/v1/app/manga/episodes/#{@work_id}/frames")
|
|
return {} if resp.blank? || resp.code.to_i == 404
|
|
JSON.parse(resp)["data"]["result"]
|
|
rescue JSON::ParserError
|
|
{}
|
|
end
|
|
|
|
def user_api_response(user_id)
|
|
return {} unless user_id.present?
|
|
resp = get("#{XML_API}/user/info?id=#{user_id}")
|
|
return {} if resp.blank? || resp.code.to_i == 404
|
|
Hash.from_xml(resp.to_s)["response"]["user"]
|
|
end
|
|
|
|
def cookies
|
|
{
|
|
skip_fetish_warning: "1",
|
|
user_session: user_session,
|
|
}
|
|
end
|
|
|
|
def get(url)
|
|
http.cookies(cookies).cache(1.minute).get(url)
|
|
end
|
|
|
|
def head(url)
|
|
http.cookies(cookies).cache(1.minute).head(url)
|
|
end
|
|
|
|
memoize :api_response, :manga_api_response, :user_api_response
|
|
end
|