refactor: Update BadgeModal component to handle badge cancellation an…

…d confirmation
apache · Aug 22, 2024 · 34684cd · 34684cd
1 parent e6aa178
commit 34684cd
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/ui/src/components/Modal/BadgeModal.tsx b/ui/src/components/Modal/BadgeModal.tsx
@@ -19,14 +19,21 @@ const BadgeModal: FC<BadgeModalProps> = ({ badge, visible }) => {
   const { t } = useTranslation('translation', { keyPrefix: 'badges.modal' });
   const { user } = loggedUserInfoStore();
   const navigate = useNavigate();
-  const { data } = useQueryNotificationStatus();
+  const { data, mutate } = useQueryNotificationStatus();
 
-  const handleCancel = async () => {
+  const handle = async () => {
     if (!data) return;
     await readNotification(badge?.notification_id);
+    await mutate({
+      ...data,
+      badge_award: null,
+    });
+  };
+  const handleCancel = async () => {
+    await handle();
   };
   const handleConfirm = async () => {
-    await readNotification(badge?.notification_id);
+    await handle();
 
     const url = `/badges/${badge?.badge_id}?username=${user.username}`;
     navigate(url);