This commit is contained in:
r888888888
2015-01-27 16:29:11 -08:00
parent 75bdc3fc63
commit 1d57d37cd5

View File

@@ -47,23 +47,22 @@ module Sources::Strategies
private private
def add_cookie(mech, name, value)
cookie = Mechanize::Cookie.new(name, value)
cookie.domain = ".twitter.com"
cookie.path = "/"
mech.cookie_jar.add(cookie)
end
def agent def agent
@agent ||= begin @agent ||= begin
mech = Mechanize.new mech = Mechanize.new
session = Cache.get("twitter-session") session = Cache.get("twitter-session")
auth_token = Cache.get("twitter-auth-token")
if session if session && auth_token
auth_token = Cache.get("twitter-auth-token") add_cookie(mech, "_twitter_sess", session)
add_cookie(mech, "auth_token", auth_token)
cookie = Mechanize::Cookie.new("_twitter_sess", session)
cookie.domain = ".twitter.com"
cookie.path = "/"
mech.cookie_jar.add(cookie)
cookie = Mechanize::Cookie.new("auth_token", auth_token)
cookie.domain = ".twitter.com"
cookie.path = "/"
mech.cookie_jar.add(cookie)
elsif Danbooru.config.twitter_login elsif Danbooru.config.twitter_login
mech.get("https://twitter.com/login") do |page| mech.get("https://twitter.com/login") do |page|