1 files changed, 107 insertions, 0 deletions
diff --git a/src/main/java/dev/tylerm/khs/database/Database.java b/src/main/java/dev/tylerm/khs/database/Database.java
new file mode 100644
index 0000000..42c4798
--- /dev/null
+++ b/src/main/java/dev/tylerm/khs/database/Database.java
@@ -0,0 +1,107 @@
+package dev.tylerm.khs.database;
+
+import java.io.ByteArrayInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.nio.Buffer;
+import java.nio.ByteBuffer;
+import java.sql.Connection;
+import java.sql.SQLException;
+import java.util.UUID;
+
+import com.google.common.io.ByteStreams;
+
+import dev.tylerm.khs.Main;
+import dev.tylerm.khs.configuration.Config;
+import dev.tylerm.khs.database.connections.DatabaseConnection;
+import dev.tylerm.khs.database.connections.MySQLConnection;
+import dev.tylerm.khs.database.connections.SQLiteConnection;
+
+public class Database {
+
+    private final GameDataTable playerInfo;
+    private final NameDataTable nameInfo;
+    private final InventoryTable inventoryInfo;
+    private final DatabaseConnection connection;
+
+    public Database(){
+
+        if(Config.databaseType.equalsIgnoreCase("SQLITE")) {
+            Main.getInstance().getLogger().info("SQLITE database chosen");
+            connection = new SQLiteConnection();
+        } else if(Config.databaseType.equalsIgnoreCase("MYSQL")) {
+            Main.getInstance().getLogger().info("MYSQL database chosen");
+            connection = new MySQLConnection();
+        } else {
+            throw new IllegalArgumentException("Invalid database type: " + Config.databaseType);
+        }
+
+        playerInfo = new GameDataTable(this);
+
+        nameInfo = new NameDataTable(this);
+
+        inventoryInfo = new InventoryTable(this);
+
+        LegacyTable legacyTable = new LegacyTable(this);
+        if(legacyTable.exists()){
+            if(legacyTable.copyData()){
+                if(!legacyTable.drop()){
+                    Main.getInstance().getLogger().severe("Failed to drop old legacy table: player_info. Some data may be duplicated!");
+                }
+            }
+        }
+    }
+
+    public GameDataTable getGameData(){
+        return playerInfo;
+    }
+
+    public NameDataTable getNameData() { return nameInfo; }
+
+    public InventoryTable getInventoryData() { return inventoryInfo; }
+
+    protected Connection connect() {
+        Connection conn = null;
+        try {
+            conn = connection.connect();
+        } catch (SQLException e) {
+            Main.getInstance().getLogger().severe(e.getMessage());
+            e.printStackTrace();
+        }
+        return conn;
+    }
+
+    protected byte[] encodeUUID(UUID uuid) {
+        try {
+            byte[] bytes = new byte[16];
+            ByteBuffer.wrap(bytes)
+                    .putLong(uuid.getMostSignificantBits())
+                    .putLong(uuid.getLeastSignificantBits());
+            InputStream is = new ByteArrayInputStream(bytes);
+            byte[] result = new byte[is.available()];
+            if (is.read(result) == -1) {
+                Main.getInstance().getLogger().severe("IO Error: Failed to read bytes from input stream");
+                return new byte[0];
+            }
+            return result;
+        } catch (IOException e) {
+            Main.getInstance().getLogger().severe("IO Error: " + e.getMessage());
+            return new byte[0];
+        }
+    }
+
+    @SuppressWarnings("UnstableApiUsage")
+    protected UUID decodeUUID(byte[] bytes) {
+        InputStream is = new ByteArrayInputStream(bytes);
+        ByteBuffer buffer = ByteBuffer.allocate(16);
+        try {
+            buffer.put(ByteStreams.toByteArray(is));
+            ((Buffer)buffer).flip();
+            return new UUID(buffer.getLong(), buffer.getLong());
+        } catch (IOException e) {
+            Main.getInstance().getLogger().severe("IO Error: " + e.getMessage());
+        }
+        return null;
+    }
+
+}