modularize comments, add smithery

core/comments.py

@@ -0,0 +1,189 @@
+"""
+Core Comments Module
+
+This module provides reusable comment management functions for Google Workspace applications.
+All Google Workspace apps (Docs, Sheets, Slides) use the Drive API for comment operations.
+"""
+
+import logging
+import asyncio
+from typing import Dict, Any
+
+from mcp import types
+from googleapiclient.errors import HttpError
+
+from auth.service_decorator import require_google_service
+from core.server import server
+from core.utils import handle_http_errors
+
+logger = logging.getLogger(__name__)
+
+
+def create_comment_tools(app_name: str, file_id_param: str):
+    """
+    Factory function to create comment management tools for a specific Google Workspace app.
+    
+    Args:
+        app_name: Name of the app (e.g., "doc", "sheet", "presentation")
+        file_id_param: Parameter name for the file ID (e.g., "document_id", "spreadsheet_id", "presentation_id")
+    
+    Returns:
+        Dict containing the four comment management functions
+    """
+    
+    @server.tool()
+    @require_google_service("drive", "drive_read")
+    @handle_http_errors(f"read_{app_name}_comments")
+    async def read_comments(
+        service,
+        user_google_email: str,
+        **kwargs
+    ) -> str:
+        """
+        Read all comments from a Google Workspace file.
+        """
+        file_id = kwargs[file_id_param]
+        logger.info(f"[read_{app_name}_comments] Reading comments for {app_name} {file_id}")
+
+        response = await asyncio.to_thread(
+            service.comments().list(
+                fileId=file_id,
+                fields="comments(id,content,author,createdTime,modifiedTime,resolved,replies(content,author,id,createdTime,modifiedTime))"
+            ).execute
+        )
+        
+        comments = response.get('comments', [])
+        
+        if not comments:
+            return f"No comments found in {app_name} {file_id}"
+        
+        output = [f"Found {len(comments)} comments in {app_name} {file_id}:\\n"]
+        
+        for comment in comments:
+            author = comment.get('author', {}).get('displayName', 'Unknown')
+            content = comment.get('content', '')
+            created = comment.get('createdTime', '')
+            resolved = comment.get('resolved', False)
+            comment_id = comment.get('id', '')
+            status = " [RESOLVED]" if resolved else ""
+            
+            output.append(f"Comment ID: {comment_id}")
+            output.append(f"Author: {author}")
+            output.append(f"Created: {created}{status}")
+            output.append(f"Content: {content}")
+            
+            # Add replies if any
+            replies = comment.get('replies', [])
+            if replies:
+                output.append(f"  Replies ({len(replies)}):")
+                for reply in replies:
+                    reply_author = reply.get('author', {}).get('displayName', 'Unknown')
+                    reply_content = reply.get('content', '')
+                    reply_created = reply.get('createdTime', '')
+                    reply_id = reply.get('id', '')
+                    output.append(f"    Reply ID: {reply_id}")
+                    output.append(f"    Author: {reply_author}")
+                    output.append(f"    Created: {reply_created}")
+                    output.append(f"    Content: {reply_content}")
+            
+            output.append("")  # Empty line between comments
+        
+        return "\\n".join(output)
+
+    @server.tool()
+    @require_google_service("drive", "drive_file")
+    @handle_http_errors(f"create_{app_name}_comment")
+    async def create_comment(
+        service,
+        user_google_email: str,
+        comment_content: str,
+        **kwargs
+    ) -> str:
+        """
+        Create a new comment on a Google Workspace file.
+        """
+        file_id = kwargs[file_id_param]
+        logger.info(f"[create_{app_name}_comment] Creating comment in {app_name} {file_id}")
+
+        body = {"content": comment_content}
+        
+        comment = await asyncio.to_thread(
+            service.comments().create(
+                fileId=file_id,
+                body=body,
+                fields="id,content,author,createdTime,modifiedTime"
+            ).execute
+        )
+        
+        comment_id = comment.get('id', '')
+        author = comment.get('author', {}).get('displayName', 'Unknown')
+        created = comment.get('createdTime', '')
+        
+        return f"Comment created successfully!\\nComment ID: {comment_id}\\nAuthor: {author}\\nCreated: {created}\\nContent: {comment_content}"
+
+    @server.tool()
+    @require_google_service("drive", "drive_file")
+    @handle_http_errors(f"reply_to_{app_name}_comment")
+    async def reply_to_comment(
+        service,
+        user_google_email: str,
+        comment_id: str,
+        reply_content: str,
+        **kwargs
+    ) -> str:
+        """
+        Reply to a specific comment in a Google Workspace file.
+        """
+        file_id = kwargs[file_id_param]
+        logger.info(f"[reply_to_{app_name}_comment] Replying to comment {comment_id} in {app_name} {file_id}")
+
+        body = {'content': reply_content}
+        
+        reply = await asyncio.to_thread(
+            service.replies().create(
+                fileId=file_id,
+                commentId=comment_id,
+                body=body,
+                fields="id,content,author,createdTime,modifiedTime"
+            ).execute
+        )
+        
+        reply_id = reply.get('id', '')
+        author = reply.get('author', {}).get('displayName', 'Unknown')
+        created = reply.get('createdTime', '')
+        
+        return f"Reply posted successfully!\\nReply ID: {reply_id}\\nAuthor: {author}\\nCreated: {created}\\nContent: {reply_content}"
+
+    @server.tool()
+    @require_google_service("drive", "drive_file")
+    @handle_http_errors(f"resolve_{app_name}_comment")
+    async def resolve_comment(
+        service,
+        user_google_email: str,
+        comment_id: str,
+        **kwargs
+    ) -> str:
+        """
+        Resolve a comment in a Google Workspace file.
+        """
+        file_id = kwargs[file_id_param]
+        logger.info(f"[resolve_{app_name}_comment] Resolving comment {comment_id} in {app_name} {file_id}")
+
+        body = {"resolved": True}
+        
+        await asyncio.to_thread(
+            service.comments().update(
+                fileId=file_id,
+                commentId=comment_id,
+                body=body
+            ).execute
+        )
+        
+        return f"Comment {comment_id} has been resolved successfully."
+
+    return {
+        'read_comments': read_comments,
+        'create_comment': create_comment,
+        'reply_to_comment': reply_to_comment,
+        'resolve_comment': resolve_comment
+    }