From 010f07373d797534a3e6287aff0fcfc2603fa52a Mon Sep 17 00:00:00 2001
From: Zamitto <167933696+zamitto@users.noreply.github.com>
Date: Tue, 23 Jul 2024 18:37:19 -0300
Subject: [PATCH] feat: add getUserFriends event

---
 src/main/events/index.ts                 |  1 +
 src/main/events/user/get-user-friends.ts | 20 +++++++++++---------
 src/preload/index.ts                     |  1 +
 src/renderer/src/declaration.d.ts        |  2 ++
 4 files changed, 15 insertions(+), 9 deletions(-)

diff --git a/src/main/events/index.ts b/src/main/events/index.ts
index dd5e3263..398235f0 100644
--- a/src/main/events/index.ts
+++ b/src/main/events/index.ts
@@ -43,6 +43,7 @@ import "./auth/sign-out";
 import "./auth/open-auth-window";
 import "./auth/get-session-hash";
 import "./user/get-user";
+import "./user/get-user-friends";
 import "./profile/get-friend-requests";
 import "./profile/get-me";
 import "./profile/update-friend-request";
diff --git a/src/main/events/user/get-user-friends.ts b/src/main/events/user/get-user-friends.ts
index 4505909a..a9394d95 100644
--- a/src/main/events/user/get-user-friends.ts
+++ b/src/main/events/user/get-user-friends.ts
@@ -8,21 +8,23 @@ export const getUserFriends = async (
   take: number,
   skip: number
 ): Promise<UserFriends> => {
-  const loggedUser = await userAuthRepository.findOne({ where: { id: 1 } });
+  try {
+    const loggedUser = await userAuthRepository.findOne({ where: { id: 1 } });
 
-  if (loggedUser?.userId == userId) {
-    return HydraApi.get(`/profile/friends`, { take, skip }).catch(
+    if (loggedUser?.userId == userId) {
+      return HydraApi.get(`/profile/friends`, { take, skip }).catch((_err) => {
+        return { totalFriends: 0, friends: [] };
+      });
+    }
+
+    return HydraApi.get(`/user/${userId}/friends`, { take, skip }).catch(
       (_err) => {
         return { totalFriends: 0, friends: [] };
       }
     );
+  } catch (err) {
+    return { totalFriends: 0, friends: [] };
   }
-
-  return HydraApi.get(`/user/${userId}/friends`, { take, skip }).catch(
-    (_err) => {
-      return { totalFriends: 0, friends: [] };
-    }
-  );
 };
 
 const getUserFriendsEvent = async (
diff --git a/src/preload/index.ts b/src/preload/index.ts
index 91722606..e6dafe0f 100644
--- a/src/preload/index.ts
+++ b/src/preload/index.ts
@@ -145,6 +145,7 @@ contextBridge.exposeInMainWorld("electron", {
 
   /* User */
   getUser: (userId: string) => ipcRenderer.invoke("getUser", userId),
+  getUserFriends: (userId: string) => ipcRenderer.invoke("getUserFriends", userId),
 
   /* Auth */
   signOut: () => ipcRenderer.invoke("signOut"),
diff --git a/src/renderer/src/declaration.d.ts b/src/renderer/src/declaration.d.ts
index bb89f84e..55fc1269 100644
--- a/src/renderer/src/declaration.d.ts
+++ b/src/renderer/src/declaration.d.ts
@@ -16,6 +16,7 @@ import type {
   UserProfile,
   FriendRequest,
   FriendRequestAction,
+  UserFriends,
 } from "@types";
 import type { DiskSpace } from "check-disk-space";
 
@@ -127,6 +128,7 @@ declare global {
 
     /* User */
     getUser: (userId: string) => Promise<UserProfile | null>;
+    getUserFriends: (userId: string) => Promise<UserFriends>;
 
     /* Profile */
     getMe: () => Promise<UserProfile | null>;