Skip to content

Add the ability to add a ⭐ to a message in the #starboard channel as well as the original message #167

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.

Sign up for GitHub

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

Closed
wants to merge 6 commits into from
Closed

Add the ability to add a ⭐ to a message in the #starboard channel as well as the original message #167

Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
737a22d
Starboard Feature Update.
venkat1701 Jan 13, 2022
dc12fd7
Starboard Code Update.
venkat1701 Jan 14, 2022
167f6f7
Merge branch 'Java-Discord:main' into krish/starboard
venkat1701 Jan 15, 2022
260b6fb
Starboard Reaction Counts Updated.
venkat1701 Jan 15, 2022
14a38cc
Null Check Updated.
venkat1701 Jan 15, 2022
a211ce6
Removed Objects#requireNonNull
venkat1701 Jan 15, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
104 changes: 80 additions & 24 deletions src/main/java/net/javadiscord/javabot/events/StarboardListener.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
import net.dv8tion.jda.api.EmbedBuilder;
import net.dv8tion.jda.api.entities.*;
import net.dv8tion.jda.api.events.message.MessageDeleteEvent;
import net.dv8tion.jda.api.events.message.react.GenericMessageReactionEvent;
import net.dv8tion.jda.api.events.message.react.MessageReactionAddEvent;
import net.dv8tion.jda.api.events.message.react.MessageReactionRemoveEvent;
import net.dv8tion.jda.api.exceptions.ErrorHandler;
Expand All @@ -19,6 +20,11 @@
import org.bson.Document;
import org.jetbrains.annotations.NotNull;

import java.util.List;
import java.util.Objects;
import java.util.concurrent.ExecutionException;
import java.util.stream.Stream;

import static com.mongodb.client.model.Filters.eq;

@Slf4j
Expand Down Expand Up @@ -53,18 +59,24 @@ void addToSB(Guild guild, MessageChannel channel, Message message, int starCount
e.printStackTrace();
}
}
msgAction.queue(sbMsg -> db.createStarboardDoc(guildId, channelId, messageId, sbMsg.getId()));
msgAction.queue(sbMsg -> {
db.createStarboardDoc(guildId, channelId, messageId, sbMsg.getId());
sbMsg.addReaction(config.getEmotes().get(0)).queue();
});
}

void updateSB(Guild guild, String channelId, String messageId, int reactionCount) {
void updateSB(Guild guild, String channelId, String messageId, int reactionCount, Message message) {
Database db = new Database();
String guildId = guild.getId();
String sbcEmbedId = db.getStarboardChannelString(guildId, channelId, messageId, "starboard_embed");
var config = Bot.config.get(guild).getStarBoard();
if (sbcEmbedId == null) return;

Bot.config.get(guild).getStarBoard().getChannel()
.retrieveMessageById(sbcEmbedId).queue((sbMsg) -> handleMessage(guild, channelId, messageId, reactionCount, sbMsg),
failure -> this.removeMessageFromStarboard(guild, messageId));
.retrieveMessageById(sbcEmbedId).queue((sbMsg) -> {
// Adding -1 to the UniqueStarCount since it should not count for the ⭐ of JavaBot.
this.handleMessage(guild, channelId, messageId, this.getUniqueStarCounts(sbMsg, message, config)-1, sbMsg);
}, failure -> this.removeMessageFromStarboard(guild, messageId));
}

private void handleMessage(Guild guild, String channelId, String messageId, int reactionCount, Message sbMsg) {
Expand Down Expand Up @@ -133,7 +145,7 @@ private void processStarboardMessage(Guild guild, String guildId, MongoCollectio
if (!db.isMessageOnStarboard(guildId, channelId, messageId) && reactionCount >= config.getReactionThreshold()) {
addToSB(guild, channel, msg, reactionCount);
} else if (db.getStarboardChannelString(guildId, channelId, messageId, "starboard_embed") != null) {
updateSB(guild, channelId, messageId, reactionCount);
updateSB(guild, channelId, messageId, reactionCount, msg);
}
}, failure -> collection.deleteOne(doc));
}
Expand All @@ -147,7 +159,7 @@ private int getReactionCountForEmote(String emote, Message msg) {
}

private void updateIfEmpty(Guild guild, String channelId, String messageId, Message msg) {
if (msg.getReactions().isEmpty()) updateSB(guild, channelId, messageId, 0);
if (msg.getReactions().isEmpty()) updateSB(guild, channelId, messageId, 0, msg);
}

@Override
Expand All @@ -166,15 +178,20 @@ public void onMessageReactionAdd(@NotNull MessageReactionAddEvent event) {
String channelId = event.getChannel().getId();
String messageId = event.getMessageId();

event.getChannel().retrieveMessageById(messageId).queue(message -> {
int reactionCount = getReactionCountForEmote(config.getEmotes().get(0), message);
if (channelId.equals(config.getChannel().getId())) {
this.updateSbReactionCounts(event, config, messageId, channelId);
}

if (db.isMessageOnStarboard(guildId, channelId, messageId)) {
updateSB(event.getGuild(), channelId, messageId, reactionCount);
} else if (reactionCount >= config.getReactionThreshold()) {
addToSB(event.getGuild(), event.getChannel(), message, reactionCount);
}
});
if (!channelId.equals(config.getChannel().getId())) {
event.getChannel().retrieveMessageById(messageId).queue(message -> {
int reactionCount = getReactionCountForEmote(config.getEmotes().get(0), message);
if (db.isMessageOnStarboard(guildId, channelId, messageId)) {
updateSB(event.getGuild(), channelId, messageId, reactionCount, message);
} else if (reactionCount >= config.getReactionThreshold()) {
addToSB(event.getGuild(), event.getChannel(), message, reactionCount);
}
});
}
}

@Override
Expand All @@ -190,21 +207,60 @@ public void onMessageReactionRemove(@NotNull MessageReactionRemoveEvent event) {
String channelId = event.getChannel().getId();
String messageId = event.getMessageId();

event.getChannel().retrieveMessageById(messageId).queue(message -> {
int reactionCount = getReactionCountForEmote(config.getEmotes().get(0), message);
if (channelId.equals(config.getChannel().getId())) {
this.updateSbReactionCounts(event, config, messageId, channelId);
}

if (db.isMessageOnStarboard(guildId, channelId, messageId)) {
updateSB(event.getGuild(), channelId, messageId, reactionCount);
} else if (reactionCount >= config.getReactionThreshold()) {
addToSB(event.getGuild(), event.getChannel(), message, reactionCount);
} else {
removeMessageFromStarboard(event.getGuild(), messageId);
}
});
if (!channelId.equals(config.getChannel().getId())) {

event.getChannel().retrieveMessageById(messageId).queue(message -> {
int reactionCount = getReactionCountForEmote(config.getEmotes().get(0), message);
if (db.isMessageOnStarboard(guildId, channelId, messageId)) {
updateSB(event.getGuild(), channelId, messageId, reactionCount, message);
} else if (reactionCount >= config.getReactionThreshold()) {
addToSB(event.getGuild(), event.getChannel(), message, reactionCount);
} else {
removeMessageFromStarboard(event.getGuild(), messageId);
}
});
}
}

@Override
public void onMessageDelete(@NotNull MessageDeleteEvent event) {
removeMessageFromStarboard(event.getGuild(), event.getMessageId());
}

private void updateSbReactionCounts(GenericMessageReactionEvent event, StarBoardConfig config, String messageId, String remoteTextChannelId) {
var starboardChannel = event.getGuild().getTextChannelById(remoteTextChannelId);
if(starboardChannel!=null){
starboardChannel.retrieveMessageById(messageId).queue(message -> {
MessageEmbed me;
if(message.getEmbeds().get(0)!=null){
me = message.getEmbeds().get(0);
String prevChannelId = me.getAuthor().getUrl().split("/")[5];
String originalMessageId = me.getAuthor().getUrl().split("/")[6];
event.getGuild().getTextChannelById(prevChannelId).retrieveMessageById(originalMessageId).queue(msg -> {
// Adding -1 to the UniqueStarCount since it should not count for the ⭐ of JavaBot.
this.updateMessage(
event.getGuild(),
prevChannelId,
this.getUniqueStarCounts(message, msg, config)-1,
message, config);
});
}
});
}
}

private int getUniqueStarCounts(Message starboardMessage, Message originalMessage, StarBoardConfig config){
try {
List<User> originalMessageUsers = originalMessage.retrieveReactionUsers(config.getEmotes().get(0)).takeAsync(this.getReactionCountForEmote(config.getEmotes().get(0), originalMessage)).get();
List<User> starboardMessageUsers = starboardMessage.retrieveReactionUsers(config.getEmotes().get(0)).takeAsync(this.getReactionCountForEmote(config.getEmotes().get(0), starboardMessage)).get();

Stream<User> stream = Stream.concat(originalMessageUsers.stream(), starboardMessageUsers.stream()).distinct();

return (int)stream.count();
}catch(InterruptedException | ExecutionException e){ return 0; }
}
}