1 files changed, 34 insertions, 1 deletions
diff --git a/src/main/java/com/juick/server/UserQueries.java b/src/main/java/com/juick/server/UserQueries.java
index 9140e8ea..ff819629 100644
--- a/src/main/java/com/juick/server/UserQueries.java
+++ b/src/main/java/com/juick/server/UserQueries.java
@@ -17,6 +17,8 @@
  */
 package com.juick.server;
 
+import com.juick.User;
+
 import java.sql.Connection;
 import java.sql.PreparedStatement;
 import java.sql.ResultSet;
@@ -25,6 +27,7 @@ import java.sql.Statement;
 import java.util.ArrayList;
 import java.util.Random;
 import java.util.UUID;
+import java.util.logging.Logger;
 
 /**
  *
@@ -34,6 +37,8 @@ public class UserQueries {
 
     static final String ABCDEF = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ";
 
+    private static Logger logger = Logger.getLogger(UserQueries.class.getName());
+
     public static String getSignUpHashByJID(Connection sql, String jid) {
         String hash = SQLHelpers.getString(sql, "SELECT loginhash FROM jids WHERE jid=? AND user_id IS NULL", jid);
 
@@ -279,13 +284,14 @@ public class UserQueries {
         PreparedStatement stmt = null;
         ResultSet rs = null;
         try {
-            stmt = sql.prepareStatement("SELECT logins.user_id,users.nick FROM logins INNER JOIN users ON logins.user_id=users.id WHERE logins.hash=?");
+            stmt = sql.prepareStatement("SELECT logins.user_id,users.nick, users.banned FROM logins INNER JOIN users ON logins.user_id=users.id WHERE logins.hash=?");
             stmt.setString(1, hash);
             rs = stmt.executeQuery();
             if (rs.first()) {
                 user = new com.juick.User();
                 user.UID = rs.getInt(1);
                 user.UName = rs.getString(2);
+                user.Banned = rs.getBoolean(3);
                 user.AuthHash = hash;
             }
         } catch (SQLException e) {
@@ -574,6 +580,33 @@ public class UserQueries {
         return users;
     }
 
+    public static boolean linkTwitterAccount(Connection sql, User user, String accessToken,
+                                             String accessTokenSecret, String screenName) {
+        try {
+            PreparedStatement stmt =
+                    sql.prepareStatement("INSERT INTO twitter(user_id,access_token,access_token_secret,uname) " +
+                            "VALUES (?,?,?,?)" +
+                            " ON DUPLICATE KEY UPDATE access_token=?,access_token_secret=?,uname=?");
+            stmt.setInt(1, user.UID);
+            stmt.setString(2, accessToken);
+            stmt.setString(3, accessTokenSecret);
+            stmt.setString(4, screenName);
+            stmt.setString(5, accessToken);
+            stmt.setString(6, accessTokenSecret);
+            stmt.setString(7, screenName);
+            if (stmt.execute()) {
+                PreparedStatement stmt2 =
+                        sql.prepareStatement("INSERT INTO subscr_users(user_id,suser_id,jid) " +
+                                "VALUES (?,1741,'juick\\@twitter.juick.com')");
+                stmt2.setInt(1, user.UID);
+                return stmt2.execute();
+            }
+        } catch (SQLException e) {
+            logger.throwing(UserQueries.class.getName(), "linkTwitterAccount", e);
+        }
+        return false;
+    }
+
     public static int getStatsIRead(Connection sql, int uid) {
         return SQLHelpers.getInt(sql, "SELECT COUNT(*) FROM subscr_users WHERE suser_id=?", uid, 0);
     }