From 68626fecd70c9fe07aa79997f7ffe384a39e3d03 Mon Sep 17 00:00:00 2001
From: Azalea <22280294+hykilpikonna@users.noreply.github.com>
Date: Tue, 5 Mar 2024 17:59:44 -0500
Subject: [PATCH] [+] Add profile picture field to aqua net user

---
 src/main/java/icu/samnyan/aqua/net/UserRegistrar.kt      | 6 +++++-
 src/main/java/icu/samnyan/aqua/net/db/AquaNetUser.kt     | 1 +
 src/main/resources/db/migration/mariadb/V1000_3__pfp.sql | 1 +
 3 files changed, 7 insertions(+), 1 deletion(-)
 create mode 100644 src/main/resources/db/migration/mariadb/V1000_3__pfp.sql

diff --git a/src/main/java/icu/samnyan/aqua/net/UserRegistrar.kt b/src/main/java/icu/samnyan/aqua/net/UserRegistrar.kt
index 0f983452..9e1e2d1a 100644
--- a/src/main/java/icu/samnyan/aqua/net/UserRegistrar.kt
+++ b/src/main/java/icu/samnyan/aqua/net/UserRegistrar.kt
@@ -208,7 +208,11 @@ class UserRegistrar(
         if (!mime.startsWith("image/")) 400 - "Invalid file type"
 
         // Save the image
-        (portraitPath / "${u.auId}.${MIMES.forName(mime)?.extension ?: "jpg"}").writeBytes(bytes)
+        val name = "${u.auId}.${MIMES.forName(mime)?.extension ?: "jpg"}"
+        async {
+            (portraitPath / name).writeBytes(bytes)
+            userRepo.save(u.apply { profilePicture = name })
+        }
 
         SUCCESS
     }
diff --git a/src/main/java/icu/samnyan/aqua/net/db/AquaNetUser.kt b/src/main/java/icu/samnyan/aqua/net/db/AquaNetUser.kt
index bb4fea77..1c6aca86 100644
--- a/src/main/java/icu/samnyan/aqua/net/db/AquaNetUser.kt
+++ b/src/main/java/icu/samnyan/aqua/net/db/AquaNetUser.kt
@@ -52,6 +52,7 @@ class AquaNetUser(
     // Profile fields
     var profileLocation: String = "",
     var profileBio: String = "",
+    var profilePicture: String = "",
 
     // Email confirmation
     var emailConfirmed: Boolean = false,
diff --git a/src/main/resources/db/migration/mariadb/V1000_3__pfp.sql b/src/main/resources/db/migration/mariadb/V1000_3__pfp.sql
new file mode 100644
index 00000000..9ccc975b
--- /dev/null
+++ b/src/main/resources/db/migration/mariadb/V1000_3__pfp.sql
@@ -0,0 +1 @@
+ALTER TABLE aqua_net_user ADD profile_picture VARCHAR(255) NULL;
\ No newline at end of file