854d587373
e.g. If you were on an html work page on pixiv, clicked a link to a different html work page on pixiv, and then clicked the bookmarklet, then it used to fetch the source from the FIRST work you were on instead of the second.
115 lines
3.0 KiB
Ruby
115 lines
3.0 KiB
Ruby
module Sources
|
|
module Strategies
|
|
class Nijie < Base
|
|
def self.url_match?(url)
|
|
url =~ /^https?:\/\/(?:.+?\.)?nijie\.info/
|
|
end
|
|
|
|
def referer_url(template)
|
|
if template.params[:ref] =~ /nijie\.info\/view\.php.+id=\d+/ && template.params[:url] =~ /pic\d+\.nijie.info\/nijie_picture\//
|
|
template.params[:ref]
|
|
else
|
|
template.params[:url]
|
|
end
|
|
end
|
|
|
|
def site_name
|
|
"Nijie"
|
|
end
|
|
|
|
def unique_id
|
|
profile_url =~ /nijie\.info\/members.php\?id=(\d+)/
|
|
"nijie" + $1.to_s
|
|
end
|
|
|
|
def get
|
|
page = agent.get(url)
|
|
|
|
if page.search("div#header-login-container").any?
|
|
# Session cache is invalid, clear it and log in normally.
|
|
Cache.delete("nijie-session")
|
|
@agent = nil
|
|
page = agent.get(url)
|
|
end
|
|
|
|
@artist_name, @profile_url = get_profile_from_page(page)
|
|
@image_url = get_image_url_from_page(page)
|
|
@tags = get_tags_from_page(page)
|
|
end
|
|
|
|
protected
|
|
|
|
def get_profile_from_page(page)
|
|
links = page.search("a.name")
|
|
|
|
if links.any?
|
|
profile_url = "http://nijie.info/" + links[0]["href"]
|
|
artist_name = links[0].text
|
|
else
|
|
profile_url = nil
|
|
artist_name = nil
|
|
end
|
|
|
|
return [artist_name, profile_url].compact
|
|
end
|
|
|
|
def get_image_url_from_page(page)
|
|
image = page.search("div#gallery a img")
|
|
|
|
if image.any?
|
|
image[0]["src"].try(:sub, %r!^//!, "http://")
|
|
else
|
|
nil
|
|
end
|
|
end
|
|
|
|
def get_tags_from_page(page)
|
|
# puts page.root.to_xhtml
|
|
|
|
links = page.search("div#view-tag a").find_all do |node|
|
|
node["href"] =~ /search\.php/
|
|
end
|
|
|
|
if links.any?
|
|
links.map do |node|
|
|
[node.inner_text, "http://nijie.info" + node.attr("href")]
|
|
end
|
|
else
|
|
[]
|
|
end
|
|
end
|
|
|
|
def agent
|
|
@agent ||= begin
|
|
mech = Mechanize.new
|
|
|
|
session = Cache.get("nijie-session")
|
|
if session
|
|
cookie = Mechanize::Cookie.new("NIJIEIJIEID", session)
|
|
cookie.domain = ".nijie.info"
|
|
cookie.path = "/"
|
|
mech.cookie_jar.add(cookie)
|
|
else
|
|
mech.get("http://nijie.info/login.php") do |page|
|
|
page.form_with(:action => "/login_int.php") do |form|
|
|
form['email'] = Danbooru.config.nijie_login
|
|
form['password'] = Danbooru.config.nijie_password
|
|
end.click_button
|
|
end
|
|
session = mech.cookie_jar.cookies.select{|c| c.name == "NIJIEIJIEID"}.first
|
|
Cache.put("nijie-session", session.value, 1.month) if session
|
|
end
|
|
|
|
# This cookie needs to be set to allow viewing of adult works while anonymous
|
|
cookie = Mechanize::Cookie.new("R18", "1")
|
|
cookie.domain = ".nijie.info"
|
|
cookie.path = "/"
|
|
mech.cookie_jar.add(cookie)
|
|
|
|
mech
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|