From 62d4f6f3fabe69b0e08195e99eedd72442bffa68 Mon Sep 17 00:00:00 2001
From: Vitaly Takmazov
Date: Thu, 13 Sep 2018 15:45:52 +0300
Subject: hide banned from search

---
 .../src/main/java/com/juick/service/MessagesServiceImpl.java  |  8 ++++----
 .../src/main/java/com/juick/service/SphinxSearchService.java  | 11 +++++++++--
 2 files changed, 13 insertions(+), 6 deletions(-)

(limited to 'juick-server/src/main/java/com/juick/service')

diff --git a/juick-server/src/main/java/com/juick/service/MessagesServiceImpl.java b/juick-server/src/main/java/com/juick/service/MessagesServiceImpl.java
index 5dba41be..fbf22d59 100644
--- a/juick-server/src/main/java/com/juick/service/MessagesServiceImpl.java
+++ b/juick-server/src/main/java/com/juick/service/MessagesServiceImpl.java
@@ -652,8 +652,8 @@ public class MessagesServiceImpl extends BaseJdbcService implements MessagesServ
 
     @Transactional(readOnly = true)
     @Override
-    public List<Integer> getSearch(final String search, final int page) {
-        return searchService.searchInAllMessages(search, page);
+    public List<Integer> getSearch(final User visitor, final String search, final int page) {
+        return searchService.searchInAllMessages(visitor, search, page);
     }
 
     @Transactional(readOnly = true)
@@ -789,8 +789,8 @@ public class MessagesServiceImpl extends BaseJdbcService implements MessagesServ
 
     @Transactional(readOnly = true)
     @Override
-    public List<Integer> getUserSearch(final int UID, final String search, final int privacy, final int page) {
-        return searchService.searchByStringAndUser(search, UID, page);
+    public List<Integer> getUserSearch(final User visitor, final int UID, final String search, final int privacy, final int page) {
+        return searchService.searchByStringAndUser(visitor, search, UID, page);
     }
 
     @Transactional(readOnly = true)
diff --git a/juick-server/src/main/java/com/juick/service/SphinxSearchService.java b/juick-server/src/main/java/com/juick/service/SphinxSearchService.java
index 40927bbb..cdd5e92d 100644
--- a/juick-server/src/main/java/com/juick/service/SphinxSearchService.java
+++ b/juick-server/src/main/java/com/juick/service/SphinxSearchService.java
@@ -17,10 +17,12 @@
 
 package com.juick.service;
 
+import com.juick.User;
 import org.apache.commons.lang3.StringUtils;
 import org.springframework.stereotype.Repository;
 import org.springframework.transaction.annotation.Transactional;
 
+import javax.inject.Inject;
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.List;
@@ -38,13 +40,16 @@ public class SphinxSearchService extends BaseJdbcService implements SearchServic
 
     private int maxResult = DEFAULT_MAX_RESULT;
 
+    @Inject
+    UserService userService;
+
     public String sortHint(String searchString) {
         boolean isOneWord = searchString.split("[^\\S\\+]+").length == 1;
         return isOneWord ? "extended:@id desc" : "extended:@weight desc, @id desc";
     }
 
     @Override
-    public List<Integer> searchInAllMessages(final String searchString, final int page) {
+    public List<Integer> searchInAllMessages(User visitor, final String searchString, final int page) {
         if (StringUtils.isBlank(searchString))
             return Collections.emptyList();
 
@@ -52,6 +57,8 @@ public class SphinxSearchService extends BaseJdbcService implements SearchServic
         sphinxQuery.put("limit", String.valueOf(maxResult));
         sphinxQuery.put("mode", "any");
         sphinxQuery.put("sort", sortHint(searchString));
+        String usersFilter = userService.getUserBLUsers(visitor.getUid()).stream().map(u -> String.valueOf(u.getUid())).collect(Collectors.joining(","));
+        sphinxQuery.put("!filter", usersFilter);
         if (page > 0) {
             sphinxQuery.put("offset", String.valueOf(page * maxResult));
         }
@@ -63,7 +70,7 @@ public class SphinxSearchService extends BaseJdbcService implements SearchServic
     }
 
     @Override
-    public List<Integer> searchByStringAndUser(final String searchString, final int userId, int page) {
+    public List<Integer> searchByStringAndUser(User visitor, final String searchString, final int userId, int page) {
         if (StringUtils.isBlank(searchString))
             return Collections.emptyList();
 
-- 
cgit v1.2.3