2 files changed, 89 insertions, 1 deletions
diff --git a/src/main/java/com/juick/http/www/Main.java b/src/main/java/com/juick/http/www/Main.java
index ced7fb98..9a22c172 100644
--- a/src/main/java/com/juick/http/www/Main.java
+++ b/src/main/java/com/juick/http/www/Main.java
@@ -59,6 +59,7 @@ public class Main extends HttpServlet implements Stream.StreamListener {
     NewMessage pagesNewMessage = new NewMessage();
     FacebookLogin loginFacebook = new FacebookLogin();
     VKontakteLogin loginVK = new VKontakteLogin();
+    TwitterAuth twitterAuth;
     SignUp signup = new SignUp();
     Settings settings = new Settings();
     RSS rss = new RSS();
@@ -76,7 +77,8 @@ public class Main extends HttpServlet implements Stream.StreamListener {
             sqlSearch = null; // init this on search, timeout is too low
 
             setupXmppComponent(conf.getProperty("xmpp_password"));
-
+            twitterAuth = new TwitterAuth(conf.getProperty("twitter_consumer_key"),
+                    conf.getProperty("twitter_consumer_secret"));
             PageTemplates.sape = new Sape(conf.getProperty("sape_user"), "juick.com", 2000, 3600);
         } catch (Exception e) {
             log(null, e);
@@ -217,6 +219,8 @@ public class Main extends HttpServlet implements Stream.StreamListener {
             loginFacebook.doGet(sql, request, response);
         } else if (uri.equals("/_vklogin")) {
             loginVK.doGet(sql, request, response);
+        } else if (uri.equals("/_twitter")) {
+            twitterAuth.doGet(sql, request, response);
         } else if (uri.equals("/signup")) {
             signup.doGet(sql, request, response);
         } else if (uri.equals("/help") || uri.equals("/help/")) {
diff --git a/src/main/java/com/juick/http/www/TwitterAuth.java b/src/main/java/com/juick/http/www/TwitterAuth.java
new file mode 100644
index 00000000..5c5581d1
--- /dev/null
+++ b/src/main/java/com/juick/http/www/TwitterAuth.java
@@ -0,0 +1,84 @@
+package com.juick.http.www;
+
+import com.github.scribejava.apis.TwitterApi;
+import com.github.scribejava.core.builder.ServiceBuilder;
+import com.github.scribejava.core.model.*;
+import com.github.scribejava.core.oauth.OAuthService;
+import com.juick.server.UserQueries;
+import org.json.JSONObject;
+
+import javax.servlet.ServletException;
+import javax.servlet.http.Cookie;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.sql.Connection;
+
+/**
+ * Created by vt on 01.12.2015.
+ */
+public class TwitterAuth {
+
+    private final static String VERIFY_URL = "https://api.twitter.com/1.1/account/verify_credentials.json";
+
+    private String consumerKey, consumerSecret;
+
+    public TwitterAuth(String consumerKey, String consumerSecret) {
+        this.consumerKey = consumerKey;
+        this.consumerSecret = consumerSecret;
+    }
+
+    protected void doGet(Connection sql, HttpServletRequest request, HttpServletResponse response)
+            throws ServletException, IOException {
+        String hash = "", request_token = "", request_token_secret = "";
+        String verifier = request.getParameter("oauth_verifier");
+        Cookie[] cookies = request.getCookies();
+        for (Cookie cookie : cookies) {
+            if (cookie.getName().equals("hash")) {
+                hash = cookie.getValue();
+            }
+            if (cookie.getName().equals("request_token")) {
+                request_token = cookie.getValue();
+            }
+            if (cookie.getName().equals("request_token_secret")) {
+                request_token_secret = cookie.getValue();
+            }
+        }
+        com.juick.User user = UserQueries.getUserByHash(sql, hash);
+        if ( user == null || user.UID == 0) {
+            response.sendError(403);
+            return;
+        }
+        OAuthService oAuthService = new ServiceBuilder()
+                .provider(TwitterApi.class)
+                .apiKey(consumerKey)
+                .apiSecret(consumerSecret)
+                .callback("http://juick.com/_twitter")
+                .build();
+
+        if (request_token.isEmpty() && request_token_secret.isEmpty()
+                && (verifier == null || verifier.isEmpty())) {
+            Token requestToken = oAuthService.getRequestToken();
+            String authUrl = oAuthService.getAuthorizationUrl(requestToken);
+            response.addCookie(new Cookie("request_token", requestToken.getToken()));
+            response.addCookie(new Cookie("request_token_secret", requestToken.getSecret()));
+            response.setStatus(HttpServletResponse.SC_MOVED_TEMPORARILY);
+            response.setHeader("Location", authUrl);
+        } else {
+            if (verifier != null && verifier.length() > 0) {
+                Token requestToken = new Token(request_token, request_token_secret);
+                Token accessToken = oAuthService.getAccessToken(requestToken, new Verifier(verifier));
+                OAuthRequest oAuthRequest = new OAuthRequest(Verb.GET, VERIFY_URL, oAuthService);
+                oAuthService.signRequest(accessToken, oAuthRequest);
+                JSONObject jsonResponse = new JSONObject(oAuthRequest.send().getBody());
+                String screenName = jsonResponse.getString("screen_name");
+                if (UserQueries.linkTwitterAccount(sql, user, accessToken.getToken(), accessToken.getSecret(), screenName)) {
+                    response.setStatus(HttpServletResponse.SC_MOVED_TEMPORARILY);
+                    response.setHeader("Location", "http://juick.com/settings");
+                } else {
+                    response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
+                }
+            }
+        }
+    }
+}