Update routes/support.py

routes/support.py

@@ -270,7 +270,7 @@ async def respond_ticket(
     user_id = await verify_user_token(user_token)
     created_at = datetime.utcnow().isoformat()
 
-    # 2. Verificar se o ticket existe
+    # 2. Buscar dados do ticket
     async with aiohttp.ClientSession() as session:
         async with session.get(
             f"{SUPABASE_URL}/rest/v1/Tickets?id=eq.{body.ticket_id}",
@@ -285,7 +285,37 @@ async def respond_ticket(
 
             ticket = ticket_data[0]
 
-    # 3. Criar a mensagem de resposta ao ticket
+    support_id = ticket.get("support_id")
+
+    # 3. Verificar se o suporte está atribuído
+    if support_id is None:
+        # Checar se o usuário é admin
+        async with aiohttp.ClientSession() as session:
+            async with session.get(
+                f"{SUPABASE_URL}/rest/v1/User?id=eq.{user_id}",
+                headers=SUPABASE_ROLE_HEADERS
+            ) as user_resp:
+                if user_resp.status != 200:
+                    raise HTTPException(status_code=403, detail="Usuário inválido")
+
+                user_data = await user_resp.json()
+                if not user_data or not user_data[0].get("is_admin", False):
+                    raise HTTPException(status_code=403, detail="Apenas administradores podem assumir tickets")
+
+        # Atribuir ticket ao suporte (usuário atual)
+        async with aiohttp.ClientSession() as session:
+            async with session.patch(
+                f"{SUPABASE_URL}/rest/v1/Tickets?id=eq.{body.ticket_id}",
+                headers={**SUPABASE_ROLE_HEADERS, "Prefer": "return=representation"},
+                json={"support_id": user_id}
+            ) as patch_resp:
+                if patch_resp.status != 200:
+                    raise HTTPException(status_code=500, detail="Erro ao atribuir o suporte ao ticket")
+    else:
+        if support_id != user_id:
+            raise HTTPException(status_code=403, detail="Ticket já está atribuído a outro suporte")
+
+    # 4. Criar a mensagem de resposta
     message_payload = {
         "user": user_id,
         "content": body.content,
@@ -293,7 +323,7 @@ async def respond_ticket(
         "ticket_id": body.ticket_id
     }
 
-    # 4. Salvar a mensagem no banco
+    # 5. Salvar a mensagem no banco
     async with aiohttp.ClientSession() as session:
         async with session.post(
             f"{SUPABASE_URL}/rest/v1/messages_tickets",
@@ -306,7 +336,7 @@ async def respond_ticket(
 
             message_data = await message_resp.json()
 
-    # 5. Retornar confirmação da resposta
+    # 6. Retornar confirmação
     return {
         "status": "response sent successfully",
         "ticket_id": body.ticket_id,