Skip to content

Commit 40e1881

Browse files
committed
Added button count check to prevent removing the button
1 parent 38cb08d commit 40e1881

File tree

2 files changed

+18
-4
lines changed

2 files changed

+18
-4
lines changed

pom.xml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
<modelVersion>4.0.0</modelVersion>
66
<groupId>com.github.ygimenez</groupId>
77
<artifactId>PaginationUtils</artifactId>
8-
<version>1.2.0</version>
8+
<version>1.2.1</version>
99
<packaging>jar</packaging>
1010

1111
<name>Pagination Utils</name>
@@ -29,7 +29,7 @@
2929
<repository>
3030
<id>github</id>
3131
<name>GitHub KuuHaKu Apache Maven Packages</name>
32-
<url>https://maven.pkg.github.com/ygimenez/PaginationUtils</url>
32+
<url>https://maven.pkg.github.com/ygimenez/Pagination-Utils</url>
3333
</repository>
3434
</distributionManagement>
3535

src/main/java/com/github/ygimenez/listener/MessageHandler.java

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import net.dv8tion.jda.api.entities.Message;
44
import net.dv8tion.jda.api.events.message.MessageDeleteEvent;
55
import net.dv8tion.jda.api.events.message.react.MessageReactionAddEvent;
6+
import net.dv8tion.jda.api.events.message.react.MessageReactionRemoveEvent;
67
import net.dv8tion.jda.api.hooks.ListenerAdapter;
78

89
import javax.annotation.Nonnull;
@@ -23,12 +24,25 @@ public void removeEvent(Message msg) {
2324

2425
@Override
2526
public void onMessageReactionAdd(@Nonnull MessageReactionAddEvent evt) {
26-
Consumer<MessageReactionAddEvent> act = events.get(evt.getGuild().getId() + evt.getMessageId());
27-
if (act != null) act.accept(evt);
27+
if (events.containsKey(evt.getGuild().getId() + evt.getMessageId()))
28+
events.get(evt.getGuild().getId() + evt.getMessageId()).accept(evt);
2829
}
2930

3031
@Override
3132
public void onMessageDelete(@Nonnull MessageDeleteEvent evt) {
3233
events.remove((evt.getChannelType().isGuild() ? evt.getGuild().getId() : evt.getPrivateChannel().getUser().getId()) + evt.getMessageId());
3334
}
35+
36+
@Override
37+
public void onMessageReactionRemove(@Nonnull MessageReactionRemoveEvent evt) {
38+
if (events.containsKey(evt.getGuild().getId() + evt.getMessageId()))
39+
evt.getChannel().retrieveMessageById(evt.getMessageId()).queue(msg -> {
40+
if (!msg.getReactions().contains(evt.getReaction())) {
41+
if (evt.getReactionEmote().isEmoji())
42+
msg.addReaction(evt.getReactionEmote().getAsCodepoints()).queue();
43+
else
44+
msg.addReaction(evt.getReactionEmote().getEmote()).queue();
45+
}
46+
});
47+
}
3448
}

0 commit comments

Comments
 (0)