Update routes/sendnotifications.py

routes/sendnotifications.py

@@ -125,29 +125,80 @@ async def send_notification(
     follower_id = await verify_user_token(user_token)
     target_user_id = data.target_user_id
 
-    if data.keyword == "follow":
-        # verifica se usuário alvo existe e tem token
-        target_user = await get_user_info(target_user_id)
-        if not target_user or not target_user.get("token_fcm"):
-            raise HTTPException(status_code=404, detail="Usuário alvo ou token FCM não encontrado")
+    if data.keyword != "follow":
+        raise HTTPException(status_code=400, detail="Unsupported keyword")
 
-        # verifica se a relação de follow existe mesmo
-        if not await check_follow_exists(follower_id, target_user_id):
-            raise HTTPException(status_code=403, detail="Relação de follow não encontrada")
-
-        # pega nome do follower para notificação
-        follower = await get_user_info(follower_id)
-        if not follower or not follower.get("name"):
-            raise HTTPException(status_code=404, detail="Usuário follower não encontrado")
-
-        follower_name = format_name(follower["name"])
-
-        title = "Novo seguidor!"
-        body = f"{follower_name} começou a seguir você."
+    async with aiohttp.ClientSession() as session:
+        # Busca usuário alvo
+        target_user_resp = await session.get(
+            f"{SUPABASE_URL}/rest/v1/User?select=name,token_fcm&id=eq.{target_user_id}",
+            headers=SUPABASE_ROLE_HEADERS
+        )
+        if target_user_resp.status != 200:
+            raise HTTPException(status_code=500, detail="Failed to fetch target user")
+        target_users = await target_user_resp.json()
+        if not target_users or not target_users[0].get("token_fcm"):
+            raise HTTPException(status_code=404, detail="Target user or FCM token not found")
+        target_user = target_users[0]
+
+        # Verifica se follow existe
+        follow_resp = await session.get(
+            f"{SUPABASE_URL}/rest/v1/followers?select=id&follower_id=eq.{follower_id}&following_id=eq.{target_user_id}",
+            headers=SUPABASE_ROLE_HEADERS
+        )
+        if follow_resp.status != 200:
+            raise HTTPException(status_code=500, detail="Failed to verify follow relationship")
+        follow_exists = len(await follow_resp.json()) > 0
+        if not follow_exists:
+            raise HTTPException(status_code=403, detail="Follow relationship does not exist")
+
+        # Busca dados do follower
+        follower_resp = await session.get(
+            f"{SUPABASE_URL}/rest/v1/User?select=name&id=eq.{follower_id}",
+            headers=SUPABASE_ROLE_HEADERS
+        )
+        if follower_resp.status != 200:
+            raise HTTPException(status_code=500, detail="Failed to fetch follower user")
+        followers = await follower_resp.json()
+        if not followers or not followers[0].get("name"):
+            raise HTTPException(status_code=404, detail="Follower user not found")
+        follower_name = format_name(followers[0]["name"])
+
+    # Monta notificação
+    title = "🎉 New Follower!"
+    body = f"{follower_name} started following you."
+
+    # Envio da notificação com collapse_key para evitar spam
+    payload = {
+        "message": {
+            "notification": {
+                "title": title,
+                "body": body,
+            },
+            "token": target_user["token_fcm"],
+            "android": {
+                "collapse_key": f"follow_{follower_id}_{target_user_id}"
+            },
+            "apns": {
+                "headers": {
+                    "apns-collapse-id": f"follow_{follower_id}_{target_user_id}"
+                }
+            }
+        }
+    }
 
-        fcm_response = await send_fcm_notification(title, body, target_user["token_fcm"])
+    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"
 
-        return {"detail": "Notificação de follow enviada com sucesso", "fcm_response": fcm_response}
+    async with aiohttp.ClientSession() as session:
+        async with session.post(url, headers=headers, json=payload) as resp:
+            resp_text = await resp.text()
+            if resp.status != 200:
+                raise HTTPException(status_code=resp.status, detail=f"FCM error: {resp_text}")
+            fcm_response = await resp.json()
 
-    else:
-        raise HTTPException(status_code=400, detail="Keyword não suportada")
+    return {"detail": "Follow notification sent successfully", "fcm_response": fcm_response}