Update routes/subscription.py

routes/subscription.py

@@ -796,15 +796,12 @@ async def stripe_webhook(request: Request):
         return {"status": "error", "message": str(e)}
         
 @router.post("/create_price")
-async def create_price(
-    data: CreatePriceRequest,
-    user_token: str = Header(None, alias="User-key")
-):
+async def create_price(data: CreatePriceRequest, user_token: str = Header(None, alias="User-key")):
     try:
         if not user_token:
             raise HTTPException(status_code=401, detail="Missing User-key header")
 
-        # 🔹 1. Validar o token e obter user_id
+        # 1. Validar o token e obter user_id
         user_id = verify_token(user_token)
         logger.info(f"🔹 User verified. user_id: {user_id}")
 
@@ -812,13 +809,13 @@ async def create_price(
         emergency_price = data.emergency_price
         consultations = data.consultations
 
-        # 🔹 2. Verificar se os valores estão dentro dos limites permitidos
+        # 2. Verificar se os valores estão dentro dos limites permitidos
         if not (500 <= amount <= 99900):
             raise HTTPException(status_code=400, detail="Amount must be between $5 and $999")
         if not (500 <= emergency_price <= 99900):
             raise HTTPException(status_code=400, detail="Emergency price must be between $5 and $999")
 
-        # 🔹 3. Consultations precisa ser 0 obrigatoriamente se não for definido
+        # 3. Consultations precisa ser 0 obrigatoriamente se não for definido
         if consultations is None:
             consultations = 0
         elif consultations < 0:
@@ -826,10 +823,10 @@ async def create_price(
 
         logger.info(f"🔹 Validated amounts: amount = {amount}, emergency_price = {emergency_price}, consultations = {consultations}")
 
-        # 🔹 4. Buscar price_id, name, avatar e bio do usuário no Supabase
+        # 4. Buscar price_id, name, avatar e bio do usuário no Supabase
         supabase_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}"
         headers = {
-            "Authorization": f"Bearer {user_token}",  # 🔹 Incluindo token correto no header
+            "Authorization": f"Bearer {user_token}",
             **SUPABASE_HEADERS
         }
         response = requests.get(supabase_url, headers=headers)
@@ -844,10 +841,10 @@ async def create_price(
 
         user = user_data[0]
         existing_price_id = user.get("price_id")
-        user_full_name = user.get("name", "Unknown User")  # 🔹 Recuperar o nome do usuário
-        user_avatar = user.get("avatar", None)  # 🔹 Recuperar a URL do avatar do usuário, se existir
+        user_full_name = user.get("name", "Unknown User")
+        user_avatar = user.get("avatar", None)
 
-        # 🔹 Criar nome abreviado do usuário (primeiro nome + inicial do segundo nome)
+        # Criar nome abreviado do usuário
         name_parts = user_full_name.split()
         if len(name_parts) > 1:
             abbreviated_name = f"{name_parts[0]} {name_parts[1][0]}."
@@ -858,60 +855,62 @@ async def create_price(
         
         logger.info(f"🔹 Existing price_id: {existing_price_id}, user_name: {abbreviated_name}")
 
-        # 🔹 5. Criar produto no Stripe com nome e imagem baseados no usuário
+        # 5. Criar produto no Stripe
         product_data = {
-            "name": product_name,  # Nome do produto formatado
+            "name": product_name,
         }
         
-        if user_avatar:  # Verificar se o avatar existe e adicionar a imagem
+        if user_avatar:
             product_data["images"] = [user_avatar]
 
         product = stripe.Product.create(**product_data)
         product_id = product.id
         logger.info(f"✅ New product created: {product_id}")
 
-        # 🔹 6. Criar novo preço no Stripe associado ao produto
+        # 6. Criar novo preço no Stripe
         price = stripe.Price.create(
             unit_amount=amount,
             currency="brl",
             recurring={"interval": "month"},
-            product=product_id  # Associando o preço ao produto
+            product=product_id
         )
         new_price_id = price.id
         logger.info(f"✅ New price created: {new_price_id}")
 
-        # 🔹 7. Se já houver um price_id, cancelar e modificar assinaturas ativas com o novo preço
+        # 7. Cancelar as assinaturas existentes e aplicar o novo price_id sem descontos
         if existing_price_id:
             subscriptions = stripe.Subscription.list(status="active")
             for sub in subscriptions.auto_paging_iter():
                 if sub["items"]["data"][0]["price"]["id"] == existing_price_id:
-                    # Atualiza o price_id antes de cancelar a assinatura
-                    stripe.Subscription.modify(sub.id, items=[{
-                        "id": sub["items"]["data"][0]["id"], 
-                        "price": new_price_id
-                    }], cancel_at_period_end=True)
+                    # Cancelar assinatura no final do período
+                    stripe.Subscription.modify(sub.id, cancel_at_period_end=True)
+                    # Aplicar o novo preço na próxima fatura
+                    stripe.Subscription.modify(
+                        sub.id,
+                        items=[{
+                            "id": sub["items"]["data"][0]["id"],
+                            "price": new_price_id
+                        }],
+                        proration_behavior="none"  # Garantir que não haja prorrogação ou descontos
+                    )
                     logger.info(f"🔹 Subscription {sub.id} set to cancel at period end and updated with new price {new_price_id}.")
 
-        # 🔹 8. Atualizar Supabase com o novo price_id e valores adicionais
+        # 8. Atualizar Supabase com o novo price_id
         update_data = {
-            "price_id": new_price_id,  # Novo price_id
-            "price": amount,  # Atualizando o valor de 'price'
-            "emergency_price": emergency_price,  # Atualizando o valor de 'emergency_price'
-            "consultations": consultations  # Atualizando o valor de 'consultations'
+            "price_id": new_price_id,
+            "price": amount,
+            "emergency_price": emergency_price,
+            "consultations": consultations
         }
         
-        # Aqui estamos incluindo o cabeçalho de autenticação com o token JWT correto
         update_headers = {
-            "Authorization": f"Bearer {user_token}",  # Incluindo token correto no header
+            "Authorization": f"Bearer {user_token}",
             "apikey": SUPABASE_KEY,
             "Content-Type": "application/json"
         }
         
         update_response = requests.patch(supabase_url, headers=update_headers, json=update_data)
         
-        # Log detalhado para verificar a resposta
-        logger.info(f"🔹 Supabase PATCH response: {update_response.status_code} - {update_response.text}")
-        
         if update_response.status_code not in [200, 204]:
             raise HTTPException(status_code=500, detail=f"Failed to update Supabase: {update_response.text}")