Files
danbooru/app/logical/source/extractor/anifty.rb
2022-04-18 16:50:26 +02:00

111 lines
3.0 KiB
Ruby

# frozen_string_literal: true
# @see Source::URL::Anifty
class Source::Extractor
class Anifty < Source::Extractor
def match?
Source::URL::Anifty === parsed_url
end
def image_urls
if parsed_url.image_url?
[parsed_url.full_image_url].compact
else
[api_response["imageURL"]].compact
end
end
def profile_url
if artist_name.present?
"https://anifty.jp/@#{username}"
else
parsed_url.profile_url || parsed_referer&.profile_url
end
end
def username
api_response.dig("creator", "userName") || artist_api_response["userName"]
end
def artist_name
api_response.dig("creator", "displayName") || artist_api_response.dig("createdTokens", 0, "creatorProfile", "displayNameEN")
end
def other_names
other_names = [username]
if api_response.present?
other_names << api_response.dig("creator", "displayNameJA")
elsif artist_api_response
other_names << artist_api_response.dig("createdTokens", 0, "creatorProfile", "displayNameJP")
end
other_names.compact.uniq
end
def artist_commentary_title
api_response["title"] || api_response["titleJA"]
end
def artist_commentary_desc
api_response["description"] || api_response["descriptionJA"]
end
def tags
# anifty marketplace uses XHR requests to filter by tags, so there's no url to get
api_response["tags"].to_a.map do |tag|
[tag["name"], "https://anifty.jp/marketplace"]
end
end
def page_url
if page_url_from_parsed_urls.present?
page_url_from_parsed_urls
elsif work_id.present?
"https://anifty.jp/creations/#{work_id}"
end
end
def page_url_from_parsed_urls
parsed_url.page_url || parsed_referer&.page_url
end
def work_id
parsed_url.work_id || parsed_referer&.work_id || work_id_from_artist_api
end
def work_id_from_artist_api
# Try to get the work ID from the artist's list of tokens
return nil unless parsed_url.file.present? && parsed_url.work_type == "creation"
artist_api_response["createdTokens"].to_a.map do |token|
if Source::URL.parse(token["imageURL"])&.file == parsed_url.file
return token["creationID"]
end
end
nil
end
def artist_hash
parsed_url.artist_hash || parsed_referer&.artist_hash
end
def api_response
return {} if work_id.blank?
resp = http.cache(1.minute).get("https://asia-northeast1-anifty-59655.cloudfunctions.net/api/v2/creations/#{work_id}")
return {} if resp.code != 200
resp.parse.with_indifferent_access
end
memoize :api_response
def artist_api_response
return {} if artist_hash.blank?
resp = http.cache(1.minute).get("https://asia-northeast1-anifty-59655.cloudfunctions.net/api/users/#{artist_hash}")
return {} if resp.code != 200
resp.parse.with_indifferent_access
end
memoize :artist_api_response
end
end