PMQueries; UserQueries.getUserByUID; UserQueries.fillUsersByID

3 files changed, 237 insertions, 0 deletions

diff --git a/src/com/juick/server/MessagesQueries.java b/src/com/juick/server/MessagesQueries.java
index 7eaaff2e3..444cd13d4 100644
--- a/src/com/juick/server/MessagesQueries.java
+++ b/src/com/juick/server/MessagesQueries.java
@@ -119,6 +119,10 @@ public class MessagesQueries {
         return ret;
     }
 
+    public static int getMessagePrivacy(Connection sql, int mid) {
+        return SQLHelpers.getInt(sql, "SELECT privacy FROM messages WHERE message_id=?", mid, -4);
+    }
+
     public static com.juick.Message getMessage(Connection sql, int mid) {
         com.juick.Message msg = null;
 
diff --git a/src/com/juick/server/PMQueries.java b/src/com/juick/server/PMQueries.java
new file mode 100644
index 000000000..42f3aefd4
--- /dev/null
+++ b/src/com/juick/server/PMQueries.java
@@ -0,0 +1,171 @@
+/*
+ * Juick
+ * Copyright (C) 2008-2011, Ugnich Anton
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as
+ * published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+package com.juick.server;
+
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.ArrayList;
+
+/**
+ *
+ * @author Ugnich Anton
+ */
+public class PMQueries {
+
+    public static boolean createPM(Connection sql, int uid_from, int uid_to, String body) {
+        boolean ret = false;
+        PreparedStatement stmt = null;
+        try {
+            stmt = sql.prepareStatement("INSERT INTO pm(user_id,user_id_to,txt) VALUES (?,?,?)");
+            stmt.setInt(1, uid_from);
+            stmt.setInt(2, uid_to);
+            stmt.setString(3, body);
+            ret = stmt.executeUpdate() > 0;
+        } catch (SQLException e) {
+            System.err.println(e);
+        } finally {
+            Utils.finishSQL(null, stmt);
+        }
+
+        if (ret) {
+            PreparedStatement stmt2 = null;
+            try {
+                stmt2 = sql.prepareStatement("INSERT INTO pm_streams(user_id,user_id_to,lastmessage,unread) VALUES (?,?,NOW(),1) ON DUPLICATE KEY UPDATE lastmessage=NOW(),unread=unread+1");
+                stmt2.setInt(1, uid_from);
+                stmt2.setInt(2, uid_to);
+                stmt2.executeUpdate();
+            } catch (SQLException e) {
+                System.err.println(e);
+            } finally {
+                Utils.finishSQL(null, stmt2);
+            }
+        }
+
+        return ret;
+    }
+
+    public static ArrayList<com.juick.User> getPMLastConversationsUsers(Connection sql, int uid, int cnt) {
+        ArrayList<com.juick.User> users = new ArrayList<com.juick.User>();
+
+        PreparedStatement stmt = null;
+        ResultSet rs = null;
+        try {
+            stmt = sql.prepareStatement("SELECT user_id,unread FROM pm_streams WHERE user_id_to=? AND unread>0 ORDER BY lastmessage DESC LIMIT " + cnt);
+            stmt.setInt(1, uid);
+            rs = stmt.executeQuery();
+            rs.beforeFirst();
+            while (rs.next()) {
+                com.juick.User u = new com.juick.User();
+                u.UID = rs.getInt(1);
+                u.MessagesCount = rs.getInt(2);
+                users.add(u);
+            }
+        } catch (SQLException e) {
+            System.err.println(e);
+        } finally {
+            Utils.finishSQL(rs, stmt);
+        }
+
+        if (users.size() < cnt) {
+            try {
+                stmt = sql.prepareStatement("SELECT user_id,user_id_to FROM pm_streams WHERE user_id=? OR user_id_to=? ORDER BY lastmessage DESC LIMIT ?");
+                stmt.setInt(1, uid);
+                stmt.setInt(2, uid);
+                stmt.setInt(3, cnt * 2);
+                rs = stmt.executeQuery();
+                rs.beforeFirst();
+                while (rs.next()) {
+                    int uuid = rs.getInt(1) + rs.getInt(2) - uid;
+                    if (!haveUserInArray(users, uuid)) {
+                        com.juick.User u = new com.juick.User();
+                        u.UID = uuid;
+                        users.add(u);
+                        if (users.size() >= cnt) {
+                            break;
+                        }
+                    }
+                }
+            } catch (SQLException e) {
+                System.err.println(e);
+            } finally {
+                Utils.finishSQL(rs, stmt);
+            }
+        }
+
+        if (!users.isEmpty()) {
+            UserQueries.fillUsersByID(sql, users);
+        }
+
+        return users;
+    }
+
+    public static boolean haveUserInArray(ArrayList<com.juick.User> arr, int uid) {
+        int s = arr.size();
+        for (int i = 0; i < s; i++) {
+            if (arr.get(i).UID == uid) {
+                return true;
+            }
+        }
+        return false;
+    }
+
+    public static ArrayList<com.juick.Message> getPMMessages(Connection sql, int uid, int uid_to) {
+        ArrayList<com.juick.Message> msgs = new ArrayList<com.juick.Message>(20);
+
+        PreparedStatement stmt = null;
+        ResultSet rs = null;
+        try {
+            stmt = sql.prepareStatement("SELECT user_id,txt,ts FROM pm WHERE (user_id=? AND user_id_to=?) OR (user_id_to=? AND user_id=?) ORDER BY ts DESC LIMIT 20");
+            stmt.setInt(1, uid);
+            stmt.setInt(2, uid_to);
+            stmt.setInt(3, uid);
+            stmt.setInt(4, uid_to);
+            rs = stmt.executeQuery();
+            rs.beforeFirst();
+            while (rs.next()) {
+                com.juick.Message msg = new com.juick.Message();
+                int uuid = rs.getInt(1);
+                msg.User = new com.juick.User();
+                msg.User.UID = uuid;
+                msg.Text = rs.getString(2);
+                msg.TimestampString = rs.getString(3);
+                msgs.add(0, msg);
+            }
+        } catch (SQLException e) {
+            System.err.println(e);
+        } finally {
+            Utils.finishSQL(rs, stmt);
+        }
+
+        PreparedStatement stmt2 = null;
+        try {
+            stmt2 = sql.prepareStatement("UPDATE pm_streams SET lastview=NOW(),unread=0 WHERE user_id_to=? AND user_id=?");
+            stmt2.setInt(1, uid);
+            stmt2.setInt(2, uid_to);
+            stmt2.executeUpdate();
+        } catch (SQLException e) {
+            System.err.println(e);
+        } finally {
+            Utils.finishSQL(null, stmt);
+        }
+
+        return msgs;
+    }
+}
diff --git a/src/com/juick/server/UserQueries.java b/src/com/juick/server/UserQueries.java
index c170b6cd7..bb4d19df3 100644
--- a/src/com/juick/server/UserQueries.java
+++ b/src/com/juick/server/UserQueries.java
@@ -79,6 +79,28 @@ public class UserQueries {
         return uid;
     }
 
+    public static com.juick.User getUserByUID(Connection sql, int uid) {
+        com.juick.User user = null;
+
+        PreparedStatement stmt = null;
+        ResultSet rs = null;
+        try {
+            stmt = sql.prepareStatement("SELECT nick FROM users WHERE id=?");
+            stmt.setInt(1, uid);
+            rs = stmt.executeQuery();
+            if (rs.first()) {
+                user = new com.juick.User();
+                user.UID = uid;
+                user.UName = rs.getString(1);
+            }
+        } catch (SQLException e) {
+            System.err.println(e);
+        } finally {
+            Utils.finishSQL(rs, stmt);
+        }
+        return user;
+    }
+
     public static com.juick.User getUserByName(Connection sql, String username) {
         com.juick.User user = null;
 
@@ -148,6 +170,42 @@ public class UserQueries {
         return users;
     }
 
+    public static boolean fillUsersByID(Connection sql, ArrayList<com.juick.User> users) {
+        boolean ret = false;
+
+        String uids = "";
+        final int usersSize = users.size();
+        for (int i = 0; i < usersSize; i++) {
+            if (i > 0) {
+                uids += ",";
+            }
+            uids += users.get(i).UID;
+        }
+
+        PreparedStatement stmt = null;
+        ResultSet rs = null;
+        try {
+            stmt = sql.prepareStatement("SELECT id,nick FROM users WHERE id IN (" + uids + ")");
+            rs = stmt.executeQuery();
+            rs.beforeFirst();
+            while (rs.next()) {
+                int uid = rs.getInt(1);
+                for (int i = 0; i < usersSize; i++) {
+                    if (users.get(i).UID == uid) {
+                        users.get(i).UName = rs.getString(2);
+                        ret = true;
+                    }
+                }
+            }
+        } catch (SQLException e) {
+            System.err.println(e);
+        } finally {
+            Utils.finishSQL(rs, stmt);
+        }
+
+        return ret;
+    }
+
     public static ArrayList<com.juick.User> getUsersByJID(Connection sql, ArrayList<String> jids) {
         ArrayList<com.juick.User> users = new ArrayList<com.juick.User>();
 
@@ -180,6 +238,10 @@ public class UserQueries {
         return SQLHelpers.getInt(sql, "SELECT user_id FROM jids WHERE jid=?", jid, 0);
     }
 
+    public static int getUIDbyName(Connection sql, String uname) {
+        return SQLHelpers.getInt(sql, "SELECT id FROM users WHERE nick=?", uname, 0);
+    }
+
     public static int getUIDbyHash(Connection sql, String hash) {
         return SQLHelpers.getInt(sql, "SELECT user_id FROM logins WHERE hash=?", hash, 0);
     }