Files
danbooru/app/logical/sources/strategies/hentai_foundry.rb
evazion b4aea72d04 sources: remove preview_urls method from base strategy.
Remove the `preview_urls` method from strategies. The only place this was used was
when doing IQDB searches, to download the thumbnail image from the source instead of
the full image.

This wasn't worth it for a few reasons:

* Thumbnails on other sites are sometimes not the size we want, which could affect
  IQDB results.
* Grabbing thumbnails is complex for some sites. You can't always just rewrite the
  image URL. Sometimes it requires extra API calls, which can be slower than just
  grabbing the full image.
* For videos and animations, thumbnails from other sites don't always match our
  thumbnails. We do smart thumbnail generation to try to avoid blank thumbnails, which
  means we don't always pick the first frame, which could affect IQDB results.

API changes:

* /iqdb_queries?search[file_url] now downloads the URL as is without any modification.
  Before it tried to change thumbnail and sample size image URLs to the full version.

* /iqdb_queries?search[url] now returns an error if the URL is for a HTML page that
  contains multiple images. Before it would grab only the first image and silently
  ignore the rest.
2022-03-11 03:22:23 -06:00

87 lines
1.9 KiB
Ruby

# frozen_string_literal: true
# @see Source::URL::HentaiFoundry
module Sources
module Strategies
class HentaiFoundry < Base
def match?
parsed_url&.site_name == "Hentai Foundry"
end
def site_name
parsed_url.site_name
end
def image_urls
image = page&.search("#picBox img")
return [] unless image
image.to_a.map { |img| URI.join(page_url, img["src"]).to_s }
end
def page_url
return nil if illust_id.blank?
if artist_name.blank?
"https://www.hentai-foundry.com/pic-#{illust_id}"
else
"https://www.hentai-foundry.com/pictures/user/#{artist_name}/#{illust_id}"
end
end
def page
return nil if page_url.blank?
response = http.cache(1.minute).get("#{page_url}?enterAgree=1")
return nil unless response.status == 200
response.parse
end
def tags
tags = page&.search(".boxbody [rel='tag']") || []
tags.map do |tag|
[tag.text, URI.join(page_url, tag.attr("href")).to_s]
end
end
def artist_name
parsed_url.username || parsed_referer&.username
end
def canonical_url
image_urls.first
end
def profile_url
return nil if artist_name.blank?
"https://www.hentai-foundry.com/user/#{artist_name}"
end
def artist_commentary_title
page&.search("#picBox .imageTitle")&.text
end
def artist_commentary_desc
page&.search("#descriptionBox .picDescript")&.to_html
end
def dtext_artist_commentary_desc
DText.from_html(artist_commentary_desc).gsub(/\A[[:space:]]+|[[:space:]]+\z/, "").gsub(/\n+/, "\n")
end
def normalize_for_source
page_url
end
def illust_id
parsed_url.work_id || parsed_referer&.work_id
end
memoize :page
end
end
end