NSFW check

yvasyliev · Jun 25, 2023 · 780ebe9 · 780ebe9
1 parent d9113cb
commit 780ebe9
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/src/main/java/com/github/yvasyliev/config/AppConfig.java b/src/main/java/com/github/yvasyliev/config/AppConfig.java
@@ -16,6 +16,7 @@
 import com.github.yvasyliev.telegram.TelegramSenderBotImpl;
 import com.github.yvasyliev.telegram.chain.FilterAuthorChain;
 import com.github.yvasyliev.telegram.chain.NOPRepeaterChain;
+import com.github.yvasyliev.telegram.chain.NsfwCheckChain;
 import com.github.yvasyliev.telegram.chain.RepeatGif;
 import com.github.yvasyliev.telegram.chain.RepeatMultiplePhotos;
 import com.github.yvasyliev.telegram.chain.RepeatNestedPost;
@@ -94,7 +95,7 @@ public RedditPostService redditPostService() {
 
     @Bean
     public SubredditPostRepeaterChain repeatNestedPost() {
-        return new RepeatNestedPost(repeatText());
+        return new RepeatNestedPost(nsfwCheckChain());
     }
 
     @Bean
@@ -141,4 +142,9 @@ public SubredditPostRepeaterChain repeatPoll() {
     public SubredditPostRepeaterChain filterAuthorChain() {
         return new FilterAuthorChain(repeatNestedPost());
     }
+
+    @Bean
+    public SubredditPostRepeaterChain nsfwCheckChain() {
+        return new NsfwCheckChain(repeatText());
+    }
 }
diff --git a/src/main/java/com/github/yvasyliev/telegram/chain/NsfwCheckChain.java b/src/main/java/com/github/yvasyliev/telegram/chain/NsfwCheckChain.java
@@ -0,0 +1,15 @@
+package com.github.yvasyliev.telegram.chain;
+
+import com.fasterxml.jackson.databind.JsonNode;
+import com.github.yvasyliev.telegram.TelegramSenderBot;
+
+public class NsfwCheckChain extends SubredditPostRepeaterChain {
+    public NsfwCheckChain(SubredditPostRepeaterChain nextChain) {
+        super(nextChain);
+    }
+
+    @Override
+    public void repeatRedditPost(JsonNode data, TelegramSenderBot telegramSenderBot, boolean needModerate) {
+        super.repeatRedditPost(data, telegramSenderBot, hasSpoiler(data));
+    }
+}