/*
* 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.User;
import org.springframework.dao.EmptyResultDataAccessException;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.RowMapper;
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.SQLException;
import java.sql.Statement;
import java.util.*;
import java.util.stream.Collectors;
/**
*
* @author Ugnich Anton
*/
public class UserQueries {
static final String ABCDEF = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ";
public static class UserMapper implements RowMapper {
@Override
public User mapRow(ResultSet rs, int rowNum) throws SQLException {
User user = new User();
user.setUID(rs.getInt(1));
user.setUName(rs.getString(2));
user.Banned = rs.getBoolean(3);
return user;
}
}
public static String getSignUpHashByJID(JdbcTemplate sql, String jid) {
String hash = sql.queryForObject("SELECT loginhash FROM jids WHERE jid=? AND user_id IS NULL",
String.class, jid);
if (hash == null) {
hash = UUID.randomUUID().toString();
sql.update("INSERT INTO jids(jid,loginhash) VALUES (?,?)", jid, hash);
}
return hash;
}
public static String getSignUpHashByTelegramID(JdbcTemplate sql, Long telegramId, String username) {
try {
return sql.queryForObject("SELECT loginhash FROM telegram WHERE tg_id=? AND user_id IS NULL",
String.class, telegramId);
} catch (EmptyResultDataAccessException e) {
String hash = UUID.randomUUID().toString();
sql.update("INSERT INTO telegram(tg_id, loginhash, tg_name) VALUES (?, ?, ?)", telegramId, hash, username);
return hash;
}
}
public static int createUser(JdbcTemplate sql, String username, String password) {
KeyHolder holder = new GeneratedKeyHolder();
sql.update(con -> {
PreparedStatement stmt = con.prepareStatement("INSERT INTO users(nick,passw) VALUES (?,?)",
Statement.RETURN_GENERATED_KEYS);
stmt.setString(1, username);
stmt.setString(2, password);
return stmt;
}, holder);
int uid = holder.getKey().intValue();
sql.update("INSERT INTO useroptions(user_id) VALUES (?)", uid);
sql.update("INSERT INTO subscr_users(user_id,suser_id) VALUES (2,?)", uid);
return uid;
}
public static Optional getUserByUID(JdbcTemplate sql, int uid) {
try {
return Optional.of(sql.queryForObject("SELECT id, nick,banned FROM users WHERE id=?",
new UserMapper(), uid));
} catch (EmptyResultDataAccessException e) {
return Optional.empty();
}
}
public static User getUserByName(JdbcTemplate sql, String username) {
try {
return sql.queryForObject("SELECT id,nick,banned FROM users WHERE nick=?",
new UserMapper(),
username);
} catch (EmptyResultDataAccessException e) {
return null;
}
}
public static User getUserByJID(JdbcTemplate sql, String jid) {
try {
return sql.queryForObject("SELECT id,nick,banned FROM users WHERE id=(SELECT user_id FROM jids WHERE jid=?)",
new UserMapper(), jid);
} catch (EmptyResultDataAccessException e) {
return null;
}
}
public static List getUsersByName(JdbcTemplate sql, List unames) {
if (!unames.isEmpty()) {
return sql.query("SELECT id,nick,banned FROM users WHERE nick IN (" + StringUtils.arrayToCommaDelimitedString(unames.toArray()) + ")",
new UserMapper());
}
return Collections.emptyList();
}
public static List getUsersByID(JdbcTemplate sql, List uids) {
if (!uids.isEmpty()) {
return sql.query("SELECT id,nick,banned FROM users WHERE id IN (" + StringUtils.arrayToCommaDelimitedString(uids.toArray()) + ")",
new UserMapper());
}
return Collections.emptyList();
}
public static boolean fillUsersByID(JdbcTemplate sql, List users) {
String uids = StringUtils.arrayToCommaDelimitedString(users.stream().map(u -> u.getUID()).collect(Collectors.toList()).toArray());
sql.query("SELECT id,nick,banned FROM users WHERE id IN (" + uids + ")",
(rs, num) -> {
User u = new User();
u.setUID(rs.getInt(1));
u.setUName(rs.getString(2));
return u;
});
return true;
}
public static List getUsersByJID(JdbcTemplate sql, List jids) {
if (!jids.isEmpty()) {
return sql.query("SELECT users.id,users.nick,jids.jid FROM users "
+ "INNER JOIN jids ON jids.user_id=users.id "
+ "WHERE jids.jid IN (" + StringUtils.arrayToCommaDelimitedString(jids.toArray()) + ")",
(rs, rowNum) -> {
com.juick.User user = new com.juick.User();
user.setUID(rs.getInt(1));
user.setUName(rs.getString(2));
user.setJID(rs.getString(3));
return user;
});
}
return Collections.emptyList();
}
public static List getJIDsbyUID(JdbcTemplate sql, int uid) {
return sql.queryForList("SELECT jid FROM jids WHERE user_id=? AND active=1", String.class, uid);
}
public static int getUIDbyJID(JdbcTemplate sql, String jid) {
try {
return sql.queryForObject("SELECT user_id FROM jids WHERE jid=?", Integer.class, jid);
} catch (EmptyResultDataAccessException e) {
return 0;
}
}
public static int getUIDbyName(JdbcTemplate sql, String uname) {
try {
return sql.queryForObject("SELECT id FROM users WHERE nick=?", Integer.class, uname);
} catch (EmptyResultDataAccessException e) {
return 0;
}
}
public static int getUIDbyHash(JdbcTemplate sql, String hash) {
try {
return sql.queryForObject("SELECT user_id FROM logins WHERE hash=?", Integer.class, hash);
} catch (EmptyResultDataAccessException e) {
return 0;
}
}
public static com.juick.User getUserByHash(JdbcTemplate sql, String hash) {
try {
User user = sql.queryForObject("SELECT logins.user_id,users.nick, users.banned FROM logins " +
"INNER JOIN users ON logins.user_id=users.id WHERE logins.hash=?",
new UserMapper(), hash);
user.setAuthHash(hash);
return user;
} catch (EmptyResultDataAccessException e) {
return new User();
}
}
public static String getHashByUID(JdbcTemplate sql, int uid) {
try {
return sql.queryForObject("SELECT hash FROM logins WHERE user_id=?", String.class, uid);
} catch (EmptyResultDataAccessException e){
String hash = generateHash(16);
sql.update(con -> {
PreparedStatement stmt = con.prepareStatement("INSERT INTO logins(user_id,hash) VALUES (?,?)");
stmt.setInt(1, uid);
stmt.setString(2, hash);
return stmt;
});
return hash;
}
}
public static String generateHash(int len) {
Random rnd = new Random();
StringBuilder sb = new StringBuilder(len);
for (int i = 0; i < len; i++) {
sb.append(ABCDEF.charAt(rnd.nextInt(ABCDEF.length())));
}
return sb.toString();
}
public static boolean checkUserNameValid(String uname) {
return uname != null && uname.length() >= 2 && uname.length() <= 16 && uname.matches("[a-zA-Z0-9\\-]+");
}
public static int checkPassword(JdbcTemplate sql, String username, String password) {
try {
String realPassword = sql.queryForObject("SELECT passw FROM users WHERE nick=?", String.class, username);
if (realPassword.equals(password)) {
User user = UserQueries.getUserByName(sql, username);
if (user != null) {
return user.getUID();
} else {
return -1;
}
} else {
return -1;
}
} catch (EmptyResultDataAccessException e) {
return -1;
}
}
public static int getUserOptionInt(JdbcTemplate sql, int uid, String option, int defaultValue) {
try {
return sql.queryForObject("SELECT " + option + " FROM useroptions WHERE user_id=?", Integer.class, uid);
} catch (EmptyResultDataAccessException e) {
return defaultValue;
}
}
public static void setUserOptionInt(JdbcTemplate sql, int uid, String option, int value) {
sql.update("UPDATE useroptions SET " + option + "=? WHERE user_id=?", value, uid);
}
public static boolean getCanMedia(JdbcTemplate sql, int uid) {
try {
int res = sql.queryForObject("SELECT users.lastphoto-UNIX_TIMESTAMP() FROM users WHERE id=?",
Integer.class, uid);
return res < 3600;
} catch (EmptyResultDataAccessException e) {
return false;
}
}
public static boolean isInWL(JdbcTemplate sql, int uid, int check) {
try {
return sql.queryForObject("SELECT 1 FROM wl_users WHERE user_id=? AND wl_user_id=?",
Integer.class, uid, check) == 1;
} catch (EmptyResultDataAccessException e) {
return false;
}
}
public static boolean isInBL(JdbcTemplate sql, int uid, int check) {
try {
return sql.queryForObject("SELECT 1 FROM bl_users WHERE user_id=? AND bl_user_id=?",
Integer.class, uid, check) == 1;
} catch (EmptyResultDataAccessException e) {
return false;
}
}
public static boolean isInBLAny(JdbcTemplate sql, int uid, int uid2) {
try {
return sql.queryForObject("SELECT 1 FROM bl_users "
+ "WHERE (user_id=? AND bl_user_id=?) "
+ "OR (user_id=? AND bl_user_id=?)", new Object[]{uid, uid2, uid2, uid}, Integer.class) == 1;
} catch (EmptyResultDataAccessException e) {
return false;
}
}
public static List checkBL(JdbcTemplate sql, int visitor, List uids) {
if (!uids.isEmpty()) {
return sql.queryForList("SELECT user_id FROM bl_users WHERE bl_user_id=? and user_id IN (" +
StringUtils.collectionToCommaDelimitedString(uids) + ")", Integer.class, visitor);
} else {
return new ArrayList<>();
}
}
public static boolean isSubscribed(JdbcTemplate sql, int uid, int check) {
try {
return sql.queryForObject("SELECT 1 FROM subscr_users WHERE suser_id=? AND user_id=?",
Integer.class, uid, check) == 1;
} catch (EmptyResultDataAccessException e) {
return false;
}
}
public static List getUserRead(JdbcTemplate sql, int uid) {
return sql.queryForList("SELECT user_id FROM subscr_users WHERE suser_id=?", Integer.class, uid);
}
public static List getUserReadLeastPopular(JdbcTemplate sql, int uid, int cnt) {
return sql.query("SELECT users.id,users.nick FROM (subscr_users " +
"INNER JOIN users_subscr ON (subscr_users.suser_id=? " +
"AND subscr_users.user_id=users_subscr.user_id)) INNER JOIN users " +
"ON subscr_users.user_id=users.id ORDER BY cnt LIMIT ?",
(rs, num) -> {
com.juick.User u = new com.juick.User();
u.setUID(rs.getInt(1));
u.setUName(rs.getString(2));
return u;
}, uid, cnt);
}
public static List getUserReaders(JdbcTemplate sql, int uid) {
return sql.query("SELECT users.id, users.nick FROM subscr_users " +
"INNER JOIN users ON subscr_users.suser_id=users.id " +
"WHERE subscr_users.user_id=? ORDER BY users.nick",
(rs, num) -> {
com.juick.User u = new com.juick.User();
u.setUID(rs.getInt(1));
u.setUName(rs.getString(2));
return u;
}, uid);
}
public static List getUserFriends(JdbcTemplate sql, int uid) {
return sql.query("SELECT users.id,users.nick FROM subscr_users " +
"INNER JOIN users ON subscr_users.user_id=users.id " +
"WHERE subscr_users.suser_id=? AND users.id!=? " +
"ORDER BY users.nick",
(rs, num) -> {
com.juick.User u = new com.juick.User();
u.setUID(rs.getInt(1));
u.setUName(rs.getString(2));
return u;
}, uid, uid);
}
public static List getUserBLUsers(JdbcTemplate sql, int uid) {
return sql.query("SELECT users.id,users.nick FROM users INNER JOIN bl_users " +
"ON(bl_users.bl_user_id=users.id) WHERE bl_users.user_id=? ORDER BY users.nick",
(rs, num) -> {
com.juick.User u = new com.juick.User();
u.setUID(rs.getInt(1));
u.setUName(rs.getString(2));
return u;
}, uid);
}
public static boolean linkTwitterAccount(JdbcTemplate sql, User user, String accessToken,
String accessTokenSecret, String screenName) {
if (sql.update("INSERT INTO twitter(user_id,access_token,access_token_secret,uname) " +
"VALUES (?,?,?,?)" +
" ON DUPLICATE KEY UPDATE access_token=?,access_token_secret=?,uname=?",
user.getUID(), accessToken, accessTokenSecret, screenName, accessToken, accessTokenSecret, screenName) > 0) {
return sql.update("INSERT INTO subscr_users(user_id,suser_id,jid) " +
"VALUES (?,1741,'juick\\@twitter.juick.com')", user.getUID()) > 0;
}
return false;
}
public static int getStatsIRead(JdbcTemplate sql, int uid) {
try {
return sql.queryForObject("SELECT COUNT(*) FROM subscr_users WHERE suser_id=?", Integer.class, uid);
} catch (EmptyResultDataAccessException e) {
return 0;
}
}
public static int getStatsMyReaders(JdbcTemplate sql, int uid) {
try {
return sql.queryForObject("SELECT COUNT(*) FROM subscr_users WHERE user_id=?", Integer.class, uid);
} catch (EmptyResultDataAccessException e) {
return 0;
}
}
public static int getStatsMessages(JdbcTemplate sql, int uid) {
try {
return sql.queryForObject("SELECT COUNT(*) FROM messages WHERE user_id=?", Integer.class, uid);
} catch (EmptyResultDataAccessException e) {
return 0;
}
}
public static int getStatsReplies(JdbcTemplate sql, int uid) {
try {
return sql.queryForObject("SELECT COUNT(*) FROM replies WHERE user_id=?", Integer.class, uid);
} catch (EmptyResultDataAccessException e) {
return 0;
}
}
public enum ActiveStatus {
Inactive,
Active
}
public static boolean setActiveStatusForJID(JdbcTemplate sql, String JID, ActiveStatus jidStatus) {
User user = getUserByJID(sql, JID);
if (user != null) {
return sql.update(con -> {
PreparedStatement preparedStatement = con.prepareStatement(
"UPDATE jids SET active=? WHERE user_id=? AND jid=?");
int newStatus = jidStatus == ActiveStatus.Active ? 1 : 0;
preparedStatement.setInt(1, newStatus);
preparedStatement.setInt(2, user.getUID());
preparedStatement.setString(3, JID);
return preparedStatement;
}) >= 0;
}
return false;
}
}