Update routes/sendnotifications.py

routes/sendnotifications.py

@@ -36,6 +36,7 @@ FCM_PROJECT_ID = "closetcoach-2d50b"
 class NotificationRequest(BaseModel):
     keyword: str
     target_user_id: str
+    reference: str = ""
 
 def short_collapse_key(follower_id: str, target_user_id: str) -> str:
     raw = f"follow:{follower_id}:{target_user_id}"
@@ -60,6 +61,27 @@ async def verify_user_token(user_token: str) -> str:
 
     return user_id
 
+async def get_post_info(feed_id: str):
+    # Busca o feed
+    feeds = await fetch_supabase("Feeds", "description,portfolios,user_id", {"id": feed_id})
+    if not feeds:
+        raise HTTPException(status_code=404, detail="Post not found")
+    
+    feed = feeds[0]
+    description = feed.get("description", "")
+    portfolio_ids = eval(feed.get("portfolios", "[]"))
+    
+    image_url = None
+    if portfolio_ids:
+        portfolio_data = await fetch_supabase("Portfolio", "image_url", {"id": portfolio_ids[0]})
+        if portfolio_data:
+            image_url = portfolio_data[0].get("image_url")
+
+    return {
+        "description": description,
+        "image_url": image_url
+    }
+
 async def fetch_supabase(table: str, select: str, filters: dict, headers=SUPABASE_ROLE_HEADERS):
     filter_query = '&'.join([f'{k}=eq.{v}' for k, v in filters.items()])
     url = f"{SUPABASE_URL}/rest/v1/{table}?select={select}&{filter_query}"
@@ -130,72 +152,70 @@ async def send_notification(
     follower_id = await verify_user_token(user_token)
     target_user_id = data.target_user_id
 
-    if data.keyword != "follow":
+    if data.keyword not in ("follow", "like"):
         raise HTTPException(status_code=400, detail="Unsupported keyword")
 
-    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
+    # Verifica se relação de follow existe
+    if data.keyword == "follow":
+        follow_exists = await check_follow_exists(follower_id, target_user_id)
         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."
+    # Busca FCM e nome do usuário de destino
+    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="Target user or FCM token not found")
+
+    # Busca nome do autor da ação
+    actor_info = await get_user_info(follower_id)
+    if not actor_info or not actor_info.get("name"):
+        raise HTTPException(status_code=404, detail="User not found")
+    actor_name = format_name(actor_info["name"])
 
     collapse_id = short_collapse_key(follower_id, target_user_id)
 
-    payload = {
-        "message": {
+    # Monta conteúdo da notificação
+    if data.keyword == "follow":
+        title = "🎉 New Follower!"
+        body = f"{actor_name} started following you."
+        image_url = None
+    elif data.keyword == "like":
+        # Pega dados do post
+        post_info = await get_post_info(data.reference)
+        title = "❤️ New Like!"
+        desc = post_info["description"]
+        body = f"{actor_name} liked your post" + (f": \"{desc}\"" if desc else ".")
+        image_url = post_info["image_url"]
+    else:
+        raise HTTPException(status_code=400, detail="Unsupported keyword")
+
+    # Monta payload
+    message = {
+        "notification": {
+            "title": title,
+            "body": body,
+        },
+        "token": target_user["token_fcm"],
+        "android": {
+            "collapse_key": collapse_id,
             "notification": {
-                "title": title,
-                "body": body,
-            },
-            "token": target_user["token_fcm"],
-            "android": {
-                "collapse_key": collapse_id,
-                "notification": {
-                    "tag": collapse_id  # 🔥 ESSENCIAL para evitar duplicata visual
-                }
-            },
-            "apns": {
-                "headers": {
-                    "apns-collapse-id": collapse_id
-                }
+                "tag": collapse_id
+            }
+        },
+        "apns": {
+            "headers": {
+                "apns-collapse-id": collapse_id
             }
         }
     }
 
+    if image_url:
+        message["notification"]["image"] = image_url
+        message["android"]["notification"]["image"] = image_url
+
+    payload = {"message": message}
+
+    # Envia via FCM
     access_token = get_access_token()
     headers = {
         "Authorization": f"Bearer {access_token}",
@@ -210,4 +230,4 @@ async def send_notification(
                 raise HTTPException(status_code=resp.status, detail=f"FCM error: {resp_text}")
             fcm_response = await resp.json()
 
-    return {"detail": "Follow notification sent successfully", "fcm_response": fcm_response}
+    return {"detail": "Notification sent successfully", "fcm_response": fcm_response}