From 00c46bc98155989859d6f0d7f3e7c3837ec88ce8 Mon Sep 17 00:00:00 2001
From: Zamitto <167933696+zamitto@users.noreply.github.com>
Date: Thu, 25 Jul 2024 23:03:11 -0300
Subject: [PATCH] feat: implement undo friendship

---
 src/main/events/index.ts                     |  1 +
 src/main/events/profile/undo-friendship.ts   | 11 +++++++++++
 src/preload/index.ts                         |  2 ++
 src/renderer/src/declaration.d.ts            |  1 +
 src/renderer/src/hooks/use-user-details.ts   |  5 +++++
 src/renderer/src/pages/user/user-content.tsx |  7 +++++--
 6 files changed, 25 insertions(+), 2 deletions(-)
 create mode 100644 src/main/events/profile/undo-friendship.ts

diff --git a/src/main/events/index.ts b/src/main/events/index.ts
index 5ae0db46..57daf51c 100644
--- a/src/main/events/index.ts
+++ b/src/main/events/index.ts
@@ -48,6 +48,7 @@ import "./user/unblock-user";
 import "./user/get-user-friends";
 import "./profile/get-friend-requests";
 import "./profile/get-me";
+import "./profile/undo-friendship";
 import "./profile/update-friend-request";
 import "./profile/update-profile";
 import "./profile/send-friend-request";
diff --git a/src/main/events/profile/undo-friendship.ts b/src/main/events/profile/undo-friendship.ts
new file mode 100644
index 00000000..d6f858f1
--- /dev/null
+++ b/src/main/events/profile/undo-friendship.ts
@@ -0,0 +1,11 @@
+import { registerEvent } from "../register-event";
+import { HydraApi } from "@main/services";
+
+const undoFriendship = async (
+  _event: Electron.IpcMainInvokeEvent,
+  userId: string
+): Promise<void> => {
+  return HydraApi.delete(`/profile/friends/${userId}`);
+};
+
+registerEvent("undoFriendship", undoFriendship);
diff --git a/src/preload/index.ts b/src/preload/index.ts
index b7c368fa..3350a340 100644
--- a/src/preload/index.ts
+++ b/src/preload/index.ts
@@ -135,6 +135,8 @@ contextBridge.exposeInMainWorld("electron", {
 
   /* Profile */
   getMe: () => ipcRenderer.invoke("getMe"),
+  undoFriendship: (userId: string) =>
+    ipcRenderer.invoke("undoFriendship", userId),
   updateProfile: (displayName: string, newProfileImagePath: string | null) =>
     ipcRenderer.invoke("updateProfile", displayName, newProfileImagePath),
   getFriendRequests: () => ipcRenderer.invoke("getFriendRequests"),
diff --git a/src/renderer/src/declaration.d.ts b/src/renderer/src/declaration.d.ts
index 564d7501..e022cffe 100644
--- a/src/renderer/src/declaration.d.ts
+++ b/src/renderer/src/declaration.d.ts
@@ -138,6 +138,7 @@ declare global {
 
     /* Profile */
     getMe: () => Promise<UserProfile | null>;
+    undoFriendship: (userId: string) => Promise<void>;
     updateProfile: (
       displayName: string,
       newProfileImagePath: string | null
diff --git a/src/renderer/src/hooks/use-user-details.ts b/src/renderer/src/hooks/use-user-details.ts
index 4b4d5b45..21690e7e 100644
--- a/src/renderer/src/hooks/use-user-details.ts
+++ b/src/renderer/src/hooks/use-user-details.ts
@@ -124,6 +124,10 @@ export function useUserDetails() {
     [fetchFriendRequests]
   );
 
+  const undoFriendship = (userId: string) => {
+    return window.electron.undoFriendship(userId);
+  };
+
   const blockUser = (userId: string) => {
     return window.electron.blockUser(userId);
   };
@@ -151,5 +155,6 @@ export function useUserDetails() {
     updateFriendRequestState,
     blockUser,
     unblockUser,
+    undoFriendship,
   };
 }
diff --git a/src/renderer/src/pages/user/user-content.tsx b/src/renderer/src/pages/user/user-content.tsx
index 8224122b..ad591790 100644
--- a/src/renderer/src/pages/user/user-content.tsx
+++ b/src/renderer/src/pages/user/user-content.tsx
@@ -50,6 +50,7 @@ export function UserContent({
     fetchFriendRequests,
     showFriendsModal,
     updateFriendRequestState,
+    undoFriendship,
     blockUser,
   } = useUserDetails();
   const { showSuccessToast, showErrorToast } = useToast();
@@ -127,9 +128,11 @@ export function UserContent({
 
   const handleUndoFriendship = (userRelation: UserRelation) => {
     const userId =
-      userRelation.AId === userProfile.id ? userRelation.BId : userRelation.AId;
+      userRelation.AId === userDetails?.id
+        ? userRelation.BId
+        : userRelation.AId;
 
-    updateFriendRequestState(userId, "CANCEL")
+    undoFriendship(userId)
       .then(updateUserProfile)
       .catch(() => {
         showErrorToast(t("try_again"));