Compare commits
No commits in common. "98a162a33b6e7ef12cce3906d6b9f4add5fba025" and "3d626bb46f3a17fbf871a90efcbcb3f3e6406d98" have entirely different histories.
98a162a33b
...
3d626bb46f
6
pom.xml
6
pom.xml
@ -30,12 +30,6 @@
|
||||
<artifactId>slf4j-simple</artifactId>
|
||||
<version>2.0.0</version>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.xerial</groupId>
|
||||
<artifactId>sqlite-jdbc</artifactId>
|
||||
<version>3.39.4.1</version>
|
||||
</dependency>
|
||||
|
||||
</dependencies>
|
||||
|
||||
<build>
|
||||
|
@ -1,23 +1,17 @@
|
||||
package wtf.beatrice.hidekobot;
|
||||
|
||||
import org.jetbrains.annotations.Nullable;
|
||||
import wtf.beatrice.hidekobot.database.DatabaseManager;
|
||||
import wtf.beatrice.hidekobot.listeners.MessageLogger;
|
||||
|
||||
public class Configuration
|
||||
{
|
||||
|
||||
|
||||
private static DatabaseManager dbManager = null;
|
||||
private static boolean verbose = false;
|
||||
private static MessageLogger verbosityLogger;
|
||||
|
||||
// todo: allow people to set their own user id
|
||||
private static final long botOwnerId = 979809420714332260L;
|
||||
|
||||
private final static String expiryTimestampFormat = "yy/MM/dd HH:mm:ss";
|
||||
private final static long expiryTimeSeconds = 60L;
|
||||
|
||||
private final static String defaultInviteLink =
|
||||
"https://discord.com/api/oauth2/authorize?client_id=%userid%&scope=bot+applications.commands&permissions=8";
|
||||
|
||||
@ -101,35 +95,4 @@ public class Configuration
|
||||
return defaultInviteLink.replace("%userid%", botApplicationId);
|
||||
}
|
||||
|
||||
/**
|
||||
* Set the already fully-initialized DatabaseManager instance, ready to be accessed and used.
|
||||
*
|
||||
* @param databaseManagerInstance the fully-initialized DatabaseManager instance.
|
||||
*/
|
||||
public static void setDatabaseManagerInstance(DatabaseManager databaseManagerInstance)
|
||||
{
|
||||
dbManager = databaseManagerInstance;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the fully-initialized DatabaseManager instance, ready to be used.
|
||||
*
|
||||
* @return the DatabaseManager instance.
|
||||
*/
|
||||
public static @Nullable DatabaseManager getDatabaseManager() { return dbManager; }
|
||||
|
||||
/**
|
||||
* Get the DateTimeFormatter string for parsing the expired messages timestamp.
|
||||
*
|
||||
* @return the String of the DateTimeFormatter format.
|
||||
*/
|
||||
public static String getExpiryTimestampFormat(){ return expiryTimestampFormat; }
|
||||
|
||||
/**
|
||||
* Get the amount of seconds after which a message expires.
|
||||
*
|
||||
* @return long value of the expiry seconds.
|
||||
*/
|
||||
public static long getExpiryTimeSeconds() { return expiryTimeSeconds; }
|
||||
|
||||
}
|
||||
|
@ -6,21 +6,17 @@ import net.dv8tion.jda.api.OnlineStatus;
|
||||
import net.dv8tion.jda.api.entities.Activity;
|
||||
import net.dv8tion.jda.api.requests.GatewayIntent;
|
||||
import sun.misc.Signal;
|
||||
import wtf.beatrice.hidekobot.database.DatabaseManager;
|
||||
import wtf.beatrice.hidekobot.listeners.ButtonInteractionListener;
|
||||
import wtf.beatrice.hidekobot.listeners.MessageListener;
|
||||
import wtf.beatrice.hidekobot.listeners.SlashCommandCompleter;
|
||||
import wtf.beatrice.hidekobot.listeners.SlashCommandListener;
|
||||
import wtf.beatrice.hidekobot.utils.ExpiredMessageRunner;
|
||||
import wtf.beatrice.hidekobot.utils.Logger;
|
||||
import wtf.beatrice.hidekobot.utils.SlashCommandsUtil;
|
||||
|
||||
import java.io.File;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
import java.util.concurrent.Executors;
|
||||
import java.util.concurrent.ScheduledExecutorService;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
public class HidekoBot
|
||||
@ -97,30 +93,6 @@ public class HidekoBot
|
||||
jda.getPresence().setStatus(OnlineStatus.ONLINE);
|
||||
jda.getPresence().setActivity(Activity.playing("Hatsune Miku: Project DIVA"));
|
||||
|
||||
// connect to database
|
||||
logger.log("Connecting to database...");
|
||||
String dbFilePath = System.getProperty("user.dir") + File.separator + "db.sqlite"; // in current directory
|
||||
DatabaseManager dbManager = new DatabaseManager(dbFilePath);
|
||||
if(dbManager.connect() && dbManager.initDb())
|
||||
{
|
||||
logger.log("Database connection initialized!");
|
||||
Configuration.setDatabaseManagerInstance(dbManager);
|
||||
|
||||
// load data here...
|
||||
|
||||
logger.log("Database data loaded into memory!");
|
||||
} else {
|
||||
logger.log("Error initializing database connection!");
|
||||
}
|
||||
|
||||
// start scheduled runnables
|
||||
ScheduledExecutorService scheduler = Executors.newSingleThreadScheduledExecutor();
|
||||
ExpiredMessageRunner task = new ExpiredMessageRunner();
|
||||
int initDelay = 5;
|
||||
int periodicDelay = 5;
|
||||
scheduler.scheduleAtFixedRate(task, initDelay, periodicDelay, TimeUnit.SECONDS);
|
||||
|
||||
|
||||
// print the bot logo.
|
||||
logger.log("\n\n" + logger.getLogo() + "\nv" + version + " - bot is ready!\n", 2);
|
||||
|
||||
|
@ -4,15 +4,10 @@ import net.dv8tion.jda.api.entities.Message;
|
||||
import net.dv8tion.jda.api.entities.MessageHistory;
|
||||
import net.dv8tion.jda.api.entities.channel.concrete.TextChannel;
|
||||
import net.dv8tion.jda.api.entities.channel.middleman.MessageChannel;
|
||||
import net.dv8tion.jda.api.entities.emoji.Emoji;
|
||||
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent;
|
||||
import net.dv8tion.jda.api.events.interaction.component.ButtonInteractionEvent;
|
||||
import net.dv8tion.jda.api.interactions.InteractionHook;
|
||||
import net.dv8tion.jda.api.interactions.commands.OptionMapping;
|
||||
import net.dv8tion.jda.api.interactions.components.buttons.Button;
|
||||
import net.dv8tion.jda.api.requests.restaction.WebhookMessageEditAction;
|
||||
import org.jetbrains.annotations.NotNull;
|
||||
import wtf.beatrice.hidekobot.Configuration;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
@ -22,11 +17,6 @@ public class ClearChatCommand
|
||||
|
||||
public void runSlashCommand(@NotNull SlashCommandInteractionEvent event)
|
||||
{
|
||||
|
||||
// run in a new thread, so we don't block the main one
|
||||
new Thread(() ->
|
||||
{
|
||||
|
||||
MessageChannel channel = event.getChannel();
|
||||
|
||||
if(!(channel instanceof TextChannel))
|
||||
@ -144,39 +134,17 @@ public class ClearChatCommand
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Button deleteButton = Button.danger("clear_delete", "Delete").withEmoji(Emoji.fromUnicode("❌"));
|
||||
|
||||
WebhookMessageEditAction<Message> webhookMessageEditAction;
|
||||
|
||||
// log having deleted the messages (or not).
|
||||
// log having deleted the messages.
|
||||
if(deleted < 1)
|
||||
{
|
||||
webhookMessageEditAction = replyInteraction.editOriginal("\uD83D\uDE22 Couldn't clear any message!");
|
||||
replyInteraction.editOriginal("\uD83D\uDE22 Couldn't clear any message!").queue();
|
||||
} else if(deleted == 1)
|
||||
{
|
||||
webhookMessageEditAction = replyInteraction.editOriginal("âś‚ Cleared 1 message!");
|
||||
replyInteraction.editOriginal("âś‚ Cleared 1 message!").queue();
|
||||
} else {
|
||||
webhookMessageEditAction = replyInteraction.editOriginal("âś‚ Cleared " + deleted + " messages!");
|
||||
replyInteraction.editOriginal("âś‚ Cleared " + deleted + " messages!").queue();
|
||||
}
|
||||
}
|
||||
|
||||
Message message = webhookMessageEditAction
|
||||
.setActionRow(deleteButton)
|
||||
.complete();
|
||||
|
||||
String replyMessageId = message.getId();
|
||||
String replyChannelId = message.getChannel().getId();
|
||||
String replyGuildId = message.getGuild().getId();
|
||||
|
||||
Configuration.getDatabaseManager().queueDisabling(replyGuildId, replyChannelId, replyMessageId);
|
||||
|
||||
}
|
||||
}).start();
|
||||
|
||||
}
|
||||
|
||||
public void deleteButton(ButtonInteractionEvent event)
|
||||
{
|
||||
event.getInteraction().getMessage().delete().queue();
|
||||
}
|
||||
}
|
||||
|
@ -1,6 +1,7 @@
|
||||
package wtf.beatrice.hidekobot.commands.slash;
|
||||
|
||||
import net.dv8tion.jda.api.entities.emoji.Emoji;
|
||||
import net.dv8tion.jda.api.events.interaction.GenericInteractionCreateEvent;
|
||||
import net.dv8tion.jda.api.events.interaction.command.SlashCommandInteractionEvent;
|
||||
import net.dv8tion.jda.api.events.interaction.component.ButtonInteractionEvent;
|
||||
import net.dv8tion.jda.api.interactions.components.ActionRow;
|
||||
@ -13,31 +14,7 @@ import java.util.List;
|
||||
public class CoinFlipCommand
|
||||
{
|
||||
|
||||
private final Button reflipButton = Button.primary("coinflip_reflip", "Flip again")
|
||||
.withEmoji(Emoji.fromFormatted("\uD83E\uDE99"));
|
||||
|
||||
public void runSlashCommand(@NotNull SlashCommandInteractionEvent event)
|
||||
{
|
||||
// perform coin flip
|
||||
event.reply(genRandom())
|
||||
.addActionRow(reflipButton)
|
||||
.queue();
|
||||
}
|
||||
|
||||
public void buttonReFlip(ButtonInteractionEvent event)
|
||||
{
|
||||
// set old message's button as disabled
|
||||
List<ActionRow> actionRows = event.getMessage().getActionRows();
|
||||
actionRows.set(0, actionRows.get(0).asDisabled());
|
||||
event.editComponents(actionRows).queue();
|
||||
|
||||
// perform coin flip
|
||||
event.getHook().sendMessage(genRandom())
|
||||
.addActionRow(reflipButton)
|
||||
.queue();
|
||||
}
|
||||
|
||||
private String genRandom()
|
||||
{
|
||||
int rand = RandomUtil.getRandomNumber(0, 1);
|
||||
String msg;
|
||||
@ -49,7 +26,32 @@ public class CoinFlipCommand
|
||||
msg = "It's **Tails**! :coin:";
|
||||
}
|
||||
|
||||
return msg;
|
||||
event.reply(msg)
|
||||
.addActionRow(Button.primary("reflip", "Flip again")
|
||||
.withEmoji(Emoji.fromFormatted("\uD83E\uDE99")) // coin emoji
|
||||
).queue();
|
||||
}
|
||||
|
||||
public void buttonReflip(ButtonInteractionEvent event)
|
||||
{
|
||||
List<ActionRow> actionRows = event.getMessage().getActionRows();
|
||||
actionRows.set(0, actionRows.get(0).asDisabled());
|
||||
event.editComponents(actionRows).queue();
|
||||
|
||||
int rand = RandomUtil.getRandomNumber(0, 1);
|
||||
String msg;
|
||||
|
||||
if(rand == 1)
|
||||
{
|
||||
msg = ":coin: It's **Heads**!";
|
||||
} else {
|
||||
msg = "It's **Tails**! :coin:";
|
||||
}
|
||||
|
||||
event.getHook().sendMessage(msg)
|
||||
.addActionRow(Button.primary("reflip", "Flip again")
|
||||
.withEmoji(Emoji.fromFormatted("\uD83E\uDE99")) // coin emoji
|
||||
).queue();
|
||||
}
|
||||
|
||||
}
|
||||
|
@ -1,266 +0,0 @@
|
||||
package wtf.beatrice.hidekobot.database;
|
||||
|
||||
import wtf.beatrice.hidekobot.Configuration;
|
||||
import wtf.beatrice.hidekobot.utils.Logger;
|
||||
|
||||
import java.sql.*;
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.format.DateTimeFormatter;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
public class DatabaseManager
|
||||
{
|
||||
|
||||
private final static String sqliteURL = "jdbc:sqlite:%path%";
|
||||
private Connection dbConnection = null;
|
||||
private final String dbPath;
|
||||
private final Logger logger;
|
||||
|
||||
public DatabaseManager(String dbPath)
|
||||
{
|
||||
this.dbPath = dbPath;
|
||||
this.logger = new Logger(getClass());
|
||||
}
|
||||
|
||||
public boolean connect()
|
||||
{
|
||||
String url = sqliteURL.replace("%path%", dbPath);
|
||||
|
||||
if(!close()) return false;
|
||||
|
||||
try {
|
||||
dbConnection = DriverManager.getConnection(url);
|
||||
logger.log("Database connection established!");
|
||||
return true;
|
||||
} catch (SQLException e) {
|
||||
e.printStackTrace();
|
||||
return false;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
public boolean close()
|
||||
{
|
||||
if (dbConnection != null)
|
||||
{
|
||||
try {
|
||||
if(!dbConnection.isClosed())
|
||||
{
|
||||
dbConnection.close();
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
e.printStackTrace();
|
||||
return false;
|
||||
}
|
||||
|
||||
dbConnection = null;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
|
||||
public boolean initDb()
|
||||
{
|
||||
List<String> newTables = new ArrayList<>();
|
||||
|
||||
newTables.add("CREATE TABLE IF NOT EXISTS pending_disabled_messages (" +
|
||||
"guild_id TEXT NOT NULL, " +
|
||||
"channel_id TEXT NOT NULL," +
|
||||
"message_id TEXT NOT NULL," +
|
||||
"expiry_timestamp TEXT NOT NULL" +
|
||||
");");
|
||||
|
||||
newTables.add("CREATE TABLE IF NOT EXISTS command_runners (" +
|
||||
"guild_id TEXT NOT NULL, " +
|
||||
"channel_id TEXT NOT NULL," + // channel the command was run in
|
||||
"message_id TEXT NOT NULL," + // message id of the bot's response
|
||||
"user_id TEXT NOT NULL" + // user who ran the command
|
||||
");");
|
||||
|
||||
for(String sql : newTables)
|
||||
{
|
||||
try (Statement stmt = dbConnection.createStatement()) {
|
||||
// execute the statement
|
||||
stmt.execute(sql);
|
||||
} catch (SQLException e) {
|
||||
e.printStackTrace();
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
public boolean queueDisabling(String guildId, String channelId, String messageId)
|
||||
{
|
||||
LocalDateTime expiryTime = LocalDateTime.now().plusSeconds(Configuration.getExpiryTimeSeconds());
|
||||
|
||||
DateTimeFormatter dateTimeFormatter = DateTimeFormatter.ofPattern(Configuration.getExpiryTimestampFormat());
|
||||
String expiryTimeFormatted = dateTimeFormatter.format(expiryTime);
|
||||
|
||||
String query = "INSERT INTO pending_disabled_messages " +
|
||||
"(guild_id, channel_id, message_id, expiry_timestamp) VALUES " +
|
||||
" (?, ?, ?, ?);";
|
||||
|
||||
try(PreparedStatement preparedStatement = dbConnection.prepareStatement(query))
|
||||
{
|
||||
preparedStatement.setString(1, guildId);
|
||||
preparedStatement.setString(2, channelId);
|
||||
preparedStatement.setString(3, messageId);
|
||||
preparedStatement.setString(4, expiryTimeFormatted);
|
||||
|
||||
preparedStatement.executeUpdate();
|
||||
|
||||
return true;
|
||||
} catch (SQLException e)
|
||||
{
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
public List<String> getQueuedExpiringMessages()
|
||||
{
|
||||
List<String> messages = new ArrayList<>();
|
||||
|
||||
String query = "SELECT message_id " +
|
||||
"FROM pending_disabled_messages ";
|
||||
|
||||
try (Statement statement = dbConnection.createStatement())
|
||||
{
|
||||
ResultSet resultSet = statement.executeQuery(query);
|
||||
if(resultSet.isClosed()) return messages;
|
||||
while(resultSet.next())
|
||||
{
|
||||
messages.add(resultSet.getString("message_id"));
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
return messages;
|
||||
}
|
||||
|
||||
public boolean untrackExpiredMessage(String messageId)
|
||||
{
|
||||
String query = "DELETE FROM pending_disabled_messages WHERE message_id = ?;";
|
||||
|
||||
try(PreparedStatement preparedStatement = dbConnection.prepareStatement(query))
|
||||
{
|
||||
preparedStatement.setString(1, messageId);
|
||||
preparedStatement.execute();
|
||||
} catch (SQLException e)
|
||||
{
|
||||
e.printStackTrace();
|
||||
return false;
|
||||
}
|
||||
|
||||
query = "DELETE FROM command_runners WHERE message_id = ?;";
|
||||
try(PreparedStatement preparedStatement = dbConnection.prepareStatement(query))
|
||||
{
|
||||
preparedStatement.setString(1, messageId);
|
||||
preparedStatement.execute();
|
||||
} catch (SQLException e)
|
||||
{
|
||||
e.printStackTrace();
|
||||
return false;
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
public String getQueuedExpiringMessageExpiryDate(String messageId)
|
||||
{
|
||||
String query = "SELECT expiry_timestamp " +
|
||||
"FROM pending_disabled_messages " +
|
||||
"WHERE message_id = ?;";
|
||||
|
||||
try(PreparedStatement preparedStatement = dbConnection.prepareStatement(query))
|
||||
{
|
||||
preparedStatement.setString(1, messageId);
|
||||
ResultSet resultSet = preparedStatement.executeQuery();
|
||||
if(resultSet.isClosed()) return null;
|
||||
while(resultSet.next())
|
||||
{
|
||||
return resultSet.getString("expiry_timestamp");
|
||||
}
|
||||
|
||||
} catch (SQLException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
public String getQueuedExpiringMessageChannel(String messageId)
|
||||
{
|
||||
String query = "SELECT channel_id " +
|
||||
"FROM pending_disabled_messages " +
|
||||
"WHERE message_id = ?;";
|
||||
|
||||
try(PreparedStatement preparedStatement = dbConnection.prepareStatement(query))
|
||||
{
|
||||
preparedStatement.setString(1, messageId);
|
||||
ResultSet resultSet = preparedStatement.executeQuery();
|
||||
if(resultSet.isClosed()) return null;
|
||||
while(resultSet.next())
|
||||
{
|
||||
return resultSet.getString("channel_id");
|
||||
}
|
||||
|
||||
} catch (SQLException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
public String getQueuedExpiringMessageGuild(String messageId)
|
||||
{
|
||||
String query = "SELECT guild_id " +
|
||||
"FROM pending_disabled_messages " +
|
||||
"WHERE message_id = ?;";
|
||||
|
||||
try(PreparedStatement preparedStatement = dbConnection.prepareStatement(query))
|
||||
{
|
||||
preparedStatement.setString(1, messageId);
|
||||
ResultSet resultSet = preparedStatement.executeQuery();
|
||||
if(resultSet.isClosed()) return null;
|
||||
while(resultSet.next())
|
||||
{
|
||||
return resultSet.getString("guild_id");
|
||||
}
|
||||
|
||||
} catch (SQLException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
/**
|
||||
* DB STRUCTURE
|
||||
* TABLE 1: pending_disabled_messages
|
||||
* ----------------------------------------------------------------------------------
|
||||
* | guild_id | channel_id | message_id | expiry_timestamp |
|
||||
* ----------------------------------------------------------------------------------
|
||||
* |39402849302 | 39402849302 | 39402849302 | 2022-11-20 22:45:53:300 |
|
||||
* ----------------------------------------------------------------------------------
|
||||
*
|
||||
*
|
||||
* TABLE 2: command_runners
|
||||
* --------------------------------------------------------------------------
|
||||
* | guild_id | channel_id | message_id | user_id |
|
||||
* --------------------------------------------------------------------------
|
||||
* | 39402849302 | 39402849302 | 39402849302 | 39402849302 |
|
||||
* --------------------------------------------------------------------------
|
||||
*
|
||||
*/
|
||||
|
||||
|
||||
}
|
@ -2,7 +2,6 @@ package wtf.beatrice.hidekobot.listeners;
|
||||
|
||||
import net.dv8tion.jda.api.events.interaction.component.ButtonInteractionEvent;
|
||||
import net.dv8tion.jda.api.hooks.ListenerAdapter;
|
||||
import wtf.beatrice.hidekobot.commands.slash.ClearChatCommand;
|
||||
import wtf.beatrice.hidekobot.commands.slash.CoinFlipCommand;
|
||||
|
||||
public class ButtonInteractionListener extends ListenerAdapter
|
||||
@ -15,10 +14,7 @@ public class ButtonInteractionListener extends ListenerAdapter
|
||||
switch (event.getComponentId().toLowerCase()) {
|
||||
|
||||
// coinflip
|
||||
case "coinflip_reflip" -> new CoinFlipCommand().buttonReFlip(event);
|
||||
|
||||
// clearchat command
|
||||
case "clear_delete" -> new ClearChatCommand().deleteButton(event);
|
||||
case "reflip" -> new CoinFlipCommand().buttonReflip(event);
|
||||
|
||||
}
|
||||
|
||||
|
@ -1,99 +0,0 @@
|
||||
package wtf.beatrice.hidekobot.utils;
|
||||
|
||||
import net.dv8tion.jda.api.entities.Guild;
|
||||
import net.dv8tion.jda.api.entities.Message;
|
||||
import net.dv8tion.jda.api.entities.channel.concrete.TextChannel;
|
||||
import net.dv8tion.jda.api.interactions.components.LayoutComponent;
|
||||
import net.dv8tion.jda.api.requests.RestAction;
|
||||
import wtf.beatrice.hidekobot.Configuration;
|
||||
import wtf.beatrice.hidekobot.HidekoBot;
|
||||
import wtf.beatrice.hidekobot.database.DatabaseManager;
|
||||
|
||||
import java.time.LocalDateTime;
|
||||
import java.time.format.DateTimeFormatter;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
public class ExpiredMessageRunner implements Runnable {
|
||||
|
||||
private final DateTimeFormatter formatter;
|
||||
private final Logger logger;
|
||||
private DatabaseManager databaseManager;
|
||||
|
||||
|
||||
public ExpiredMessageRunner()
|
||||
{
|
||||
String format = Configuration.getExpiryTimestampFormat();
|
||||
formatter = DateTimeFormatter.ofPattern(format);
|
||||
databaseManager = Configuration.getDatabaseManager();
|
||||
logger = new Logger(getClass());
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
public void run() {
|
||||
|
||||
databaseManager = Configuration.getDatabaseManager();
|
||||
if(databaseManager == null) return;
|
||||
|
||||
List<String> expiringMessages = Configuration.getDatabaseManager().getQueuedExpiringMessages();
|
||||
if(expiringMessages == null || expiringMessages.isEmpty()) return;
|
||||
|
||||
LocalDateTime now = LocalDateTime.now();
|
||||
|
||||
for(String messageId : expiringMessages)
|
||||
{
|
||||
logger.log("ID: " + messageId);
|
||||
String expiryTimestamp = databaseManager.getQueuedExpiringMessageExpiryDate(messageId);
|
||||
if(expiryTimestamp == null || expiryTimestamp.equals("")) continue; //todo: idk count it as expired already?
|
||||
|
||||
|
||||
LocalDateTime expiryDate = LocalDateTime.parse(expiryTimestamp, formatter);
|
||||
if(now.isAfter(expiryDate))
|
||||
{
|
||||
disableExpired(messageId);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
}
|
||||
|
||||
private void disableExpired(String messageId)
|
||||
{
|
||||
String guildId = databaseManager.getQueuedExpiringMessageGuild(messageId);
|
||||
String channelId = databaseManager.getQueuedExpiringMessageChannel(messageId);
|
||||
|
||||
|
||||
|
||||
Guild guild = HidekoBot.getAPI().getGuildById(guildId);
|
||||
if(guild == null) return; //todo count it as done/solved/removed? we probably got kicked
|
||||
TextChannel textChannel = guild.getTextChannelById(channelId);
|
||||
if(textChannel == null) return; //todo count it as done/solved/removed? channel was probably deleted
|
||||
|
||||
RestAction<Message> retrieveAction = textChannel.retrieveMessageById(messageId);
|
||||
|
||||
|
||||
|
||||
retrieveAction.queue(
|
||||
|
||||
message -> {
|
||||
if(message == null) return; //todo count it as done/solved/removed? message was probably deleted
|
||||
|
||||
List<LayoutComponent> components = message.getComponents();
|
||||
List<LayoutComponent> newComponents = new ArrayList<>();
|
||||
for (LayoutComponent component : components)
|
||||
{
|
||||
component = component.asDisabled();
|
||||
newComponents.add(component);
|
||||
}
|
||||
|
||||
message.editMessageComponents(newComponents).queue();
|
||||
databaseManager.untrackExpiredMessage(messageId);
|
||||
},
|
||||
|
||||
(error) -> {
|
||||
databaseManager.untrackExpiredMessage(messageId);
|
||||
});
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue
Block a user