/*
* Copyright (C) 2008-2020, Juick
*
* 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.service;
import com.juick.model.Message;
import com.juick.model.Tag;
import com.juick.model.User;
import com.juick.model.NotifyOpts;
import com.juick.util.MessageUtils;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.collections4.IteratorUtils;
import org.apache.commons.collections4.ListUtils;
import org.springframework.dao.DataIntegrityViolationException;
import org.springframework.dao.DuplicateKeyException;
import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;
import org.springframework.stereotype.Repository;
import org.springframework.transaction.annotation.Transactional;
import javax.annotation.Nonnull;
import javax.inject.Inject;
import java.util.Collections;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;
/**
* Created by aalexeev on 11/13/16.
*/
@Repository
public class SubscriptionServiceImpl extends BaseJdbcService implements SubscriptionService {
@Inject
private UserService userService;
@Inject
private MessagesService messagesService;
@Inject
private TagService tagService;
@Transactional(readOnly = true)
@Override
public List<User> getSubscribedUsers(final int uid, final Message msg) {
int mid = msg.getMid();
User author = messagesService.getMessageAuthor(mid);
List<User> subscribers = userService.getUserReaders(uid);
List<User> mentionedUsers = userService.getUsersByName(MessageUtils.getMentions(msg).stream()
.map(u -> u.substring(1))
.toList()).stream()
.filter(u -> !userService.isInBL(u.getUid(), msg.getUser().getUid()))
.toList();
List<User> users = ListUtils.union(subscribers, mentionedUsers);
List<Integer> tags = tagService.getMessageTagsIDs(mid);
List<String> tagsStr = tagService.getMessageTags(mid).stream().map(t -> t.getTag().getName())
.toList();
Set<Integer> set = users.stream()
.filter(u -> !u.isBanned())
.map(User::getUid)
.filter(u -> Collections.disjoint(tagService.getUserBLTags(u), tagsStr))
.collect(Collectors.toSet());
if (!tags.isEmpty()) {
List<Integer> tagUsers = getNamedParameterJdbcTemplate().queryForList(
"SELECT st.suser_id FROM subscr_tags st " +
"WHERE st.tag_id IN (:ids) AND st.suser_id != :uid " +
" AND NOT EXISTS (SELECT 1 FROM bl_users bu WHERE bu.bl_user_id = :authorUid and st.suser_id = bu.user_id)"
+
" AND NOT EXISTS (SELECT 1 FROM bl_tags bt WHERE bt.tag_id IN (:ids) and st.suser_id = bt.user_id)",
new MapSqlParameterSource()
.addValue("ids", tags)
.addValue("uid", uid)
.addValue("authorUid", author.getUid()),
Integer.class);
set.addAll(tagUsers);
}
return userService.getUsersByID(set);
}
@Override
public List<User> getUsersSubscribedToComments(@Nonnull final Message msg, @Nonnull final Message reply) {
return getUsersSubscribedToComments(msg, reply, false);
}
@Transactional(readOnly = true)
@Override
public List<User> getUsersSubscribedToComments(@Nonnull final Message msg, @Nonnull final Message reply,
boolean blacklisted) {
List<User> subscribers = userService.getUsersByID(getJdbcTemplate().queryForList(
"SELECT suser_id FROM subscr_messages WHERE message_id=? AND suser_id!=?",
Integer.class,
msg.getMid(), reply.getUser().getUid()));
List<User> mentionedUsers = userService.getUsersByName(MessageUtils.getMentions(reply).stream()
.map(u -> u.substring(1)).toList());
List<User> users = IteratorUtils.toList(CollectionUtils.union(subscribers, mentionedUsers).iterator());
if (!users.isEmpty()) {
return users.stream()
.filter(u -> blacklisted || !u.isBanned() && !userService.isReplyToBL(u, reply))
.toList();
}
return Collections.emptyList();
}
@Override
public List<User> getUsersSubscribedToUserRecommendations(final int uid, final Message msg) {
List<String> msgTags = tagService.getMessageTags(msg.getMid()).stream().map(t -> t.getTag().getName())
.toList();
if (msg.getLikes() == 1) {
return userService.getUserReaders(uid).stream()
.filter(u -> !u.equals(msg.getUser()))
.filter(u -> !userService.isInBLAny(u.getUid(), msg.getUser().getUid()))
.filter(u -> Collections.disjoint(tagService.getUserBLTags(u.getUid()),
msgTags))
.toList();
}
return Collections.emptyList();
}
@Transactional
@Override
public boolean subscribeMessage(final Message message, final User user) {
try {
boolean result = getJdbcTemplate().update(
"INSERT INTO subscr_messages(suser_id, message_id) VALUES (?, ?)",
user.getUid(), message.getMid()) == 1;
messagesService.setLastReadComment(user, message.getMid(), message.getReplies());
return result;
} catch (DataIntegrityViolationException e) {
return false;
}
}
@Transactional
@Override
public boolean unSubscribeMessage(final int mid, final int vuid) {
return getJdbcTemplate().update(
"DELETE FROM subscr_messages WHERE message_id=? AND suser_id=?", mid, vuid) > 0;
}
@Transactional
@Override
public boolean subscribeUser(final User user, final User toUser) {
try {
return getJdbcTemplate().update(
"INSERT INTO subscr_users(user_id,suser_id) VALUES (?,?)", toUser.getUid(),
user.getUid()) == 1;
} catch (DuplicateKeyException e) {
return true;
}
}
@Transactional
@Override
public boolean unSubscribeUser(final User user, final User fromUser) {
return getJdbcTemplate().update(
"DELETE FROM subscr_users WHERE suser_id=? AND user_id=?", user.getUid(),
fromUser.getUid()) > 0;
}
@Transactional
@Override
public boolean subscribeTag(final User user, final Tag toTag) {
try {
return getJdbcTemplate().update(
"INSERT INTO subscr_tags(tag_id,suser_id) VALUES (?,?)", toTag.TID,
user.getUid()) == 1;
} catch (DuplicateKeyException e) {
return true;
}
}
@Transactional
@Override
public boolean unSubscribeTag(final User user, final Tag toTag) {
return getJdbcTemplate().update(
"DELETE FROM subscr_tags WHERE tag_id=? AND suser_id=?", toTag.TID, user.getUid()) > 0;
}
@Transactional(readOnly = true)
@Override
public List<String> getSubscribedTags(User user) {
return getJdbcTemplate().queryForList("SELECT tags.name FROM subscr_tags INNER JOIN tags " +
"ON(tags.tag_id = subscr_tags.tag_id) " +
"WHERE subscr_tags.suser_id=? ORDER BY tags.name", String.class, user.getUid());
}
@Transactional(readOnly = true)
@Override
public NotifyOpts getNotifyOptions(final User user) {
List<NotifyOpts> list = getJdbcTemplate().query(
"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());
return list.isEmpty() ? new NotifyOpts() : list.get(0);
}
@Transactional
@Override
public boolean setNotifyOptions(final User user, final NotifyOpts options) {
int jnotify = getJdbcTemplate().update(
"UPDATE useroptions SET jnotify=? WHERE user_id=?",
options.isRepliesEnabled() ? 1 : 0,
user.getUid());
int subscr_notify = getJdbcTemplate().update(
"UPDATE useroptions SET subscr_notify=? WHERE user_id=?",
options.isSubscriptionsEnabled() ? 1 : 0,
user.getUid());
int recommendations = getJdbcTemplate().update(
"UPDATE useroptions SET recommendations=? WHERE user_id=?",
options.isRecommendationsEnabled() ? 1 : 0,
user.getUid());
return jnotify > 0 && subscr_notify > 0 && recommendations > 0;
}
}