148 lines
4.6 KiB
Ruby
148 lines
4.6 KiB
Ruby
module Sources
|
|
module Strategies
|
|
class NicoSeiga < Base
|
|
def self.url_match?(url)
|
|
url =~ /^https?:\/\/(?:\w+\.)?nico(?:seiga|video)\.jp/
|
|
end
|
|
|
|
def referer_url
|
|
if @referer_url =~ /seiga\.nicovideo\.jp\/seiga\/im\d+/ && @url =~ /http:\/\/lohas\.nicoseiga\.jp\/priv\//
|
|
@referer_url
|
|
else
|
|
@url
|
|
end
|
|
end
|
|
|
|
def site_name
|
|
"Nico Seiga"
|
|
end
|
|
|
|
def unique_id
|
|
profile_url =~ /\/illust\/(\d+)/
|
|
"nicoseiga" + $1
|
|
end
|
|
|
|
def get
|
|
page = agent.get(normalized_url)
|
|
|
|
if page.search("a#link_btn_login").any?
|
|
# Session cache is invalid, clear it and log in normally.
|
|
Cache.delete("nico-seiga-session")
|
|
@agent = nil
|
|
page = agent.get(normalized_url)
|
|
end
|
|
|
|
@artist_name, @profile_url = get_profile_from_page(page)
|
|
@image_url = get_image_url_from_page(page)
|
|
|
|
# Log out before getting the tags.
|
|
# The reason for this is that if you're logged in and viewing a non-adult-rated work, the tags will be added with javascript after the page has loaded meaning we can't extract them easily.
|
|
# This does not apply if you're logged out (or if you're viewing an adult-rated work).
|
|
agent.cookie_jar.clear!
|
|
agent.get(normalized_url) do |page|
|
|
@tags = get_tags_from_page(page)
|
|
end
|
|
end
|
|
|
|
protected
|
|
|
|
def get_profile_from_page(page)
|
|
links = page.search("li a").select {|x| x["href"] =~ /user\/illust/}
|
|
|
|
if links.any?
|
|
profile_url = "http://seiga.nicovideo.jp" + links[0]["href"]
|
|
artist_name = links[0].search("span")[0].children[0].text
|
|
else
|
|
profile_url = nil
|
|
artist_name = nil
|
|
end
|
|
|
|
return [artist_name, profile_url].compact
|
|
end
|
|
|
|
def get_image_url_from_page(page)
|
|
link = page.search("a#illust_link")
|
|
|
|
if link.any?
|
|
image_url = "http://seiga.nicovideo.jp" + link[0]["href"]
|
|
page = agent.get(image_url) # need to follow this redirect while logged in or it won't work
|
|
if page.is_a?(Mechanize::Image)
|
|
return page.uri.to_s
|
|
end
|
|
images = page.search("img").select {|x| x["src"] =~ /\/priv\//}
|
|
if images.any?
|
|
image_url = "http://lohas.nicoseiga.jp" + images[0]["src"]
|
|
end
|
|
else
|
|
image_url = nil
|
|
end
|
|
|
|
return image_url
|
|
end
|
|
|
|
def get_tags_from_page(page)
|
|
links = page.search("a.tag")
|
|
|
|
links.map do |node|
|
|
[node.text, "http://seiga.nicovideo.jp" + node.attr("href")]
|
|
end
|
|
end
|
|
|
|
def normalized_url
|
|
@normalized_url ||= begin
|
|
if url =~ %r!\Ahttp://lohas\.nicoseiga\.jp/o/[a-f0-9]+/\d+/(\d+)!
|
|
"http://seiga.nicovideo.jp/seiga/im#{$1}"
|
|
elsif url =~ %r{\Ahttp://lohas\.nicoseiga\.jp/priv/(\d+)\?e=\d+&h=[a-f0-9]+}i
|
|
"http://seiga.nicovideo.jp/seiga/im#{$1}"
|
|
elsif url =~ %r{\Ahttp://lohas\.nicoseiga\.jp/priv/[a-f0-9]+/\d+/(\d+)}i
|
|
"http://seiga.nicovideo.jp/seiga/im#{$1}"
|
|
elsif url =~ %r{\Ahttp://lohas\.nicoseiga\.jp/priv/(\d+)}i
|
|
"http://seiga.nicovideo.jp/seiga/im#{$1}"
|
|
elsif url =~ %r{\Ahttp://lohas\.nicoseiga\.jp//?thumb/(\d+)}i
|
|
"http://seiga.nicovideo.jp/seiga/im#{$1}"
|
|
elsif url =~ %r{/seiga/im\d+}
|
|
url
|
|
else
|
|
nil
|
|
end
|
|
end
|
|
end
|
|
|
|
def agent
|
|
@agent ||= begin
|
|
mech = Mechanize.new
|
|
|
|
session = Cache.get("nico-seiga-session")
|
|
if session
|
|
cookie = Mechanize::Cookie.new("user_session", session)
|
|
cookie.domain = ".nicovideo.jp"
|
|
cookie.path = "/"
|
|
mech.cookie_jar.add(cookie)
|
|
else
|
|
mech.get("https://secure.nicovideo.jp/secure/login_form") do |page|
|
|
page.form_with do |form|
|
|
form["mail_tel"] = Danbooru.config.nico_seiga_login
|
|
form["password"] = Danbooru.config.nico_seiga_password
|
|
end.click_button
|
|
end
|
|
session = mech.cookie_jar.cookies.select{|c| c.name == "user_session"}.first
|
|
if session
|
|
Cache.put("nico-seiga-session", session.value, 1.month)
|
|
else
|
|
raise "Session not found"
|
|
end
|
|
end
|
|
|
|
# This cookie needs to be set to allow viewing of adult works
|
|
cookie = Mechanize::Cookie.new("skip_fetish_warning", "1")
|
|
cookie.domain = "seiga.nicovideo.jp"
|
|
cookie.path = "/"
|
|
mech.cookie_jar.add(cookie)
|
|
|
|
mech
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|