44 lines
1.2 KiB
Ruby
44 lines
1.2 KiB
Ruby
module Downloads
|
|
module Strategies
|
|
class NicoSeiga < Base
|
|
def rewrite(url, headers)
|
|
if url =~ %r{https?://lohas\.nicoseiga\.jp} || url =~ %r{https?://seiga\.nicovideo\.jp}
|
|
url, headers = rewrite_headers(url, headers)
|
|
url, headers = rewrite_html_pages(url, headers)
|
|
url, headers = rewrite_thumbnails(url, headers)
|
|
end
|
|
|
|
return [url, headers]
|
|
end
|
|
|
|
protected
|
|
def rewrite_headers(url, headers)
|
|
headers["Referer"] = "http://seiga.nicovideo.jp"
|
|
return [url, headers]
|
|
end
|
|
|
|
def rewrite_html_pages(url, headers)
|
|
# example: http://seiga.nicovideo.jp/seiga/im1389842
|
|
|
|
if url =~ %r{https?://seiga\.nicovideo\.jp/seiga/im\d+}
|
|
source = ::Sources::Strategies::NicoSeiga.new(url)
|
|
source.get
|
|
return [source.image_url, headers]
|
|
else
|
|
return [url, headers]
|
|
end
|
|
end
|
|
|
|
def rewrite_thumbnails(url, headers)
|
|
if url =~ %r{/thumb/\d+}
|
|
source = ::Sources::Strategies::NicoSeiga.new(url)
|
|
source.get
|
|
return [source.image_url, headers]
|
|
end
|
|
|
|
return [url, headers]
|
|
end
|
|
end
|
|
end
|
|
end
|