aboutsummaryrefslogtreecommitdiff
path: root/src/com/juick/server/PMQueries.java
diff options
context:
space:
mode:
Diffstat (limited to 'src/com/juick/server/PMQueries.java')
-rw-r--r--src/com/juick/server/PMQueries.java171
1 files changed, 171 insertions, 0 deletions
diff --git a/src/com/juick/server/PMQueries.java b/src/com/juick/server/PMQueries.java
new file mode 100644
index 00000000..42f3aefd
--- /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;
+ }
+}