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

Cache seiga and nijie sessions

Browse Source
This commit is contained in:
Toks
2014-10-05 12:11:08 -04:00
parent 4b9605075b
commit f4529e73e3
2 changed files with 30 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
app/logical/sources/strategies/nico_seiga.rb
View File

@@ -103,11 +103,21 @@ module Sources
@agent ||= begin @agent ||= begin
mech = Mechanize.new mech = Mechanize.new
mech.get("https://secure.nicovideo.jp/secure/login_form") do |page| session = Cache.get("nico-seiga-session")
page.form_with do |form| if session
form["mail_tel"] = Danbooru.config.nico_seiga_login cookie = Mechanize::Cookie.new("user_session", session)
form["password"] = Danbooru.config.nico_seiga_password cookie.domain = ".nicovideo.jp"
end.click_button 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
Cache.put("nico-seiga-session", session.value, 1.month) if session
end end
# This cookie needs to be set to allow viewing of adult works # This cookie needs to be set to allow viewing of adult works

20
app/logical/sources/strategies/nijie.rb
View File

@@ -76,11 +76,21 @@ module Sources
@agent ||= begin @agent ||= begin
mech = Mechanize.new mech = Mechanize.new
mech.get("http://nijie.info/login.php") do |page| session = Cache.get("nijie-session")
page.form_with(:action => "/login_int.php") do |form| if session
form['email'] = Danbooru.config.nijie_login cookie = Mechanize::Cookie.new("NIJIEIJIEID", session)
form['password'] = Danbooru.config.nijie_password cookie.domain = ".nijie.info"
end.click_button 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 end
# This cookie needs to be set to allow viewing of adult works while anonymous # This cookie needs to be set to allow viewing of adult works while anonymous