Update routes/subscription.py

routes/subscription.py

@@ -32,6 +32,9 @@ SUPABASE_HEADERS = {
     "Content-Type": "application/json"
 }
 
+class EmergencyPaymentRequest(BaseModel):
+    id: str  # ID do estilista
+
 class UserIDRequest(BaseModel):
     user_id: str
 
@@ -466,7 +469,82 @@ async def create_price(
     except Exception as e:
         logger.error(f"❌ Error creating price: {e}")
         raise HTTPException(status_code=500, detail=f"Error creating price: {str(e)}")
-        
+
+@router.post("/emergency_checkout_session")
+def emergency_checkout_session(
+    data: EmergencyPaymentRequest,
+    user_token: str = Header(None, alias="User-key")
+):
+    try:
+        if not user_token:
+            raise HTTPException(status_code=401, detail="Missing User-key header")
+
+        # 🔹 1. Validate client token and get user_id
+        user_id = verify_token(user_token)
+
+        # 🔹 2. Check if user has an active subscription
+        response_subscription = requests.get(
+            f"{SUPABASE_URL}/rest/v1/Subscriptions?customer_id=eq.{user_id}&stylist_id=eq.{data.id}&active=eq.true",
+            headers=SUPABASE_HEADERS
+        )
+        subscription_data = response_subscription.json()
+        if not subscription_data:
+            raise HTTPException(status_code=403, detail="User is not eligible for emergency checkout")
+
+        # 🔹 3. Get stylist data from Supabase
+        response_stylist = requests.get(
+            f"{SUPABASE_URL}/rest/v1/User?id=eq.{data.id}",
+            headers=SUPABASE_HEADERS
+        )
+        stylist_data = response_stylist.json()
+        if not stylist_data:
+            raise HTTPException(status_code=404, detail="Stylist not found")
+
+        stylist = stylist_data[0]
+        stylist_id = stylist.get("id")
+        stylist_stripe_id = stylist.get("stripe_id")
+        emergency_price = stylist.get("emergency_price")
+
+        if not stylist_stripe_id or emergency_price is None:
+            raise HTTPException(status_code=400, detail="Stylist profile is incomplete")
+
+        # 🔹 4. Get client data from Supabase
+        response_user = requests.get(
+            f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}",
+            headers=SUPABASE_HEADERS
+        )
+        user_data = response_user.json()
+        if not user_data:
+            raise HTTPException(status_code=404, detail="Client not found")
+
+        user = user_data[0]
+        user_stripe_id = user.get("stripe_id")
+
+        if not user_stripe_id:
+            raise HTTPException(status_code=400, detail="Client does not have a Stripe Customer ID")
+
+        # 🔹 5. Create a one-time payment intent
+        payment_intent = stripe.PaymentIntent.create(
+            amount=emergency_price,
+            currency="usd",
+            customer=user_stripe_id,
+            application_fee_amount=int(emergency_price * 0.20),  # 20% para a plataforma
+            transfer_data={
+                "destination": stylist_stripe_id,  # 80% para o estilista
+            },
+        )
+
+        return {
+            "message": "Emergency checkout session created successfully!",
+            "clientSecret": payment_intent.client_secret,
+            "paymentIntentId": payment_intent.id,
+            "price": f"{emergency_price / 100:.2f} USD"
+        }
+
+    except Exception as e:
+        logger.error(f"Error creating emergency checkout session: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+
 @router.post("/create_checkout_session")
 def create_checkout_session(
     data: SubscriptionRequest,