Update routes/subscription.py

routes/subscription.py

@@ -8,8 +8,11 @@ import os
 import requests
 import asyncio
 import jwt
+import hashlib
 from fastapi import APIRouter, HTTPException, Request, Header
 from pydantic import BaseModel
+from google.oauth2 import service_account
+from google.auth.transport.requests import Request as GoogleRequest
 
 router = APIRouter()
 
@@ -25,6 +28,10 @@ SUPABASE_URL = "https://ussxqnifefkgkaumjann.supabase.co"
 SUPABASE_KEY = os.getenv("SUPA_KEY")  # Lendo do ambiente
 SUPABASE_ROLE_KEY = os.getenv("SUPA_SERVICE_KEY")
 
+# Firebase config para notificações push
+SERVICE_ACCOUNT_FILE = './closetcoach-2d50b-firebase-adminsdk-fbsvc-7fcccbacb1.json'
+FCM_PROJECT_ID = "closetcoach-2d50b"
+
 if not stripe.api_key or not SUPABASE_KEY or not SUPABASE_ROLE_KEY:
     raise ValueError("❌ STRIPE_KEY, SUPA_KEY ou SUPA_SERVICE_KEY não foram definidos no ambiente!")
 
@@ -61,6 +68,133 @@ class CreatePriceRequest(BaseModel):
     emergency_price: int  # Valor de emergência (ex: 500 para R$5,00)
     consultations: int  # Número de consultas (ex: 3)
 
+# ==================== FUNÇÕES DE NOTIFICAÇÃO PUSH ====================
+
+def short_collapse_key(keyword: str, sender_id: str, receiver_id: str) -> str:
+    """Gera uma chave de colapso curta para notificações"""
+    raw = f"{keyword}:{sender_id}:{receiver_id}"
+    return hashlib.sha1(raw.encode()).hexdigest()[:20]
+
+async def fetch_supabase_async(table: str, select: str, filters: dict, headers=SUPABASE_ROLE_HEADERS):
+    """Função assíncrona para buscar dados do Supabase"""
+    filter_query = '&'.join([f'{k}=eq.{v}' for k, v in filters.items()])
+    url = f"{SUPABASE_URL}/rest/v1/{table}?select={select}&{filter_query}&order=created_at.desc"
+
+    async with aiohttp.ClientSession() as session:
+        async with session.get(url, headers=headers) as resp:
+            if resp.status != 200:
+                detail = await resp.text()
+                raise HTTPException(status_code=500, detail=f"Supabase error: {detail}")
+            return await resp.json()
+
+def format_name(full_name: str) -> str:
+    """Formata o nome para exibição (Nome + inicial do sobrenome)"""
+    parts = full_name.strip().split()
+    if len(parts) == 1:
+        return parts[0]
+    return f"{parts[0]} {parts[1][0].upper()}."
+
+async def get_user_info_async(user_id: str):
+    """Busca informações do usuário de forma assíncrona"""
+    users = await fetch_supabase_async("User", "name,token_fcm", {"id": user_id})
+    if not users:
+        return None
+    return users[0]
+
+def get_access_token():
+    """Obtém token de acesso para Firebase Cloud Messaging"""
+    credentials = service_account.Credentials.from_service_account_file(
+        SERVICE_ACCOUNT_FILE
+    )
+    scoped_credentials = credentials.with_scopes(
+        ['https://www.googleapis.com/auth/firebase.messaging']
+    )
+    scoped_credentials.refresh(GoogleRequest())
+    return scoped_credentials.token
+
+async def send_push_notification(sender_id: str, target_user_id: str, keyword: str = "changeprice"):
+    """Envia notificação push para um usuário específico"""
+    try:
+        # Buscar informações do usuário alvo
+        target_user = await get_user_info_async(target_user_id)
+        if not target_user or not target_user.get("token_fcm"):
+            logger.warning(f"⚠️ FCM token not found for user {target_user_id}")
+            return False
+
+        # Buscar informações do remetente (estilista)
+        actor_info = await get_user_info_async(sender_id)
+        if not actor_info or not actor_info.get("name"):
+            logger.warning(f"⚠️ Actor info not found for user {sender_id}")
+            return False
+        
+        actor_name = format_name(actor_info["name"])
+        collapse_id = short_collapse_key(keyword, sender_id, target_user_id)
+
+        # Configurar conteúdo da notificação
+        title = "⚠️ Subscription Price Changed"
+        body = f"{actor_name} changed your subscription price. Your subscription was automatically canceled. Please check the chat with {actor_name} for reactivation options and more info."
+
+        # Montar mensagem FCM
+        message = {
+            "notification": {
+                "title": title,
+                "body": body,
+            },
+            "token": target_user["token_fcm"],
+            "android": {
+                "collapse_key": collapse_id,
+                "notification": {
+                    "tag": collapse_id
+                }
+            },
+            "apns": {
+                "headers": {
+                    "apns-collapse-id": collapse_id
+                }
+            }
+        }
+
+        payload = {"message": message}
+
+        # Enviar notificação
+        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 aiohttp.ClientSession() as session:
+            async with session.post(url, headers=headers, json=payload) as resp:
+                if resp.status == 200:
+                    logger.info(f"✅ Push notification sent to user {target_user_id}")
+                    return True
+                else:
+                    resp_text = await resp.text()
+                    logger.error(f"❌ FCM error for user {target_user_id}: {resp_text}")
+                    return False
+
+    except Exception as e:
+        logger.error(f"❌ Error sending push notification to {target_user_id}: {e}")
+        return False
+
+async def send_bulk_push_notifications(sender_id: str, target_user_ids: list):
+    """Envia notificações push para múltiplos usuários"""
+    if not target_user_ids:
+        logger.info("📭 No users to notify")
+        return
+
+    logger.info(f"📤 Sending push notifications to {len(target_user_ids)} users")
+    
+    # Enviar notificações em paralelo
+    tasks = [send_push_notification(sender_id, user_id) for user_id in target_user_ids]
+    results = await asyncio.gather(*tasks, return_exceptions=True)
+    
+    success_count = sum(1 for result in results if result is True)
+    logger.info(f"✅ Successfully sent {success_count}/{len(target_user_ids)} push notifications")
+
+# ==================== FUNÇÕES ORIGINAIS ====================
+
 def get_active_subscribers_by_price_id(price_id: str) -> list:
     """
     Retorna uma lista de customer_ids que têm uma assinatura ativa com o price_id fornecido.
@@ -977,17 +1111,31 @@ async def create_price(data: CreatePriceRequest, user_token: str = Header(None,
         if update_response.status_code not in [200, 204]:
             raise HTTPException(status_code=500, detail=f"Failed to update user: {update_response.text}")
 
-        # 🔥 Cria notificações para os afetados
+        # 🔥 Cria notificações na base de dados para os afetados
         create_notifications_for_price_change(affected_users, stylist_id=user_id)
 
-        logger.info(f"✅ User updated and notifications sent")
+        # 🚀 NOVA FUNCIONALIDADE: Enviar notificações push para todos os afetados
+        if affected_users:
+            try:
+                await send_bulk_push_notifications(sender_id=user_id, target_user_ids=affected_users)
+                logger.info(f"🔔 Push notifications process completed for {len(affected_users)} users")
+            except Exception as push_error:
+                logger.error(f"⚠️ Error sending push notifications: {push_error}")
+                # Não falha a operação principal se as notificações push falharem
+
+        logger.info(f"✅ User updated, notifications created and push notifications sent")
 
-        return {"message": "Price created and user updated successfully!", "price_id": new_price_id}
+        return {
+            "message": "Price created and user updated successfully!", 
+            "price_id": new_price_id,
+            "affected_users_count": len(affected_users),
+            "notifications_sent": True
+        }
 
     except Exception as e:
         logger.error(f"❌ Error creating price: {e}")
         raise HTTPException(status_code=500, detail=f"Error creating price: {str(e)}")
-                
+        
 @router.post("/emergency_checkout_session")
 def emergency_checkout_session(
     data: EmergencyPaymentRequest,