Added reminders (in a new table for now)

This commit is contained in:
Stijn
2022-09-21 21:49:42 +02:00
parent 83bb43f696
commit d8102e4baa
12 changed files with 674 additions and 4 deletions
@@ -1,5 +1,7 @@
package com.alttd.buttonManager;
import com.alttd.buttonManager.buttons.remindMeConfirm.ButtonRemindMeCancel;
import com.alttd.buttonManager.buttons.remindMeConfirm.ButtonRemindMeConfirm;
import com.alttd.buttonManager.buttons.suggestionReview.ButtonSuggestionReviewAccept;
import com.alttd.buttonManager.buttons.suggestionReview.ButtonSuggestionReviewDeny;
import com.alttd.util.Util;
@@ -22,7 +24,9 @@ public class ButtonManager extends ListenerAdapter {
public ButtonManager() {
buttons = List.of(
new ButtonSuggestionReviewAccept(),
new ButtonSuggestionReviewDeny());
new ButtonSuggestionReviewDeny(),
new ButtonRemindMeCancel(),
new ButtonRemindMeConfirm());
}
@Override
@@ -0,0 +1,25 @@
package com.alttd.buttonManager.buttons.remindMeConfirm;
import com.alttd.buttonManager.DiscordButton;
import com.alttd.util.Util;
import net.dv8tion.jda.api.events.interaction.component.ButtonInteractionEvent;
import net.dv8tion.jda.api.interactions.components.buttons.Button;
public class ButtonRemindMeCancel extends DiscordButton {
@Override
public String getButtonId() {
return "remind_me_cancel";
}
@Override
public void execute(ButtonInteractionEvent event) {
ButtonRemindMeConfirm.removeReminder(event.getUser().getIdLong());
event.replyEmbeds(Util.genericSuccessEmbed("Success", "Cancelled your reminder!"))
.setEphemeral(true).queue();
}
@Override
public Button getButton() {
return Button.danger(getButtonId(), "Cancel");
}
}
@@ -0,0 +1,81 @@
package com.alttd.buttonManager.buttons.remindMeConfirm;
import com.alttd.buttonManager.DiscordButton;
import com.alttd.database.queries.QueriesReminders.QueriesReminders;
import com.alttd.database.queries.QueriesReminders.Reminder;
import com.alttd.reminders.ReminderScheduler;
import com.alttd.util.Util;
import net.dv8tion.jda.api.events.interaction.component.ButtonInteractionEvent;
import net.dv8tion.jda.api.interactions.components.buttons.Button;
import java.util.HashMap;
public class ButtonRemindMeConfirm extends DiscordButton {
private static final HashMap<Long, Reminder> unconfirmedReminders = new HashMap<>();
public static synchronized void putReminder(long id, Reminder reminder) {
unconfirmedReminders.put(id, reminder);
}
public static synchronized Reminder removeReminder(long id) {
return unconfirmedReminders.get(id);
}
@Override
public String getButtonId() {
return "remind_me_confirm";
}
@Override
public void execute(ButtonInteractionEvent event) {
Reminder reminder = removeReminder(event.getUser().getIdLong());
if (storeReminder(reminder, event)) {
event.replyEmbeds(Util.genericSuccessEmbed("Success", "Your reminder was successfully created!"))
.setEphemeral(true).queue();
}
}
private boolean storeReminder(Reminder reminder, ButtonInteractionEvent event) {
if (reminder == null) {
event.replyEmbeds(Util.genericErrorEmbed("Error", "Unable to retrieve reminder data for this button"))
.setEphemeral(true).queue();
return false;
}
int id = QueriesReminders.storeReminder(reminder);
if (id == 0) {
event.replyEmbeds(Util.genericErrorEmbed("Error", "Unable to store reminder in the database"))
.setEphemeral(true).queue();
return false;
}
reminder = new Reminder(
id,
reminder.title(),
reminder.description(),
reminder.userId(),
reminder.guildId(),
reminder.channelId(),
reminder.messageId(),
reminder.shouldRepeat(),
reminder.creationDate(),
reminder.remindDate());
ReminderScheduler instance = ReminderScheduler.getInstance(event.getJDA());
if (instance == null) {
QueriesReminders.removeReminder(reminder.id());
event.replyEmbeds(Util.genericErrorEmbed("Error", "Unable to start reminder, removing it from the database..."))
.setEphemeral(true).queue();
return false;
}
instance.addReminder(reminder);
return true;
}
@Override
public Button getButton() {
return Button.success(getButtonId(), "Confirm");
}
}