1 files changed, 25 insertions, 0 deletions
diff --git a/juick-core/src/main/java/com/juick/server/SubscriptionsQueries.java b/juick-core/src/main/java/com/juick/server/SubscriptionsQueries.java
index cedd2fd4..9a09a5cd 100644
--- a/juick-core/src/main/java/com/juick/server/SubscriptionsQueries.java
+++ b/juick-core/src/main/java/com/juick/server/SubscriptionsQueries.java
@@ -6,6 +6,8 @@ package com.juick.server;
 
 import com.juick.Tag;
 import com.juick.User;
+import com.juick.server.helpers.NotifyOpts;
+import org.springframework.dao.EmptyResultDataAccessException;
 import org.springframework.jdbc.core.JdbcTemplate;
 import org.springframework.util.StringUtils;
 
@@ -97,4 +99,27 @@ public class SubscriptionsQueries {
                 toTag.TID, user.getUID()) > 0;
     }
 
+    public static NotifyOpts getNotifyOptions(JdbcTemplate sql, User user) {
+        try {
+            return sql.queryForObject("SELECT jnotify,subscr_notify,recommendations FROM useroptions WHERE user_id=?",
+                    (rs, num) -> {
+                        NotifyOpts options = new NotifyOpts();
+                        options.setRepliesEnabled(rs.getInt(1) > 0);
+                        options.setSubscriptionsEnabled(rs.getInt(2) > 0);
+                        options.setRecommendationsEnabled(rs.getInt(3) > 0);
+                        return options;
+                    }, user.getUID());
+        } catch (EmptyResultDataAccessException e) {
+            return new NotifyOpts();
+        }
+    }
+
+    public static boolean setNotifyOptions(JdbcTemplate sql, User user, NotifyOpts options) {
+        return sql.update("UPDATE useroptions SET jnotify=? WHERE user_id=?", options.isRepliesEnabled() ? 1 : 0,
+                user.getUID()) > 0 &&
+                sql.update("UPDATE useroptions SET subscr_notify=? WHERE user_id=?", options.isSubscriptionsEnabled() ? 1 : 0,
+                        user.getUID()) > 0 &&
+                sql.update("UPDATE useroptions SET recommendations=? WHERE user_id=?", options.isRecommendationsEnabled() ? 1 : 0,
+                        user.getUID()) > 0;
+    }
 }