/* * 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 . */ package com.juick.server; import com.juick.Message; import com.juick.User; import com.juick.server.helpers.PrivacyOpts; import org.springframework.dao.EmptyResultDataAccessException; import org.springframework.dao.IncorrectResultSizeDataAccessException; import org.springframework.jdbc.core.ConnectionCallback; import org.springframework.jdbc.core.JdbcTemplate; import org.springframework.jdbc.support.GeneratedKeyHolder; import org.springframework.jdbc.support.KeyHolder; import org.springframework.util.StringUtils; import java.sql.PreparedStatement; import java.sql.ResultSet; import java.sql.Statement; import java.sql.Types; import java.util.ArrayList; import java.util.Collections; import java.util.List; import java.util.logging.Logger; /** * * @author Ugnich Anton */ public class MessagesQueries { private static final Logger LOGGER = Logger.getLogger(MessagesQueries.class.getName()); public static int createMessage(JdbcTemplate sql, int uid, String txt, String attachment, List tags) { KeyHolder holder = new GeneratedKeyHolder(); sql.update(con -> { PreparedStatement stmt = con.prepareStatement("INSERT INTO messages(user_id,attach) VALUES (?,?)", Statement.RETURN_GENERATED_KEYS); stmt.setInt(1, uid); if (attachment != null) { stmt.setString(2, attachment); } else { stmt.setNull(2, Types.VARCHAR); } return stmt; }, holder); int mid = holder.getKey().intValue(); if (mid > 0) { String tagsNames = ""; String tagsIDs = ""; for (int i = 0; i < tags.size(); i++) { if (i > 0) { tagsNames += " "; tagsIDs += ","; } tagsNames += tags.get(i).Name; tagsIDs += "(" + mid + "," + tags.get(i).TID + ")"; } if (tags.size() > 0) { sql.execute("INSERT INTO messages_tags(message_id,tag_id) VALUES " + tagsIDs); } final String finalTagsNames = tagsNames; sql.update(con -> { PreparedStatement stmt = con.prepareStatement("INSERT INTO messages_txt(message_id,tags,txt) " + "VALUES (?,?,?)", Statement.NO_GENERATED_KEYS); stmt.setInt(1, mid); if (finalTagsNames.isEmpty()) { stmt.setNull(2, Types.VARCHAR); } else { stmt.setString(2, finalTagsNames); } stmt.setString(3, txt); return stmt; }); } return mid; } public static int createReply(JdbcTemplate sql, int mid, int rid, int uid, String txt, String attachment) { int ridnew = getReplyIDIncrement(sql, mid); sql.update( con -> { PreparedStatement stmt = con.prepareStatement("INSERT INTO replies(message_id,reply_id,user_id," + "replyto,attach,txt) VALUES (?,?,?,?,?,?)", Statement.NO_GENERATED_KEYS); stmt.setInt(1, mid); stmt.setInt(2, ridnew); stmt.setInt(3, uid); stmt.setInt(4, rid); if (attachment != null) { stmt.setString(5, attachment); } else { stmt.setNull(5, Types.VARCHAR); } stmt.setString(6, txt); return stmt; }); if (ridnew > 0) { sql.update("UPDATE messages SET replies=replies+1 WHERE message_id=?", mid); } return ridnew; } public static int getReplyIDIncrement(JdbcTemplate sql, int mid) { return sql.execute((ConnectionCallback) conn -> { conn.setAutoCommit(false); try (PreparedStatement ps = conn.prepareStatement("UPDATE messages SET maxreplyid=maxreplyid+1 WHERE message_id=?")) { ps.setInt(1, mid); if (ps.executeUpdate() != 1) { throw new IncorrectResultSizeDataAccessException("Cannot find a message to update: " + mid, 1, 0); } } try (PreparedStatement ps = conn.prepareStatement("SELECT maxreplyid FROM messages WHERE message_id=?")) { ps.setInt(1, mid); try (ResultSet resultSet = ps.executeQuery()) { if (resultSet.next()) { int rv = resultSet.getInt(1); conn.commit(); return rv; } else { throw new IncorrectResultSizeDataAccessException("while getting getReplyIDIncrement, mid=" + mid, 1, 0); } } } }); } public static boolean recommendMessage(JdbcTemplate sql, int mid, int vuid) { boolean res = sql.update("INSERT IGNORE INTO favorites(user_id,message_id) VALUES (" + vuid + "," + mid + ")") == 1; if (res) { sql.update("UPDATE messages SET likes=likes+1 WHERE message_id=?", mid); } return res; } public static boolean canViewThread(JdbcTemplate sql, int mid, int uid) { PrivacyOpts privacyOpts; try { privacyOpts = sql.queryForObject("SELECT user_id,privacy FROM messages WHERE messages.message_id=?", (rs, rowNum) -> { PrivacyOpts res = new PrivacyOpts(); res.setUid(rs.getInt(1)); res.setPrivacy(rs.getInt(2)); return res; }, mid); } catch (EmptyResultDataAccessException e) { return true; } return privacyOpts.getPrivacy() >= 0 || uid == privacyOpts.getUid() || ((privacyOpts.getPrivacy() == -1 || privacyOpts.getPrivacy() == -2) && uid > 0 && UserQueries.isInWL(sql, privacyOpts.getUid(), uid)); } public static boolean isReadOnly(JdbcTemplate sql, int mid) { try { return sql.queryForObject("SELECT readonly FROM messages WHERE message_id=?", new Object[]{mid}, Integer.class) == 1; } catch (EmptyResultDataAccessException e) { return false; } } public static boolean isSubscribed(JdbcTemplate sql, int uid, int mid) { try { return sql.queryForObject("SELECT 1 FROM subscr_messages WHERE suser_id=? AND message_id=?", new Object[]{uid, mid}, Integer.class) == 1; } catch (EmptyResultDataAccessException e) { return false; } } public static int getMessagePrivacy(JdbcTemplate sql, int mid) { try { return sql.queryForObject("SELECT privacy FROM messages WHERE message_id=?", new Object[]{mid}, Integer.class); } catch (EmptyResultDataAccessException e) { return -4; } } public static com.juick.Message getMessage(JdbcTemplate sql, int mid) { try { return sql.queryForObject("SELECT messages.user_id,users.nick," + "TIMESTAMPDIFF(MINUTE,messages.ts,NOW())," + "messages.ts," + "messages.readonly,messages.privacy,messages.replies," + "messages.attach,messages.place_id,messages.lat," + "messages.lon,messages.likes,messages.hidden," + "txt.tags,txt.repliesby,txt.txt FROM messages " + "INNER JOIN users ON messages.user_id=users.id " + "INNER JOIN messages_txt AS txt " + "ON messages.message_id=txt.message_id " + "WHERE messages.message_id=?", new Object[]{mid}, (rs, rowNum) -> { Message msg = new Message(); msg.setMID(mid); msg.setUser(new User()); msg.getUser().setUID(rs.getInt(1)); msg.getUser().setUName(rs.getString(2)); msg.TimeAgo = rs.getInt(3); msg.setDate(rs.getTimestamp(4)); msg.ReadOnly = rs.getBoolean(5); msg.Privacy = rs.getInt(6); msg.FriendsOnly = msg.Privacy < 0; msg.Replies = rs.getInt(7); msg.AttachmentType = rs.getString(8); //if (rs.getInt(9) > 0) { // msg.Place = PlacesQueries.getPlace(sql, rs.getInt(9)); //} if (rs.getDouble(10) != 0) { msg.Place = new com.juick.Place(); msg.Place.lat = rs.getDouble(10); msg.Place.lon = rs.getDouble(11); } msg.Likes = rs.getInt(12); msg.Hidden = rs.getBoolean(13); msg.parseTags(rs.getString(14)); msg.RepliesBy = rs.getString(15); msg.setText(rs.getString(16)); return msg; }); } catch (EmptyResultDataAccessException e) { return null; } } public static com.juick.Message getReply(JdbcTemplate sql, int mid, int rid) { try { return sql.queryForObject("SELECT replies.user_id,users.nick," + "replies.replyto,replies.ts," + "replies.attach,replies.txt FROM replies INNER JOIN users " + "ON replies.user_id=users.id " + "WHERE replies.message_id=? AND replies.reply_id=?", new Object[]{mid, rid}, (rs, num) -> { Message msg = new Message(); msg.setMID(mid); msg.setRID(rid); msg.setUser(new User()); msg.getUser().setUID(rs.getInt(1)); msg.getUser().setUName(rs.getString(2)); msg.ReplyTo = rs.getInt(3); msg.setDate(rs.getTimestamp(4)); msg.AttachmentType = rs.getString(5); msg.setText(rs.getString(6)); return msg; }); } catch (EmptyResultDataAccessException e) { return null; } } public static User getMessageAuthor(JdbcTemplate sql, int mid) { try { return sql.queryForObject("SELECT messages.user_id,users.nick " + "FROM messages INNER JOIN users " + "ON messages.user_id=users.id WHERE messages.message_id=?", new Object[]{mid}, (rs, num) -> { User res = new com.juick.User(); res.setUID(rs.getInt(1)); res.setUName(rs.getString(2)); return res; }); } catch (EmptyResultDataAccessException e) { return null; } } public static List getMessageTags(JdbcTemplate sql, int mid) { return sql.query("SELECT tags.tag_id,synonym_id,name,stat_messages FROM tags " + "INNER JOIN messages_tags ON (messages_tags.message_id=? AND messages_tags.tag_id=tags.tag_id)", new Object[]{mid}, (rs, num) -> { com.juick.Tag t = new com.juick.Tag(); t.TID = rs.getInt(1); t.SynonymID = rs.getInt(2); t.Name = rs.getString(3); t.UsageCnt = rs.getInt(4); return t; }); } public static List getMessageTagsIDs(JdbcTemplate sql, int mid) { return sql.queryForList("SELECT tag_id FROM messages_tags WHERE message_id=?", new Object[] {mid}, Integer.class); } public static List getMessageRecommendations(JdbcTemplate sql, int mid) { return sql.queryForList("SELECT users.nick FROM favorites INNER JOIN users " + "ON (favorites.message_id=? AND favorites.user_id=users.id)", new Object[] {mid}, String.class); } public static List getAll(JdbcTemplate sql, int visitor_uid, int before) { if (visitor_uid > 1) { if (before > 0) { return sql.queryForList("SELECT message_id FROM messages WHERE message_id0 OR user_id=?) AND user_id NOT IN (SELECT bl_user_id FROM bl_users WHERE user_id=?)" + " AND user_id NOT IN (SELECT id from users WHERE banned=1) ORDER BY message_id DESC LIMIT 20", new Object[]{before, visitor_uid, visitor_uid}, Integer.class); } else { return sql.queryForList("SELECT message_id FROM messages WHERE hidden=0" + " AND (privacy>0 OR user_id=?)" + " AND user_id NOT IN (SELECT bl_user_id FROM bl_users WHERE user_id=?)" + " AND user_id NOT IN (SELECT id from users WHERE banned=1) ORDER BY message_id DESC LIMIT 20", new Object[]{visitor_uid, visitor_uid}, Integer.class); } } else { if (before > 0) { return sql.queryForList("SELECT message_id FROM messages WHERE message_id0 AND user_id NOT IN (SELECT id from users WHERE banned=1) " + " ORDER BY message_id DESC LIMIT 20", new Object[]{before}, Integer.class); } else { return sql.queryForList("SELECT message_id FROM messages WHERE hidden=0 AND privacy>0" + " AND user_id NOT IN (SELECT id from users WHERE banned=1) ORDER BY message_id DESC LIMIT 20", Integer.class); } } } public static List getTag(JdbcTemplate sql, int tid, int visitor_uid, int before, int cnt) { if (before > 0) { return sql.queryForList("SELECT message_id FROM (tags INNER JOIN messages_tags " + "ON ((tags.synonym_id=? OR tags.tag_id=?) AND tags.tag_id=messages_tags.tag_id)) " + "INNER JOIN messages USING(message_id) WHERE messages.message_id0 OR messages.user_id=?) ORDER BY message_id DESC LIMIT ?", new Object[]{tid, tid, before, visitor_uid, cnt}, Integer.class); } else { return sql.queryForList("SELECT message_id FROM (tags INNER JOIN messages_tags " + "ON ((tags.synonym_id=? OR tags.tag_id=?) AND tags.tag_id=messages_tags.tag_id)) " + "INNER JOIN messages USING(message_id) WHERE messages.privacy>0 OR messages.user_id=? " + "ORDER BY message_id DESC LIMIT ?", new Object[]{tid, tid, visitor_uid, cnt}, Integer.class); } } public static List getTags(JdbcTemplate sql, String tids, int visitor_uid, int before, int cnt) { if (before > 0) { return sql.queryForList("SELECT messages.message_id FROM messages_tags " + "INNER JOIN messages USING(message_id) WHERE messages_tags.tag_id IN (" + tids + ") " + "AND messages.message_id0 OR messages.user_id=?) " + "ORDER BY messages.message_id DESC LIMIT ?", new Object[]{before, visitor_uid, cnt}, Integer.class); } else { return sql.queryForList("SELECT messages.message_id FROM messages_tags " + "INNER JOIN messages USING(message_id) WHERE messages_tags.tag_id IN (" + tids + ") " + "AND (messages.privacy>0 OR messages.user_id=?) " + "ORDER BY messages.message_id DESC LIMIT ?", new Object[]{visitor_uid, cnt}, Integer.class); } } public static List getPlace(JdbcTemplate sql, int place_id, int visitor_uid, int before) { if (before > 0) { return sql.queryForList("SELECT message_id FROM messages WHERE place_id=? AND message_id0 OR user_id=?) ORDER BY message_id DESC LIMIT 20", new Object[]{place_id, before, visitor_uid}, Integer.class); } else { return sql.queryForList("SELECT message_id FROM messages WHERE place_id=? AND (privacy>0 OR user_id=?) " + "ORDER BY message_id DESC LIMIT 20", new Object[]{place_id, visitor_uid}, Integer.class); } } public static List getMyFeed(JdbcTemplate sql, int uid, int before) { List mids; if (before > 0) { mids = sql.queryForList("SELECT message_id FROM messages " + "INNER JOIN subscr_users ON (subscr_users.suser_id=? AND subscr_users.user_id=messages.user_id) " + "WHERE message_id=0 OR (privacy>=-2 AND privacy<=-1 AND messages.user_id " + "IN (SELECT user_id FROM wl_users WHERE wl_user_id=?))) ORDER BY message_id DESC LIMIT 20", Integer.class, uid, before, uid); } else { mids = sql.queryForList("SELECT message_id FROM messages " + "INNER JOIN subscr_users ON (subscr_users.suser_id=? " + "AND subscr_users.user_id=messages.user_id) " + "WHERE (privacy>=0 OR (privacy>=-2 AND privacy<=-1 AND messages.user_id " + "IN (SELECT user_id FROM wl_users WHERE wl_user_id=?))) ORDER BY message_id DESC LIMIT 20", Integer.class, uid, uid); } if (before > 0) { mids.addAll(sql.queryForList("SELECT message_id FROM messages " + "WHERE user_id=? AND message_id getPrivate(JdbcTemplate sql, int uid, int before) { if (before > 0) { return sql.queryForList("SELECT message_id FROM messages WHERE user_id=? AND privacy<0 AND message_id getDiscussions(JdbcTemplate sql, int uid, int before) { if (before > 0) { return sql.queryForList("SELECT message_id FROM subscr_messages WHERE suser_id=? AND message_id getRecommended(JdbcTemplate sql, int uid, int before) { if (before > 0) { return sql.queryForList("SELECT message_id FROM favorites WHERE user_id " + "IN (SELECT user_id FROM subscr_users WHERE suser_id=?) AND message_id getPopular(JdbcTemplate sql, int visitor_uid, int before) { if (before > 0) { return sql.queryForList("SELECT message_id FROM messages WHERE message_id0 " + "AND popular>0 AND user_id NOT IN (SELECT bl_user_id FROM bl_users WHERE user_id=?) " + "ORDER BY message_id DESC LIMIT 20", Integer.class, before, visitor_uid); } else { return sql.queryForList("SELECT message_id FROM messages WHERE privacy>0 " + "AND popular>0 AND user_id NOT IN (SELECT bl_user_id FROM bl_users WHERE user_id=?) " + "ORDER BY message_id DESC LIMIT 20", Integer.class, visitor_uid); } } public static List getPhotos(JdbcTemplate sql, int visitor_uid, int before) { if (before > 0) { return sql.queryForList("SELECT message_id FROM messages WHERE message_id0 OR user_id=?) " + "AND attach IS NOT NULL AND user_id NOT IN (SELECT id from users WHERE banned=1) " + "AND user_id NOT IN (SELECT bl_user_id FROM bl_users WHERE user_id=?) " + "ORDER BY message_id DESC LIMIT 20", Integer.class, before, visitor_uid, visitor_uid); } else { return sql.queryForList("SELECT message_id FROM messages WHERE (privacy>0 OR user_id=?) " + "AND attach IS NOT NULL AND user_id NOT IN (SELECT id from users WHERE banned=1) " + "AND user_id NOT IN (SELECT bl_user_id FROM bl_users WHERE user_id=?) " + "ORDER BY message_id DESC LIMIT 20", Integer.class, visitor_uid, visitor_uid); } } public static List getSearch(JdbcTemplate sql, JdbcTemplate sqlSearch, String search, int before) { List mids; if (before > 0) { mids = sqlSearch.queryForList("SELECT id AS message_id FROM messages WHERE MATCH(?) AND id 0) { return sql.queryForList("SELECT message_id FROM messages WHERE message_id " + "IN (" + StringUtils.arrayToCommaDelimitedString(mids.toArray()) + ") AND privacy>0 ORDER BY message_id DESC LIMIT 20", Integer.class); } return mids; } public static List getUserBlog(JdbcTemplate sql, int UID, int privacy, int before) { if (before > 0) { return sql.queryForList("SELECT message_id FROM messages WHERE user_id=? AND message_id=" + privacy + " ORDER BY message_id DESC LIMIT 20", new Object[]{UID, before}, Integer.class); } else { return sql.queryForList("SELECT message_id FROM messages WHERE user_id=? AND privacy>=" + privacy + " ORDER BY message_id DESC LIMIT 20", new Object[]{UID}, Integer.class); } } public static List getUserTag(JdbcTemplate sql, int UID, int TID, int privacy, int before) { if (before > 0) { return sql.queryForList("SELECT messages.message_id FROM messages_tags INNER JOIN messages " + "USING(message_id) WHERE messages.user_id=? AND messages_tags.tag_id=? " + "AND messages.message_id=" + privacy + " ORDER BY messages.message_id DESC LIMIT 20", new Object[] {UID, TID, before}, Integer.class); } else { return sql.queryForList("SELECT messages.message_id FROM messages_tags INNER JOIN messages " + "USING(message_id) WHERE messages.user_id=? AND messages_tags.tag_id=? " + "AND messages.privacy>=" + privacy + " ORDER BY messages.message_id DESC LIMIT 20", new Object[]{UID, TID}, Integer.class); } } public static List getUserRecommendations(JdbcTemplate sql, int UID, int before) { if (before > 0) { return sql.queryForList("SELECT message_id FROM favorites " + "WHERE user_id=? AND message_id getUserPhotos(JdbcTemplate sql, int UID, int privacy, int before) { if (before > 0) { return sql.queryForList("SELECT message_id FROM messages WHERE user_id=? AND message_id=" + privacy + " AND attach IS NOT NULL ORDER BY message_id DESC LIMIT 20", new Object[] {UID, before}, Integer.class); } else { return sql.queryForList("SELECT message_id FROM messages WHERE user_id=? AND privacy>=" + privacy + " AND attach IS NOT NULL ORDER BY message_id DESC LIMIT 20", new Object[] {UID}, Integer.class); } } public static List getUserSearch(JdbcTemplate sql, JdbcTemplate sqlSearch, int UID, String search, int privacy, int before) { List mids; if (before > 0) { mids = sqlSearch.queryForList("SELECT id AS message_id FROM messages WHERE user_id=? AND MATCH(?) AND id 0) { return sql.queryForList("SELECT message_id FROM messages WHERE message_id IN (" + StringUtils.arrayToCommaDelimitedString(mids.toArray()) + ") AND privacy>=" + privacy + " ORDER BY message_id DESC", Integer.class); } return mids; } public static List getMessages(JdbcTemplate sql, List mids) { if (!mids.isEmpty()) { return sql.query("SELECT messages.message_id,messages.user_id,users.nick," + "messages_txt.tags,messages.readonly,messages.privacy,messages_txt.txt," + "TIMESTAMPDIFF(MINUTE,messages.ts,NOW())," + "messages.ts,messages.replies," + "messages_txt.repliesby,messages.attach,messages.lat," + "messages.lon,messages.likes " + "FROM (messages INNER JOIN messages_txt " + "ON messages.message_id=messages_txt.message_id) " + "INNER JOIN users ON messages.user_id=users.id " + "WHERE messages.message_id " + "IN (" + StringUtils.arrayToCommaDelimitedString(mids.toArray()) + ") " + "ORDER BY messages.message_id DESC", (rs, rowNum) -> { com.juick.Message msg = new com.juick.Message(); msg.setUser(new User()); msg.setMID(rs.getInt(1)); msg.getUser().setUID(rs.getInt(2)); msg.getUser().setUName(rs.getString(3)); if (rs.getString(4) != null) { msg.parseTags(rs.getString(4)); } msg.ReadOnly = rs.getInt(5) == 1; msg.Privacy = rs.getInt(6); msg.FriendsOnly = msg.Privacy < 0; msg.setText(rs.getString(7)); msg.TimeAgo = rs.getInt(8); msg.setDate(rs.getTimestamp(9)); msg.Replies = rs.getInt(10); msg.RepliesBy = rs.getString(11); msg.AttachmentType = rs.getString(12); if (rs.getDouble(13) != 0) { msg.Place = new com.juick.Place(); msg.Place.lat = rs.getDouble(13); msg.Place.lon = rs.getDouble(14); } msg.Likes = rs.getInt(15); return msg; }); } return Collections.emptyList(); } public static List getReplies(JdbcTemplate sql, int mid) { return sql.query("SELECT replies.reply_id,replies.replyto,replies.user_id,users.nick,users.banned,replies.txt," + "TIMESTAMPDIFF(MINUTE,replies.ts,NOW()),replies.ts,replies.attach FROM replies INNER JOIN users " + "ON replies.user_id=users.id WHERE replies.message_id=? ORDER BY replies.reply_id ASC", new Object[]{mid}, (rs, num) -> { com.juick.Message msg = new com.juick.Message(); msg.setMID(mid); msg.setRID(rs.getInt(1)); msg.ReplyTo = rs.getInt(2); msg.setUser(new User()); msg.getUser().setUID(rs.getInt(3)); msg.getUser().setUName(rs.getString(4)); msg.getUser().Banned = rs.getBoolean(5); msg.setText(rs.getString(6)); msg.TimeAgo = rs.getInt(7); msg.setDate(rs.getTimestamp(8)); msg.AttachmentType = rs.getString(9); return msg; }); } public static boolean setMessagePopular(JdbcTemplate sql, int mid, int popular) { boolean ret; switch (popular) { case -2: ret = sql.update("UPDATE messages SET hidden=1 WHERE message_id=?", mid) > 0; break; case -1: ret = sql.update("UPDATE messages SET popular=0 WHERE message_id=?", mid) > 0; break; default: ret = sql.update("UPDATE messages SET popular=? WHERE message_id=?", popular, mid) > 0; break; } if (popular == -1) { ret = sql.update("INSERT INTO top_ignore_messages VALUES (?)", mid) > 0; } return ret; } public static boolean setMessagePrivacy(JdbcTemplate sql, int mid) { return sql.update("UPDATE messages SET privacy=1 WHERE message_id=?", mid) > 0; } public static boolean deleteMessage(JdbcTemplate sql, int uid, int mid) { if (sql.update("DELETE FROM messages WHERE message_id=? AND user_id=?", mid, uid) > 0) { sql.update("DELETE FROM messages_txt WHERE message_id=?", mid); sql.update("DELETE FROM replies WHERE message_id=?", mid); sql.update("DELETE FROM subscr_messages WHERE message_id=?", mid); sql.update("DELETE FROM messages_tags WHERE message_id=?", mid); return true; } return false; } }