1 files changed, 72 insertions, 0 deletions
diff --git a/src/service/friend.ts b/src/service/friend.ts
new file mode 100644
index 0000000..c7f4b6b
--- /dev/null
+++ b/src/service/friend.ts
@@ -0,0 +1,72 @@
+/* Financer - Pocket Money Tracker
+ * Copyright (C) 2025  Vidhu Kant Sharma <vidhukant@vidhukant.com>
+
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+import prisma from "../util/prisma";
+
+const friend = async (userId: Number, friendId: Number) => {
+    // will return error P2002 if already friends
+    return prisma.friend.createMany({
+        data: [
+            // @ts-ignore
+            { userId: userId, friendId: friendId },
+
+            // @ts-ignore
+            { userId: friendId, friendId: userId }
+        ]
+    })
+}
+
+const unfriend = async (userId: Number, friendId: Number) => {
+    return prisma.friend.deleteMany({
+        where: {
+            OR: [
+                // @ts-ignore
+                { userId: userId },
+
+                // @ts-ignore
+                { userId: friendId },
+            ]
+        }
+    })
+}
+
+const list = async (userId: Number) => {
+    const friendsRaw = await prisma.friend.findMany({
+        // @ts-ignore
+        where: { userId: userId },
+        include: { friend: true }
+    })
+
+    // TODO: check if no friends
+
+    // @ts-ignore
+    return friendsRaw.map((x: any) => {
+        const friend = x.friend;
+        delete friend["password"];
+        delete friend["refreshTokenVersion"];
+        delete friend["createdAt"];
+        delete friend["updatedAt"];
+
+        return friend;
+    })
+}
+
+export {
+    friend,
+    unfriend,
+    list
+}
+\ No newline at end of file