misc/chomens-bot-java
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

BIG feat,fix,refactor: use database and some minor refactoring which also fixes memory issues and * related to PersistentDataUtilities

Browse Source
This commit is contained in:
ChomeNS
2024-12-14 15:39:24 +07:00
parent d09b9b37db
commit 621f06f2e5
18 changed files with 656 additions and 530 deletions
Download Patch File Download Diff File Expand all files Collapse all files

130
src/main/java/me/chayapak1/chomens_bot/plugins/MailPlugin.java
View File

@@ -1,80 +1,126 @@
package me.chayapak1.chomens_bot.plugins;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.JsonNodeFactory;
import me.chayapak1.chomens_bot.Bot;
import me.chayapak1.chomens_bot.Main;
import me.chayapak1.chomens_bot.data.Mail;
import me.chayapak1.chomens_bot.data.PlayerEntry;
import me.chayapak1.chomens_bot.util.ColorUtilities;
import me.chayapak1.chomens_bot.util.PersistentDataUtilities;
import net.kyori.adventure.text.Component;
import net.kyori.adventure.text.format.NamedTextColor;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
public class MailPlugin extends PlayersPlugin.Listener {
public static final ObjectMapper objectMapper = new ObjectMapper();
private static final String CREATE_TABLE = "CREATE TABLE IF NOT EXISTS mails (sentBy VARCHAR(255), sentTo VARCHAR(255), timeSent BIGINT, server VARCHAR(255), contents TEXT);";
private static final String INSERT_MAIL = "INSERT INTO mails (sentBy, sentTo, timeSent, server, contents) VALUES (?, ?, ?, ?, ?);";
private static final String LIST_MAILS = "SELECT * FROM mails;";
private static final String REMOVE_MAIL = "DELETE FROM mails WHERE sentTo = ?;";
static {
if (Main.database != null) {
DatabasePlugin.executorService.submit(() -> {
try {
Main.database.execute(CREATE_TABLE);
} catch (SQLException e) {
e.printStackTrace();
}
});
}
}
private final Bot bot;
public static ArrayNode mails = (ArrayNode) PersistentDataUtilities.getOrDefault("mails", JsonNodeFactory.instance.arrayNode());
public MailPlugin (Bot bot) {
this.bot = bot;
if (Main.database == null) return;
bot.players.addListener(this);
}
@Override
public void playerJoined(PlayerEntry target) {
final String name = target.profile.getName();
DatabasePlugin.executorService.submit(() -> {
final String name = target.profile.getName();
int sendToTargetSize = 0;
int sendToTargetSize = 0;
boolean shouldSend = false;
for (JsonNode mailNode : mails.deepCopy()) {
try {
final Mail mail = objectMapper.treeToValue(mailNode, Mail.class);
final List<Mail> mails = list();
for (Mail mail : mails) {
if (!mail.sentTo.equals(name)) continue;
shouldSend = true;
sendToTargetSize++;
} catch (JsonProcessingException e) {
e.printStackTrace();
}
}
if (shouldSend) {
final Component component = Component.translatable(
"You have %s new mail%s!\n" +
"Do %s or %s to read",
Component.text(sendToTargetSize).color(NamedTextColor.GREEN),
Component.text((sendToTargetSize > 1) ? "s" : ""),
Component.text(bot.config.commandSpyPrefixes.get(0) + "mail read").color(ColorUtilities.getColorByString(bot.config.colorPalette.primary)),
Component.text(bot.config.prefixes.get(0) + "mail read").color(ColorUtilities.getColorByString(bot.config.colorPalette.primary))
).color(NamedTextColor.GOLD);
if (sendToTargetSize > 0) {
final Component component = Component.translatable(
"You have %s new mail%s!\n" +
"Run %s or %s to read",
Component.text(sendToTargetSize).color(NamedTextColor.GREEN),
Component.text((sendToTargetSize > 1) ? "s" : ""),
Component.text(bot.config.commandSpyPrefixes.get(0) + "mail read").color(ColorUtilities.getColorByString(bot.config.colorPalette.primary)),
Component.text(bot.config.prefixes.get(0) + "mail read").color(ColorUtilities.getColorByString(bot.config.colorPalette.primary))
).color(NamedTextColor.GOLD);
bot.chat.tellraw(component, target.profile.getId());
}
bot.chat.tellraw(component, target.profile.getId());
}
});
}
public void send (Mail mail) {
mails.add(objectMapper.valueToTree(mail));
try {
final PreparedStatement statement = bot.database.connection.prepareStatement(INSERT_MAIL);
PersistentDataUtilities.put("mails", mails);
}
statement.setString(1, mail.sentBy);
statement.setString(2, mail.sentTo);
statement.setLong(3, mail.timeSent);
statement.setString(4, mail.server);
statement.setString(5, mail.contents);
public void remove (JsonNode mail) {
for (int i = 0; i < mails.size(); i++) {
final JsonNode currentNode = mails.get(i);
if (currentNode.equals(mail)) {
mails.remove(i);
break;
}
statement.executeUpdate();
} catch (SQLException e) {
e.printStackTrace();
}
}
public void clear (String sentTo) {
try {
final PreparedStatement statement = bot.database.connection.prepareStatement(REMOVE_MAIL);
statement.setString(1, sentTo);
statement.executeUpdate();
} catch (SQLException e) {
e.printStackTrace();
}
}
public List<Mail> list () {
final List<Mail> output = new ArrayList<>();
try (ResultSet result = bot.database.query(LIST_MAILS)) {
if (result == null) return output;
while (result.next()) {
final Mail mail = new Mail(
result.getString("sentBy"),
result.getString("sentTo"),
result.getLong("timeSent"),
result.getString("server"),
result.getString("contents")
);
output.add(mail);
}
} catch (SQLException e) {
e.printStackTrace();
}
return output;
}
}