72 lines
2.0 KiB
Ruby
72 lines
2.0 KiB
Ruby
module Downloads
|
|
module Strategies
|
|
class Pixiv < Base
|
|
def rewrite(url, headers)
|
|
if url =~ /https?:\/\/(?:\w+\.)?pixiv\.net/
|
|
url, headers = rewrite_headers(url, headers)
|
|
url, headers = rewrite_html_pages(url, headers)
|
|
url, headers = rewrite_small_images(url, headers)
|
|
url, headers = rewrite_small_manga_pages(url, headers)
|
|
end
|
|
|
|
return [url, headers]
|
|
end
|
|
|
|
protected
|
|
def rewrite_headers(url, headers)
|
|
headers["Referer"] = "http://www.pixiv.net"
|
|
return [url, headers]
|
|
end
|
|
|
|
def rewrite_html_pages(url, headers)
|
|
# example: http://www.pixiv.net/member_illust.php?mode=big&illust_id=23828655
|
|
|
|
if url =~ %r!illust_id=\d+!
|
|
source = ::Sources::Strategies::Pixiv.new(url)
|
|
source.get
|
|
return [source.image_url, headers]
|
|
else
|
|
return [url, headers]
|
|
end
|
|
end
|
|
|
|
def rewrite_small_images(url, headers)
|
|
if url =~ %r!(/img/.+?/.+?)_m.+$!
|
|
match = $1
|
|
url.sub!(match + "_m", match)
|
|
end
|
|
|
|
return [url, headers]
|
|
end
|
|
|
|
def rewrite_small_manga_pages(url, headers)
|
|
if url =~ %r!(\d+_p\d+)\.!
|
|
match = $1
|
|
repl = match.sub(/_p/, "_big_p")
|
|
big_url = url.sub(match, repl)
|
|
if http_exists?(big_url)
|
|
url = big_url
|
|
end
|
|
end
|
|
|
|
return [url, headers]
|
|
end
|
|
|
|
def http_exists?(url)
|
|
# example: http://img01.pixiv.net/img/as-special/15649262_big_p2.jpg
|
|
exists = false
|
|
uri = URI.parse(url)
|
|
Net::HTTP.start(uri.host, uri.port) do |http|
|
|
headers = {"Referer" => "http://www.pixiv.net", "User-Agent" => "#{Danbooru.config.app_name}/#{Danbooru.config.version}"}
|
|
http.request_head(uri.request_uri, headers) do |res|
|
|
if res.is_a?(Net::HTTPSuccess)
|
|
exists = true
|
|
end
|
|
end
|
|
end
|
|
exists
|
|
end
|
|
end
|
|
end
|
|
end
|