Update routes/subscription.py

routes/subscription.py

@@ -58,26 +58,29 @@ def create_checkout_session(data: SubscriptionRequest):
             success_url="https://yourdomain.com/success",
             cancel_url="https://yourdomain.com/cancel",
             payment_method_types=["card"],
-            mode="subscription",
+            mode="payment",  # 🔥 Alterado de "subscription" para "payment"
             line_items=[
                 {
                     "price_data": {
                         "currency": "brl",
                         "product_data": {
-                            "name": f"{stylist_name}'s Subscription",
-                            "description": f"✔ {consultations} video call consultations per month",
+                            "name": f"{stylist_name}'s Service",
+                            "description": f"✔ {consultations} consultations",
                             "images": [stylist_avatar],
                         },
                         "unit_amount": stylist_price,
-                        "recurring": {"interval": "month"}
                     },
                     "quantity": 1
                 }
             ],
-            subscription_data={
+            payment_intent_data={  # 🔥 Adicionando transferência automática para o estilista
+                "application_fee_amount": int(stylist_price * 0.2),  # 🔹 20% fica para a plataforma
+                "transfer_data": {
+                    "destination": stylist_stripe_id  # 🔥 Enviando direto para o estilista
+                },
                 "metadata": {
-                    "stylist_id": stylist_stripe_id,  # Agora enviando o ID do Stripe correto
-                    "user_id": data.user_id,  # ID do cliente que está assinando
+                    "stylist_id": stylist_stripe_id,
+                    "user_id": data.user_id,
                     "consultations_per_month": consultations
                 }
             }
@@ -100,6 +103,7 @@ async def stripe_webhook(request: Request):
     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}")
@@ -107,42 +111,46 @@ async def stripe_webhook(request: Request):
 
     logger.info(f"🔹 Event received: {event['type']}")
 
-    if event["type"] == "invoice.payment_succeeded":
-        invoice = event["data"]["object"]
-        subscription_id = invoice.get("subscription")
+    # Verifica se o evento é de uma sessão de checkout completada
+    if event["type"] == "checkout.session.completed":
+        session = event["data"]["object"]
+        payment_intent_id = session.get("payment_intent")
 
-        subscription = stripe.Subscription.retrieve(subscription_id)
-        user_id = subscription.metadata.get("user_id")  # ID do cliente
-        stylist_id = subscription.metadata.get("stylist_id")  # ID do estilista no Stripe
-        consultations_per_month = subscription.metadata.get("consultations_per_month")
+        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"]
 
-        amount = invoice["amount_paid"]
+        # 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
 
-        # 🔹 Transferência do pagamento para o estilista
-        try:
-            transfer = stripe.Transfer.create(
-                amount=stylist_share,
-                currency="brl",
-                destination=stylist_id,
-                description=f"Payment for stylist {stylist_id} - Subscription",
-            )
-            logger.info(f"💸 Transfer successful: R${stylist_share / 100:.2f} BRL to {stylist_id}")
-            transfer_status = "Completed"
-        except Exception as e:
-            logger.error(f"🚨 Transfer error for stylist {stylist_id}: {e}")
-            transfer_status = "Failed"
+        # Aqui, o pagamento já foi transferido para o estilista, então não há mais necessidade de fazer uma transferência manual
+        logger.info(f"💸 Payment of R${amount / 100:.2f} BRL received for stylist {stylist_id}.")
+        logger.info(f"💸 Stylist share: R${stylist_share / 100:.2f} BRL, Platform share: R${platform_share / 100:.2f} BRL.")
+
+        # Aqui você pode atualizar seu banco de dados com as informações do pagamento
+        # Exemplo de atualização do banco com o valor e o status do pagamento
+        # update_payment_status(user_id, stylist_id, amount, stylist_share, platform_share)
 
         return {
             "status": "Payment processed successfully!",
-            "user_id": user_id,  # Retornando o ID do usuário que assinou
-            "stylist_id": stylist_id,
+            "user_id": user_id,  # Retorna o ID do cliente
+            "stylist_id": stylist_id,  # Retorna o ID do estilista
             "consultations_per_month": consultations_per_month,
             "total_paid": amount / 100,
             "stylist_share": stylist_share / 100,
-            "platform_share": platform_share / 100,
-            "transfer_status": transfer_status
+            "platform_share": platform_share / 100
         }
 
     return {"status": "Event received, no action needed."}