Update routes/subscription.py

routes/subscription.py

@@ -473,17 +473,16 @@ def cancel_subscription(data: CancelSubscriptionRequest):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-# 🚀 Verificação de Assinatura
 @router.post("/check_subscription")
 def check_subscription(
     data: CheckSubscriptionRequest,
-    user_token: str = Header(None, alias="User-key")  # Recebendo o token do cabeçalho
+    user_token: str = Header(None, alias="User-key")
 ):
     try:
         if not user_token:
             raise HTTPException(status_code=401, detail="Missing User-key header")
         
-        # 🔹 Validar o token do cliente e obter user_id (cliente)
+        # 🔹 Validar o token e obter user_id do cliente
         user_id = verify_token(user_token)
 
         # 🔹 Buscar o stripe_id do usuário no Supabase
@@ -499,82 +498,73 @@ def check_subscription(
         if not user_stripe_id:
             raise HTTPException(status_code=404, detail="Stripe customer not found for user")
 
-        # 🔹 Buscar todas as assinaturas ativas para o usuário (cliente) com stripe_id
+        # 🔹 Buscar todas as assinaturas ativas para o cliente
         subscriptions = stripe.Subscription.list(
-            customer=user_stripe_id,  # Usando stripe_id do cliente para buscar assinaturas
-            status="active",          # Assinaturas ativas
-            expand=["data.items"]     # Expande os dados para pegar os itens e o metadata
+            customer=user_stripe_id,
+            expand=["data.items"]
         )
 
-        # Logando a resposta do Stripe para verificar sua estrutura
         logger.info(f"Stripe response data: {subscriptions}")
 
         if not subscriptions or "data" not in subscriptions or not subscriptions["data"]:
-            return {
-                "status": "inactive",
-                "message": "No active subscription found for this stylist."
-            }
+            return {"status": "inactive", "message": "No active subscription found for this stylist."}
 
-        # 🔹 Verificar se existe uma assinatura ativa para o estilista
         for subscription in subscriptions["data"]:
-            # Verifica se o stylist_id está presente no metadata da assinatura
             if subscription["metadata"].get("stylist_id") == data.stylist_id:
-                # Se encontrou a assinatura ativa, vamos registrar a assinatura no Supabase
-                subscription_id = subscription["id"]  # ID da assinatura
-                price_id = subscription["items"]["data"][0]["price"]["id"]  # ID do preço da assinatura
-                
-                # 🔹 Definir o fuso horário de Nova York usando pytz
+                subscription_id = subscription["id"]
+                price_id = subscription["items"]["data"][0]["price"]["id"]
+                status = subscription.get("status")
+                cancel_at_period_end = subscription.get("cancel_at_period_end", False)
+                canceled_at = subscription.get("canceled_at")
+                current_period_end = subscription.get("current_period_end")
+
+                # 🔹 Converte timestamps para data legível
                 nyc_tz = pytz.timezone('America/New_York')
-                created_at = datetime.now().astimezone(nyc_tz)  # Hora atual de Nova York
+                canceled_at_date = (
+                    datetime.utcfromtimestamp(canceled_at).replace(tzinfo=pytz.utc).astimezone(nyc_tz).isoformat()
+                    if canceled_at else None
+                )
+                expiration_date = (
+                    datetime.utcfromtimestamp(current_period_end).replace(tzinfo=pytz.utc).astimezone(nyc_tz).isoformat()
+                    if current_period_end else None
+                )
 
-                # 🔹 Inserir dados na tabela Subscriptions no Supabase
+                # 🔹 Atualiza o banco de dados Supabase com os dados de cancelamento
                 subscription_data = {
-                    "stylist_id": data.stylist_id,  # ID do estilista
-                    "customer_id": user_id,         # ID do cliente
-                    "active": True,                 # Assinatura ativa
-                    "created_at": created_at.isoformat(),  # Hora atual em formato ISO 8601
-                    "sub_id": subscription_id,      # ID da assinatura
-                    "price_id": price_id            # ID do preço
-                }
-
-                # Configuração do cabeçalho de autenticação para o Supabase
-                supabase_headers = {
-                    "Authorization": f"Bearer {SUPABASE_KEY}",
-                    "apikey": SUPABASE_KEY,
-                    "Content-Type": "application/json"
+                    "stylist_id": data.stylist_id,
+                    "customer_id": user_id,
+                    "active": status == "active",
+                    "sub_id": subscription_id,
+                    "price_id": price_id,
+                    "canceled": cancel_at_period_end or (status == "canceled"),
+                    "canceled_date": canceled_at_date
                 }
 
-                # Inserir nova linha na tabela Subscriptions
-                subscription_url = f"{SUPABASE_URL}/rest/v1/Subscriptions"
                 response_subscription = requests.post(
-                    subscription_url,
-                    headers=supabase_headers,
+                    f"{SUPABASE_URL}/rest/v1/Subscriptions",
+                    headers=SUPABASE_HEADERS,
                     json=subscription_data
                 )
 
-                # Verificando a resposta da inserção
                 if response_subscription.status_code == 201:
-                    logger.info(f"✅ Subscription added successfully for user {user_id}")
+                    logger.info(f"✅ Subscription updated successfully for user {user_id}")
                 else:
-                    logger.error(f"❌ Failed to add subscription: {response_subscription.status_code} - {response_subscription.text}")
+                    logger.error(f"❌ Failed to update subscription: {response_subscription.status_code} - {response_subscription.text}")
 
                 return {
-                    "status": "active",
-                    "subscription_id": subscription["id"],
-                    "consultations_per_month": subscription["metadata"].get("consultations_per_month", None)
+                    "status": status,
+                    "subscription_id": subscription_id,
+                    "consultations_per_month": subscription["metadata"].get("consultations_per_month", None),
+                    "cancel_at_period_end": cancel_at_period_end,
+                    "canceled_at": canceled_at_date,
+                    "expiration_date": expiration_date
                 }
 
-        # Caso não tenha encontrado assinatura ativa para o estilista
-        return {
-            "status": "inactive",
-            "message": "No active subscription found for this stylist."
-        }
+        return {"status": "inactive", "message": "No active subscription found for this stylist."}
 
     except stripe.error.StripeError as e:
-        # Captura erros específicos do Stripe
         logger.error(f"Stripe error: {str(e)}")
         raise HTTPException(status_code=500, detail=f"Stripe error: {str(e)}")
     except Exception as e:
-        # Captura outros erros genéricos
         logger.error(f"Error checking subscription: {str(e)}")
         raise HTTPException(status_code=500, detail="Error checking subscription.")