Merge pull request #33 from Nayjest/exp_comment_from_code

post comment from app

Author: Nayjest

.github/workflows/ai-code-review.yml

@@ -7,7 +7,7 @@ jobs:
     steps:
     - uses: actions/checkout@v4
       with: { fetch-depth: 0 }
-    - name: Set up Python 3.11
+    - name: Set up Python
       uses: actions/setup-python@v5
       with: { python-version: "3.13" }
     - name: Install AI Code Review tool
@@ -22,17 +22,9 @@ jobs:
         LLM_API_KEY: ${{ secrets.LLM_API_KEY }}
         LLM_API_TYPE: openai
         MODEL: "gpt-4.1"
+        GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
       run: |
         ai-code-review
+        ai-code-review github-comment --token "$GITHUB_TOKEN"
     - uses: actions/upload-artifact@v4
-      with: { name: ai-code-review-results, path: code-review-report.txt }
-    - name: Comment on PR with review
-      uses: actions/github-script@v7
-      with:
-        script: |
-          await github.rest.issues.createComment({
-            owner: context.repo.owner,
-            repo: context.repo.repo,
-            issue_number: context.issue.number,
-            body: require('fs').readFileSync('code-review-report.txt', 'utf8')
-          });
+      with: { name: ai-code-review-results, path: code-review-report.txt, code-review-report.json }

README.md

@@ -37,30 +37,22 @@ jobs:
     steps:
     - uses: actions/checkout@v4
       with: { fetch-depth: 0 }
-    - name: Set up Python 3.11
+    - name: Set up Python
       uses: actions/setup-python@v5
       with: { python-version: "3.13" }
     - name: Install AI Code Review tool
-      run: pip install ai-code-review==0.3.4
+      run: pip install ai-code-review==0.4.0
     - name: Run AI code review
       env:
         LLM_API_KEY: ${{ secrets.LLM_API_KEY }}
         LLM_API_TYPE: openai
         MODEL: "gpt-4.1"
+        GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
       run: |
         ai-code-review
+        ai-code-review github-comment --token ${{ secrets.GITHUB_TOKEN }}
     - uses: actions/upload-artifact@v4
-      with: { name: ai-code-review-results, path: code-review-report.txt }
-    - name: Comment on PR with review
-      uses: actions/github-script@v7
-      with:
-        script: |
-          await github.rest.issues.createComment({
-            owner: context.repo.owner,
-            repo: context.repo.repo,
-            issue_number: context.issue.number,
-            body: require('fs').readFileSync('code-review-report.txt', 'utf8')
-          });
+      with: { name: ai-code-review-results, path: code-review-report.txt, code-review-report.json }
 ```
 
 > ⚠️ Make sure to add `LLM_API_KEY` to your repository’s GitHub secrets.

ai_code_review/cli.py

@@ -1,4 +1,5 @@
 import asyncio
+import logging
 import sys
 import os
 import shutil
@@ -9,10 +10,11 @@
 from .core import review
 from .report_struct import Report
 from git import Repo
+import requests
 
 from .constants import ENV_CONFIG_FILE
 from .bootstrap import bootstrap
-
+from .project_config import ProjectConfig
 
 app = async_typer.AsyncTyper(
     pretty_exceptions_show_locals=False,
@@ -52,3 +54,59 @@ async def remote(url=typer.Option(), branch=typer.Option()):
         await review()
     finally:
         os.chdir(prev_dir)
+
+
+@app.async_command(help="Leave a GitHub PR comment with the review.")
+async def github_comment(
+    token: str = typer.Option(
+        os.environ.get("GITHUB_TOKEN", ""), help="GitHub token (or set GITHUB_TOKEN env var)"
+    ),
+):
+    """
+    Leaves a comment with the review on the current GitHub pull request.
+    """
+    file = "code-review-report.txt"
+    if not os.path.exists(file):
+        print(f"Review file not found: {file}")
+        raise typer.Exit(4)
+
+    with open(file, "r", encoding="utf-8") as f:
+        body = f.read()
+
+    if not token:
+        print("GitHub token is required (--token or GITHUB_TOKEN env var).")
+        raise typer.Exit(1)
+
+    github_env = ProjectConfig.load().prompt_vars["github_env"]
+    repo = github_env.get("github_repo", "")
+    pr_env_val = github_env.get("github_pr_number", "")
+    logging.info(f"github_pr_number = {pr_env_val}")
+
+    # e.g. could be "refs/pull/123/merge" or a direct number
+    if "/" in pr_env_val and "pull" in pr_env_val:
+        # refs/pull/123/merge
+        try:
+            pr_num_candidate = pr_env_val.strip("/").split("/")
+            idx = pr_num_candidate.index("pull")
+            pr = int(pr_num_candidate[idx + 1])
+        except Exception:
+            pr = 0
+    else:
+        try:
+            pr = int(pr_env_val)
+        except Exception:
+            pr = 0
+
+    api_url = f"https://api.github.com/repos/{repo}/issues/{pr}/comments"
+    headers = {
+        "Authorization": f"token {token}",
+        "Accept": "application/vnd.github+json",
+    }
+    data = {"body": body}
+
+    resp = requests.post(api_url, headers=headers, json=data)
+    if 200 <= resp.status_code < 300:
+        logging.info(f"Posted review comment to PR #{pr} in {repo}")
+    else:
+        logging.error(f"Failed to post comment: {resp.status_code} {resp.reason}\n{resp.text}")
+        raise typer.Exit(5)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ai-code-review"
-version = "0.3.4"
+version = "0.4.0"
 description = "LLM-agnostic GitHub AI Code Review Tool with integration to GitHub actions"
 authors = ["Nayjest <mail@vitaliy.in>"]
 readme = "README.md"