Update routes/subscription.py

routes/subscription.py

@@ -140,6 +140,31 @@ async def create_price(
         logger.error(f"❌ Error creating/updating price: {e}")
         raise HTTPException(status_code=500, detail="Error creating/updating price.")
 
+@router.post("/webhook")
+async def stripe_webhook(request: Request):
+    try:
+        payload = await request.json()
+
+        event_type = payload.get("type")
+        if event_type == "invoice.payment_succeeded":
+            invoice = payload.get("data", {}).get("object", {})
+            amount_paid = invoice.get("amount_paid", 0)  # Valor total pago (em centavos)
+
+            # Cálculo de divisão
+            stylist_amount = int(amount_paid * 0.8)  # 80% para o estilista
+            platform_amount = int(amount_paid * 0.2)  # 20% para a plataforma
+
+            # Logando os valores
+            logger.info(f"✅ Pagamento bem-sucedido! Valor total: R$ {amount_paid / 100:.2f}")
+            logger.info(f"💰 Estilista recebe: R$ {stylist_amount / 100:.2f}")
+            logger.info(f"🏛️ Plataforma fica com: R$ {platform_amount / 100:.2f}")
+
+        return {"status": "success"}
+
+    except Exception as e:
+        logger.error(f"❌ Erro no webhook: {str(e)}")
+        return {"status": "error", "message": str(e)}
+        
 @router.post("/create_checkout_session")
 def create_checkout_session(
     data: SubscriptionRequest,
@@ -225,95 +250,6 @@ def create_checkout_session(
         logger.error(f"Error creating checkout session: {e}")
         raise HTTPException(status_code=500, detail="Error creating checkout session.")
         
-### **WEBHOOK PARA PROCESSAR PAGAMENTOS**
-@router.post("/webhook")
-async def stripe_webhook(request: Request):
-    payload = await request.body()
-    headers = dict(request.headers)
-    event = None
-    
-    try:
-        # Tenta construir o evento Stripe a partir do payload recebido
-        event = stripe.Event.construct_from(json.loads(payload), stripe.api_key)
-    except Exception as e:
-        logger.error(f"⚠️ Error processing webhook: {e}")
-        raise HTTPException(status_code=400, detail="Webhook error")
-
-    logger.info(f"🔹 Event received: {event['type']}")
-
-    if event["type"] == "checkout.session.completed":
-        session = event["data"]["object"]
-        payment_intent_id = session.get("payment_intent")
-
-        # Verifica se o PaymentIntent ID está presente
-        if not payment_intent_id:
-            logger.error("⚠️ No PaymentIntent ID found in session.")
-            # Verifica se a sessão de checkout tem um PaymentIntent associado
-            if "subscription" in session:
-                subscription_id = session["subscription"]
-                subscription = stripe.Subscription.retrieve(subscription_id)
-                payment_intent_id = subscription.latest_invoice.payment_intent
-                logger.info(f"🔹 Found PaymentIntent from subscription: {payment_intent_id}")
-            else:
-                raise HTTPException(status_code=400, detail="PaymentIntent ID missing.")
-
-        try:
-            # Recupera o PaymentIntent associado à sessão
-            payment_intent = stripe.PaymentIntent.retrieve(payment_intent_id)
-        except Exception as e:
-            logger.error(f"⚠️ Error retrieving PaymentIntent: {e}")
-            raise HTTPException(status_code=400, detail="Error retrieving PaymentIntent")
-
-        # Extrai os dados do metadata
-        user_id = session.metadata.get("user_id")  # ID do cliente
-        stylist_id = session.metadata.get("stylist_id")  # ID do estilista no Stripe
-        consultations_per_month = session.metadata.get("consultations_per_month")  # Consultas por mês
-
-        # Verifica o valor pago
-        amount = payment_intent["amount_received"]
-
-        # Define a divisão do pagamento entre a plataforma e o estilista
-        stylist_share = int(amount * 0.8)  # 80% para o estilista
-        platform_share = amount - stylist_share  # 20% para a plataforma
-
-        # Realizando a transferência para o estilista
-        try:
-            transfer = stripe.Transfer.create(
-                amount=stylist_share,  # Montante para o estilista (80%)
-                currency="brl",  # Moeda
-                destination=stylist_id,  # Conta do estilista
-                description="Payment for styling service"
-            )
-            logger.info(f"💸 Transfer to stylist {stylist_id} successful: R${stylist_share / 100:.2f} BRL.")
-        except Exception as e:
-            logger.error(f"⚠️ Error transferring funds to stylist: {e}")
-            raise HTTPException(status_code=500, detail="Error transferring funds to stylist.")
-
-        # Transferência para a plataforma
-        try:
-            transfer_platform = stripe.Transfer.create(
-                amount=platform_share,  # Montante para a plataforma (20%)
-                currency="brl",  # Moeda
-                destination="your_platform_stripe_account_id",  # Substitua com sua conta Stripe da plataforma
-                description="Platform share"
-            )
-            logger.info(f"💸 Transfer to platform successful: R${platform_share / 100:.2f} BRL.")
-        except Exception as e:
-            logger.error(f"⚠️ Error transferring funds to platform: {e}")
-            raise HTTPException(status_code=500, detail="Error transferring funds to platform.")
-
-        return {
-            "status": "Payment processed successfully!",
-            "user_id": user_id,
-            "stylist_id": stylist_id,
-            "consultations_per_month": consultations_per_month,
-            "total_paid": amount / 100,
-            "stylist_share": stylist_share / 100,
-            "platform_share": platform_share / 100
-        }
-
-    return {"status": "Event received, no action needed."}
-
 ### **CANCELAMENTO DE ASSINATURA**
 class CancelSubscriptionRequest(BaseModel):
     subscription_id: str