Merge pull request #24 from natifridman/get-comments

Get Jira comments to issue

Author: natifridman

README.md

@@ -6,22 +6,23 @@ A Model Context Protocol (MCP) server that provides access to JIRA issue data st
 
 This MCP server connects to Snowflake to query JIRA data and provides three main tools for interacting with the data:
 
-- **`list_issues`** - Query and filter JIRA issues with various criteria
-- **`get_issue_details`** - Get detailed information for a specific issue by key
-- **`get_project_summary`** - Get statistics and summaries for all projects
+- **`list_jira_issues`** - Query and filter JIRA issues with various criteria
+- **`get_jira_issue_details`** - Get detailed information for a specific issue by key
+- **`get_jira_project_summary`** - Get statistics and summaries for all projects
 
 ## Features
 
 ### Data Sources
 The server connects to Snowflake and queries the following tables:
 - `JIRA_ISSUE_NON_PII` - Main issue data (non-personally identifiable information)
 - `JIRA_LABEL_RHAI` - Issue labels and tags
+- `JIRA_COMMENT_NON_PII` - Issue comments (non-personally identifiable information)
 
 **Note**: Table names are expected to exist in your configured Snowflake database and schema.
 
 ### Available Tools
 
-#### 1. List Issues (`list_issues`)
+#### 1. List Issues (`list_jira_issues`)
 Query JIRA issues with optional filtering:
 - **Project filtering** - Filter by project key (e.g., 'SMQE', 'OSIM')
 - **Issue type filtering** - Filter by issue type ID
@@ -30,15 +31,16 @@ Query JIRA issues with optional filtering:
 - **Text search** - Search in summary and description fields
 - **Result limiting** - Control number of results returned (default: 50)
 
-#### 2. Get Issue Details (`get_issue_details`)
+#### 2. Get Issue Details (`get_jira_issue_details`)
 Retrieve comprehensive information for a specific JIRA issue by its key (e.g., 'SMQE-1280'), including:
 - Basic issue information (summary, description, status, priority)
 - Timestamps (created, updated, due date, resolution date)
 - Time tracking (original estimate, current estimate, time spent)
 - Metadata (votes, watches, environment, components)
 - Associated labels
+- Comments (with comment body, creation/update timestamps, and role level)
 
-#### 3. Get Project Summary (`get_project_summary`)
+#### 3. Get Project Summary (`get_jira_project_summary`)
 Generate statistics across all projects:
 - Total issue counts per project
 - Status distribution per project
@@ -206,25 +208,25 @@ Example configuration to add to VS Code Continue:
 ### Query Issues by Project
 ```python
 # List all issues from the SMQE project
-result = await list_issues(project="SMQE", limit=10)
+result = await list_jira_issues(project="SMQE", limit=10)
 ```
 
 ### Search Issues by Text
 ```python
 # Search for issues containing "authentication" in summary or description
-result = await list_issues(search_text="authentication", limit=20)
+result = await list_jira_issues(search_text="authentication", limit=20)
 ```
 
 ### Get Specific Issue Details
 ```python
 # Get detailed information for a specific issue
-result = await get_issue_details(issue_key="SMQE-1280")
+result = await get_jira_issue_details(issue_key="SMQE-1280")
 ```
 
 ### Get Project Overview
 ```python
 # Get statistics for all projects
-result = await get_project_summary()
+result = await get_jira_project_summary()
 ```
 
 ## Monitoring

src/database.py

@@ -171,4 +171,57 @@ async def get_issue_labels(issue_ids: List[str], snowflake_token: Optional[str]
     except Exception as e:
         logger.error(f"Error fetching labels: {str(e)}")
 
-    return labels_data 
+    return labels_data
+
+async def get_issue_comments(issue_ids: List[str], snowflake_token: Optional[str] = None) -> Dict[str, List[Dict[str, Any]]]:
+    """Get comments for given issue IDs from Snowflake"""
+    if not issue_ids:
+        return {}
+    
+    comments_data = {}
+    
+    try:
+        # Sanitize and validate issue IDs (should be numeric)
+        sanitized_ids = []
+        for issue_id in issue_ids:
+            # Ensure issue IDs are numeric to prevent injection
+            if isinstance(issue_id, (str, int)) and str(issue_id).isdigit():
+                sanitized_ids.append(str(issue_id))
+        
+        if not sanitized_ids:
+            return {}
+        
+        # Create comma-separated list for IN clause
+        ids_str = "'" + "','".join(sanitized_ids) + "'"
+        
+        sql = f"""
+        SELECT ID, ISSUEID, ROLELEVEL, BODY, CREATED, UPDATED
+        FROM JIRA_COMMENT_NON_PII 
+        WHERE ISSUEID IN ({ids_str}) AND BODY IS NOT NULL
+        ORDER BY ISSUEID, CREATED ASC
+        """
+        
+        rows = await execute_snowflake_query(sql, snowflake_token)
+        columns = ["ID", "ISSUEID", "ROLELEVEL", "BODY", "CREATED", "UPDATED"]
+        
+        for row in rows:
+            row_dict = format_snowflake_row(row, columns)
+            issue_id = str(row_dict.get("ISSUEID"))
+            
+            if issue_id:
+                if issue_id not in comments_data:
+                    comments_data[issue_id] = []
+                
+                comment = {
+                    "id": row_dict.get("ID"),
+                    "role_level": row_dict.get("ROLELEVEL"),
+                    "body": row_dict.get("BODY"),
+                    "created": row_dict.get("CREATED"),
+                    "updated": row_dict.get("UPDATED")
+                }
+                comments_data[issue_id].append(comment)
+    
+    except Exception as e:
+        logger.error(f"Error fetching comments: {str(e)}")
+    
+    return comments_data 

src/tools.py

@@ -8,7 +8,8 @@
     execute_snowflake_query,
     format_snowflake_row,
     sanitize_sql_value,
-    get_issue_labels
+    get_issue_labels,
+    get_issue_comments
 )
 from metrics import track_tool_usage
 
@@ -183,7 +184,7 @@ async def get_jira_issue_details(issue_key: str) -> Dict[str, Any]:
             issue_key: The JIRA issue key (e.g., 'SMQE-1280')
         
         Returns:
-            Dictionary containing detailed issue information
+            Dictionary containing detailed issue information including comments
         """
         try:
             # Get the Snowflake token
@@ -252,6 +253,10 @@ async def get_jira_issue_details(issue_key: str) -> Dict[str, Any]:
             labels_data = await get_issue_labels([str(issue['id'])], snowflake_token)
             issue['labels'] = labels_data.get(str(issue['id']), [])
 
+            # Get comments for this issue
+            comments_data = await get_issue_comments([str(issue['id'])], snowflake_token)
+            issue['comments'] = comments_data.get(str(issue['id']), [])
+            
             return issue
 
         except Exception as e: