Update routes/subscription.py

routes/subscription.py

@@ -60,6 +60,48 @@ class CreatePriceRequest(BaseModel):
     emergency_price: int  # Valor de emergência (ex: 500 para R$5,00)
     consultations: int  # Número de consultas (ex: 3)
 
+def get_active_subscribers_by_price_id(price_id: str) -> list:
+    """
+    Retorna uma lista de user_ids que têm uma assinatura ativa com o price_id fornecido.
+    """
+    response = requests.get(
+        f"{SUPABASE_URL}/rest/v1/Subscriptions?price_id=eq.{price_id}&select=user_id",
+        headers=SUPABASE_ROLE_HEADERS
+    )
+    if response.status_code == 200:
+        data = response.json()
+        return [entry["user_id"] for entry in data if "user_id" in entry]
+    logger.warning(f"⚠️ Failed to fetch subscribers: {response.status_code} - {response.text}")
+    return []
+
+def create_notifications_for_price_change(user_ids: list, stylist_id: str):
+    """
+    Cria registros de notificações na tabela 'Notifications' para todos os usuários afetados.
+    """
+    if not user_ids:
+        return
+
+    now = datetime.now(pytz.timezone("America/Sao_Paulo")).isoformat()
+    payload = [{
+        "type": "changeprice",
+        "target_user": uid,
+        "created_at": now,
+        "user_reference": stylist_id,
+        "read": False,
+        "post_reference": None
+    } for uid in user_ids]
+
+    response = requests.post(
+        f"{SUPABASE_URL}/rest/v1/Notifications",
+        headers=SUPABASE_ROLE_HEADERS,
+        json=payload
+    )
+
+    if response.status_code not in [200, 201]:
+        logger.warning(f"⚠️ Failed to create notifications: {response.status_code} - {response.text}")
+    else:
+        logger.info(f"✅ Created {len(payload)} notifications for affected users.")
+        
 def get_subscription_from_db(subscription_id: str):
     """
     Busca a assinatura na tabela 'Subscriptions' com base no subscription_id.
@@ -837,35 +879,33 @@ async def create_price(data: CreatePriceRequest, user_token: str = Header(None,
         if not user_token:
             raise HTTPException(status_code=401, detail="Missing User-key header")
 
+        # 1. Validar o token e obter user_id
         user_id = verify_token(user_token)
         logger.info(f"🔹 User verified. user_id: {user_id}")
 
         amount = data.amount
         emergency_price = data.emergency_price
-        consultations = data.consultations
+        consultations = data.consultations or 0
 
         if not (500 <= amount <= 99900):
             raise HTTPException(status_code=400, detail="Amount must be between $5 and $999")
         if not (500 <= emergency_price <= 99900):
             raise HTTPException(status_code=400, detail="Emergency price must be between $5 and $999")
+        if consultations < 0:
+            raise HTTPException(status_code=400, detail="Consultations must be >= 0")
 
-        if consultations is None:
-            consultations = 0
-        elif consultations < 0:
-            raise HTTPException(status_code=400, detail="Consultations must be greater than or equal to 0")
-
-        logger.info(f"🔹 Validated amounts: amount = {amount}, emergency_price = {emergency_price}, consultations = {consultations}")
+        logger.info(f"🔹 Validated inputs")
 
+        # 4. Buscar dados do usuário
         supabase_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}"
         headers = {
             "Authorization": f"Bearer {user_token}",
             **SUPABASE_HEADERS
         }
         response = requests.get(supabase_url, headers=headers)
-        logger.info(f"🔹 Supabase GET response: {response.status_code} - {response.text}")
 
         if response.status_code != 200:
-            raise HTTPException(status_code=500, detail=f"Failed to fetch user from Supabase: {response.text}")
+            raise HTTPException(status_code=500, detail=f"Failed to fetch user: {response.text}")
 
         user_data = response.json()
         if not user_data:
@@ -873,28 +913,21 @@ async def create_price(data: CreatePriceRequest, user_token: str = Header(None,
 
         user = user_data[0]
         existing_price_id = user.get("price_id")
-        user_full_name = user.get("name", "Unknown User")
-        user_avatar = user.get("avatar", None)
+        user_full_name = user.get("name", "Unknown")
+        user_avatar = user.get("avatar")
 
         name_parts = user_full_name.split()
-        if len(name_parts) > 1:
-            abbreviated_name = f"{name_parts[0]} {name_parts[1][0]}."
-        else:
-            abbreviated_name = name_parts[0]
-
+        abbreviated_name = f"{name_parts[0]} {name_parts[1][0]}." if len(name_parts) > 1 else name_parts[0]
         product_name = f"{abbreviated_name} Subscription"
-        logger.info(f"🔹 Existing price_id: {existing_price_id}, user_name: {abbreviated_name}")
 
-        product_data = {
-            "name": product_name,
-        }
+        logger.info(f"🔹 Creating Stripe product")
 
+        product_data = {"name": product_name}
         if user_avatar:
             product_data["images"] = [user_avatar]
 
         product = stripe.Product.create(**product_data)
         product_id = product.id
-        logger.info(f"✅ New product created: {product_id}")
 
         price = stripe.Price.create(
             unit_amount=amount,
@@ -903,19 +936,14 @@ async def create_price(data: CreatePriceRequest, user_token: str = Header(None,
             product=product_id
         )
         new_price_id = price.id
-        logger.info(f"✅ New price created: {new_price_id}")
+        logger.info(f"✅ Stripe product and price created: {new_price_id}")
 
-        # Coletar e logar todos os clientes com assinatura ativa do preço antigo
-        affected_customers = []
+        affected_users = []
 
         if existing_price_id:
             subscriptions = stripe.Subscription.list(status="active")
             for sub in subscriptions.auto_paging_iter():
                 if sub["items"]["data"][0]["price"]["id"] == existing_price_id:
-                    customer_id = sub["customer"]
-                    affected_customers.append(customer_id)
-
-                    # Cancelar no final do período e atualizar para novo preço
                     stripe.Subscription.modify(sub.id, cancel_at_period_end=True)
                     stripe.Subscription.modify(
                         sub.id,
@@ -925,10 +953,11 @@ async def create_price(data: CreatePriceRequest, user_token: str = Header(None,
                         }],
                         proration_behavior="none"
                     )
-                    logger.info(f"🔹 Subscription {sub.id} updated with new price {new_price_id} (customer: {customer_id})")
-
-        logger.info(f"📢 Affected customer IDs with old price: {affected_customers}")
+            # 🔥 Buscar todos os usuários afetados no Supabase
+            affected_users = get_active_subscribers_by_price_id(existing_price_id)
+            logger.info(f"🔹 Found {len(affected_users)} affected users.")
 
+        # Atualiza o usuário no Supabase
         update_data = {
             "price_id": new_price_id,
             "price": amount,
@@ -945,14 +974,14 @@ async def create_price(data: CreatePriceRequest, user_token: str = Header(None,
         update_response = requests.patch(supabase_url, headers=update_headers, json=update_data)
 
         if update_response.status_code not in [200, 204]:
-            raise HTTPException(status_code=500, detail=f"Failed to update Supabase: {update_response.text}")
+            raise HTTPException(status_code=500, detail=f"Failed to update user: {update_response.text}")
 
-        logger.info(f"✅ Successfully updated user {user_id} with new price_id and prices")
-        return {
-            "message": "Price created and user updated successfully!",
-            "price_id": new_price_id,
-            "affected_customers": affected_customers
-        }
+        # 🔥 Cria notificações para os afetados
+        create_notifications_for_price_change(affected_users, stylist_id=user_id)
+
+        logger.info(f"✅ User updated and notifications sent")
+
+        return {"message": "Price created and user updated successfully!", "price_id": new_price_id}
 
     except Exception as e:
         logger.error(f"❌ Error creating price: {e}")