From 8a53111350b695b53ee9e7b790d3dea5c8dc09a2 Mon Sep 17 00:00:00 2001
From: stjn <stjn.d2000@hotmail.com>
Date: Wed, 3 Nov 2021 16:54:22 +0100
Subject: [PATCH] Added more untested queries

---
 .../com/alttd/commands/database/Queries.java  | 94 +++++++++++++++++++
 1 file changed, 94 insertions(+)
 create mode 100644 src/main/java/com/alttd/commands/database/Queries.java

diff --git a/src/main/java/com/alttd/commands/database/Queries.java b/src/main/java/com/alttd/commands/database/Queries.java
new file mode 100644
index 0000000..e9c6419
--- /dev/null
+++ b/src/main/java/com/alttd/commands/database/Queries.java
@@ -0,0 +1,94 @@
+package com.alttd.commands.database;
+
+import com.alttd.objects.VillagerType;
+import com.alttd.util.Logger;
+
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.UUID;
+
+public class Queries {
+    /** NOTE: run async
+     * Get the points a user has for a villager type
+     * @param uuid Uuid for the user you want to get the points for
+     * @param villagerType Type of villager you want to get the points for
+     * @return The amount of points a user has for the given villager type
+     */
+    public static int getUserPoints(UUID uuid, VillagerType villagerType) {
+        String sql = "SELECT points FROM user_points " +
+                "WHERE UUID = ? " +
+                "AND villager_type = ?;";
+        int points = 0;
+
+        try {
+            PreparedStatement preparedStatement = Database.connection.prepareStatement(sql);
+            preparedStatement.setString(1, uuid.toString());
+            preparedStatement.setString(2, villagerType.getName());
+
+            ResultSet resultSet = preparedStatement.executeQuery();
+            if (resultSet != null && resultSet.next()) {
+                points = resultSet.getInt("points");
+            }
+        } catch (SQLException e) {
+            e.printStackTrace();
+        }
+        return (points);
+    }
+
+    /** NOTE: run async
+     * Add a specified amount of points to the user for the given villager type
+     * @param uuid Uuid for the user you want to add the points to
+     * @param villagerType Type of villager you want to add the points to
+     * @param points The amount of points to add
+     * @return success
+     */
+    public static boolean updateUserPoints(UUID uuid, VillagerType villagerType, int points) {
+        String sql = "UPDATE Points = SET user_points = user_points + ? " +
+                "WHERE UUID = ? " +
+                "AND villager_type = ?;";
+
+        try {
+            PreparedStatement preparedStatement = Database.connection.prepareStatement(sql);
+            preparedStatement.setInt(1, points);
+            preparedStatement.setString(2, uuid.toString());
+            preparedStatement.setString(3, villagerType.getName());
+
+            if (preparedStatement.executeUpdate() == 0)
+                return createUserPointsEntry(uuid, villagerType, points);
+        } catch (SQLException e) {
+            e.printStackTrace();
+            Logger.warning("Unable to add % points to %" +
+                    " for villager type %", String.valueOf(points), uuid.toString(), villagerType.getName());
+            return (false);
+        }
+        return (true);
+    }
+
+    /** NOTE: run async
+     * Create a new user entry
+     * @param uuid Uuid for the user you want to create an entry for
+     * @param villagerType Type of villager you want to use in that entry
+     * @param points The amount of points to set the start to
+     * @return success
+     */
+    public static boolean createUserPointsEntry(UUID uuid, VillagerType villagerType, int points) {
+        String sql = "INSERT INTO Points " +
+                "(uuid, villager_type, points) " +
+                "(?, ?, ?)";
+
+        try {
+            PreparedStatement preparedStatement = Database.connection.prepareStatement(sql);
+            preparedStatement.setString(1, uuid.toString());
+            preparedStatement.setString(2, villagerType.getName());
+            preparedStatement.setInt(3, points);
+
+        } catch (SQLException e) {
+            e.printStackTrace();
+            Logger.warning("Unable to create point entry for %" +
+                    " for villager type %", uuid.toString(), villagerType.getName());
+            return (false);
+        }
+        return (true);
+    }
+}