Skip to content

Commit e753509

Browse files
added database
1 parent 178a2ae commit e753509

File tree

3 files changed

+17
-2
lines changed

3 files changed

+17
-2
lines changed

application/src/main/java/org/togetherjava/tjbot/features/Features.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
package org.togetherjava.tjbot.features;
22

33
import net.dv8tion.jda.api.JDA;
4+
45
import org.togetherjava.tjbot.config.Config;
56
import org.togetherjava.tjbot.db.Database;
67
import org.togetherjava.tjbot.features.basic.PingCommand;
@@ -139,7 +140,7 @@ public static Collection<Feature> createFeatures(JDA jda, Database database, Con
139140
features.add(new GuildLeaveCloseThreadListener(config));
140141
features.add(new LeftoverBookmarksListener(bookmarksSystem));
141142
features.add(new HelpThreadCreatedListener(helpSystemHelper));
142-
features.add(new Starboard(config));
143+
features.add(new Starboard(config, database));
143144

144145
// Message context commands
145146

application/src/main/java/org/togetherjava/tjbot/features/basic/Starboard.java

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,17 +16,22 @@
1616

1717
import org.togetherjava.tjbot.config.Config;
1818
import org.togetherjava.tjbot.config.StarboardConfig;
19+
import org.togetherjava.tjbot.db.Database;
1920
import org.togetherjava.tjbot.features.EventReceiver;
2021

2122
import java.util.Optional;
2223

24+
import static org.togetherjava.tjbot.db.generated.tables.StarboardMessages.STARBOARD_MESSAGES;
25+
2326
public class Starboard extends ListenerAdapter implements EventReceiver {
2427

2528
private static final Logger logger = LoggerFactory.getLogger(Starboard.class);
2629
private final StarboardConfig config;
30+
private final Database database;
2731

28-
public Starboard(Config config) {
32+
public Starboard(Config config, Database database) {
2933
this.config = config.getStarboard();
34+
this.database = database;
3035
}
3136

3237
@Override
@@ -42,6 +47,12 @@ public void onMessageReactionAdd(@NotNull MessageReactionAddEvent event) {
4247
config.getChannelName());
4348
return;
4449
}
50+
if (database.read(context -> context.selectFrom(STARBOARD_MESSAGES)
51+
.fetch("message_id")
52+
.contains(event.getMessageIdLong()))) {
53+
database.write(context -> context.newRecord(STARBOARD_MESSAGES)
54+
.setMessageId(event.getMessageIdLong()));
55+
}
4556
event.retrieveMessage()
4657
.flatMap(
4758
message -> starboardChannel.orElseThrow().sendMessageEmbeds(formEmbed(message)))
Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
CREATE TABLE starboard_messages
2+
(message_id BIGINT NOT NULL PRIMARY KEY
3+
)

0 commit comments

Comments
 (0)