Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement Quotes Board #1029

Open
wants to merge 12 commits into
base: develop
Choose a base branch
from
4 changes: 4 additions & 0 deletions application/config.json.template
Original file line number Diff line number Diff line change
Expand Up @@ -115,5 +115,9 @@
"fallbackChannelPattern": "java-news-and-changes",
"pollIntervalInMinutes": 10
},
"coolMessagesConfig": {
"minimumReactions": 1,
"boardChannelPattern": "quotes"
},
"memberCountCategoryPattern": "Info"
}
Original file line number Diff line number Diff line change
Expand Up @@ -46,6 +46,7 @@ public final class Config {
private final RSSFeedsConfig rssFeedsConfig;
private final String selectRolesChannelPattern;
private final String memberCountCategoryPattern;
private final CoolMessagesBoardConfig coolMessagesConfig;

@SuppressWarnings("ConstructorWithTooManyParameters")
@JsonCreator(mode = JsonCreator.Mode.PROPERTIES)
Expand Down Expand Up @@ -94,7 +95,9 @@ private Config(@JsonProperty(value = "token", required = true) String token,
required = true) FeatureBlacklistConfig featureBlacklistConfig,
@JsonProperty(value = "rssConfig", required = true) RSSFeedsConfig rssFeedsConfig,
@JsonProperty(value = "selectRolesChannelPattern",
required = true) String selectRolesChannelPattern) {
required = true) String selectRolesChannelPattern,
@JsonProperty(value = "coolMessagesConfig",
required = true) CoolMessagesBoardConfig coolMessagesConfig) {
this.token = Objects.requireNonNull(token);
this.githubApiKey = Objects.requireNonNull(githubApiKey);
this.databasePath = Objects.requireNonNull(databasePath);
Expand Down Expand Up @@ -127,6 +130,7 @@ private Config(@JsonProperty(value = "token", required = true) String token,
this.featureBlacklistConfig = Objects.requireNonNull(featureBlacklistConfig);
this.rssFeedsConfig = Objects.requireNonNull(rssFeedsConfig);
this.selectRolesChannelPattern = Objects.requireNonNull(selectRolesChannelPattern);
this.coolMessagesConfig = Objects.requireNonNull(coolMessagesConfig);
}

/**
Expand Down Expand Up @@ -401,6 +405,15 @@ public String getSelectRolesChannelPattern() {
return selectRolesChannelPattern;
}

/**
* The configuration of the cool messages config.
*
* @return configuration of cool messages config
*/
public CoolMessagesBoardConfig getCoolMessagesConfig() {
return coolMessagesConfig;
}

/**
* Gets the pattern matching the category that is used to display the total member count.
*
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
package org.togetherjava.tjbot.config;

import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.annotation.JsonRootName;

import java.util.Objects;

/**
* Configuration for the cool messages board feature, see
* {@link org.togetherjava.tjbot.features.basic.CoolMessagesBoardManager}.
*/
@JsonRootName("coolMessagesConfig")
public record CoolMessagesBoardConfig(
@JsonProperty(value = "minimumReactions", required = true) int minimumReactions,
@JsonProperty(value = "boardChannelPattern", required = true) String boardChannelPattern,
@JsonProperty(value = "reactionEmoji", required = true) String reactionEmoji) {

/**
* Creates a CoolMessagesBoardConfig.
*
* @param minimumReactions the minimum amount of reactions
* @param boardChannelPattern the pattern for the board channel
* @param reactionEmoji the emoji with which users should react to
*/
public CoolMessagesBoardConfig {
Objects.requireNonNull(boardChannelPattern);
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
import org.togetherjava.tjbot.config.FeatureBlacklist;
import org.togetherjava.tjbot.config.FeatureBlacklistConfig;
import org.togetherjava.tjbot.db.Database;
import org.togetherjava.tjbot.features.basic.CoolMessagesBoardManager;
import org.togetherjava.tjbot.features.basic.MemberCountDisplayRoutine;
import org.togetherjava.tjbot.features.basic.PingCommand;
import org.togetherjava.tjbot.features.basic.RoleSelectCommand;
Expand Down Expand Up @@ -145,6 +146,7 @@ public static Collection<Feature> createFeatures(JDA jda, Database database, Con
features.add(new CodeMessageManualDetection(codeMessageHandler));
features.add(new SlashCommandEducator());
features.add(new PinnedNotificationRemover(config));
features.add(new CoolMessagesBoardManager(config));

// Event receivers
features.add(new RejoinModerationRoleListener(actionsStore, config));
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
import net.dv8tion.jda.api.events.message.MessageDeleteEvent;
import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
import net.dv8tion.jda.api.events.message.MessageUpdateEvent;
import net.dv8tion.jda.api.events.message.react.MessageReactionAddEvent;

import java.util.regex.Pattern;

Expand Down Expand Up @@ -56,4 +57,13 @@ public interface MessageReceiver extends Feature {
* message that was deleted
*/
void onMessageDeleted(MessageDeleteEvent event);

/**
* Triggered by the core system whenever a new reaction was added to a message in a text channel
* of a guild the bot has been added to.
*
* @param event the event that triggered this, containing information about the corresponding
* reaction that was added
*/
void onMessageReactionAdd(MessageReactionAddEvent event);
}
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
import net.dv8tion.jda.api.events.message.MessageDeleteEvent;
import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
import net.dv8tion.jda.api.events.message.MessageUpdateEvent;
import net.dv8tion.jda.api.events.message.react.MessageReactionAddEvent;

import java.util.regex.Pattern;

Expand Down Expand Up @@ -57,4 +58,10 @@ public void onMessageUpdated(MessageUpdateEvent event) {
public void onMessageDeleted(MessageDeleteEvent event) {
// Adapter does not react by default, subclasses may change this behavior
}

@SuppressWarnings("NoopMethodInAbstractClass")
@Override
public void onMessageReactionAdd(MessageReactionAddEvent event) {
// Adapter does not react by default, subclasses may change this behavior
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,141 @@
package org.togetherjava.tjbot.features.basic;

import net.dv8tion.jda.api.EmbedBuilder;
import net.dv8tion.jda.api.JDA;
import net.dv8tion.jda.api.entities.Message;
import net.dv8tion.jda.api.entities.MessageEmbed;
import net.dv8tion.jda.api.entities.MessageReaction;
import net.dv8tion.jda.api.entities.User;
import net.dv8tion.jda.api.entities.channel.concrete.TextChannel;
import net.dv8tion.jda.api.entities.emoji.Emoji;
import net.dv8tion.jda.api.events.message.react.MessageReactionAddEvent;
import net.dv8tion.jda.api.requests.restaction.MessageCreateAction;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import org.togetherjava.tjbot.config.Config;
import org.togetherjava.tjbot.config.CoolMessagesBoardConfig;
import org.togetherjava.tjbot.features.MessageReceiverAdapter;

import java.awt.Color;
import java.util.Collections;
import java.util.Optional;
import java.util.function.Predicate;
import java.util.regex.Pattern;

/**
* Manager for the cool messages board. It appends highly-voted text messages to a separate channel
* where members of the guild can see a list of all of them.
*/
public final class CoolMessagesBoardManager extends MessageReceiverAdapter {

private static final Logger logger = LoggerFactory.getLogger(CoolMessagesBoardManager.class);
private Emoji coolEmoji;
private final Predicate<String> boardChannelNamePredicate;
private final CoolMessagesBoardConfig config;

public CoolMessagesBoardManager(Config config) {
this.config = config.getCoolMessagesConfig();
this.coolEmoji = Emoji.fromUnicode(this.config.reactionEmoji());

boardChannelNamePredicate =
Pattern.compile(this.config.boardChannelPattern()).asMatchPredicate();
}

@Override
public void onMessageReactionAdd(MessageReactionAddEvent event) {
final MessageReaction messageReaction = event.getReaction();
int originalReactionsCount = messageReaction.hasCount() ? messageReaction.getCount() : 0;
boolean isCoolEmoji = messageReaction.getEmoji().equals(coolEmoji);
long guildId = event.getGuild().getIdLong();
Optional<TextChannel> boardChannel = getBoardChannel(event.getJDA(), guildId);

if (boardChannel.isEmpty()) {
logger.warn(
"Could not find board channel with pattern '{}' in server with ID '{}'. Skipping reaction handling...",
this.config.boardChannelPattern(), guildId);
return;
}

// If the bot has already reacted to this message, then this means that
// the message has been quoted to the cool messages board, so skip it.
if (hasBotReacted(event.getJDA(), messageReaction)) {
return;
}

final int newReactionsCount = originalReactionsCount + 1;
if (isCoolEmoji && newReactionsCount >= config.minimumReactions()) {
event.retrieveMessage()
.queue(message -> message.addReaction(coolEmoji)
.flatMap(v -> insertCoolMessage(boardChannel.get(), message))
.queue(),
e -> logger.warn("Tried to retrieve cool message but got: {}",
e.getMessage()));
}
}

/**
* Gets the board text channel where the quotes go to, wrapped in an optional.
*
* @param jda the JDA
* @param guildId the guild ID
* @return the board text channel
*/
private Optional<TextChannel> getBoardChannel(JDA jda, long guildId) {
return jda.getGuildById(guildId)
Taz03 marked this conversation as resolved.
Show resolved Hide resolved
.getTextChannelCache()
.stream()
.filter(channel -> boardChannelNamePredicate.test(channel.getName()))
.findAny();
}

/**
* Inserts a message to the specified text channel
*
* @return a {@link MessageCreateAction} of the call to make
*/
private static MessageCreateAction insertCoolMessage(TextChannel boardChannel,
Message message) {
return boardChannel.sendMessageEmbeds(Collections.singleton(createQuoteEmbed(message)));
}

/**
* Wraps a text message into a properly formatted quote message used for the board text channel.
*/
private static MessageEmbed createQuoteEmbed(Message message) {
final User author = message.getAuthor();
EmbedBuilder embedBuilder = new EmbedBuilder();

// If the message contains image(s), include the first one
var firstImageAttachment = message.getAttachments()
.stream()
.parallel()
.filter(Message.Attachment::isImage)
.findAny()
.orElse(null);

if (firstImageAttachment != null) {
embedBuilder.setThumbnail(firstImageAttachment.getUrl());
}

return embedBuilder.setDescription(message.getContentDisplay())
.appendDescription("%n%n[Jump to Message](%s)".formatted(message.getJumpUrl()))
.setColor(Color.orange)
.setAuthor(author.getName(), null, author.getAvatarUrl())
.setTimestamp(message.getTimeCreated())
.build();
}

/**
* Checks a {@link MessageReaction} to see if the bot has reacted to it.
*/
private boolean hasBotReacted(JDA jda, MessageReaction messageReaction) {
if (!coolEmoji.equals(messageReaction.getEmoji())) {
return false;
}

return messageReaction.retrieveUsers()
.parallelStream()
.anyMatch(user -> jda.getSelfUser().getIdLong() == user.getIdLong());
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
import net.dv8tion.jda.api.events.message.MessageDeleteEvent;
import net.dv8tion.jda.api.events.message.MessageReceivedEvent;
import net.dv8tion.jda.api.events.message.MessageUpdateEvent;
import net.dv8tion.jda.api.events.message.react.MessageReactionAddEvent;
import net.dv8tion.jda.api.hooks.ListenerAdapter;
import net.dv8tion.jda.api.interactions.callbacks.IReplyCallback;
import net.dv8tion.jda.api.interactions.components.ComponentInteraction;
Expand Down Expand Up @@ -238,6 +239,14 @@ public void onMessageDelete(final MessageDeleteEvent event) {
}
}

@Override
public void onMessageReactionAdd(final MessageReactionAddEvent event) {
if (event.isFromGuild()) {
getMessageReceiversSubscribedTo(event.getChannel())
.forEach(messageReceiver -> messageReceiver.onMessageReactionAdd(event));
}
}

private Stream<MessageReceiver> getMessageReceiversSubscribedTo(Channel channel) {
String channelName = channel.getName();
return channelNameToMessageReceiver.entrySet()
Expand Down
Loading