Update routes/subscription.py

routes/subscription.py

@@ -58,37 +58,62 @@ async def create_price(
         if not amount or not user_id:
             raise HTTPException(status_code=400, detail="Amount and user_id are required")
 
-        # 🔹 Criar 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 price_id, autenticando com User-key
+        # 🔹 Buscar 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}",  # 🔥 Passando token do usuário
+            "Authorization": f"Bearer {user_token}",  # 🔥 Token do usuário
             "Content-Type": "application/json"
         }
 
-        response = requests.patch(
+        response = requests.get(supabase_url, headers=supabase_headers)
+        user_data = response.json()
+
+        if not user_data:
+            raise HTTPException(status_code=404, detail="User not found in Supabase")
+
+        user = user_data[0]
+        price_id = user.get("price_id")  # Pega o price_id salvo
+
+        # 🔹 Se o usuário já tem um price_id salvo, tentar atualizar no Stripe
+        if price_id:
+            try:
+                price = stripe.Price.modify(
+                    price_id,
+                    active=False  # Desativa o antigo antes de criar um novo
+                )
+                logger.info(f"🔄 Deactivated old price {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 se não existir um válido
+        if not price_id:
+            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}")
+        else:
+            price = stripe.Price.retrieve(price_id)
+            logger.info(f"✅ Reusing existing price {price.id} for user {user_id}")
+
+        # 🔹 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 response: {response.status_code}, {response.text}")
+        logger.info(f"🔹 Supabase update response: {update_response.status_code}, {update_response.text}")
 
-        if response.status_code not in [200, 204]:
-            raise HTTPException(status_code=500, detail=f"Failed to update Supabase: {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 successfully!", "price_id": price.id}
+        return {"message": "Price created/updated successfully!", "price_id": price.id}
 
     except Exception as e:
         logger.error(f"❌ Error creating price: {e}")