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")
+    user_token: str = Header(None, alias="User-key")  # Pegando o token do Header
 ):
     try:
         if not user_token:
@@ -58,12 +58,12 @@ 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 atual do usuário no Supabase
+        # 🔹 1. Buscar o price_id atual 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}",
+            "Authorization": f"Bearer {user_token}",  # 🔥 Passando token do usuário
             "Content-Type": "application/json"
         }
 
@@ -71,66 +71,51 @@ async def create_price(
         user_data = response.json()
 
         if not user_data:
-            raise HTTPException(status_code=404, detail="User not found in Supabase")
+            raise HTTPException(status_code=404, detail="User not found")
 
         user = user_data[0]
-        price_id = user.get("price_id")  # Pega o price_id salvo
+        current_price_id = user.get("price_id")  # 🔍 Pegamos o price_id atual do usuário
 
-        # 🔹 Se o usuário já tem um price_id salvo, tentar buscar no Stripe
-        current_price_valid = False
-        if price_id:
+        # 🔹 2. Se o price_id existir, tentar atualizar
+        if current_price_id:
             try:
-                price = stripe.Price.retrieve(price_id)
-                
-                if price.unit_amount == amount:  # Se o preço for igual, podemos reutilizar
-                    current_price_valid = True
-                    logger.info(f"✅ Reusing existing price {price_id} (same amount) for user {user_id}")
-                else:
-                    # 🔥 1º TENTATIVA: Tentar desativar o preço antigo
-                    try:
-                        stripe.Price.modify(price_id, active=False)
-                        logger.info(f"🔄 Deactivated old price {price_id}")
-                    except stripe.error.InvalidRequestError:
-                        logger.warning(f"⚠️ Failed to deactivate price {price_id}, trying to archive...")
-                    
-                        # 🔥 2º TENTATIVA: Se não pode desativar, arquiva
-                        try:
-                            stripe.Price.modify(price_id, metadata={"archived": "true"})
-                            logger.info(f"📦 Archived price {price_id}")
-                        except Exception as e:
-                            logger.error(f"❌ Could not archive price {price_id}: {e}")
-
+                stripe.Price.modify(
+                    current_price_id,
+                    active=True,  # Mantém ativo
+                    metadata={"updated_at": "now"}  # Apenas um metadado de controle
+                )
+                logger.info(f"🔄 Price updated successfully: {current_price_id}")
+                return {"message": "Price updated successfully!", "price_id": current_price_id}
             except stripe.error.InvalidRequestError:
-                logger.warning(f"⚠️ Invalid price_id ({price_id}), creating a new one...")
-                price_id = None  # Se for inválido, vamos criar um novo
-
-        # 🔹 Criar um novo preço apenas se necessário
-        if not current_price_valid:
-            price = stripe.Price.create(
-                unit_amount=amount,
-                currency="brl",
-                recurring={"interval": "month"},
-                product_data={"name": "Custom Subscription Price"}
-            )
-            logger.info(f"✅ Created new price {price.id} for user {user_id}")
-
-        # 🔹 Atualizar Supabase com o novo price_id
+                logger.warning(f"⚠️ Invalid price_id: {current_price_id}, creating a new one...")
+
+        # 🔹 3. Se o price_id não existir ou for inválido, criar um novo
+        price = stripe.Price.create(
+            unit_amount=amount,
+            currency="brl",
+            recurring={"interval": "month"},
+            product_data={"name": "Custom Subscription Price"}
+        )
+
+        logger.info(f"✅ New price created: {price.id} for user {user_id}")
+
+        # 🔹 4. Atualizar Supabase com o novo price_id
         update_response = requests.patch(
             supabase_url,
             headers=supabase_headers,
             json={"price_id": price.id}
         )
 
-        logger.info(f"🔹 Supabase update response: {update_response.status_code}, {update_response.text}")
+        logger.info(f"🔹 Supabase 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}")
 
-        return {"message": "Price created/updated successfully!", "price_id": price.id}
+        return {"message": "New price created successfully!", "price_id": 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):