HidekoBot/src/main/java/wtf/beatrice/hidekobot/database/DatabaseManager.java

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    |
     * --------------------------------------------------------------------------
     *
     */


}