Update routes/notifications.py

routes/notifications.py

@@ -4,6 +4,7 @@ from fastapi import APIRouter, HTTPException, Header
 from pydantic import BaseModel
 from google.oauth2 import service_account
 from google.auth.transport.requests import Request
+from datetime import datetime
 
 router = APIRouter()
 
@@ -25,14 +26,12 @@ SUPABASE_HEADERS = {
 SERVICE_ACCOUNT_FILE = './closetcoach-2d50b-firebase-adminsdk-fbsvc-7fcccbacb1.json'
 FCM_PROJECT_ID = "closetcoach-2d50b"
 
-# Modelo da requisição
 class SimpleNotification(BaseModel):
-    target: str  # "all" ou user_id
+    target: str  # email or "all"
     title: str
     content: str
     image_url: str = ""
 
-# Função para obter token do Firebase
 def get_fcm_access_token():
     credentials = service_account.Credentials.from_service_account_file(
         SERVICE_ACCOUNT_FILE
@@ -43,7 +42,6 @@ def get_fcm_access_token():
     scoped.refresh(Request())
     return scoped.token
 
-# Verifica o token de usuário no Supabase
 async def get_user_id_from_token(user_token: str) -> str:
     url = f"{SUPABASE_URL}/auth/v1/user"
     headers = {
@@ -59,27 +57,50 @@ async def get_user_id_from_token(user_token: str) -> str:
             data = await resp.json()
             return data.get("id")
 
-# Busca informações de um usuário pelo ID
-async def get_user(user_id: str):
-    url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}&select=id,token_fcm,manage_notifications"
+async def get_user_by_email(email: str):
+    url = f"{SUPABASE_URL}/rest/v1/User?email=eq.{email}&select=id,token_fcm"
     async with aiohttp.ClientSession() as session:
         async with session.get(url, headers=SUPABASE_HEADERS) as resp:
             if resp.status != 200:
-                raise HTTPException(status_code=resp.status, detail="Failed to fetch user info")
+                raise HTTPException(status_code=500, detail="Failed to fetch target user from database")
             data = await resp.json()
             return data[0] if data else None
 
+async def get_user_by_id(user_id: str):
+    url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}&select=id,manage_notifications"
+    async with aiohttp.ClientSession() as session:
+        async with session.get(url, headers=SUPABASE_HEADERS) as resp:
+            if resp.status != 200:
+                raise HTTPException(status_code=500, detail="Failed to fetch sender user from database")
+            data = await resp.json()
+            return data[0] if data else None
+
+async def log_notification(send_by: str, title: str, content: str, target_id: str | None, image_url: str):
+    payload = {
+        "send_by": send_by,
+        "title": title,
+        "content": content,
+        "target": target_id,
+        "image_url": image_url or None,
+        "created_at": datetime.utcnow().isoformat()
+    }
+    url = f"{SUPABASE_URL}/rest/v1/fcm_notifications"
+    async with aiohttp.ClientSession() as session:
+        async with session.post(url, headers=SUPABASE_HEADERS, json=payload) as resp:
+            if resp.status not in (200, 201):
+                detail = await resp.text()
+                raise HTTPException(status_code=500, detail=f"Failed to log notification: {detail}")
+
 @router.post("/send-global-notification")
 async def send_global_notification(
     payload: SimpleNotification,
     user_token: str = Header(..., alias="User-key")
 ):
-    # Verifica o ID do usuário autenticado
     sender_id = await get_user_id_from_token(user_token)
-    sender = await get_user(sender_id)
+    sender = await get_user_by_id(sender_id)
 
     if not sender or not sender.get("manage_notifications"):
-        raise HTTPException(status_code=403, detail="Not authorized to send notifications")
+        raise HTTPException(status_code=403, detail="You are not authorized to send notifications.")
 
     message = {
         "notification": {
@@ -88,17 +109,25 @@ async def send_global_notification(
         }
     }
 
+    target_user_id = None
+
     if payload.image_url:
         message["notification"]["image"] = payload.image_url
 
-    if payload.target == "all":
+    if payload.target.lower() == "all":
         message["topic"] = "all"
     else:
-        target_user = await get_user(payload.target)
-        if not target_user or not target_user.get("token_fcm"):
-            raise HTTPException(status_code=404, detail="Target user or FCM token not found")
+        target_user = await get_user_by_email(payload.target.lower().strip())
+
+        if not target_user:
+            raise HTTPException(status_code=404, detail="The provided email does not correspond to any user.")
+        if not target_user.get("token_fcm"):
+            raise HTTPException(status_code=400, detail="The target user does not have a registered FCM token.")
+
         message["token"] = target_user["token_fcm"]
+        target_user_id = target_user["id"]
 
+    # Envia para o Firebase
     access_token = get_fcm_access_token()
     headers = {
         "Authorization": f"Bearer {access_token}",
@@ -111,4 +140,14 @@ async def send_global_notification(
             response_text = await resp.text()
             if resp.status != 200:
                 raise HTTPException(status_code=resp.status, detail=f"FCM error: {response_text}")
-            return {"detail": "Notification sent successfully"}
+
+    # Salvar notificação no banco
+    await log_notification(
+        send_by=sender_id,
+        title=payload.title,
+        content=payload.content,
+        target_id=target_user_id,
+        image_url=payload.image_url
+    )
+
+    return {"detail": "Notification sent and logged successfully."}