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

save twitter auth token

Browse Source
This commit is contained in:
r888888888
2015-01-24 13:41:09 -08:00
parent b3336e8657
commit 4ac89201df
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
app/logical/sources/strategies/twitter.rb
View File

@@ -14,6 +14,7 @@ module Sources::Strategies
def get
agent.get(url) do |page|
puts page.body
@artist_name, @profile_url = get_profile_from_page(page)
@image_url = get_image_url_from_page(page)
end
@@ -52,10 +53,18 @@ module Sources::Strategies
session = Cache.get("twitter-session")
if session
auth_token = Cache.get("twitter-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
mech.get("https://twitter.com/login") do |page|
page.form_with(:action => "https://twitter.com/sessions") do |form|
@@ -65,6 +74,8 @@ module Sources::Strategies
end
session = mech.cookie_jar.cookies.select{|c| c.name == "_twitter_sess"}.first
Cache.put("twitter-session", session.value, 1.month) if session
auth_token = mech.cookie_jar.cookies.select{|c| c.name == "auth_token"}.first
Cache.put("twitter-auth-token", auth_token.value, 1.month) if auth_token
end
mech