juick-www: fix bl

1 files changed, 9 insertions, 1 deletions

diff --git a/juick-www/src/main/java/com/juick/www/controllers/Home.java b/juick-www/src/main/java/com/juick/www/controllers/Home.java
index cde9e635..22cf4b90 100644
--- a/juick-www/src/main/java/com/juick/www/controllers/Home.java
+++ b/juick-www/src/main/java/com/juick/www/controllers/Home.java
@@ -43,6 +43,7 @@ import javax.inject.Inject;
 import java.io.IOException;
 import java.net.URLEncoder;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  *
@@ -165,7 +166,14 @@ public class Home {
         model.addAttribute("visitor", visitor);
         model.addAttribute("readonly", messagesService.isReadonly());
         model.addAttribute("noindex", paramShow == null && paramBefore == 0);
-        model.addAttribute("msgs", messagesService.getMessages(mids));
+        List<com.juick.Message> msgs = messagesService.getMessages(mids);
+
+        if (visitor.getUid() != 0) {
+            List<Integer> blUIDs = userService.checkBL(visitor.getUid(),
+                    msgs.stream().map(m -> m.getUser().getUid()).collect(Collectors.toList()));
+            msgs.forEach(m -> m.ReadOnly |= blUIDs.contains(m.getUser().getUid()));
+        }
+        model.addAttribute("msgs", msgs);
         model.addAttribute("tags", tagService.getPopularTags());
         model.addAttribute("headers", head);
         model.addAttribute("showAdv",