broadcast message

App/Messages/MessagesRoute.py

@@ -1,4 +1,13 @@
 # App/Messages/Routes.py
+
+### In App/Messages/Routes.py
+
+from fastapi import APIRouter, HTTPException, Depends
+from typing import List
+import asyncio
+from App.Users.Model import User
+from App.Users.dependencies import get_admin_user, UserType
+from pydantic import BaseModel
 from fastapi import APIRouter, HTTPException, Depends
 from typing import List
 from .Model import Message
@@ -27,6 +36,31 @@ message_router = APIRouter(tags=["Messages"], prefix="/messages")
 logging.basicConfig(level=logging.WARNING)
 
 
+class BroadcastMessageRequest(BaseModel):
+    message: str
+
+
+@message_router.post("/broadcast")
+async def broadcast_message(request: BroadcastMessageRequest):
+    # # Ensure the requesting user is an admin
+    # if current_user.user_type != UserType.ADMIN:
+    #     raise HTTPException(
+    #         status_code=403,
+    #         detail="User does not have permission to perform a broadcast.",
+    #     )
+
+    # Fetch all users
+    users = await User.all()
+
+    # Create a list of coroutines for sending the message
+    tasks = [user.send_message(request.message) for user in users]
+
+    # Run them concurrently
+    await asyncio.gather(*tasks)
+
+    return {"status": 200, "message": "Broadcast message sent to all users."}
+
+
 @message_router.get("/", response_model=List[MessageResponse])
 async def get_all_messages(current_user: User = Depends(get_admin_user)):
     # Check if the current user is an admin