From 4f7702a77f9bf9e18ca18487bfbc7634e567f671 Mon Sep 17 00:00:00 2001
From: Untone <anton.rewin@gmail.com>
Date: Thu, 16 Nov 2023 18:34:37 +0300
Subject: [PATCH] add-sse-entity

---
 src/context/inbox.tsx         | 8 ++++----
 src/context/notifications.tsx | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/context/inbox.tsx b/src/context/inbox.tsx
index 44cf75e2..e90dcbc3 100644
--- a/src/context/inbox.tsx
+++ b/src/context/inbox.tsx
@@ -32,12 +32,12 @@ export const InboxProvider = (props: { children: JSX.Element }) => {
   const handleMessage = (sseMessage) => {
     console.log('[context.inbox]:', sseMessage)
     // TODO: handle all action types: create update delete join left
-    if (['create', 'update', 'delete'].includes(sseMessage.action)) {
+    if (sseMessage.entity == 'message') {
       const relivedMessage = sseMessage.payload
       setMessages((prev) => [...prev, relivedMessage])
-    } else if (['left', 'join'].includes(sseMessage.action)) {
-      // TODO: set chat members
-      console.debug(sseMessage)
+    } else if (sseMessage.entity == 'chat') {
+      const relivedChat = sseMessage.payload
+      setChats((prev) => [...prev, relivedChat])
     }
   }
 
diff --git a/src/context/notifications.tsx b/src/context/notifications.tsx
index f0476e43..c7e28da2 100644
--- a/src/context/notifications.tsx
+++ b/src/context/notifications.tsx
@@ -104,7 +104,7 @@ export const NotificationsProvider = (props: { children: JSX.Element }) => {
         },
         onmessage(event) {
           const m: SSEMessage = JSON.parse(event.data)
-          if (m.entity === 'chat') {
+          if (m.entity === 'chat' || m.entity == 'message') {
             console.log('[context.notifications] Received message:', m)
             messageHandler()(m)
           } else {