chore: workflows improvements (#15)

* chore: updat smithery to stdio

* chore: add labels to container

* chore: update function docs

Author: saidsef

.github/workflows/ci.yml

@@ -79,8 +79,13 @@ jobs:
       uses: docker/setup-buildx-action@v3
       env:
         DOCKER_BUILDKIT: 1
+    - name: Docker meta
+      id: meta
+      uses: docker/metadata-action@v5
+      with:
+        images: ${{ env.REGISTRY }}/${{ env.OWNER }}/${{ env.REPO_NAME }}
     - name: Build and push Docker image
-      uses: docker/build-push-action@v5
+      uses: docker/build-push-action@v6
       env:
         DOCKER_BUILD_SUMMARY: true
         DOCKER_BUILD_CHECKS_ANNOTATIONS: true
@@ -94,14 +99,15 @@ jobs:
         pull: true
         push: true
         sbom: true
-        provenance: true
+        provenance: mode=max
         cache-from: type=gha
         cache-to: type=gha,mode=max
         tags: ${{ env.REGISTRY }}/${{ env.OWNER }}/${{ env.REPO_NAME }}:${{ env.TAG != 'main' && env.TAG || env.DATE }}
+        labels: ${{ steps.meta.outputs.labels }}
 
   auto-approve:
     runs-on: ubuntu-latest
-    needs: [build]
+    needs: [build, review]
     if: ${{ github.event_name == 'pull_request' }}
     steps:
     - name: Auto Approve PR

github_integration.py

@@ -21,7 +21,7 @@
 import requests
 import traceback
 from os import getenv
-from typing import Dict, Any
+from typing import Dict, Any, Optional
 
 GITHUB_TOKEN = getenv('GITHUB_TOKEN')
 
@@ -115,6 +115,38 @@ def get_pr_content(self, repo_owner: str, repo_name: str, pr_number: int) -> Dic
             logging.error(f"Error fetching PR content: {str(e)}")
             traceback.print_exc()
             return None
+
+    def add_pr_comments(self, repo_owner: str, repo_name: str, pr_number: int, comment: str) -> Dict[str, Any]:
+        """Add a comment to a pull request.
+
+        Args:
+            repo_owner: The owner of the GitHub repository
+            repo_name: The name of the GitHub repository
+            pr_number: The number of the pull request to comment on
+            comment: The content of the comment
+
+        Returns:
+            A dictionary containing the added comment's metadata
+        """
+        logging.info(f"Adding comment to PR {repo_owner}/{repo_name}#{pr_number}")
+
+        # Construct the comments URL
+        comments_url = f"https://api.github.com/repos/{repo_owner}/{repo_name}/issues/{pr_number}/comments"
+
+        try:
+            # Add the comment
+            response = requests.post(comments_url, headers=self._get_headers(), json={'body': comment})
+            response.raise_for_status()
+            comment_data = response.json()
+
+            logging.info(f"Comment added successfully")
+            return comment_data
+
+        except Exception as e:
+            logging.error(f"Error adding comment: {str(e)}")
+            traceback.print_exc()
+            return None
+
     def update_pr_description(self, repo_owner: str, repo_name: str, pr_number: int, new_title: str, new_description: str) -> Dict[str, Any]:
         """Update the description of a pull request.
 
@@ -130,7 +162,6 @@ def update_pr_description(self, repo_owner: str, repo_name: str, pr_number: int,
         """
         logging.info(f"Updating PR description for {repo_owner}/{repo_name}#{pr_number}")
 
-
         # Construct the PR URL
         pr_url = self._get_pr_url(repo_owner, repo_name, pr_number)
         try:
@@ -149,49 +180,54 @@ def update_pr_description(self, repo_owner: str, repo_name: str, pr_number: int,
             traceback.print_exc()
             return None
 
-    def create_issue(self, repo_owner: str, repo_name: str, title: str, body: str) -> Dict[str, Any]:
+    def create_issue(self, repo_owner: str, repo_name: str, title: str, body: str, labels: list[str]) -> Dict[str, Any]:
         """Create a new issue in the specified GitHub repository.
         
         Args:
             repo_owner: The owner of the GitHub repository
             repo_name: The name of the GitHub repository
             title: The title of the issue
-            body: The body content of the issue
+            body: The body content of the issue, this should include description, why, details, references
+                  if there is a PR number, add resolve by PR number
+            labels: A list of labels to apply to the issue
         Returns:
             A dictionary containing the created issue's metadata
         """
         logging.info(f"Creating issue in {repo_owner}/{repo_name}")
 
         # Construct the issues URL
         issues_url = f"https://api.github.com/repos/{repo_owner}/{repo_name}/issues"
-        
+
         try:
             # Create the issue
+            issue_labels = ['mcp'] if not labels else labels + ['mcp']
             response = requests.post(issues_url, headers=self._get_headers(), json={
                 'title': title,
                 'body': body,
-                'labels': ['mcp']
+                'labels': issue_labels
             })
             response.raise_for_status()
             issue_data = response.json()
 
             logging.info(f"Issue created successfully")
             return issue_data
-            
+
         except Exception as e:
             logging.error(f"Error creating issue: {str(e)}")
             traceback.print_exc()
             return None
 
-    def update_issue(self, repo_owner: str, repo_name: str, issue_number: int, title: str, body: str, state: str = 'open') -> Dict[str, Any]:
+    def update_issue(self, repo_owner: str, repo_name: str, issue_number: int, title: str, body: str, labels: list[str] = [], state: str = 'open') -> Dict[str, Any]:
         """Update an existing issue in the specified GitHub repository.
-        
+
         Args:
             repo_owner: The owner of the GitHub repository
             repo_name: The name of the GitHub repository
             issue_number: The number of the issue to update
             title: The new title of the issue
             body: The new body content of the issue
+            labels: A list of labels to apply to the issue
+            state: The new state of the issue (open/closed)
         Returns:
             A dictionary containing the updated issue's metadata
         """
@@ -205,6 +241,7 @@ def update_issue(self, repo_owner: str, repo_name: str, issue_number: int, title
             response = requests.patch(issue_url, headers=self._get_headers(), json={
                 'title': title,
                 'body': body,
+                'labels': labels,
                 'state': state
             })
             response.raise_for_status()
@@ -218,40 +255,40 @@ def update_issue(self, repo_owner: str, repo_name: str, issue_number: int, title
 
     def get_latest_sha(self, repo_owner: str, repo_name: str) -> Optional[str]:
         """Fetch the latest commit SHA from the specified GitHub repository.
-        
+
         Args:
             repo_owner: The owner of the GitHub repository
             repo_name: The name of the GitHub repository
         Returns:
             The latest commit SHA as a string, or None if no commits are found
         """
         logging.info(f"Fetching latest commit SHA for {repo_owner}/{repo_name}")
-        
+
         # Construct the commits URL
         commits_url = f"https://api.github.com/repos/{repo_owner}/{repo_name}/commits"
-        
+
         try:
             # Fetch the latest commit
             response = requests.get(commits_url, headers=self._get_headers())
             response.raise_for_status()
             commits_data = response.json()
-            
+
             if commits_data:
                 latest_sha = commits_data[0]['sha']
                 logging.info(f"Latest commit SHA fetched successfully")
                 return latest_sha
             else:
                 logging.warning("No commits found in the repository")
                 return None
-            
+
         except Exception as e:
             logging.error(f"Error fetching latest commit SHA: {str(e)}")
             traceback.print_exc()
             return None
-        
+
     def create_tag(self, repo_owner: str, repo_name: str, tag_name: str, message: str) -> Dict[str, Any]:
         """Create a new tag in the specified GitHub repository.
-        
+
         Args:
             repo_owner: The owner of the GitHub repository
             repo_name: The name of the GitHub repository
@@ -268,7 +305,7 @@ def create_tag(self, repo_owner: str, repo_name: str, tag_name: str, message: st
             latest_sha = self.get_latest_sha(repo_owner, repo_name)
             if not latest_sha:
                 raise ValueError("Failed to fetch the latest commit SHA")
-            
+
             # Create the tag
             response = requests.post(tags_url, headers=self._get_headers(), json={
                 'ref': f'refs/tags/{tag_name}',
@@ -277,18 +314,18 @@ def create_tag(self, repo_owner: str, repo_name: str, tag_name: str, message: st
             })
             response.raise_for_status()
             tag_data = response.json()
-            
+
             logging.info(f"Tag created successfully")
             return tag_data
-            
+
         except Exception as e:
             logging.error(f"Error creating tag: {str(e)}")
             traceback.print_exc()
             return None
 
     def create_release(self, repo_owner: str, repo_name: str, tag_name: str, release_name: str, body: str) -> Dict[str, Any]:
         """Create a new release in the specified GitHub repository.
-        
+
         Args:
             repo_owner: The owner of the GitHub repository
             repo_name: The name of the GitHub repository
@@ -299,10 +336,10 @@ def create_release(self, repo_owner: str, repo_name: str, tag_name: str, release
             A dictionary containing the created release's metadata
         """
         logging.info(f"Creating release {release_name} in {repo_owner}/{repo_name}")
-        
+
         # Construct the releases URL
         releases_url = f"https://api.github.com/repos/{repo_owner}/{repo_name}/releases"
-        
+
         try:
             # Create the release
             response = requests.post(releases_url, headers=self._get_headers(), json={
@@ -315,10 +352,10 @@ def create_release(self, repo_owner: str, repo_name: str, tag_name: str, release
             })
             response.raise_for_status()
             release_data = response.json()
-            
+
             logging.info(f"Release created successfully")
             return release_data
-            
+
         except Exception as e:
             logging.error(f"Error creating release: {str(e)}")
             traceback.print_exc()

issues_pr_analyser.py

@@ -128,7 +128,7 @@ async def update_github_pr_description(repo_owner: str, repo_name: str, pr_numbe
                 return error_msg
 
         @self.mcp.tool()
-        async def create_github_issue(repo_owner: str, repo_name: str, title: str, body: str) -> str:
+        async def create_github_issue(repo_owner: str, repo_name: str, title: str, body: str, labels: list[str]) -> str:
             """
             Create a GitHub issue.
             Prefix the title with one of chore, feat, or fix.
@@ -137,13 +137,14 @@ async def create_github_issue(repo_owner: str, repo_name: str, title: str, body:
                 repo_owner: The owner of the GitHub repository
                 repo_name: The name of the GitHub repository
                 title: The title of the issue, one of feat, fix, chore, docs, refactor, style, test
-                body: The body of the issue
+                body: The body content of the issue, this should include description, why, details, references
+                labels: The labels to add to the issue
             Returns:
                 A message indicating the success or failure of the issue creation
             """
             logging.info(f"Creating GitHub issue: {title}")
             try:
-                issue = self.gi.create_issue(repo_owner, repo_name, title, body)
+                issue = self.gi.create_issue(repo_owner, repo_name, title, body, labels)
                 logging.info(f"GitHub issue '{title}' created successfully!")
                 return f"GitHub issue '{title}' created successfully!"
             except Exception as e:
@@ -153,7 +154,7 @@ async def create_github_issue(repo_owner: str, repo_name: str, title: str, body:
                 return error_msg
 
         @self.mcp.tool()
-        async def update_github_issue(repo_owner: str, repo_name: str, issue_number: int, title: str, body: str, state: str) -> str:
+        async def update_github_issue(repo_owner: str, repo_name: str, issue_number: int, title: str, body: str, labels: list[str], state: str) -> str:
             """
             Update a GitHub issue.
             Prefix the title with one of chore, feat, or fix.
@@ -163,14 +164,16 @@ async def update_github_issue(repo_owner: str, repo_name: str, issue_number: int
                 repo_name: The name of the GitHub repository
                 issue_number: The number of the issue to update
                 title: The new title of the issue, one of feat, fix, chore, docs, refactor, style, test
-                body: The new body of the issue
+                body: The body content of the issue, this should include description, why, details, references
+                      if there is a PR number, add resolve by PR number
+                labels: The labels to add to the issue
                 state: The new state of the issue (open/closed)
             Returns:
                 A message indicating the success or failure of the issue update
             """
             logging.info(f"Updating GitHub issue #{issue_number}: {title}")
             try:
-                issue = self.gi.update_issue(repo_owner, repo_name, issue_number, title, body, state)
+                issue = self.gi.update_issue(repo_owner, repo_name, issue_number, title, body, labels, state)
                 logging.info(f"GitHub issue #{issue_number} updated successfully!")
                 return f"GitHub issue #{issue_number} updated successfully!"
             except Exception as e:

smithery.yaml

@@ -1,7 +1,21 @@
----
-version: 1
-start:
-  command: ["python3", "issues_pr_analyser.py"]
-  env:
-    ENABLE_SSE: true
-  port: 8080
+# Smithery configuration file: https://smithery.ai/docs/use/session-config
+
+startCommand:
+  type: stdio
+  configSchema:
+    type: object
+    required: ["githubToken"]
+    properties:
+      githubToken:
+        type: string
+        description: The GitHub token to use for authentication.
+        default: "foo"
+  commandFunction: |-
+    (config) => ({ 
+      command: 'python', 
+      args: ['issues_pr_analyser.py'],
+      env: {
+        GITHUB_TOKEN: config.githubToken,
+      },
+    })
+  exampleConfig: {}