7f3b98969f
Refactors things such that Sources::Site has a normalize_for_artist_finder! method that delegates to the strategy for the appropriate site. This way any site that needs to normalize URLs for the artist finder can do so.
63 lines
1.6 KiB
Ruby
63 lines
1.6 KiB
Ruby
# encoding: UTF-8
|
|
|
|
module Sources
|
|
class Site
|
|
attr_reader :url, :strategy
|
|
delegate :get, :referer_url, :site_name, :artist_name, :profile_url, :image_url, :tags, :artist_record, :unique_id, :page_count, :to => :strategy
|
|
|
|
def self.strategies
|
|
[Strategies::Pixiv, Strategies::NicoSeiga, Strategies::DeviantArt, Strategies::Nijie]
|
|
end
|
|
|
|
def initialize(url)
|
|
@url = url
|
|
|
|
Site.strategies.each do |strategy|
|
|
if strategy.url_match?(url)
|
|
@strategy = strategy.new(url)
|
|
break
|
|
end
|
|
end
|
|
end
|
|
|
|
def normalize_for_artist_finder!
|
|
if available?
|
|
return strategy.normalize_for_artist_finder!
|
|
else
|
|
return url
|
|
end
|
|
end
|
|
|
|
def translated_tags
|
|
untranslated_tags = tags
|
|
untranslated_tags = untranslated_tags.map(&:first)
|
|
untranslated_tags = untranslated_tags.map do |tag|
|
|
if tag =~ /\A(\S+?)_?\d+users入り\Z/
|
|
$1
|
|
else
|
|
tag
|
|
end
|
|
end
|
|
WikiPage.other_names_match(untranslated_tags).map{|wiki_page| [wiki_page.title, wiki_page.category_name]}
|
|
end
|
|
|
|
def to_json
|
|
return {
|
|
:artist_name => artist_name,
|
|
:profile_url => profile_url,
|
|
:image_url => image_url,
|
|
:tags => tags,
|
|
:translated_tags => translated_tags,
|
|
:danbooru_name => artist_record.try(:first).try(:name),
|
|
:danbooru_id => artist_record.try(:first).try(:id),
|
|
:unique_id => unique_id,
|
|
:page_count => page_count
|
|
}.to_json
|
|
end
|
|
|
|
def available?
|
|
strategy.present?
|
|
end
|
|
end
|
|
end
|