Update routes/subscription.py

routes/subscription.py

@@ -19,7 +19,7 @@ class PriceRequest(BaseModel):
 
 class SubscriptionRequest(BaseModel):
     user_id: str
-    stylist_id: str
+    stylist_id: str  # This must be the Stripe connected account ID (acct_xxxxx)
     price_id: str  # Now using a pre-created price
 
 class CancelSubscriptionRequest(BaseModel):
@@ -88,7 +88,7 @@ def create_checkout_session(data: SubscriptionRequest):
         logger.error(f"Error creating checkout session: {e}")
         raise HTTPException(status_code=500, detail="Error creating checkout session.")
 
-### **3️⃣ WEBHOOK TO PROCESS PAYMENTS** ###
+### **3️⃣ WEBHOOK TO PROCESS PAYMENTS & TRANSFER FUNDS** ###
 @router.post("/webhook")
 async def stripe_webhook(request: Request):
     payload = await request.body()
@@ -104,39 +104,43 @@ 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")
+        invoice = event["data"]["object"]
+        subscription_id = invoice.get("subscription")  # Get subscription ID
+
+        # Retrieve subscription to get metadata
+        subscription = stripe.Subscription.retrieve(subscription_id)
+        user_id = subscription.metadata.get("user_id")
+        stylist_id = subscription.metadata.get("stylist_id")  # Must be a Stripe connected account ID
+
+        amount = invoice["amount_paid"]
+        stylist_share = int(amount * 0.8)  # 80% for stylist
+        platform_share = amount - stylist_share  # 20% for platform
+
+        # 🔹 Transfer the funds to the stylist
+        try:
+            transfer = stripe.Transfer.create(
+                amount=stylist_share,  # Value in CENTS
+                currency="usd",
+                destination=stylist_id,  # The Stripe Connected Account ID (acct_xxxx)
+                description=f"Payment for stylist {stylist_id} - Subscription",
+            )
+            logger.info(f"💸 Transfer successful: ${stylist_share / 100:.2f} USD to {stylist_id}")
+            transfer_status = "Completed"
+        except Exception as e:
+            logger.error(f"🚨 Transfer error for stylist {stylist_id}: {e}")
+            transfer_status = "Failed"
 
-    # Retrieve subscription metadata
-    subscription = stripe.Subscription.retrieve(subscription_id)
-    user_id = subscription.metadata.get("user_id")
-    stylist_id = subscription.metadata.get("stylist_id")
-
-    amount = invoice["amount_paid"]
-    stylist_share = int(amount * 0.8)  # 80% for stylist
-    platform_share = amount - stylist_share  # 20% for platform
+        return {
+            "status": "Payment processed successfully!",
+            "user_id": user_id,
+            "stylist_id": stylist_id,
+            "total_paid": amount / 100,  # Convert cents to dollars
+            "stylist_share": stylist_share / 100,
+            "platform_share": platform_share / 100,
+            "transfer_status": transfer_status
+        }
 
-    # 🔹 Adicionar transferência para o estilista
-    try:
-        transfer = stripe.Transfer.create(
-            amount=stylist_share,  # Valor em CENTAVOS
-            currency="usd",
-            destination=stylist_id,  # Conta Stripe do estilista (Connected Account ID)
-            description=f"Payment for stylist {stylist_id} - Subscription",
-        )
-        logger.info(f"💸 Transferência realizada: {stylist_share / 100:.2f} USD para {stylist_id}")
-    except Exception as e:
-        logger.error(f"🚨 Erro ao transferir para o estilista {stylist_id}: {e}")
-
-    return {
-        "status": "Payment processed and transferred successfully!",
-        "user_id": user_id,
-        "stylist_id": stylist_id,
-        "total_paid": amount / 100,
-        "stylist_share": stylist_share / 100,
-        "platform_share": platform_share / 100,
-        "transfer_status": "Completed" if "transfer" in locals() else "Failed"
-    }
+    return {"status": "Event received, no action needed."}
 
 ### **4️⃣ CANCEL SUBSCRIPTION** ###
 @router.post("/cancel_subscription")