Update routes/sendnotifications.py

routes/sendnotifications.py

@@ -1,16 +1,18 @@
-from fastapi import APIRouter
+from fastapi import APIRouter, HTTPException
+from pydantic import BaseModel
 import httpx
-import json
 from google.oauth2 import service_account
 from google.auth.transport.requests import Request
 
 router = APIRouter()
 
-# Caminho para o arquivo da conta de serviço correta
-SERVICE_ACCOUNT_FILE = './closetcoach-2d50b-firebase-adminsdk-fbsvc-7fcccbacb1.json'
+# Configurações Supabase
+SUPABASE_URL = "https://your-project.supabase.co"
+SUPABASE_API_KEY = "your-supabase-anon-or-service-role-key"
 
-# Token de teste (fixo por enquanto)
-TEST_DEVICE_TOKEN = 'cnL5SYv0RF-VHoVo8JoAvv:APA91bGeOzUhAicTfoboPXI1immeb4ueR_vBL5Pf1Kky5cltKfvw76EBnh7VDhviyvRn4qhV5IjfZieQKJXe-_kYxi0ZJOJJBrfXdeIwVsLQ-0bi2ElNwds'
+# Firebase config
+SERVICE_ACCOUNT_FILE = './closetcoach-2d50b-firebase-adminsdk-fbsvc-7fcccb1.json'
+FCM_PROJECT_ID = "closetcoach-2d50b"
 
 def get_access_token():
     credentials = service_account.Credentials.from_service_account_file(
@@ -22,35 +24,109 @@ def get_access_token():
     scoped_credentials.refresh(Request())
     return scoped_credentials.token
 
-@router.get("/send-test-notification")
-async def send_test_notification():
-    try:
-        access_token = get_access_token()
+class NotificationRequest(BaseModel):
+    keyword: str
+    initiator_id: str   # quem iniciou a ação (ex: quem seguiu)
+    target_user_id: str # quem recebe a notificação
 
-        headers = {
-            'Authorization': f'Bearer {access_token}',
-            'Content-Type': 'application/json',
-        }
+async def supabase_select(table: str, select: str, filters: dict):
+    headers = {
+        "apikey": SUPABASE_API_KEY,
+        "Authorization": f"Bearer {SUPABASE_API_KEY}",
+        "Content-Type": "application/json",
+        "Accept": "application/json",
+    }
+    filter_query = '&'.join([f'{key}=eq.{value}' for key, value in filters.items()])
+    url = f"{SUPABASE_URL}/rest/v1/{table}?select={select}&{filter_query}"
 
-        payload = {
-            "message": {
-                "notification": {
-                    "title": "Teste ClosetCoach",
-                    "body": "Esta é uma notificação de teste 🚀",
-                },
-                "token": TEST_DEVICE_TOKEN
-            }
-        }
+    async with httpx.AsyncClient() as client:
+        response = await client.get(url, headers=headers)
+        response.raise_for_status()
+        return response.json()
 
-        url = "https://fcm.googleapis.com/v1/projects/closetcoach-2d50b/messages:send"
+def format_name(full_name: str) -> str:
+    parts = full_name.strip().split()
+    if len(parts) == 1:
+        return parts[0]
+    first_name = parts[0]
+    second_initial = parts[1][0].upper() if len(parts) > 1 else ''
+    return f"{first_name} {second_initial}."
 
-        async with httpx.AsyncClient() as client:
-            response = await client.post(url, headers=headers, json=payload)
+async def check_follow_exists(follower_id: str, following_id: str) -> bool:
+    filters = {
+        "follower_id": follower_id,
+        "following_id": following_id
+    }
+    result = await supabase_select("followers", "id", filters)
+    return len(result) > 0
 
-        return {
-            "status_code": response.status_code,
-            "response": response.json()
+async def get_user_info(user_id: str):
+    users = await supabase_select("User", "name,token_fcm", {"id": user_id})
+    if not users:
+        return None
+    return users[0]
+
+def build_notification_payload(title: str, body: str, token: str) -> dict:
+    return {
+        "message": {
+            "notification": {
+                "title": title,
+                "body": body,
+            },
+            "token": token
         }
+    }
+
+async def send_fcm_notification(payload: dict):
+    access_token = get_access_token()
+    headers = {
+        'Authorization': f'Bearer {access_token}',
+        'Content-Type': 'application/json',
+    }
+    url = f"https://fcm.googleapis.com/v1/projects/{FCM_PROJECT_ID}/messages:send"
+    async with httpx.AsyncClient() as client:
+        response = await client.post(url, headers=headers, json=payload)
+    return response
+
+@router.post("/send-notification")
+async def send_notification(data: NotificationRequest):
+    keyword = data.keyword.lower()
+
+    # Pega info do initiator (ex: quem fez a ação)
+    initiator = await get_user_info(data.initiator_id)
+    if not initiator:
+        raise HTTPException(status_code=404, detail="Initiator user not found")
+
+    # Pega info do target (quem vai receber notificação)
+    target = await get_user_info(data.target_user_id)
+    if not target or not target.get("token_fcm"):
+        raise HTTPException(status_code=404, detail="Target user or token not found")
+
+    # Aqui definimos regras para cada keyword
+    if keyword == "follow":
+        # Verifica se realmente está seguindo
+        if not await check_follow_exists(data.initiator_id, data.target_user_id):
+            raise HTTPException(status_code=403, detail="Follow relationship not found")
+
+        # Monta mensagem
+        follower_name = format_name(initiator["name"])
+        title = "Novo seguidor!"
+        body = f"{follower_name} começou a seguir você."
+
+    elif keyword == "like":
+        # Exemplo: notificação de like
+        liker_name = format_name(initiator["name"])
+        title = "Nova curtida!"
+        body = f"{liker_name} curtiu sua publicação."
+
+    else:
+        raise HTTPException(status_code=400, detail="Keyword não suportada")
+
+    payload = build_notification_payload(title, body, target["token_fcm"])
+
+    response = await send_fcm_notification(payload)
+
+    if response.status_code != 200:
+        raise HTTPException(status_code=response.status_code, detail=response.text)
 
-    except Exception as e:
-        return {"error": str(e)}
+    return {"detail": "Notification sent successfully"}