Update routes/subscription.py

routes/subscription.py

@@ -359,19 +359,67 @@ def cancel_subscription(data: CancelSubscriptionRequest):
 
 # 🚀 Verificação de Assinatura
 @router.post("/check_subscription")
-def check_subscription(data: CheckSubscriptionRequest):
+def check_subscription(
+    data: CheckSubscriptionRequest,
+    user_token: str = Header(None, alias="User-key")  # Recebendo o token do cabeçalho
+):
     try:
-        # Buscar todas as assinaturas ativas para o usuário (cliente)
+        if not user_token:
+            raise HTTPException(status_code=401, detail="Missing User-key header")
+        
+        # 🔹 Validar o token do cliente e obter user_id (cliente)
+        user_id = verify_token(user_token)
+
+        # 🔹 Buscar todas as assinaturas ativas para o usuário (cliente)
         subscriptions = stripe.Subscription.list(
-            customer=data.user_id,  # ID do cliente
-            status="active",         # Assinaturas ativas
-            expand=["data.items"]    # Expande os dados para pegar os itens e o metadata
+            customer=user_id,  # ID do cliente obtido do token
+            status="active",    # Assinaturas ativas
+            expand=["data.items"]  # Expande os dados para pegar os itens e o metadata
         )
 
-        # Verificar se existe uma assinatura ativa para o estilista
+        # 🔹 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
+                nyc_tz = pytz.timezone('America/New_York')
+                created_at = datetime.now(nyc_tz)  # Hora atual de Nova York
+
+                # 🔹 Inserir dados na tabela Subscriptions no Supabase
+                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"
+                }
+
+                # Inserir nova linha na tabela Subscriptions (não precisamos definir 'id' explicitamente)
+                subscription_url = f"{SUPABASE_URL}/rest/v1/Subscriptions"
+                response_subscription = requests.post(
+                    subscription_url,
+                    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}")
+                else:
+                    logger.error(f"❌ Failed to add subscription: {response_subscription.status_code} - {response_subscription.text}")
+
                 return {
                     "status": "active",
                     "subscription_id": subscription.id,