Update routes/sendnotifications.py

routes/sendnotifications.py

@@ -146,6 +146,18 @@ async def send_fcm_notification(title: str, body: str, token: str):
                 raise HTTPException(status_code=resp.status, detail=f"FCM error: {resp_text}")
             return await resp.json()
 
+async def check_subscription_exists(customer_id: str, stylist_id: str) -> bool:
+    result = await fetch_supabase(
+        "Subscriptions",
+        "id",
+        {
+            "customer_id": customer_id,
+            "stylist_id": stylist_id,
+            "active": "true"
+        }
+    )
+    return len(result) > 0
+
 @router.post("/send-notification")
 async def send_notification(
     data: NotificationRequest,
@@ -153,28 +165,32 @@ async def send_notification(
 ):
     follower_id = await verify_user_token(user_token)
 
-    if data.keyword not in ("follow", "like"):
+    if data.keyword not in ("follow", "like", "subscriber"):
         raise HTTPException(status_code=400, detail="Unsupported keyword")
 
-    # Determina o usuário de destino
     if data.keyword == "like":
         post_info = await get_post_info(data.reference)
         target_user_id = post_info["user_id"]
     else:
         target_user_id = data.target_user_id
 
-    # 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 FCM e nome do usuário de destino
+    if data.keyword == "subscriber":
+        subscription_exists = await check_subscription_exists(
+            customer_id=follower_id,
+            stylist_id=target_user_id
+        )
+        if not subscription_exists:
+            return {"detail": "No active subscription found, notification not sent"}
+
     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")
@@ -182,7 +198,6 @@ async def send_notification(
 
     collapse_id = short_collapse_key(data.keyword, follower_id, target_user_id)
 
-    # Monta conteúdo da notificação
     if data.keyword == "follow":
         title = "🎉 New Follower!"
         body = f"{actor_name} started following you."
@@ -192,10 +207,13 @@ async def send_notification(
         title = "❤️ New Like!"
         body = f"{actor_name} liked your post" + (f": \"{desc}\"" if desc else ".")
         image_url = post_info["image_url"]
+    elif data.keyword == "subscriber":
+        title = "💼 New Subscriber!"
+        body = f"{actor_name} just subscribed to your styling services. Check out their profile!"
+        image_url = None
     else:
         raise HTTPException(status_code=400, detail="Unsupported keyword")
 
-    # Monta payload
     message = {
         "notification": {
             "title": title,
@@ -221,7 +239,6 @@ async def send_notification(
 
     payload = {"message": message}
 
-    # Envia via FCM
     access_token = get_access_token()
     headers = {
         "Authorization": f"Bearer {access_token}",