jaculabilis/danbooru
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

artstation: fix source to page url after download.

Browse Source
This commit is contained in:
evazion
2017-06-06 13:01:49 -05:00
parent a68ff3c347
commit 215569fbe0
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app/logical/downloads/file.rb
View File

@@ -57,7 +57,7 @@ module Downloads
def after_download(src)
src = fix_twitter_sources(src)
if options[:referer_url].present?
src = set_source_to_referer(src)
src = set_source_to_referer(src, options[:referer_url])
end
src
end
@@ -137,12 +137,13 @@ module Downloads
end
end
def set_source_to_referer(src)
def set_source_to_referer(src, referer)
if Sources::Strategies::Nijie.url_match?(src) ||
Sources::Strategies::Twitter.url_match?(src) ||
Sources::Strategies::Tumblr.url_match?(src) ||
Sources::Strategies::Pawoo.url_match?(src)
strategy = Sources::Site.new(src, :referer_url => options[:referer_url])
Sources::Strategies::Pawoo.url_match?(src) ||
Sources::Strategies::ArtStation.url_match?(src) || Sources::Strategies::ArtStation.url_match?(referer)
strategy = Sources::Site.new(src, :referer_url => referer)
strategy.referer_url
else
src