From 1d57d37cd52510d307698b30e577c1e2da7651c3 Mon Sep 17 00:00:00 2001
From: r888888888 <r888888888@gmail.com>
Date: Tue, 27 Jan 2015 16:29:11 -0800
Subject: [PATCH] refactor

---
 app/logical/sources/strategies/twitter.rb | 23 +++++++++++------------
 1 file changed, 11 insertions(+), 12 deletions(-)

diff --git a/app/logical/sources/strategies/twitter.rb b/app/logical/sources/strategies/twitter.rb
index 762635c31..c23807742 100644
--- a/app/logical/sources/strategies/twitter.rb
+++ b/app/logical/sources/strategies/twitter.rb
@@ -47,23 +47,22 @@ module Sources::Strategies
 
     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
       @agent ||= begin
         mech = Mechanize.new
         session = Cache.get("twitter-session")
+        auth_token = Cache.get("twitter-auth-token")
 
-        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)
+        if session && auth_token
+          add_cookie(mech, "_twitter_sess", session)
+          add_cookie(mech, "auth_token", auth_token)
 
         elsif Danbooru.config.twitter_login
           mech.get("https://twitter.com/login") do |page|