Update routes/subscription.py

routes/subscription.py

@@ -46,7 +46,7 @@ from fastapi import Header  # Import para pegar headers
 @router.post("/create_price")
 async def create_price(
     data: CreatePriceRequest,
-    user_token: str = Header(None, alias="User-key")  # Pegando o token do Header
+    user_token: str = Header(None, alias="User-key")
 ):
     try:
         if not user_token:
@@ -58,11 +58,11 @@ async def create_price(
         if not amount or not user_id:
             raise HTTPException(status_code=400, detail="Amount and user_id are required")
 
-        # 🔹 Buscar `price_id` salvo no Supabase
+        # 🔹 Buscar `price_id` do usuário no Supabase
         supabase_url = f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}"
         supabase_headers = {
             "apikey": SUPABASE_KEY,
-            "Authorization": f"Bearer {user_token}",  # 🔥 Passando token do usuário
+            "Authorization": f"Bearer {user_token}",
             "Content-Type": "application/json"
         }
 
@@ -75,39 +75,60 @@ async def create_price(
         user = user_data[0]
         existing_price_id = user.get("price_id")
 
-        # 🔥 Se já existir um price_id, desativar o preço antigo e criar um novo
+        # 🔹 Se já existir um price_id, tentar atualizar
         if existing_price_id:
             try:
-                stripe.Price.modify(existing_price_id, active=False)  # Desativando preço antigo
-                logger.info(f"🔹 Disabled old price: {existing_price_id}")
-            except Exception as e:
-                logger.warning(f"⚠️ Could not disable old price: {e}")
-
-        # 🔹 Criar novo preço no Stripe
-        price = stripe.Price.create(
-            unit_amount=amount,
-            currency="brl",
-            recurring={"interval": "month"},
-            product_data={"name": "Custom Subscription Price"}
-        )
-
-        logger.info(f"✅ Price created successfully: {price.id} for user {user_id}")
-
-        # 🔹 Atualizar Supabase com novo `price_id`
-        response_update = requests.patch(
+                stripe.Price.modify(
+                    existing_price_id,
+                    unit_amount=amount,
+                    metadata={"updated_at": "now"}  # Apenas para gerar um evento de atualização
+                )
+                logger.info(f"✅ Price {existing_price_id} updated successfully.")
+                updated_price_id = existing_price_id
+            except stripe.error.InvalidRequestError:
+                logger.warning(f"⚠️ Failed to update price {existing_price_id}, creating a new one instead.")
+                updated_price_id = None
+        else:
+            updated_price_id = None
+
+        # 🔹 Se não conseguiu atualizar, criar um novo preço
+        if not updated_price_id:
+            price = stripe.Price.create(
+                unit_amount=amount,
+                currency="brl",
+                recurring={"interval": "month"},
+                product_data={"name": "Custom Subscription Price"}
+            )
+            updated_price_id = price.id
+            logger.info(f"✅ New price created: {updated_price_id}")
+
+            # 🔹 Pausar todas as assinaturas associadas ao preço antigo
+            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:
+                        stripe.Subscription.modify(sub.id, pause_collection={"behavior": "void"})
+                        logger.info(f"🔹 Subscription {sub.id} paused.")
+
+                # 🔹 Desativar o preço antigo
+                stripe.Price.modify(existing_price_id, active=False)
+                logger.info(f"🚫 Price {existing_price_id} deactivated.")
+
+        # 🔹 Atualizar Supabase com o novo `price_id`
+        update_response = requests.patch(
             supabase_url,
             headers=supabase_headers,
-            json={"price_id": price.id}
+            json={"price_id": updated_price_id}
         )
 
-        if response_update.status_code not in [200, 204]:
-            raise HTTPException(status_code=500, detail=f"Failed to update Supabase: {response_update.text}")
+        if update_response.status_code not in [200, 204]:
+            raise HTTPException(status_code=500, detail=f"Failed to update Supabase: {update_response.text}")
 
-        return {"message": "Price created successfully!", "price_id": price.id}
+        return {"message": "Price created or updated successfully!", "price_id": updated_price_id}
 
     except Exception as e:
-        logger.error(f"❌ Error creating price: {e}")
-        raise HTTPException(status_code=500, detail="Error creating price.")
+        logger.error(f"❌ Error creating/updating price: {e}")
+        raise HTTPException(status_code=500, detail="Error creating/updating price.")
         
 @router.post("/create_checkout_session")
 def create_checkout_session(data: SubscriptionRequest):