pullwise 0.1.0__tar.gz → 0.2.0__tar.gz

{pullwise-0.1.0 → pullwise-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pullwise
-Version: 0.1.0
+Version: 0.2.0
 Summary: AI-powered PR descriptions, labels, and code review using OpenAI
 License: MIT
 Project-URL: Homepage, https://github.com/albertusreza/pr-pilot
@@ -21,6 +21,11 @@ Requires-Dist: pytest; extra == "dev"
 
 # pr-pilot ✈️
 
+[![PyPI](https://img.shields.io/pypi/v/pullwise)](https://pypi.org/project/pullwise/)
+[![Downloads](https://img.shields.io/pypi/dm/pullwise)](https://pypi.org/project/pullwise/)
+[![CI](https://github.com/albertusreza/pr-pilot/actions/workflows/ci.yml/badge.svg)](https://github.com/albertusreza/pr-pilot/actions)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+
 **Stop writing PR descriptions.** Let AI do it.
 
 `pr-pilot` is a GitHub Action + CLI that analyzes your diff and commit history, then automatically writes a clear, structured PR description — with a summary, change list, test plan, and labels.
@@ -87,7 +92,7 @@ Add `OPENAI_API_KEY` to your repo secrets and you're done. Works with any langua
 ### As a CLI
 
 ```bash
-pip install pr-pilot
+pip install pullwise
 export OPENAI_API_KEY=sk-...
 
 # Generate a description for your current branch
@@ -103,6 +108,50 @@ pr-pilot review
 pr-pilot describe --markdown pr_description.md
 ```
 
+## Features
+
+| Command | What it does |
+|---|---|
+| `pr-pilot describe` | Generate a structured PR description from your diff |
+| `pr-pilot review` | Get a senior-engineer-style code review in the terminal |
+| `pr-pilot comment` | Post an AI review as a GitHub PR comment (updates on re-run, no spam) |
+| `pr-pilot changelog` | Generate a `CHANGELOG.md` entry from commits since last tag |
+
+## Usage
+
+```bash
+# Generate PR description (terminal)
+pr-pilot describe --base main
+
+# Get a quick code review in the terminal
+pr-pilot review --base main
+
+# Post review as a GitHub PR comment
+pr-pilot comment --repo owner/repo --pr 42
+
+# Generate CHANGELOG entry (print to stdout)
+pr-pilot changelog
+
+# Write directly into CHANGELOG.md
+pr-pilot changelog --output CHANGELOG.md
+```
+
+### Auto-comment on every PR (GitHub Action)
+
+```yaml
+# .github/workflows/pr-review.yml
+- uses: actions/checkout@v4
+  with:
+    fetch-depth: 0
+- run: pip install pullwise
+- run: pr-pilot comment --repo ${{ github.repository }} --pr ${{ github.event.pull_request.number }}
+  env:
+    OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+    GITHUB_TOKEN: ${{ github.token }}
+```
+
+The comment updates itself on every new push — no duplicate comments.
+
 ## Options
 
 | Input | Default | Description |

{pullwise-0.1.0 → pullwise-0.2.0}/README.md

@@ -1,5 +1,10 @@
 # pr-pilot ✈️
 
+[![PyPI](https://img.shields.io/pypi/v/pullwise)](https://pypi.org/project/pullwise/)
+[![Downloads](https://img.shields.io/pypi/dm/pullwise)](https://pypi.org/project/pullwise/)
+[![CI](https://github.com/albertusreza/pr-pilot/actions/workflows/ci.yml/badge.svg)](https://github.com/albertusreza/pr-pilot/actions)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+
 **Stop writing PR descriptions.** Let AI do it.
 
 `pr-pilot` is a GitHub Action + CLI that analyzes your diff and commit history, then automatically writes a clear, structured PR description — with a summary, change list, test plan, and labels.
@@ -66,7 +71,7 @@ Add `OPENAI_API_KEY` to your repo secrets and you're done. Works with any langua
 ### As a CLI
 
 ```bash
-pip install pr-pilot
+pip install pullwise
 export OPENAI_API_KEY=sk-...
 
 # Generate a description for your current branch
@@ -82,6 +87,50 @@ pr-pilot review
 pr-pilot describe --markdown pr_description.md
 ```
 
+## Features
+
+| Command | What it does |
+|---|---|
+| `pr-pilot describe` | Generate a structured PR description from your diff |
+| `pr-pilot review` | Get a senior-engineer-style code review in the terminal |
+| `pr-pilot comment` | Post an AI review as a GitHub PR comment (updates on re-run, no spam) |
+| `pr-pilot changelog` | Generate a `CHANGELOG.md` entry from commits since last tag |
+
+## Usage
+
+```bash
+# Generate PR description (terminal)
+pr-pilot describe --base main
+
+# Get a quick code review in the terminal
+pr-pilot review --base main
+
+# Post review as a GitHub PR comment
+pr-pilot comment --repo owner/repo --pr 42
+
+# Generate CHANGELOG entry (print to stdout)
+pr-pilot changelog
+
+# Write directly into CHANGELOG.md
+pr-pilot changelog --output CHANGELOG.md
+```
+
+### Auto-comment on every PR (GitHub Action)
+
+```yaml
+# .github/workflows/pr-review.yml
+- uses: actions/checkout@v4
+  with:
+    fetch-depth: 0
+- run: pip install pullwise
+- run: pr-pilot comment --repo ${{ github.repository }} --pr ${{ github.event.pull_request.number }}
+  env:
+    OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+    GITHUB_TOKEN: ${{ github.token }}
+```
+
+The comment updates itself on every new push — no duplicate comments.
+
 ## Options
 
 | Input | Default | Description |

pullwise-0.2.0/pr_pilot/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.2.0"

pullwise-0.2.0/pr_pilot/analyzer.py

@@ -0,0 +1,232 @@
+from __future__ import annotations
+import json
+import subprocess
+from dataclasses import dataclass, field
+
+from openai import OpenAI
+
+from .templates import (
+    DESCRIBE_SYSTEM, DESCRIBE_USER, LABEL_SYSTEM, REVIEW_SYSTEM,
+    CHANGELOG_SYSTEM, CHANGELOG_USER, REVIEW_COMMENT_HEADER, REVIEW_COMMENT_TEMPLATE,
+)
+
+_MAX_DIFF_CHARS = 24_000   # stay well within context limits
+_MODEL = "gpt-4o"
+
+
+@dataclass
+class PRDescription:
+    title: str
+    summary: str
+    changes: list[str]
+    breaking: bool
+    breaking_notes: str | None
+    test_plan: str
+    labels: list[str]
+
+    def to_markdown(self) -> str:
+        lines = [
+            f"## Summary\n{self.summary}\n",
+            "## Changes",
+        ]
+        for c in self.changes:
+            lines.append(f"- {c}")
+        if self.breaking:
+            lines.append(f"\n## ⚠️ Breaking Changes\n{self.breaking_notes}")
+        lines.append(f"\n## Test Plan\n{self.test_plan}")
+        return "\n".join(lines)
+
+
+def _git(*args: str) -> str:
+    result = subprocess.run(["git"] + list(args), capture_output=True, text=True)
+    return result.stdout.strip()
+
+
+def get_diff(base: str = "main") -> str:
+    diff = _git("diff", f"{base}...HEAD")
+    if len(diff) > _MAX_DIFF_CHARS:
+        diff = diff[:_MAX_DIFF_CHARS] + "\n\n[diff truncated — showing first 24k chars]"
+    return diff
+
+
+def get_commits(base: str = "main") -> str:
+    return _git("log", f"{base}...HEAD", "--oneline", "--no-merges")
+
+
+def get_branch() -> str:
+    return _git("rev-parse", "--abbrev-ref", "HEAD")
+
+
+def describe_pr(api_key: str, base: str = "main", model: str = _MODEL) -> PRDescription:
+    client = OpenAI(api_key=api_key)
+    diff = get_diff(base)
+    commits = get_commits(base)
+    branch = get_branch()
+
+    if not diff and not commits:
+        raise ValueError(f"No changes found between '{branch}' and '{base}'")
+
+    user_msg = DESCRIBE_USER.format(
+        branch=branch, base=base, commits=commits or "(none)", diff=diff or "(empty)"
+    )
+    resp = client.chat.completions.create(
+        model=model,
+        max_tokens=1024,
+        messages=[
+            {"role": "system", "content": DESCRIBE_SYSTEM},
+            {"role": "user", "content": user_msg},
+        ],
+    )
+    raw = resp.choices[0].message.content or "{}"
+    raw = raw.strip()
+    if raw.startswith("```"):
+        raw = "\n".join(raw.splitlines()[1:])
+    if raw.endswith("```"):
+        raw = "\n".join(raw.splitlines()[:-1])
+
+    data = json.loads(raw.strip())
+    return PRDescription(
+        title=data.get("title", branch),
+        summary=data.get("summary", ""),
+        changes=data.get("changes", []),
+        breaking=data.get("breaking", False),
+        breaking_notes=data.get("breaking_notes"),
+        test_plan=data.get("test_plan", ""),
+        labels=data.get("labels", ["feature"]),
+    )
+
+
+def suggest_labels(api_key: str, title: str, body: str, model: str = _MODEL) -> list[str]:
+    client = OpenAI(api_key=api_key)
+    resp = client.chat.completions.create(
+        model=model,
+        max_tokens=64,
+        messages=[
+            {"role": "system", "content": LABEL_SYSTEM},
+            {"role": "user", "content": f"Title: {title}\n\nBody: {body[:2000]}"},
+        ],
+    )
+    raw = resp.choices[0].message.content or "[]"
+    return json.loads(raw.strip())
+
+
+def review_pr(api_key: str, base: str = "main", model: str = _MODEL) -> str:
+    client = OpenAI(api_key=api_key)
+    diff = get_diff(base)
+    if not diff:
+        return "No changes to review."
+    resp = client.chat.completions.create(
+        model=model,
+        max_tokens=1024,
+        messages=[
+            {"role": "system", "content": REVIEW_SYSTEM},
+            {"role": "user", "content": f"Diff:\n\n{diff}"},
+        ],
+    )
+    return (resp.choices[0].message.content or "").strip()
+
+
+def review_pr_as_comment(api_key: str, base: str = "main", model: str = _MODEL) -> str:
+    """Return a GitHub-flavoured markdown comment with the AI review."""
+    body = review_pr(api_key, base=base, model=model)
+    return REVIEW_COMMENT_TEMPLATE.format(header=REVIEW_COMMENT_HEADER, body=body)
+
+
+# ── Changelog ────────────────────────────────────────────────────────────────
+
+@dataclass
+class ChangelogEntry:
+    version_bump: str        # "patch" | "minor" | "major"
+    highlights: str
+    added: list[str]
+    changed: list[str]
+    fixed: list[str]
+    removed: list[str]
+    security: list[str]
+
+    def to_markdown(self, new_version: str, date: str) -> str:
+        lines = [f"## [{new_version}] - {date}", f"\n_{self.highlights}_\n"]
+        sections = [
+            ("Added", self.added),
+            ("Changed", self.changed),
+            ("Fixed", self.fixed),
+            ("Removed", self.removed),
+            ("Security", self.security),
+        ]
+        for title, items in sections:
+            if items:
+                lines.append(f"\n### {title}")
+                for item in items:
+                    lines.append(f"- {item}")
+        return "\n".join(lines)
+
+
+def _get_current_version() -> str:
+    """Read version from pyproject.toml, setup.cfg, or package __init__."""
+    try:
+        tag = _git("describe", "--tags", "--abbrev=0")
+        return tag.lstrip("v") if tag else "0.0.0"
+    except Exception:
+        return "0.0.0"
+
+
+def _get_commits_since_tag() -> str:
+    tag = _git("describe", "--tags", "--abbrev=0")
+    if tag:
+        return _git("log", f"{tag}...HEAD", "--oneline", "--no-merges")
+    return _git("log", "--oneline", "--no-merges", "-30")
+
+
+def _bump_version(current: str, bump: str) -> str:
+    parts = current.split(".")
+    try:
+        major, minor, patch = int(parts[0]), int(parts[1]), int(parts[2].split("-")[0])
+    except (IndexError, ValueError):
+        return "0.1.0"
+    if bump == "major":
+        return f"{major + 1}.0.0"
+    if bump == "minor":
+        return f"{major}.{minor + 1}.0"
+    return f"{major}.{minor}.{patch + 1}"
+
+
+def generate_changelog(api_key: str, model: str = _MODEL) -> tuple[ChangelogEntry, str]:
+    """Return (ChangelogEntry, new_version_string)."""
+    import datetime
+    client = OpenAI(api_key=api_key)
+    current = _get_current_version()
+    commits = _get_commits_since_tag()
+    diff = get_diff(base="HEAD~10") if not commits else get_diff(base=f"v{current}" if current != "0.0.0" else "HEAD~10")
+
+    if not commits:
+        raise ValueError("No commits found since last tag. Nothing to changelog.")
+
+    user_msg = CHANGELOG_USER.format(
+        current_version=current, commits=commits, diff=diff[:12_000] or "(empty)"
+    )
+    resp = client.chat.completions.create(
+        model=model,
+        max_tokens=1024,
+        messages=[
+            {"role": "system", "content": CHANGELOG_SYSTEM},
+            {"role": "user", "content": user_msg},
+        ],
+    )
+    raw = (resp.choices[0].message.content or "{}").strip()
+    if raw.startswith("```"):
+        raw = "\n".join(raw.splitlines()[1:])
+    if raw.endswith("```"):
+        raw = "\n".join(raw.splitlines()[:-1])
+
+    data = json.loads(raw.strip())
+    entry = ChangelogEntry(
+        version_bump=data.get("version", "patch"),
+        highlights=data.get("highlights", ""),
+        added=data.get("added", []),
+        changed=data.get("changed", []),
+        fixed=data.get("fixed", []),
+        removed=data.get("removed", []),
+        security=data.get("security", []),
+    )
+    new_version = _bump_version(current, entry.version_bump)
+    return entry, new_version

{pullwise-0.1.0 → pullwise-0.2.0}/pr_pilot/cli.py

@@ -3,7 +3,7 @@ import argparse
 import os
 import sys
 
-from .analyzer import describe_pr, suggest_labels, review_pr
+from .analyzer import describe_pr, suggest_labels, review_pr, review_pr_as_comment, generate_changelog
 
 _RESET = "\033[0m"
 _BOLD  = "\033[1m"
@@ -52,6 +52,54 @@ def cmd_review(args: argparse.Namespace) -> None:
     print()
 
 
+def cmd_comment(args: argparse.Namespace) -> None:
+    """Post AI review as a PR comment on GitHub."""
+    from .github_client import upsert_comment
+    from .templates import REVIEW_COMMENT_HEADER
+
+    repo   = args.repo or os.environ.get("GITHUB_REPOSITORY", "")
+    pr_num = args.pr or int(os.environ.get("PR_NUMBER", "0"))
+
+    if not repo or not pr_num:
+        print("pr-pilot comment: --repo and --pr are required (or set GITHUB_REPOSITORY / PR_NUMBER)", file=sys.stderr)
+        sys.exit(1)
+
+    print(f"\n  {_DIM}Generating review for PR #{pr_num}...{_RESET}\n")
+    comment_body = review_pr_as_comment(_key(), base=args.base, model=args.model)
+    url = upsert_comment(repo, pr_num, comment_body, REVIEW_COMMENT_HEADER)
+    print(f"  {_GREEN}✓{_RESET} Review posted: {url}\n")
+
+
+def cmd_changelog(args: argparse.Namespace) -> None:
+    import datetime
+    print(f"\n  {_DIM}Generating changelog from commits...{_RESET}\n")
+    entry, new_version = generate_changelog(_key(), model=args.model)
+    today = datetime.date.today().isoformat()
+    md = entry.to_markdown(new_version, today)
+
+    print(f"{_BOLD}Suggested version bump:{_RESET} {entry.version_bump}  →  {_GREEN}{new_version}{_RESET}")
+    print(f"{_BOLD}Highlights:{_RESET} {entry.highlights}\n")
+    print(md)
+
+    if args.output:
+        changelog_path = args.output
+        import pathlib
+        p = pathlib.Path(changelog_path)
+        if p.exists():
+            existing = p.read_text()
+            # Insert after the first line (# Changelog header) if present
+            if existing.startswith("# "):
+                header, rest = existing.split("\n", 1)
+                new_content = f"{header}\n\n{md}\n{rest}"
+            else:
+                new_content = f"{md}\n\n{existing}"
+        else:
+            new_content = f"# Changelog\n\n{md}\n"
+        p.write_text(new_content)
+        print(f"\n  {_GREEN}✓{_RESET} Prepended to {changelog_path}")
+    print()
+
+
 def cmd_action(args: argparse.Namespace) -> None:
     """Run as a GitHub Action — reads env vars set by the Actions runner."""
     import json as _json
@@ -107,6 +155,21 @@ def main() -> None:
     p_rev.add_argument("--model", default="gpt-4o", help="OpenAI model to use")
     p_rev.set_defaults(func=cmd_review)
 
+    # --- comment ---
+    p_com = sub.add_parser("comment", help="Post an AI review as a GitHub PR comment")
+    p_com.add_argument("--base", default="main", help="Base branch to diff against (default: main)")
+    p_com.add_argument("--model", default="gpt-4o", help="OpenAI model to use")
+    p_com.add_argument("--repo", default=None, help="GitHub repo slug (e.g. owner/repo)")
+    p_com.add_argument("--pr", type=int, default=None, help="PR number")
+    p_com.set_defaults(func=cmd_comment)
+
+    # --- changelog ---
+    p_cl = sub.add_parser("changelog", help="Generate a CHANGELOG.md entry from commits since last tag")
+    p_cl.add_argument("--model", default="gpt-4o", help="OpenAI model to use")
+    p_cl.add_argument("--output", default=None, metavar="FILE",
+                      help="Prepend entry to FILE (e.g. CHANGELOG.md). Prints to stdout if omitted.")
+    p_cl.set_defaults(func=cmd_changelog)
+
     # --- action (internal, called by entrypoint.sh) ---
     p_act = sub.add_parser("action", help="Run as a GitHub Action (internal)")
     p_act.add_argument("--model", default="gpt-4o")

{pullwise-0.1.0 → pullwise-0.2.0}/pr_pilot/github_client.py

@@ -78,3 +78,14 @@ def add_labels(repo: str, pr_number: int, labels: list[str]) -> None:
 
 def post_comment(repo: str, pr_number: int, body: str) -> None:
     _api("POST", f"/repos/{repo}/issues/{pr_number}/comments", {"body": body})
+
+
+def upsert_comment(repo: str, pr_number: int, body: str, marker: str) -> str:
+    """Post or update a comment that contains `marker`. Returns the comment URL."""
+    comments = _api("GET", f"/repos/{repo}/issues/{pr_number}/comments?per_page=100")
+    for comment in comments:
+        if marker in (comment.get("body") or ""):
+            data = _api("PATCH", f"/repos/{repo}/issues/comments/{comment['id']}", {"body": body})
+            return data.get("html_url", "")
+    data = _api("POST", f"/repos/{repo}/issues/{pr_number}/comments", {"body": body})
+    return data.get("html_url", "")

{pullwise-0.1.0 → pullwise-0.2.0}/pr_pilot/templates.py

@@ -48,3 +48,46 @@ Given a diff, summarize:
 
 Be direct and specific. No flattery. Format as plain markdown.
 """
+
+REVIEW_COMMENT_HEADER = "<!-- pr-pilot-review -->"
+
+REVIEW_COMMENT_TEMPLATE = """\
+{header}
+### 🤖 pr-pilot review
+
+{body}
+
+---
+<sub>Generated by [pr-pilot](https://github.com/albertusreza/pr-pilot) · [pullwise](https://pypi.org/project/pullwise/) on PyPI</sub>
+"""
+
+CHANGELOG_SYSTEM = """\
+You are a technical writer generating a CHANGELOG entry for a software release.
+Given git commits and a diff summary, write a well-structured CHANGELOG section.
+
+Return a JSON object with exactly these keys:
+{
+  "version": "suggested semver bump: patch | minor | major",
+  "highlights": "one-sentence release summary",
+  "added": ["list of new features added"],
+  "changed": ["list of changes to existing functionality"],
+  "fixed": ["list of bug fixes"],
+  "removed": ["list of removed features"],
+  "security": ["list of security fixes"]
+}
+
+Rules:
+- Only include keys with at least one item (empty lists are OK, they'll be filtered)
+- Follow Keep a Changelog format: https://keepachangelog.com
+- Be specific and user-facing. Omit internal refactors unless they affect behavior.
+- Return ONLY the JSON object, no markdown fences
+"""
+
+CHANGELOG_USER = """\
+Version to release from: {current_version}
+Commits since last tag:
+{commits}
+
+Diff summary (may be truncated):
+{diff}
+"""

{pullwise-0.1.0 → pullwise-0.2.0}/pullwise.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pullwise
-Version: 0.1.0
+Version: 0.2.0
 Summary: AI-powered PR descriptions, labels, and code review using OpenAI
 License: MIT
 Project-URL: Homepage, https://github.com/albertusreza/pr-pilot
@@ -21,6 +21,11 @@ Requires-Dist: pytest; extra == "dev"
 
 # pr-pilot ✈️
 
+[![PyPI](https://img.shields.io/pypi/v/pullwise)](https://pypi.org/project/pullwise/)
+[![Downloads](https://img.shields.io/pypi/dm/pullwise)](https://pypi.org/project/pullwise/)
+[![CI](https://github.com/albertusreza/pr-pilot/actions/workflows/ci.yml/badge.svg)](https://github.com/albertusreza/pr-pilot/actions)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+
 **Stop writing PR descriptions.** Let AI do it.
 
 `pr-pilot` is a GitHub Action + CLI that analyzes your diff and commit history, then automatically writes a clear, structured PR description — with a summary, change list, test plan, and labels.
@@ -87,7 +92,7 @@ Add `OPENAI_API_KEY` to your repo secrets and you're done. Works with any langua
 ### As a CLI
 
 ```bash
-pip install pr-pilot
+pip install pullwise
 export OPENAI_API_KEY=sk-...
 
 # Generate a description for your current branch
@@ -103,6 +108,50 @@ pr-pilot review
 pr-pilot describe --markdown pr_description.md
 ```
 
+## Features
+
+| Command | What it does |
+|---|---|
+| `pr-pilot describe` | Generate a structured PR description from your diff |
+| `pr-pilot review` | Get a senior-engineer-style code review in the terminal |
+| `pr-pilot comment` | Post an AI review as a GitHub PR comment (updates on re-run, no spam) |
+| `pr-pilot changelog` | Generate a `CHANGELOG.md` entry from commits since last tag |
+
+## Usage
+
+```bash
+# Generate PR description (terminal)
+pr-pilot describe --base main
+
+# Get a quick code review in the terminal
+pr-pilot review --base main
+
+# Post review as a GitHub PR comment
+pr-pilot comment --repo owner/repo --pr 42
+
+# Generate CHANGELOG entry (print to stdout)
+pr-pilot changelog
+
+# Write directly into CHANGELOG.md
+pr-pilot changelog --output CHANGELOG.md
+```
+
+### Auto-comment on every PR (GitHub Action)
+
+```yaml
+# .github/workflows/pr-review.yml
+- uses: actions/checkout@v4
+  with:
+    fetch-depth: 0
+- run: pip install pullwise
+- run: pr-pilot comment --repo ${{ github.repository }} --pr ${{ github.event.pull_request.number }}
+  env:
+    OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+    GITHUB_TOKEN: ${{ github.token }}
+```
+
+The comment updates itself on every new push — no duplicate comments.
+
 ## Options
 
 | Input | Default | Description |

{pullwise-0.1.0 → pullwise-0.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pullwise"
-version = "0.1.0"
+version = "0.2.0"
 description = "AI-powered PR descriptions, labels, and code review using OpenAI"
 readme = "README.md"
 license = { text = "MIT" }

pullwise-0.2.0/tests/test_analyzer.py

@@ -0,0 +1,162 @@
+from __future__ import annotations
+import json
+from unittest.mock import MagicMock, patch
+from pr_pilot.analyzer import (
+    describe_pr, suggest_labels, review_pr, PRDescription,
+    review_pr_as_comment, generate_changelog, ChangelogEntry,
+)
+from pr_pilot.templates import REVIEW_COMMENT_HEADER
+
+
+def _mock_openai_response(content: str) -> MagicMock:
+    choice = MagicMock()
+    choice.message.content = content
+    resp = MagicMock()
+    resp.choices = [choice]
+    return resp
+
+
+_SAMPLE_DESC = {
+    "title": "Add dark mode support",
+    "summary": "Implements a dark mode toggle that persists via localStorage.",
+    "changes": ["Add ThemeToggle component", "Add useTheme hook", "Update global CSS variables"],
+    "breaking": False,
+    "breaking_notes": None,
+    "test_plan": "Toggle dark mode, reload page — preference should persist.",
+    "labels": ["feature"],
+}
+
+
+@patch("pr_pilot.analyzer.get_diff", return_value="diff --git a/theme.css ...")
+@patch("pr_pilot.analyzer.get_commits", return_value="abc123 Add dark mode")
+@patch("pr_pilot.analyzer.get_branch", return_value="feat/dark-mode")
+def test_describe_pr_basic(mock_branch, mock_commits, mock_diff):
+    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
+        MockOpenAI.return_value.chat.completions.create.return_value = \
+            _mock_openai_response(json.dumps(_SAMPLE_DESC))
+        desc = describe_pr("fake-key", base="main")
+    assert desc.title == "Add dark mode support"
+    assert desc.breaking is False
+    assert "feature" in desc.labels
+    assert len(desc.changes) == 3
+
+
+@patch("pr_pilot.analyzer.get_diff", return_value="diff --git a/theme.css ...")
+@patch("pr_pilot.analyzer.get_commits", return_value="abc123 Add dark mode")
+@patch("pr_pilot.analyzer.get_branch", return_value="feat/dark-mode")
+def test_describe_pr_strips_markdown_fence(mock_branch, mock_commits, mock_diff):
+    raw = f"```json\n{json.dumps(_SAMPLE_DESC)}\n```"
+    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
+        MockOpenAI.return_value.chat.completions.create.return_value = \
+            _mock_openai_response(raw)
+        desc = describe_pr("fake-key")
+    assert desc.title == "Add dark mode support"
+
+
+def test_describe_pr_to_markdown():
+    desc = PRDescription(**_SAMPLE_DESC)
+    md = desc.to_markdown()
+    assert "## Summary" in md
+    assert "## Changes" in md
+    assert "## Test Plan" in md
+    assert "ThemeToggle" in md
+
+
+def test_suggest_labels():
+    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
+        MockOpenAI.return_value.chat.completions.create.return_value = \
+            _mock_openai_response('["bug", "security"]')
+        labels = suggest_labels("fake-key", "Fix XSS in comment input", "sanitize user input")
+    assert "bug" in labels
+    assert "security" in labels
+
+
+@patch("pr_pilot.analyzer.get_diff", return_value="- old line\n+ new line")
+@patch("pr_pilot.analyzer.get_commits", return_value="")
+@patch("pr_pilot.analyzer.get_branch", return_value="fix/xss")
+def test_review_pr(mock_branch, mock_commits, mock_diff):
+    review_text = "This PR sanitizes user input to prevent XSS attacks."
+    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
+        MockOpenAI.return_value.chat.completions.create.return_value = \
+            _mock_openai_response(review_text)
+        result = review_pr("fake-key")
+    assert "XSS" in result
+
+
+# ── Auto-comment tests ────────────────────────────────────────────────────────
+
+@patch("pr_pilot.analyzer.get_diff", return_value="+ new feature code")
+@patch("pr_pilot.analyzer.get_commits", return_value="")
+@patch("pr_pilot.analyzer.get_branch", return_value="feat/x")
+def test_review_pr_as_comment_contains_marker(mock_branch, mock_commits, mock_diff):
+    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
+        MockOpenAI.return_value.chat.completions.create.return_value = \
+            _mock_openai_response("Looks good overall.")
+        comment = review_pr_as_comment("fake-key")
+    assert REVIEW_COMMENT_HEADER in comment
+    assert "pr-pilot review" in comment
+    assert "Looks good overall." in comment
+
+
+@patch("pr_pilot.analyzer.get_diff", return_value="")
+@patch("pr_pilot.analyzer.get_commits", return_value="")
+@patch("pr_pilot.analyzer.get_branch", return_value="main")
+def test_review_pr_as_comment_no_diff(mock_branch, mock_commits, mock_diff):
+    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
+        MockOpenAI.return_value.chat.completions.create.return_value = \
+            _mock_openai_response("No changes to review.")
+        comment = review_pr_as_comment("fake-key")
+    assert "No changes to review." in comment
+
+
+# ── Changelog tests ───────────────────────────────────────────────────────────
+
+_SAMPLE_CHANGELOG = {
+    "version": "minor",
+    "highlights": "Adds dark mode and fixes login bug.",
+    "added": ["Dark mode toggle with localStorage persistence"],
+    "changed": ["Login flow now uses refresh tokens"],
+    "fixed": ["Session expiry on mobile Safari"],
+    "removed": [],
+    "security": [],
+}
+
+
+@patch("pr_pilot.analyzer.get_diff", return_value="diff --git a/theme.css ...")
+@patch("pr_pilot.analyzer.get_commits", return_value="")
+@patch("pr_pilot.analyzer.get_branch", return_value="main")
+@patch("pr_pilot.analyzer._get_commits_since_tag", return_value="abc123 Add dark mode\ndef456 Fix login")
+@patch("pr_pilot.analyzer._get_current_version", return_value="1.2.3")
+def test_generate_changelog(mock_ver, mock_commits_tag, mock_branch, mock_commits, mock_diff):
+    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
+        MockOpenAI.return_value.chat.completions.create.return_value = \
+            _mock_openai_response(json.dumps(_SAMPLE_CHANGELOG))
+        entry, new_version = generate_changelog("fake-key")
+    assert entry.version_bump == "minor"
+    assert new_version == "1.3.0"
+    assert "Dark mode" in entry.added[0]
+    assert entry.security == []
+
+
+def test_changelog_entry_to_markdown():
+    entry = ChangelogEntry(
+        version_bump="patch",
+        highlights="Bug fixes.",
+        added=[],
+        changed=[],
+        fixed=["Fix crash on startup"],
+        removed=[],
+        security=[],
+    )
+    md = entry.to_markdown("1.0.1", "2026-06-02")
+    assert "## [1.0.1] - 2026-06-02" in md
+    assert "### Fixed" in md
+    assert "Fix crash on startup" in md
+    assert "### Added" not in md   # empty section should be omitted
+
+
+def test_version_bump():
+    from pr_pilot.analyzer import _bump_version
+    assert _bump_version("1.2.3", "patch") == "1.2.4"
+    assert _bump_version("1.2.3", "minor") == "1.3.0"
+    assert _bump_version("1.2.3", "major") == "2.0.0"

pullwise-0.1.0/pr_pilot/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.1.0"

pullwise-0.1.0/pr_pilot/analyzer.py

@@ -1,123 +0,0 @@
-from __future__ import annotations
-import json
-import subprocess
-from dataclasses import dataclass, field
-
-from openai import OpenAI
-
-from .templates import DESCRIBE_SYSTEM, DESCRIBE_USER, LABEL_SYSTEM, REVIEW_SYSTEM
-
-_MAX_DIFF_CHARS = 24_000   # stay well within context limits
-_MODEL = "gpt-4o"
-
-
-@dataclass
-class PRDescription:
-    title: str
-    summary: str
-    changes: list[str]
-    breaking: bool
-    breaking_notes: str | None
-    test_plan: str
-    labels: list[str]
-
-    def to_markdown(self) -> str:
-        lines = [
-            f"## Summary\n{self.summary}\n",
-            "## Changes",
-        ]
-        for c in self.changes:
-            lines.append(f"- {c}")
-        if self.breaking:
-            lines.append(f"\n## ⚠️ Breaking Changes\n{self.breaking_notes}")
-        lines.append(f"\n## Test Plan\n{self.test_plan}")
-        return "\n".join(lines)
-
-
-def _git(*args: str) -> str:
-    result = subprocess.run(["git"] + list(args), capture_output=True, text=True)
-    return result.stdout.strip()
-
-
-def get_diff(base: str = "main") -> str:
-    diff = _git("diff", f"{base}...HEAD")
-    if len(diff) > _MAX_DIFF_CHARS:
-        diff = diff[:_MAX_DIFF_CHARS] + "\n\n[diff truncated — showing first 24k chars]"
-    return diff
-
-
-def get_commits(base: str = "main") -> str:
-    return _git("log", f"{base}...HEAD", "--oneline", "--no-merges")
-
-
-def get_branch() -> str:
-    return _git("rev-parse", "--abbrev-ref", "HEAD")
-
-
-def describe_pr(api_key: str, base: str = "main", model: str = _MODEL) -> PRDescription:
-    client = OpenAI(api_key=api_key)
-    diff = get_diff(base)
-    commits = get_commits(base)
-    branch = get_branch()
-
-    if not diff and not commits:
-        raise ValueError(f"No changes found between '{branch}' and '{base}'")
-
-    user_msg = DESCRIBE_USER.format(
-        branch=branch, base=base, commits=commits or "(none)", diff=diff or "(empty)"
-    )
-    resp = client.chat.completions.create(
-        model=model,
-        max_tokens=1024,
-        messages=[
-            {"role": "system", "content": DESCRIBE_SYSTEM},
-            {"role": "user", "content": user_msg},
-        ],
-    )
-    raw = resp.choices[0].message.content or "{}"
-    raw = raw.strip()
-    if raw.startswith("```"):
-        raw = "\n".join(raw.splitlines()[1:])
-    if raw.endswith("```"):
-        raw = "\n".join(raw.splitlines()[:-1])
-
-    data = json.loads(raw.strip())
-    return PRDescription(
-        title=data.get("title", branch),
-        summary=data.get("summary", ""),
-        changes=data.get("changes", []),
-        breaking=data.get("breaking", False),
-        breaking_notes=data.get("breaking_notes"),
-        test_plan=data.get("test_plan", ""),
-        labels=data.get("labels", ["feature"]),
-    )
-
-
-def suggest_labels(api_key: str, title: str, body: str, model: str = _MODEL) -> list[str]:
-    client = OpenAI(api_key=api_key)
-    resp = client.chat.completions.create(
-        model=model,
-        max_tokens=64,
-        messages=[
-            {"role": "system", "content": LABEL_SYSTEM},
-            {"role": "user", "content": f"Title: {title}\n\nBody: {body[:2000]}"},
-        ],
-    )
-    raw = resp.choices[0].message.content or "[]"
-    return json.loads(raw.strip())
-
-
-def review_pr(api_key: str, base: str = "main", model: str = _MODEL) -> str:
-    client = OpenAI(api_key=api_key)
-    diff = get_diff(base)
-    if not diff:
-        return "No changes to review."
-    resp = client.chat.completions.create(
-        model=model,
-        max_tokens=1024,
-        messages=[
-            {"role": "system", "content": REVIEW_SYSTEM},
-            {"role": "user", "content": f"Diff:\n\n{diff}"},
-        ],
-    )
-    return (resp.choices[0].message.content or "").strip()

pullwise-0.1.0/tests/test_analyzer.py

@@ -1,79 +0,0 @@
-from __future__ import annotations
-import json
-from unittest.mock import MagicMock, patch
-from pr_pilot.analyzer import describe_pr, suggest_labels, review_pr, PRDescription
-
-
-def _mock_openai_response(content: str) -> MagicMock:
-    choice = MagicMock()
-    choice.message.content = content
-    resp = MagicMock()
-    resp.choices = [choice]
-    return resp
-
-
-_SAMPLE_DESC = {
-    "title": "Add dark mode support",
-    "summary": "Implements a dark mode toggle that persists via localStorage.",
-    "changes": ["Add ThemeToggle component", "Add useTheme hook", "Update global CSS variables"],
-    "breaking": False,
-    "breaking_notes": None,
-    "test_plan": "Toggle dark mode, reload page — preference should persist.",
-    "labels": ["feature"],
-}
-
-
-@patch("pr_pilot.analyzer.get_diff", return_value="diff --git a/theme.css ...")
-@patch("pr_pilot.analyzer.get_commits", return_value="abc123 Add dark mode")
-@patch("pr_pilot.analyzer.get_branch", return_value="feat/dark-mode")
-def test_describe_pr_basic(mock_branch, mock_commits, mock_diff):
-    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
-        MockOpenAI.return_value.chat.completions.create.return_value = \
-            _mock_openai_response(json.dumps(_SAMPLE_DESC))
-        desc = describe_pr("fake-key", base="main")
-    assert desc.title == "Add dark mode support"
-    assert desc.breaking is False
-    assert "feature" in desc.labels
-    assert len(desc.changes) == 3
-
-
-@patch("pr_pilot.analyzer.get_diff", return_value="diff --git a/theme.css ...")
-@patch("pr_pilot.analyzer.get_commits", return_value="abc123 Add dark mode")
-@patch("pr_pilot.analyzer.get_branch", return_value="feat/dark-mode")
-def test_describe_pr_strips_markdown_fence(mock_branch, mock_commits, mock_diff):
-    raw = f"```json\n{json.dumps(_SAMPLE_DESC)}\n```"
-    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
-        MockOpenAI.return_value.chat.completions.create.return_value = \
-            _mock_openai_response(raw)
-        desc = describe_pr("fake-key")
-    assert desc.title == "Add dark mode support"
-
-
-def test_describe_pr_to_markdown():
-    desc = PRDescription(**_SAMPLE_DESC)
-    md = desc.to_markdown()
-    assert "## Summary" in md
-    assert "## Changes" in md
-    assert "## Test Plan" in md
-    assert "ThemeToggle" in md
-
-
-def test_suggest_labels():
-    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
-        MockOpenAI.return_value.chat.completions.create.return_value = \
-            _mock_openai_response('["bug", "security"]')
-        labels = suggest_labels("fake-key", "Fix XSS in comment input", "sanitize user input")
-    assert "bug" in labels
-    assert "security" in labels
-
-
-@patch("pr_pilot.analyzer.get_diff", return_value="- old line\n+ new line")
-@patch("pr_pilot.analyzer.get_commits", return_value="")
-@patch("pr_pilot.analyzer.get_branch", return_value="fix/xss")
-def test_review_pr(mock_branch, mock_commits, mock_diff):
-    review_text = "This PR sanitizes user input to prevent XSS attacks."
-    with patch("pr_pilot.analyzer.OpenAI") as MockOpenAI:
-        MockOpenAI.return_value.chat.completions.create.return_value = \
-            _mock_openai_response(review_text)
-        result = review_pr("fake-key")
-    assert "XSS" in result