Create notifications.py

routes/notifications.py

@@ -0,0 +1,98 @@
+import os
+import aiohttp
+from fastapi import APIRouter, HTTPException, Header
+from pydantic import BaseModel
+from google.oauth2 import service_account
+from google.auth.transport.requests import Request
+
+router = APIRouter()
+
+# Supabase Config
+SUPABASE_URL = "https://ussxqnifefkgkaumjann.supabase.co"
+SUPABASE_KEY = os.getenv("SUPA_KEY")
+SUPABASE_SERVICE_KEY = os.getenv("SUPA_SERVICE_KEY")
+
+if not SUPABASE_KEY or not SUPABASE_SERVICE_KEY:
+    raise ValueError("❌ SUPA_KEY or SUPA_SERVICE_KEY not set in environment!")
+
+SUPABASE_HEADERS = {
+    "apikey": SUPABASE_SERVICE_KEY,
+    "Authorization": f"Bearer {SUPABASE_SERVICE_KEY}",
+    "Content-Type": "application/json"
+}
+
+# Firebase config
+SERVICE_ACCOUNT_FILE = './closetcoach-2d50b-firebase-adminsdk-fbsvc-7fcccbacb1.json'
+FCM_PROJECT_ID = "closetcoach-2d50b"
+
+# Request body
+class SimpleNotification(BaseModel):
+    target: str  # "all" or user ID
+    title: str
+    content: str
+    image_url: str = ""
+
+# Firebase Auth
+def get_fcm_access_token():
+    credentials = service_account.Credentials.from_service_account_file(
+        SERVICE_ACCOUNT_FILE
+    )
+    scoped = credentials.with_scopes(
+        ['https://www.googleapis.com/auth/firebase.messaging']
+    )
+    scoped.refresh(Request())
+    return scoped.token
+
+# Fetch user info from Supabase
+async def get_user(user_id: str):
+    url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}&select=id,token_fcm,manage_notifications"
+    async with aiohttp.ClientSession() as session:
+        async with session.get(url, headers=SUPABASE_HEADERS) as resp:
+            if resp.status != 200:
+                raise HTTPException(status_code=resp.status, detail="Failed to fetch user info")
+            data = await resp.json()
+            return data[0] if data else None
+
+@router.post("/send-global-notification")
+async def send_global_notification(
+    payload: SimpleNotification,
+    user_id: str = Header(..., alias="User-id")
+):
+    # Autorização
+    sender = await get_user(user_id)
+    if not sender or not sender.get("manage_notifications"):
+        raise HTTPException(status_code=403, detail="Not authorized to send notifications")
+
+    message = {
+        "notification": {
+            "title": payload.title,
+            "body": payload.content
+        }
+    }
+
+    if payload.image_url:
+        message["notification"]["image"] = payload.image_url
+
+    if payload.target == "all":
+        message["topic"] = "all"
+    else:
+        # Busca o token FCM do usuário de destino
+        target_user = await get_user(payload.target)
+        if not target_user or not target_user.get("token_fcm"):
+            raise HTTPException(status_code=404, detail="Target user or FCM token not found")
+        message["token"] = target_user["token_fcm"]
+
+    # Enviar para Firebase
+    access_token = get_fcm_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={"message": message}) as resp:
+            response_text = await resp.text()
+            if resp.status != 200:
+                raise HTTPException(status_code=resp.status, detail=f"FCM error: {response_text}")
+            return {"detail": "Notification sent successfully"}