added route to fetch shared emails

potpie-ai · Oct 29, 2024 · 532a62b · 532a62b
1 parent 2dac7bd
commit 532a62b
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/app/modules/conversations/access/access_service.py b/app/modules/conversations/access/access_service.py
@@ -53,3 +53,9 @@ async def share_chat(self, conversation_id: str, recipient_emails: List[str] = N
                 return conversation_id
         return conversation_id
 
+    async def get_shared_emails(self, conversation_id: str) -> List[str]:
+        chat = self.db.query(Conversation).filter_by(id=conversation_id).first()
+        if not chat:
+            raise ShareChatServiceError("Chat not found.")
+
+        return chat.shared_with_emails or []
diff --git a/app/modules/conversations/conversations_router.py b/app/modules/conversations/conversations_router.py
@@ -166,3 +166,15 @@ async def share_chat(
         )
     except ShareChatServiceError as e:
         raise HTTPException(status_code=400, detail=str(e))
+
+@router.get("/conversations/{conversation_id}/shared-emails", response_model=List[str])
+async def get_shared_emails(
+    conversation_id: str,
+    db: Session = Depends(get_db),
+):
+    service = ShareChatService(db)
+    try:
+        shared_emails = await service.get_shared_emails(conversation_id)
+        return shared_emails
+    except ShareChatServiceError as e:
+        raise HTTPException(status_code=404, detail=str(e))