sse-endpoint

services/inbox/sse.py

@@ -1,13 +1,14 @@
 from sse_starlette.sse import EventSourceResponse
 from starlette.requests import Request
-from resolvers.inbox.messages import messages_generator_by_user
+from resolvers.inbox.messages import message_generator
 from base.exceptions import Unauthorized
 
 
 async def sse_messages(request: Request):
-    print(f'[SSE] {request.scope}')
+    print(f'[SSE] {request.scope}')  # FIXME: DOES NOT HAPPEN TO BE CALLED
     # https://github.com/enisdenjo/graphql-sse/blob/master/PROTOCOL.md
     if request['user']:
-        return EventSourceResponse(messages_generator_by_user(request['user'].user_id))
+        event_generator = message_generator(None, request.scope)
+        return EventSourceResponse(event_generator)
     else:
         raise Unauthorized("Please login")