372 lines
13 KiB
Java
372 lines
13 KiB
Java
package com.alttd.chat.database;
|
|
|
|
import com.alttd.chat.data.ChatUser;
|
|
import com.alttd.chat.data.Party;
|
|
import com.alttd.chat.data.ServerWrapper;
|
|
import com.alttd.chat.handlers.ChatUserManager;
|
|
import com.alttd.chat.util.ALogger;
|
|
|
|
import java.sql.Connection;
|
|
import java.sql.PreparedStatement;
|
|
import java.sql.ResultSet;
|
|
import java.sql.SQLException;
|
|
import java.util.ArrayList;
|
|
import java.util.HashMap;
|
|
import java.util.List;
|
|
import java.util.UUID;
|
|
|
|
public class Queries {
|
|
|
|
//Create Tables
|
|
|
|
public static void createTables() {
|
|
List<String> tables = new ArrayList<>();
|
|
tables.add("CREATE TABLE IF NOT EXISTS ignored_users (`uuid` VARCHAR(36) NOT NULL, `ignored_uuid` VARCHAR(36) NOT NULL, PRIMARY KEY (`uuid`, `ignored_uuid`))");
|
|
tables.add("CREATE TABLE IF NOT EXISTS parties (`id` INT NOT NULL AUTO_INCREMENT, `owner_uuid` VARCHAR(36) NOT NULL, `party_name` VARCHAR(36) NOT NULL, `password` VARCHAR(36), PRIMARY KEY (`id`))");
|
|
tables.add("CREATE TABLE IF NOT EXISTS chat_users (`uuid` VARCHAR(36) NOT NULL, `party_id` INT NOT NULL, `toggled_chat` BIT(1) DEFAULT b'0', `force_tp` BIT(1) DEFAULT b'1', `toggled_gc` BIT(1) DEFAULT b'0', PRIMARY KEY (`uuid`))");
|
|
|
|
try {
|
|
Connection connection = DatabaseConnection.getConnection();
|
|
|
|
for (String query : tables) {
|
|
connection.prepareStatement(query).execute();
|
|
}
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------
|
|
|
|
//Nicknames
|
|
|
|
public static String getNickname(UUID uuid) {
|
|
// View has been created.
|
|
String query = "SELECT nickname FROM nicknames WHERE uuid = ?";
|
|
|
|
try {
|
|
Connection connection = DatabaseConnection.getConnection();
|
|
|
|
PreparedStatement statement = connection.prepareStatement(query);
|
|
|
|
statement.setString(1, uuid.toString());
|
|
|
|
ResultSet resultSet = statement.executeQuery();
|
|
|
|
if (resultSet.next()) {
|
|
return resultSet.getString("nickname");
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
return null;
|
|
}
|
|
|
|
//-----------------------------------------
|
|
|
|
//Ignore
|
|
|
|
public static void getIgnoredUsers() { //TODO store this in a user object or something? -> ChatPlayer
|
|
HashMap<UUID, ArrayList<UUID>> ignoredUsers = new HashMap<>(); //TODO Replace with a proper way/location to store this in -> a list<UUID> in ChatPlayer?
|
|
String query = "SELECT * FROM ignored_users";
|
|
|
|
try {
|
|
Connection connection = DatabaseConnection.getConnection();
|
|
|
|
ResultSet resultSet = connection.prepareStatement(query).executeQuery();
|
|
|
|
while (resultSet.next()) {
|
|
UUID uuid = UUID.fromString(resultSet.getString("uuid"));
|
|
UUID ignoredUuid = UUID.fromString(resultSet.getString("ignored_uuid"));
|
|
ArrayList<UUID> uuids;
|
|
|
|
if (ignoredUsers.containsKey(uuid)) {
|
|
uuids = ignoredUsers.get(uuid);
|
|
} else {
|
|
uuids = new ArrayList<>();
|
|
}
|
|
|
|
uuids.add(ignoredUuid);
|
|
ignoredUsers.put(uuid, uuids);
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public static void ignoreUser(UUID uuid, UUID ignoredUuid) {
|
|
//TODO this should be called from a function that already adds the user to the live ignored list, so this just adds it to the database
|
|
|
|
String query = "INSERT INTO ignored_users (uuid, ignored_uuid) VALUES (?, ?)";
|
|
|
|
try {
|
|
Connection connection = DatabaseConnection.getConnection();
|
|
PreparedStatement statement = connection.prepareStatement(query);
|
|
|
|
statement.setString(1, uuid.toString());
|
|
statement.setString(2, ignoredUuid.toString());
|
|
|
|
statement.execute();
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public static void unIgnoreUser(UUID uuid, UUID ignoredUuid) {
|
|
//TODO this should be called from a function that already removes the user from the live ignored list, so this just removes it from the database
|
|
|
|
String query = "DELETE FROM ignored_users WHERE uuid = ? AND ignored_uuid = ?";
|
|
|
|
try {
|
|
Connection connection = DatabaseConnection.getConnection();
|
|
PreparedStatement statement = connection.prepareStatement(query);
|
|
|
|
statement.setString(1, uuid.toString());
|
|
statement.setString(2, ignoredUuid.toString());
|
|
|
|
statement.execute();
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------
|
|
|
|
//Party
|
|
|
|
public static void getParties() {
|
|
HashMap<Integer, Party> parties = new HashMap<>(); //TODO Replace with a proper way/location to store this in
|
|
String query = "SELECT * FROM parties";
|
|
|
|
try {
|
|
Connection connection = DatabaseConnection.getConnection();
|
|
|
|
ResultSet resultSet = connection.prepareStatement(query).executeQuery();
|
|
|
|
while (resultSet.next()) {
|
|
|
|
int id = resultSet.getInt("id");
|
|
UUID ownerUuid = UUID.fromString(resultSet.getString("owner_uuid"));
|
|
String partyName = resultSet.getString("party_name");
|
|
String password = resultSet.getString("password");
|
|
|
|
parties.put(id, new Party(id, ownerUuid, partyName, password));
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
|
|
getChatUsers(parties); //TODO This parameter should be temporary, it should access the actual list of parties normally
|
|
}
|
|
|
|
public static Party addParty(UUID partyOwner, String partyName, String password) {
|
|
String query = "INSERT INTO parties (owner_uuid, party_name, password) VALUES (?, ?, ?)";
|
|
|
|
try {
|
|
Connection connection = DatabaseConnection.getConnection();
|
|
PreparedStatement statement = connection.prepareStatement(query);
|
|
|
|
statement.setString(1, partyOwner.toString());
|
|
statement.setString(2, partyName);
|
|
statement.setString(3, password.isEmpty() ? null : password);
|
|
|
|
statement.execute();
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
|
|
return getParty(partyOwner, partyName, password);
|
|
}
|
|
|
|
private static Party getParty(UUID partyOwner, String partyName, String password) {
|
|
String query = "SELECT * FROM parties WHERE owner_uuid = ? AND party_name = ?";
|
|
|
|
try {
|
|
Connection connection = DatabaseConnection.getConnection();
|
|
|
|
PreparedStatement statement = connection.prepareStatement(query);
|
|
|
|
statement.setString(1, partyOwner.toString());
|
|
statement.setString(2, partyName);
|
|
|
|
ResultSet resultSet = statement.executeQuery();
|
|
|
|
if (resultSet.next()) {
|
|
int id = resultSet.getInt("id");
|
|
|
|
return new Party(id, partyOwner, partyName, password);
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
public static void setPartyOwner(UUID uuid, int id) {
|
|
setStringWhereId("UPDATE parties set owner_uuid = ? WHERE id = ?", uuid.toString(), id);
|
|
}
|
|
|
|
public static void setPartyName(String name, int id) {
|
|
setStringWhereId("UPDATE parties set party_name = ? WHERE id = ?", name, id);
|
|
}
|
|
|
|
public static void setPartyPassword(String password, int id) {
|
|
setStringWhereId("UPDATE parties set password = ? WHERE id = ?", password, id);
|
|
}
|
|
|
|
private static void setStringWhereId(String query, String string, int id) {
|
|
try {
|
|
Connection connection = DatabaseConnection.getConnection();
|
|
PreparedStatement statement = connection.prepareStatement(query);
|
|
|
|
statement.setString(1, string);
|
|
statement.setInt(2, id);
|
|
|
|
statement.execute();
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public static void removeParty(int id) {
|
|
String query = "DELETE FROM parties WHERE id = ?";
|
|
|
|
try {
|
|
Connection connection = DatabaseConnection.getConnection();
|
|
PreparedStatement statement = connection.prepareStatement(query);
|
|
|
|
statement.setInt(1, id);
|
|
|
|
statement.execute();
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------
|
|
|
|
private static void getChatUsers(HashMap<Integer, Party> parties) { //TODO Get parties from cache somewhere
|
|
String query = "SELECT * FROM chat_users";
|
|
|
|
try {
|
|
Connection connection = DatabaseConnection.getConnection();
|
|
|
|
ResultSet resultSet = connection.prepareStatement(query).executeQuery();
|
|
|
|
while (resultSet.next()) {
|
|
|
|
UUID uuid = UUID.fromString(resultSet.getString("uuid"));
|
|
int partyId = resultSet.getInt("party_id");
|
|
boolean toggled_chat = resultSet.getInt("toggled_chat") == 1;
|
|
boolean force_tp = resultSet.getInt("force_tp") == 1;
|
|
boolean toggle_Gc = resultSet.getInt("toggled_gc") == 1;
|
|
|
|
if (partyId == 0) {
|
|
continue;
|
|
}
|
|
|
|
Party party = parties.get(partyId);
|
|
|
|
if (party == null) {
|
|
//TODO log this properly
|
|
ALogger.error("INCORRECT LOGGING: party was empty, the party id stored in the database with user " + uuid + " was invalid.");
|
|
System.out.println("INCORRECT LOGGING: party was empty, the party id stored in the database with user " + uuid + " was invalid.");
|
|
continue;
|
|
}
|
|
|
|
party.addUser(new ChatUser(uuid, partyId, toggled_chat, force_tp, toggle_Gc));
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public static void loadChatUsers() { //TODO Get parties from cache somewhere
|
|
String query = "SELECT * FROM chat_users";
|
|
|
|
try {
|
|
Connection connection = DatabaseConnection.getConnection();
|
|
|
|
ResultSet resultSet = connection.prepareStatement(query).executeQuery();
|
|
|
|
while (resultSet.next()) {
|
|
|
|
UUID uuid = UUID.fromString(resultSet.getString("uuid"));
|
|
int partyId = resultSet.getInt("party_id");
|
|
boolean toggled_chat = resultSet.getInt("toggled_chat") == 1;
|
|
boolean force_tp = resultSet.getInt("force_tp") == 1;
|
|
boolean toggle_Gc = resultSet.getInt("toggled_gc") == 1;
|
|
// could do a constructor for chatuser to accept the record?
|
|
ChatUserManager.addUser(new ChatUser(uuid, partyId, toggled_chat, force_tp, toggle_Gc));
|
|
}
|
|
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public static void addUser(ChatUser user) {
|
|
String query = "INSERT INTO chat_users (uuid, party_id, toggled_chat, force_tp, toggled_gc) VALUES (?, ?, ?, ?, ?)";
|
|
|
|
try {
|
|
Connection connection = DatabaseConnection.getConnection();
|
|
PreparedStatement statement = connection.prepareStatement(query);
|
|
|
|
statement.setString(1, user.getUuid().toString());
|
|
statement.setInt(2, user.getPartyId());
|
|
statement.setInt(3, user.toggledPartyChat() ? 1 : 0);
|
|
statement.setInt(4, user.ForceTp() ? 1 : 0);
|
|
statement.setInt(5, user.isGcOn() ? 1 : 0);
|
|
|
|
statement.execute();
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public static void setPartyChatState(boolean toggledChat, UUID uuid) {
|
|
setBitWhereId("UPDATE chat_users set toggled_chat = ? WHERE uuid = ?", toggledChat, uuid);
|
|
}
|
|
|
|
public static void setForceTpState(boolean forceTp, UUID uuid) {
|
|
setBitWhereId("UPDATE chat_users set force_tp = ? WHERE uuid = ?", forceTp, uuid);
|
|
}
|
|
|
|
private static void setBitWhereId(String query, boolean bool, UUID uuid) {
|
|
try {
|
|
Connection connection = DatabaseConnection.getConnection();
|
|
PreparedStatement statement = connection.prepareStatement(query);
|
|
|
|
statement.setInt(1, bool ? 1 : 0);
|
|
statement.setString(2, uuid.toString());
|
|
|
|
statement.execute();
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
public static void removeUser(UUID uuid) {
|
|
String query = "DELETE FROM chat_users WHERE uuid = ?";
|
|
|
|
try {
|
|
Connection connection = DatabaseConnection.getConnection();
|
|
PreparedStatement statement = connection.prepareStatement(query);
|
|
|
|
statement.setString(1, uuid.toString());
|
|
|
|
statement.execute();
|
|
} catch (SQLException e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------
|
|
}
|