Premium users

1 files changed, 38 insertions, 19 deletions

diff --git a/src/main/java/com/juick/service/UserServiceImpl.java b/src/main/java/com/juick/service/UserServiceImpl.java
index 932be19c..a7ea5c5f 100644
--- a/src/main/java/com/juick/service/UserServiceImpl.java
+++ b/src/main/java/com/juick/service/UserServiceImpl.java
@@ -65,6 +65,7 @@ public class UserServiceImpl extends BaseJdbcService implements UserService {
                 user.setSeen(seen.toInstant());
             }
             user.setVerified(rs.getLong(6) > 0);
+            user.setPremium(rs.getInt(7) > 0);
             return user;
         }
     }
@@ -113,7 +114,7 @@ public class UserServiceImpl extends BaseJdbcService implements UserService {
     public Optional<User> getUserByUID(final int uid) {
         var list = getJdbcTemplate().query("""
                 SELECT DISTINCT u.id, u.nick, u.passw, u.banned, u.last_seen,
-                COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified
+                COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified, premium
                 FROM users u LEFT JOIN facebook f ON f.user_id = u.id
                 LEFT JOIN vk ON u.id = vk.user_id LEFT JOIN telegram t ON u.id = t.user_id
                 LEFT JOIN emails e ON e.user_id = u.id WHERE u.id = ?""", new UserMapper(), uid);
@@ -127,7 +128,7 @@ public class UserServiceImpl extends BaseJdbcService implements UserService {
         if (StringUtils.isNotBlank(username)) {
             var list = getJdbcTemplate().query("""
                     SELECT DISTINCT u.id, u.nick, u.passw, u.banned, u.last_seen,
-                    COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified
+                    COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified, premium
                     FROM users u LEFT JOIN facebook f ON f.user_id = u.id
                     LEFT JOIN vk ON u.id = vk.user_id LEFT JOIN telegram t ON u.id = t.user_id
                     LEFT JOIN emails e ON e.user_id = u.id
@@ -152,7 +153,7 @@ public class UserServiceImpl extends BaseJdbcService implements UserService {
             try {
                 List<User> list = getJdbcTemplate().query(
                         "SELECT DISTINCT u.id, u.nick, u.passw, u.banned, u.last_seen, " +
-                                "COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified " +
+                                "COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified, premium " +
                                 "FROM users u LEFT JOIN facebook f ON f.user_id = u.id " +
                                 "LEFT JOIN vk ON u.id = vk.user_id LEFT JOIN telegram t ON u.id = t.user_id " +
                                 "LEFT JOIN emails e ON e.user_id = u.id " +
@@ -178,7 +179,7 @@ public class UserServiceImpl extends BaseJdbcService implements UserService {
         if (StringUtils.isNotBlank(jid)) {
             List<User> list = getJdbcTemplate().query(
                     "SELECT DISTINCT u.id, u.nick, u.passw, u.banned, u.last_seen," +
-                            "COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified " +
+                            "COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified, premium " +
                             "FROM users u LEFT JOIN facebook f ON f.user_id = u.id " +
                             "LEFT JOIN vk ON u.id = vk.user_id LEFT JOIN telegram t ON u.id = t.user_id " +
                             "LEFT JOIN emails e ON e.user_id = u.id " +
@@ -200,7 +201,7 @@ public class UserServiceImpl extends BaseJdbcService implements UserService {
 
         return getNamedParameterJdbcTemplate().query(
                 "SELECT DISTINCT u.id, u.nick, u.passw, u.banned, u.last_seen," +
-                        "COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified " +
+                        "COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified, premium " +
                         "FROM users u LEFT JOIN facebook f ON f.user_id = u.id " +
                         "LEFT JOIN vk ON u.id = vk.user_id LEFT JOIN telegram t ON u.id = t.user_id " +
                         "LEFT JOIN emails e ON e.user_id = u.id " +
@@ -217,7 +218,7 @@ public class UserServiceImpl extends BaseJdbcService implements UserService {
 
         return getNamedParameterJdbcTemplate().query(
                 "SELECT DISTINCT u.id, u.nick, u.passw, u.banned, u.last_seen," +
-                        "COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified " +
+                        "COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified, premium " +
                         "FROM users u LEFT JOIN facebook f ON f.user_id = u.id " +
                         "LEFT JOIN vk ON u.id = vk.user_id LEFT JOIN telegram t ON u.id = t.user_id " +
                         "LEFT JOIN emails e ON e.user_id = u.id " +
@@ -252,7 +253,7 @@ public class UserServiceImpl extends BaseJdbcService implements UserService {
         if (StringUtils.isNotBlank(hash)) {
             List<User> list = getJdbcTemplate().query(
                     "SELECT DISTINCT logins.user_id, u.nick, u.passw, u.banned, u.last_seen," +
-                            "COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified " +
+                            "COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified, premium " +
                             "FROM logins INNER JOIN users u ON logins.user_id = u.id " +
                             "LEFT JOIN facebook f ON f.user_id = u.id " +
                             "LEFT JOIN vk ON u.id = vk.user_id LEFT JOIN telegram t ON u.id = t.user_id " +
@@ -290,7 +291,7 @@ public class UserServiceImpl extends BaseJdbcService implements UserService {
         if (StringUtils.isNotBlank(username)) {
             List<User> list = getJdbcTemplate().query(
                     "SELECT DISTINCT u.id, u.nick, u.passw, u.banned, u.last_seen," +
-                            "COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified " +
+                            "COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified, premium " +
                             "FROM users u LEFT JOIN facebook f ON f.user_id = u.id " +
                             "LEFT JOIN vk ON u.id = vk.user_id LEFT JOIN telegram t ON u.id = t.user_id " +
                             "LEFT JOIN emails e ON e.user_id = u.id " +
@@ -719,7 +720,7 @@ public class UserServiceImpl extends BaseJdbcService implements UserService {
         List<User> list = getJdbcTemplate().query(
                 """
                         SELECT DISTINCT u.id, u.nick, u.passw, u.banned, u.last_seen,
-                        COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified
+                        COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified, premium
                         FROM users u LEFT JOIN facebook f ON f.user_id = u.id
                         LEFT JOIN vk ON u.id = vk.user_id LEFT JOIN telegram t ON u.id = t.user_id
                         LEFT JOIN emails e ON e.user_id = u.id
@@ -732,14 +733,14 @@ public class UserServiceImpl extends BaseJdbcService implements UserService {
 
     @Transactional(readOnly = true)
     @Override
-    public Optional<Pair<String, String>> getVkTokens(final int uid) {
-        List<Optional<Pair<String, String>>> list = getJdbcTemplate().query(
-                "SELECT vk_id, access_token FROM vk WHERE user_id = ? AND crosspost = 1",
-                (rs, num) -> Optional.of(Pair.of(rs.getString(1), rs.getString(2))),
-                uid);
-
-        return list.isEmpty() ?
-                Optional.empty() : list.get(0);
+    public List<Pair<String, String>> getVkTokens(List<Integer> uids) {
+        return getNamedParameterJdbcTemplate().query(
+                """
+                        SELECT vk_id, access_token FROM vk WHERE crosspost = 1 AND access_token <> ''"""
+                        + (uids.isEmpty() ? "" : " AND user_id IN (:uids)"),
+                new MapSqlParameterSource()
+                        .addValue("uids", uids),
+                (rs, num) -> Pair.of(rs.getString(1), rs.getString(2)));
     }
 
     @Transactional
@@ -753,7 +754,7 @@ public class UserServiceImpl extends BaseJdbcService implements UserService {
     public Optional<User> getUserByFacebookId(long facebookId) {
         List<User> list = getJdbcTemplate().query(
                 "SELECT DISTINCT u.id, u.nick, u.passw, u.banned, u.last_seen, " +
-                        "COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified " +
+                        "COALESCE(f.fb_id, vk.vk_id, t.tg_id, e.user_id, 0) AS verified, premium " +
                         "FROM users u LEFT JOIN facebook f ON f.user_id = u.id " +
                         "LEFT JOIN vk ON u.id = vk.user_id LEFT JOIN telegram t ON u.id = t.user_id " +
                         "LEFT JOIN emails e ON e.user_id = u.id WHERE f.fb_id = ?", new UserMapper(), facebookId);
@@ -789,6 +790,18 @@ public class UserServiceImpl extends BaseJdbcService implements UserService {
         return getJdbcTemplate().update("INSERT INTO vk(vk_id,loginhash,access_token,vk_name,vk_link) VALUES (?,?,?,?,?)",
                 vkID, loginhash, token, vkName, vkLink) > 0;
     }
+    @Transactional
+    @Override
+    public boolean updateVkUser(long vkID, String token, String vkName, String vkLink) {
+        return getJdbcTemplate().update("UPDATE vk SET access_token=?,vk_name=?,vk_link=? WHERE vk_id=?",
+                token, vkName, vkLink, vkID) > 0;
+    }
+    @Transactional
+    @Override
+    public boolean updateVkToken(long userId, String token) {
+        return getJdbcTemplate().update("UPDATE vk SET access_token=? WHERE user_id=?",
+                token, userId) > 0;
+    }
 
     @Transactional(readOnly = true)
     @Override
@@ -864,7 +877,7 @@ public class UserServiceImpl extends BaseJdbcService implements UserService {
     @Transactional(readOnly = true)
     @Override
     public boolean canDeleteTelegramUser(User user) {
-        return getEmails(user).size() > 0 || getFbCrossPostStatus(user.getUid()).isConnected() || getVkTokens(user.getUid()).isPresent();
+        return getEmails(user).size() > 0 || getFbCrossPostStatus(user.getUid()).isConnected() || !getVkTokens(List.of(user.getUid())).isEmpty();
     }
     private static class TokenMapper implements RowMapper<ExternalToken> {
 
@@ -918,5 +931,11 @@ public class UserServiceImpl extends BaseJdbcService implements UserService {
     public boolean deleteToken(final String serviceType, final String token) {
         return getJdbcTemplate().update("DELETE FROM user_services WHERE regid=? AND service_type=?", token, serviceType) > 0;
     }
+
+    @Transactional
+    @Override
+    public void setPremium(final Integer uid, boolean isPremium) {
+        getJdbcTemplate().update("UPDATE users SET premium=? WHERE id=?", isPremium ? 1 : 0, uid);
+    }
 }