workspace-mcp 1.0.0__py3-none-any.whl → 1.0.2__py3-none-any.whl

gchat/chat_tools.py

@@ -13,11 +13,13 @@ from googleapiclient.errors import HttpError
 # Auth & server utilities
 from auth.service_decorator import require_google_service
 from core.server import server
+from core.utils import handle_http_errors
 
 logger = logging.getLogger(__name__)
 
 @server.tool()
 @require_google_service("chat", "chat_read")
+@handle_http_errors("list_spaces")
 async def list_spaces(
     service,
     user_google_email: str,
@@ -32,44 +34,37 @@ async def list_spaces(
     """
     logger.info(f"[list_spaces] Email={user_google_email}, Type={space_type}")
 
-    try:
-        # Build filter based on space_type
-        filter_param = None
-        if space_type == "room":
-            filter_param = "spaceType = SPACE"
-        elif space_type == "dm":
-            filter_param = "spaceType = DIRECT_MESSAGE"
+    # Build filter based on space_type
+    filter_param = None
+    if space_type == "room":
+        filter_param = "spaceType = SPACE"
+    elif space_type == "dm":
+        filter_param = "spaceType = DIRECT_MESSAGE"
 
-        request_params = {"pageSize": page_size}
-        if filter_param:
-            request_params["filter"] = filter_param
+    request_params = {"pageSize": page_size}
+    if filter_param:
+        request_params["filter"] = filter_param
 
-        response = await asyncio.to_thread(
-            service.spaces().list(**request_params).execute
-        )
+    response = await asyncio.to_thread(
+        service.spaces().list(**request_params).execute
+    )
 
-        spaces = response.get('spaces', [])
-        if not spaces:
-            return f"No Chat spaces found for type '{space_type}'."
+    spaces = response.get('spaces', [])
+    if not spaces:
+        return f"No Chat spaces found for type '{space_type}'."
 
-        output = [f"Found {len(spaces)} Chat spaces (type: {space_type}):"]
-        for space in spaces:
-            space_name = space.get('displayName', 'Unnamed Space')
-            space_id = space.get('name', '')
-            space_type_actual = space.get('spaceType', 'UNKNOWN')
-            output.append(f"- {space_name} (ID: {space_id}, Type: {space_type_actual})")
+    output = [f"Found {len(spaces)} Chat spaces (type: {space_type}):"]
+    for space in spaces:
+        space_name = space.get('displayName', 'Unnamed Space')
+        space_id = space.get('name', '')
+        space_type_actual = space.get('spaceType', 'UNKNOWN')
+        output.append(f"- {space_name} (ID: {space_id}, Type: {space_type_actual})")
 
-        return "\n".join(output)
-
-    except HttpError as e:
-        logger.error(f"API error in list_spaces: {e}", exc_info=True)
-        raise Exception(f"API error: {e}")
-    except Exception as e:
-        logger.exception(f"Unexpected error in list_spaces: {e}")
-        raise Exception(f"Unexpected error: {e}")
+    return "\n".join(output)
 
 @server.tool()
 @require_google_service("chat", "chat_read")
+@handle_http_errors("get_messages")
 async def get_messages(
     service,
     user_google_email: str,
@@ -85,48 +80,41 @@ async def get_messages(
     """
     logger.info(f"[get_messages] Space ID: '{space_id}' for user '{user_google_email}'")
 
-    try:
-        # Get space info first
-        space_info = await asyncio.to_thread(
-            service.spaces().get(name=space_id).execute
-        )
-        space_name = space_info.get('displayName', 'Unknown Space')
-
-        # Get messages
-        response = await asyncio.to_thread(
-            service.spaces().messages().list(
-                parent=space_id,
-                pageSize=page_size,
-                orderBy=order_by
-            ).execute
-        )
-
-        messages = response.get('messages', [])
-        if not messages:
-            return f"No messages found in space '{space_name}' (ID: {space_id})."
-
-        output = [f"Messages from '{space_name}' (ID: {space_id}):\n"]
-        for msg in messages:
-            sender = msg.get('sender', {}).get('displayName', 'Unknown Sender')
-            create_time = msg.get('createTime', 'Unknown Time')
-            text_content = msg.get('text', 'No text content')
-            msg_name = msg.get('name', '')
-
-            output.append(f"[{create_time}] {sender}:")
-            output.append(f"  {text_content}")
-            output.append(f"  (Message ID: {msg_name})\n")
-
-        return "\n".join(output)
-
-    except HttpError as error:
-        logger.error(f"[get_messages] API error for space {space_id}: {error}", exc_info=True)
-        raise Exception(f"API error accessing space {space_id}: {error}")
-    except Exception as e:
-        logger.exception(f"[get_messages] Unexpected error for space {space_id}: {e}")
-        raise Exception(f"Unexpected error accessing space {space_id}: {e}")
+    # Get space info first
+    space_info = await asyncio.to_thread(
+        service.spaces().get(name=space_id).execute
+    )
+    space_name = space_info.get('displayName', 'Unknown Space')
+
+    # Get messages
+    response = await asyncio.to_thread(
+        service.spaces().messages().list(
+            parent=space_id,
+            pageSize=page_size,
+            orderBy=order_by
+        ).execute
+    )
+
+    messages = response.get('messages', [])
+    if not messages:
+        return f"No messages found in space '{space_name}' (ID: {space_id})."
+
+    output = [f"Messages from '{space_name}' (ID: {space_id}):\n"]
+    for msg in messages:
+        sender = msg.get('sender', {}).get('displayName', 'Unknown Sender')
+        create_time = msg.get('createTime', 'Unknown Time')
+        text_content = msg.get('text', 'No text content')
+        msg_name = msg.get('name', '')
+
+        output.append(f"[{create_time}] {sender}:")
+        output.append(f"  {text_content}")
+        output.append(f"  (Message ID: {msg_name})\n")
+
+    return "\n".join(output)
 
 @server.tool()
 @require_google_service("chat", "chat_write")
+@handle_http_errors("send_message")
 async def send_message(
     service,
     user_google_email: str,
@@ -142,39 +130,32 @@ async def send_message(
     """
     logger.info(f"[send_message] Email: '{user_google_email}', Space: '{space_id}'")
 
-    try:
-        message_body = {
-            'text': message_text
-        }
-
-        # Add thread key if provided (for threaded replies)
-        request_params = {
-            'parent': space_id,
-            'body': message_body
-        }
-        if thread_key:
-            request_params['threadKey'] = thread_key
-
-        message = await asyncio.to_thread(
-            service.spaces().messages().create(**request_params).execute
-        )
+    message_body = {
+        'text': message_text
+    }
 
-        message_name = message.get('name', '')
-        create_time = message.get('createTime', '')
+    # Add thread key if provided (for threaded replies)
+    request_params = {
+        'parent': space_id,
+        'body': message_body
+    }
+    if thread_key:
+        request_params['threadKey'] = thread_key
 
-        msg = f"Message sent to space '{space_id}' by {user_google_email}. Message ID: {message_name}, Time: {create_time}"
-        logger.info(f"Successfully sent message to space '{space_id}' by {user_google_email}")
-        return msg
+    message = await asyncio.to_thread(
+        service.spaces().messages().create(**request_params).execute
+    )
 
-    except HttpError as e:
-        logger.error(f"API error in send_message: {e}", exc_info=True)
-        raise Exception(f"API error: {e}")
-    except Exception as e:
-        logger.exception(f"Unexpected error in send_message: {e}")
-        raise Exception(f"Unexpected error: {e}")
+    message_name = message.get('name', '')
+    create_time = message.get('createTime', '')
+
+    msg = f"Message sent to space '{space_id}' by {user_google_email}. Message ID: {message_name}, Time: {create_time}"
+    logger.info(f"Successfully sent message to space '{space_id}' by {user_google_email}")
+    return msg
 
 @server.tool()
 @require_google_service("chat", "chat_read")
+@handle_http_errors("search_messages")
 async def search_messages(
     service,
     user_google_email: str,
@@ -190,65 +171,57 @@ async def search_messages(
     """
     logger.info(f"[search_messages] Email={user_google_email}, Query='{query}'")
 
-    try:
-        # If specific space provided, search within that space
-        if space_id:
-            response = await asyncio.to_thread(
-                service.spaces().messages().list(
-                    parent=space_id,
-                    pageSize=page_size,
-                    filter=f'text:"{query}"'
-                ).execute
-            )
-            messages = response.get('messages', [])
-            context = f"space '{space_id}'"
-        else:
-            # Search across all accessible spaces (this may require iterating through spaces)
-            # For simplicity, we'll search the user's spaces first
-            spaces_response = await asyncio.to_thread(
-                service.spaces().list(pageSize=100).execute
-            )
-            spaces = spaces_response.get('spaces', [])
-
-            messages = []
-            for space in spaces[:10]:  # Limit to first 10 spaces to avoid timeout
-                try:
-                    space_messages = await asyncio.to_thread(
-                        service.spaces().messages().list(
-                            parent=space.get('name'),
-                            pageSize=5,
-                            filter=f'text:"{query}"'
-                        ).execute
-                    )
-                    space_msgs = space_messages.get('messages', [])
-                    for msg in space_msgs:
-                        msg['_space_name'] = space.get('displayName', 'Unknown')
-                    messages.extend(space_msgs)
-                except HttpError:
-                    continue  # Skip spaces we can't access
-            context = "all accessible spaces"
-
-        if not messages:
-            return f"No messages found matching '{query}' in {context}."
-
-        output = [f"Found {len(messages)} messages matching '{query}' in {context}:"]
-        for msg in messages:
-            sender = msg.get('sender', {}).get('displayName', 'Unknown Sender')
-            create_time = msg.get('createTime', 'Unknown Time')
-            text_content = msg.get('text', 'No text content')
-            space_name = msg.get('_space_name', 'Unknown Space')
-
-            # Truncate long messages
-            if len(text_content) > 100:
-                text_content = text_content[:100] + "..."
-
-            output.append(f"- [{create_time}] {sender} in '{space_name}': {text_content}")
-
-        return "\n".join(output)
-
-    except HttpError as e:
-        logger.error(f"API error in search_messages: {e}", exc_info=True)
-        raise Exception(f"API error: {e}")
-    except Exception as e:
-        logger.exception(f"Unexpected error in search_messages: {e}")
-        raise Exception(f"Unexpected error: {e}")
+    # If specific space provided, search within that space
+    if space_id:
+        response = await asyncio.to_thread(
+            service.spaces().messages().list(
+                parent=space_id,
+                pageSize=page_size,
+                filter=f'text:"{query}"'
+            ).execute
+        )
+        messages = response.get('messages', [])
+        context = f"space '{space_id}'"
+    else:
+        # Search across all accessible spaces (this may require iterating through spaces)
+        # For simplicity, we'll search the user's spaces first
+        spaces_response = await asyncio.to_thread(
+            service.spaces().list(pageSize=100).execute
+        )
+        spaces = spaces_response.get('spaces', [])
+
+        messages = []
+        for space in spaces[:10]:  # Limit to first 10 spaces to avoid timeout
+            try:
+                space_messages = await asyncio.to_thread(
+                    service.spaces().messages().list(
+                        parent=space.get('name'),
+                        pageSize=5,
+                        filter=f'text:"{query}"'
+                    ).execute
+                )
+                space_msgs = space_messages.get('messages', [])
+                for msg in space_msgs:
+                    msg['_space_name'] = space.get('displayName', 'Unknown')
+                messages.extend(space_msgs)
+            except HttpError:
+                continue  # Skip spaces we can't access
+        context = "all accessible spaces"
+
+    if not messages:
+        return f"No messages found matching '{query}' in {context}."
+
+    output = [f"Found {len(messages)} messages matching '{query}' in {context}:"]
+    for msg in messages:
+        sender = msg.get('sender', {}).get('displayName', 'Unknown Sender')
+        create_time = msg.get('createTime', 'Unknown Time')
+        text_content = msg.get('text', 'No text content')
+        space_name = msg.get('_space_name', 'Unknown Space')
+
+        # Truncate long messages
+        if len(text_content) > 100:
+            text_content = text_content[:100] + "..."
+
+        output.append(f"- [{create_time}] {sender} in '{space_name}': {text_content}")
+
+    return "\n".join(output)