Update routes/users.py

routes/users.py

@@ -203,20 +203,18 @@ async def get_recent_users_endpoint(
         logger.error(f"❌ Erro ao obter usuários: {str(e)}")
         raise HTTPException(status_code=500, detail=str(e))
 
-from datetime import datetime
-
-from datetime import datetime
-
 @router.get("/admin/user-transfers")
 async def get_user_transfer_history(
     user_token: str = Header(None, alias="User-key"),
     user_id: str = Query(..., description="ID of the user in Supabase")
 ):
     """
-    Returns the last 10 payments of the user with metadata, stylist info and formatted status.
+    Returns financial information for a user:
+    - If the user is a customer (stripe_id starts with 'cus_'), returns last 10 charges.
+    - If the user is a seller (stripe_id starts with 'acct_'), returns total revenue and upcoming payout info.
     """
     try:
-        # Verify permissions
+        # Verify admin permission
         await verify_token_with_permissions(user_token, "view_users")
 
         # Get user's stripe_id
@@ -230,77 +228,105 @@ async def get_user_transfer_history(
                     raise HTTPException(status_code=404, detail="Stripe ID not found for this user")
                 stripe_id = data[0]["stripe_id"]
 
-        # Fetch charges from Stripe
-        all_charges = stripe.Charge.list(customer=stripe_id, limit=100)
-        total_count = len(all_charges["data"])
-        recent_charges = sorted(all_charges["data"], key=lambda c: c["created"], reverse=True)[:10]
+        # Se for estilista (Stripe Account)
+        if stripe_id.startswith("acct_"):
+            # List all transfers to this connected account
+            transfers = stripe.Transfer.list(destination=stripe_id, limit=100)
+            total_revenue = sum([t["amount"] for t in transfers["data"]]) / 100
 
-        # Status mapping (id → label + color)
-        status_map = {
-            "succeeded": {"label": "Succeeded", "color": "#15803d"},
-            "pending": {"label": "Pending", "color": "#92400e"},
-            "failed": {"label": "Failed", "color": "#991b1b"},
-            "canceled": {"label": "Canceled", "color": "#374151"},
-        }
+            # Próximo pagamento (caso exista)
+            upcoming_payouts = stripe.Payout.list(
+                destination=None,  # Para connected accounts, isso vem automaticamente
+                stripe_account=stripe_id,
+                limit=1,
+                arrival_date={"gte": int(datetime.now().timestamp())}
+            )
 
-        # Helper to fetch stylist info
-        async def fetch_stylist(stylist_id: str) -> Optional[Dict[str, Any]]:
-            url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{stylist_id}&select=name,email,avatar"
-            async with aiohttp.ClientSession() as session:
-                async with session.get(url, headers=SUPABASE_HEADERS) as response:
-                    if response.status == 200:
-                        data = await response.json()
-                        if data:
-                            return data[0]
-            return None
+            upcoming = upcoming_payouts["data"][0] if upcoming_payouts["data"] else None
+
+            return {
+                "stripe_id": stripe_id,
+                "role": "stylist",
+                "total_revenue": total_revenue,
+                "upcoming_payout": {
+                    "amount": upcoming["amount"] / 100 if upcoming else None,
+                    "arrival_date": datetime.fromtimestamp(upcoming["arrival_date"]).isoformat() if upcoming else None,
+                },
+                "transfers_count": len(transfers["data"])
+            }
 
-        # Collect stylist tasks
-        transfers = []
-        tasks = []
+        # Se for cliente (Stripe Customer)
+        elif stripe_id.startswith("cus_"):
+            all_charges = stripe.Charge.list(customer=stripe_id, limit=100)
+            total_count = len(all_charges["data"])
+            recent_charges = sorted(all_charges["data"], key=lambda c: c["created"], reverse=True)[:10]
 
-        for charge in recent_charges:
-            metadata = charge.get("metadata", {})
-            stylist_id = metadata.get("stylist_id")
-            task = fetch_stylist(stylist_id) if stylist_id else None
-            tasks.append(task)
+            # Status mapping (id → label + dark color)
+            status_map = {
+                "succeeded": {"label": "Succeeded", "color": "#15803d"},
+                "pending": {"label": "Pending", "color": "#92400e"},
+                "failed": {"label": "Failed", "color": "#991b1b"},
+                "canceled": {"label": "Canceled", "color": "#374151"},
+            }
 
-        stylist_data_list = await asyncio.gather(*[t if t else asyncio.sleep(0) for t in tasks])
+            async def fetch_stylist(stylist_id: str) -> Optional[Dict[str, Any]]:
+                url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{stylist_id}&select=name,email,avatar"
+                async with aiohttp.ClientSession() as session:
+                    async with session.get(url, headers=SUPABASE_HEADERS) as response:
+                        if response.status == 200:
+                            data = await response.json()
+                            if data:
+                                return data[0]
+                return None
 
-        for i, charge in enumerate(recent_charges):
-            metadata = charge.get("metadata", {})
-            stylist_info = stylist_data_list[i] if i < len(stylist_data_list) else None
+            # Buscar stylists
+            tasks = []
+            for charge in recent_charges:
+                metadata = charge.get("metadata", {})
+                stylist_id = metadata.get("stylist_id")
+                task = fetch_stylist(stylist_id) if stylist_id else None
+                tasks.append(task)
+            stylist_data_list = await asyncio.gather(*[t if t else asyncio.sleep(0) for t in tasks])
 
-            status_id = charge["status"]
-            status_info = status_map.get(status_id, {
-                "label": "Unknown",
-                "color": "#9ca3af"
-            })
+            # Formatando
+            transfers = []
+            for i, charge in enumerate(recent_charges):
+                metadata = charge.get("metadata", {})
+                stylist_info = stylist_data_list[i] if i < len(stylist_data_list) else None
+                status_id = charge["status"]
+                status_info = status_map.get(status_id, {
+                    "label": "Unknown",
+                    "color": "#1f2937"
+                })
+                transfers.append({
+                    "id": charge["id"],
+                    "amount": charge["amount"] / 100,
+                    "currency": charge["currency"].upper(),
+                    "status_info": {
+                        "id": status_id,
+                        "label": status_info["label"],
+                        "color": status_info["color"]
+                    },
+                    "created_at": datetime.fromtimestamp(charge["created"]).isoformat(),
+                    "description": charge.get("description"),
+                    "payment_method": charge.get("payment_method_details", {}).get("type"),
+                    "receipt_url": charge.get("receipt_url"),
+                    "metadata": metadata,
+                    "stylist": stylist_info
+                })
 
-            transfers.append({
-                "id": charge["id"],
-                "amount": charge["amount"] / 100,
-                "currency": charge["currency"].upper(),
-                "status_info": {
-                    "id": status_id,
-                    "label": status_info["label"],
-                    "color": status_info["color"]
-                },
-                "created_at": datetime.fromtimestamp(charge["created"]).isoformat(),
-                "description": charge.get("description"),
-                "payment_method": charge.get("payment_method_details", {}).get("type"),
-                "receipt_url": charge.get("receipt_url"),
-                "metadata": metadata,
-                "stylist": stylist_info
-            })
+            return {
+                "stripe_id": stripe_id,
+                "role": "customer",
+                "transfers": transfers,
+                "count": total_count
+            }
 
-        return {
-            "stripe_id": stripe_id,
-            "transfers": transfers,
-            "count": total_count
-        }
+        else:
+            raise HTTPException(status_code=400, detail="Invalid stripe_id format.")
 
     except HTTPException as he:
         raise he
     except Exception as e:
         logger.error(f"❌ Error retrieving transfers: {str(e)}")
-        raise HTTPException(status_code=500, detail="Internal error while fetching transfer history.")
+        raise HTTPException(status_code=500, detail="Internal error while fetching financial data.")