593 lines
19 KiB
Java
593 lines
19 KiB
Java
package wtf.beatrice.hidekobot.datasources;
|
|
|
|
import net.dv8tion.jda.api.entities.Message;
|
|
import net.dv8tion.jda.api.entities.User;
|
|
import net.dv8tion.jda.api.entities.channel.ChannelType;
|
|
import org.slf4j.LoggerFactory;
|
|
import wtf.beatrice.hidekobot.Cache;
|
|
|
|
import java.sql.*;
|
|
import java.time.LocalDateTime;
|
|
import java.time.format.DateTimeFormatter;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
public class DatabaseSource {
|
|
|
|
private static final org.slf4j.Logger LOGGER = LoggerFactory.getLogger(DatabaseSource.class);
|
|
private static final String JDBC_URL = "jdbc:sqlite:%path%";
|
|
private Connection dbConnection = null;
|
|
private final String dbPath;
|
|
|
|
public DatabaseSource(String dbPath) {
|
|
this.dbPath = dbPath;
|
|
}
|
|
|
|
private void logException(SQLException e) {
|
|
LOGGER.error("Database Exception", e);
|
|
}
|
|
|
|
public boolean connect() {
|
|
String url = JDBC_URL.replace("%path%", dbPath);
|
|
|
|
if (!close()) return false;
|
|
|
|
try {
|
|
dbConnection = DriverManager.getConnection(url);
|
|
LOGGER.info("Database connection established!");
|
|
return true;
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
return false;
|
|
}
|
|
|
|
}
|
|
|
|
|
|
public boolean close() {
|
|
if (dbConnection != null) {
|
|
try {
|
|
if (!dbConnection.isClosed()) {
|
|
dbConnection.close();
|
|
}
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
return false;
|
|
}
|
|
|
|
dbConnection = null;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
/*
|
|
* 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 | channel_type |
|
|
* --------------------------------------------------------------------------------------------
|
|
* | 39402849302 | 39402849302 | 39402849302 | 39402849302 | PRIVATE |
|
|
* --------------------------------------------------------------------------------------------
|
|
*
|
|
* TABLE 3: urban_dictionary
|
|
* -----------------------------------------------------------------------------------------------------
|
|
* | message_id | page | meanings | examples | contributors | dates | term |
|
|
* -----------------------------------------------------------------------------------------------------
|
|
* | 39402849302 | 0 | base64 | base64 | base64 | base64 | miku |
|
|
* -----------------------------------------------------------------------------------------------------
|
|
*/
|
|
|
|
//todo: javadocs
|
|
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,
|
|
message_id TEXT NOT NULL,
|
|
user_id TEXT NOT NULL,
|
|
channel_type TEXT NOT NULL);
|
|
""");
|
|
|
|
newTables.add("""
|
|
CREATE TABLE IF NOT EXISTS urban_dictionary (
|
|
message_id TEXT NOT NULL,
|
|
page INTEGER NOT NULL,
|
|
meanings TEXT NOT NULL,
|
|
examples TEXT NOT NULL,
|
|
contributors TEXT NOT NULL,
|
|
dates TEXT NOT NULL,
|
|
term TEXT NOT NULL
|
|
);
|
|
""");
|
|
|
|
for (String sql : newTables) {
|
|
try (Statement stmt = dbConnection.createStatement()) {
|
|
// execute the statement
|
|
stmt.execute(sql);
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
public boolean trackRanCommandReply(Message message, User user) {
|
|
String userId = user.getId();
|
|
String guildId;
|
|
|
|
ChannelType channelType = message.getChannelType();
|
|
if (!(channelType.isGuild())) {
|
|
guildId = userId;
|
|
} else {
|
|
guildId = message.getGuild().getId();
|
|
}
|
|
|
|
String channelId = message.getChannel().getId();
|
|
String messageId = message.getId();
|
|
|
|
|
|
String query = """
|
|
INSERT INTO command_runners
|
|
(guild_id, channel_id, message_id, user_id, channel_type) VALUES
|
|
(?, ?, ?, ?, ?);
|
|
""";
|
|
|
|
try (PreparedStatement preparedStatement = dbConnection.prepareStatement(query)) {
|
|
preparedStatement.setString(1, guildId);
|
|
preparedStatement.setString(2, channelId);
|
|
preparedStatement.setString(3, messageId);
|
|
preparedStatement.setString(4, userId);
|
|
preparedStatement.setString(5, channelType.name());
|
|
|
|
preparedStatement.executeUpdate();
|
|
|
|
return true;
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
public boolean isUserTrackedFor(String userId, String messageId) {
|
|
String trackedUserId = getTrackedReplyUserId(messageId);
|
|
if (trackedUserId == null) return false;
|
|
return userId.equals(trackedUserId);
|
|
}
|
|
|
|
public ChannelType getTrackedMessageChannelType(String messageId) {
|
|
String query = """
|
|
""SELECT channel_type
|
|
FROM command_runners
|
|
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()) {
|
|
String channelTypeName = resultSet.getString("channel_type");
|
|
return ChannelType.valueOf(channelTypeName);
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
}
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
public String getTrackedReplyUserId(String messageId) {
|
|
String query = """
|
|
""SELECT user_id
|
|
FROM command_runners
|
|
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("user_id");
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
public boolean queueDisabling(Message message) {
|
|
String messageId = message.getId();
|
|
String channelId = message.getChannel().getId();
|
|
String guildId;
|
|
|
|
ChannelType channelType = message.getChannelType();
|
|
if (!(channelType.isGuild())) {
|
|
guildId = "PRIVATE";
|
|
} else {
|
|
guildId = message.getGuild().getId();
|
|
}
|
|
|
|
LocalDateTime expiryTime = LocalDateTime.now().plusSeconds(Cache.getExpiryTimeSeconds());
|
|
|
|
DateTimeFormatter dateTimeFormatter = DateTimeFormatter.ofPattern(Cache.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) {
|
|
logException(e);
|
|
}
|
|
|
|
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) {
|
|
logException(e);
|
|
}
|
|
|
|
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) {
|
|
logException(e);
|
|
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) {
|
|
logException(e);
|
|
return false;
|
|
}
|
|
|
|
query = "DELETE FROM urban_dictionary WHERE message_id = ?;";
|
|
try (PreparedStatement preparedStatement = dbConnection.prepareStatement(query)) {
|
|
preparedStatement.setString(1, messageId);
|
|
preparedStatement.execute();
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
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) {
|
|
logException(e);
|
|
}
|
|
|
|
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) {
|
|
logException(e);
|
|
}
|
|
|
|
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) {
|
|
logException(e);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
public boolean trackUrban(String meanings, String examples,
|
|
String contributors, String dates,
|
|
Message message, String term) {
|
|
|
|
String query = """
|
|
INSERT INTO urban_dictionary
|
|
(message_id, page, meanings, examples, contributors, dates, term) VALUES
|
|
(?, ?, ?, ?, ?, ?, ?);
|
|
""";
|
|
|
|
try (PreparedStatement preparedStatement = dbConnection.prepareStatement(query)) {
|
|
preparedStatement.setString(1, message.getId());
|
|
preparedStatement.setInt(2, 0);
|
|
preparedStatement.setString(3, meanings);
|
|
preparedStatement.setString(4, examples);
|
|
preparedStatement.setString(5, contributors);
|
|
preparedStatement.setString(6, dates);
|
|
preparedStatement.setString(7, term);
|
|
|
|
preparedStatement.executeUpdate();
|
|
|
|
return true;
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
public int getUrbanPage(String messageId) {
|
|
String query = """
|
|
SELECT page
|
|
FROM urban_dictionary
|
|
WHERE message_id = ?;
|
|
""";
|
|
|
|
try (PreparedStatement preparedStatement = dbConnection.prepareStatement(query)) {
|
|
preparedStatement.setString(1, messageId);
|
|
ResultSet resultSet = preparedStatement.executeQuery();
|
|
if (resultSet.isClosed()) return 0;
|
|
while (resultSet.next()) {
|
|
return resultSet.getInt("page");
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
public String getUrbanMeanings(String messageId) {
|
|
String query = """
|
|
SELECT meanings
|
|
FROM urban_dictionary
|
|
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("meanings");
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
public String getUrbanExamples(String messageId) {
|
|
String query = """
|
|
SELECT examples
|
|
FROM urban_dictionary
|
|
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("examples");
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
public String getUrbanContributors(String messageId) {
|
|
String query = """
|
|
SELECT contributors
|
|
FROM urban_dictionary
|
|
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("contributors");
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
public String getUrbanDates(String messageId) {
|
|
String query = """
|
|
SELECT dates
|
|
FROM urban_dictionary
|
|
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("dates");
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
public String getUrbanTerm(String messageId) {
|
|
String query = """
|
|
SELECT term
|
|
FROM urban_dictionary
|
|
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("term");
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
public boolean setUrbanPage(String messageId, int page) {
|
|
String query = """
|
|
UPDATE urban_dictionary
|
|
SET page = ?
|
|
WHERE message_id = ?;
|
|
""";
|
|
|
|
try (PreparedStatement preparedStatement = dbConnection.prepareStatement(query)) {
|
|
preparedStatement.setInt(1, page);
|
|
preparedStatement.setString(2, messageId);
|
|
preparedStatement.executeUpdate();
|
|
|
|
return true;
|
|
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
public boolean resetExpiryTimestamp(String messageId) {
|
|
LocalDateTime expiryTime = LocalDateTime.now().plusSeconds(Cache.getExpiryTimeSeconds());
|
|
|
|
DateTimeFormatter dateTimeFormatter = DateTimeFormatter.ofPattern(Cache.getExpiryTimestampFormat());
|
|
String expiryTimeFormatted = dateTimeFormatter.format(expiryTime);
|
|
|
|
String query = """
|
|
UPDATE pending_disabled_messages
|
|
SET expiry_timestamp = ?
|
|
WHERE message_id = ?;
|
|
""";
|
|
|
|
try (PreparedStatement preparedStatement = dbConnection.prepareStatement(query)) {
|
|
preparedStatement.setString(1, expiryTimeFormatted);
|
|
preparedStatement.setString(2, messageId);
|
|
preparedStatement.executeUpdate();
|
|
|
|
return true;
|
|
|
|
} catch (SQLException e) {
|
|
logException(e);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
|
|
}
|