diff --git a/src/main/java/com/github/yvasyliev/config/AppConfig.java b/src/main/java/com/github/yvasyliev/config/AppConfig.java index 6145607..07ef379 100644 --- 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()); + } } \ No newline at end of file diff --git a/src/main/java/com/github/yvasyliev/telegram/chain/NsfwCheckChain.java b/src/main/java/com/github/yvasyliev/telegram/chain/NsfwCheckChain.java new file mode 100644 index 0000000..71dde1f --- /dev/null +++ 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)); + } +}