From 57b40ee91dcae3628cd706d11833fa560f85f11b Mon Sep 17 00:00:00 2001
From: BenCodez <benbergen12@gmail.com>
Date: Sun, 30 Apr 2023 16:52:53 -0400
Subject: [PATCH] Fix minor issues with FLAT

---
 .../src/com/bencodez/advancedcore/api/user/UserData.java  | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/AdvancedCore/src/com/bencodez/advancedcore/api/user/UserData.java b/AdvancedCore/src/com/bencodez/advancedcore/api/user/UserData.java
index 4ddb3d0e87..3f2beaed89 100644
--- a/AdvancedCore/src/com/bencodez/advancedcore/api/user/UserData.java
+++ b/AdvancedCore/src/com/bencodez/advancedcore/api/user/UserData.java
@@ -207,9 +207,12 @@ public ArrayList<String> getKeys(boolean waitForCache) {
 		return keys;
 	}
 
+	@SuppressWarnings("deprecation")
 	public ArrayList<String> getKeys(UserStorage storage, boolean waitForCache) {
 		ArrayList<String> keys = new ArrayList<String>();
-		if (storage.equals(UserStorage.MYSQL)) {
+		if (storage.equals(UserStorage.FLAT)) {
+			keys = new ArrayList<String>(getData(user.getUUID()).getConfigurationSection("").getKeys(false));
+		} else if (storage.equals(UserStorage.MYSQL)) {
 			List<Column> col = getMySqlRow();
 			if (col != null && !col.isEmpty()) {
 				for (Column c : col) {
@@ -395,11 +398,14 @@ private void setData(final String uuid, final String path, final Object value) {
 		FileThread.getInstance().getThread().setData(this, uuid, path, value);
 	}
 
+	@SuppressWarnings("deprecation")
 	public void remove() {
 		if (user.getPlugin().getStorageType().equals(UserStorage.MYSQL)) {
 			user.getPlugin().getMysql().deletePlayer(user.getUUID());
 		} else if (user.getPlugin().getStorageType().equals(UserStorage.SQLITE)) {
 			user.getPlugin().getSQLiteUserTable().delete(new Column("uuid", new DataValueString(user.getUUID())));
+		} else if (user.getPlugin().getStorageType().equals(UserStorage.FLAT)) {
+			FileThread.getInstance().getThread().deletePlayerFile(user.getUUID());
 		}
 		user.clearCache();
 	}