✨ feat: Add CI checks functionality for pull requests with rendering support

Author: SigureMo

src/gh_llm/commands/pr.py

@@ -5,6 +5,7 @@
 from gh_llm.github_api import GitHubClient
 from gh_llm.pager import DEFAULT_PAGE_SIZE, TimelinePager
 from gh_llm.render import (
+    render_checks_section,
     render_event_detail,
     render_expand_hints,
     render_header,
@@ -60,6 +61,12 @@ def register_pr_parser(subparsers: Any) -> None:
     review_expand_parser.add_argument("--page-size", type=int, help="timeline entries per page")
     review_expand_parser.set_defaults(handler=cmd_pr_review_expand)
 
+    checks_parser = pr_subparsers.add_parser("checks", help="show CI checks for the pull request")
+    checks_parser.add_argument("--pr", help="PR number/url/branch")
+    checks_parser.add_argument("--repo", help="repository in OWNER/REPO format")
+    checks_parser.add_argument("--all", action="store_true", help="show all checks including passed")
+    checks_parser.set_defaults(handler=cmd_pr_checks)
+
     thread_reply_parser = pr_subparsers.add_parser(
         "thread-reply", help="reply to a pull request review thread"
     )
@@ -141,6 +148,14 @@ def cmd_pr_view(args: Any) -> int:
     print()
     for line in render_expand_hints(context, shown_pages):
         print(line)
+    checks = client.fetch_checks(meta.ref) if meta.state == "OPEN" else []
+    for line in render_checks_section(
+        context=context,
+        checks=checks,
+        show_all=False,
+        is_open=(meta.state == "OPEN"),
+    ):
+        print(line)
     for line in render_pr_actions(context):
         print(line)
 
@@ -223,6 +238,21 @@ def cmd_pr_review_expand(args: Any) -> int:
     return 0
 
 
+def cmd_pr_checks(args: Any) -> int:
+    client = GitHubClient()
+    pager = TimelinePager(client)
+    context, meta = _resolve_context_and_meta(client=client, pager=pager, args=args)
+    checks = client.fetch_checks(meta.ref)
+    for line in render_checks_section(
+        context=context,
+        checks=checks,
+        show_all=bool(args.all),
+        is_open=(meta.state == "OPEN"),
+    ):
+        print(line)
+    return 0
+
+
 def cmd_pr_thread_reply(args: Any) -> int:
     client = GitHubClient()
     if args.repo is not None and args.pr is None:

src/gh_llm/github_api.py

@@ -8,7 +8,7 @@
 from typing import cast
 from urllib.parse import urlparse
 
-from gh_llm.models import PageInfo, PullRequestMeta, PullRequestRef, TimelineEvent, TimelinePage
+from gh_llm.models import CheckItem, PageInfo, PullRequestMeta, PullRequestRef, TimelineEvent, TimelinePage
 
 MAX_INLINE_TEXT = 8000
 MAX_INLINE_LINES = 200
@@ -244,6 +244,41 @@ class ReferenceSubject:
 }
 """.strip()
 
+CHECKS_QUERY = """
+query($owner:String!,$name:String!,$number:Int!){
+  repository(owner:$owner,name:$name){
+    pullRequest(number:$number){
+      commits(last:1){
+        nodes{
+          commit{
+            statusCheckRollup{
+              contexts(first:100){
+                nodes{
+                  __typename
+                  ... on CheckRun{
+                    name
+                    status
+                    conclusion
+                    detailsUrl
+                    databaseId
+                  }
+                  ... on StatusContext{
+                    context
+                    state
+                    targetUrl
+                    description
+                  }
+                }
+              }
+            }
+          }
+        }
+      }
+    }
+  }
+}
+""".strip()
+
 
 class GitHubClient:
     def __init__(self) -> None:
@@ -508,6 +543,65 @@ def _get_viewer_login(self) -> str:
         login = _as_optional_str(payload.get("login"))
         return login or ""
 
+    def fetch_checks(self, ref: PullRequestRef) -> list[CheckItem]:
+        payload = _run_graphql_payload(
+            CHECKS_QUERY,
+            {"owner": ref.owner, "name": ref.name, "number": ref.number},
+        )
+        data_obj = _as_dict(payload.get("data"), context="graphql data")
+        repo_obj = _as_dict(data_obj.get("repository"), context="repository")
+        pr_obj = _as_dict(repo_obj.get("pullRequest"), context="pullRequest")
+        commits_obj = _as_dict(pr_obj.get("commits"), context="commits")
+        nodes = _as_list(commits_obj.get("nodes"))
+        if not nodes:
+            return []
+        head = _as_dict(nodes[0], context="commit node")
+        commit_obj = _as_dict(head.get("commit"), context="commit")
+        rollup_obj = _as_dict_optional(commit_obj.get("statusCheckRollup"))
+        if rollup_obj is None:
+            return []
+        contexts_obj = _as_dict_optional(rollup_obj.get("contexts"))
+        if contexts_obj is None:
+            return []
+
+        items: list[CheckItem] = []
+        for raw in _as_list(contexts_obj.get("nodes")):
+            node = _as_dict(raw, context="check context")
+            typename = _as_optional_str(node.get("__typename")) or ""
+            if typename == "CheckRun":
+                name = (_as_optional_str(node.get("name")) or "").strip() or "(unnamed check run)"
+                status = _as_optional_str(node.get("status")) or "UNKNOWN"
+                conclusion = _as_optional_str(node.get("conclusion"))
+                label = f"{status}/{(conclusion or 'NONE')}"
+                details_url = _as_optional_str(node.get("detailsUrl"))
+                run_id, job_id = _extract_actions_run_and_job_ids(details_url)
+                items.append(
+                    CheckItem(
+                        name=name,
+                        kind="check-run",
+                        status=label,
+                        passed=_is_check_run_passed(status=status, conclusion=conclusion),
+                        details_url=details_url,
+                        run_id=run_id,
+                        job_id=job_id,
+                    )
+                )
+                continue
+            if typename == "StatusContext":
+                name = (_as_optional_str(node.get("context")) or "").strip() or "(unnamed status)"
+                state = _as_optional_str(node.get("state")) or "UNKNOWN"
+                items.append(
+                    CheckItem(
+                        name=name,
+                        kind="status-context",
+                        status=state,
+                        passed=(state == "SUCCESS"),
+                        details_url=_as_optional_str(node.get("targetUrl")),
+                        run_id=None,
+                    )
+                )
+        return items
+
 def _run_graphql_connection(query: str, variables: dict[str, str | int]) -> dict[str, object]:
     payload = _run_graphql_payload(query, variables)
     data_obj = _as_dict(payload.get("data"), context="graphql data")
@@ -931,7 +1025,7 @@ def _render_review_thread_block(
                 viewer_login=viewer_login,
             )
         )
-    lines.append(f"  🆔 thread_id: {thread_id}")
+    lines.append(f"  ◌ thread_id: {thread_id}")
     lines.append("  ⌨ reply_body: '<reply>'")
     lines.append(
         f"  ⏎ Reply via gh-llm: `gh-llm pr thread-reply {thread_id} --body '<reply>' --pr {ref.number} --repo {ref.owner}/{ref.name}`"
@@ -981,7 +1075,7 @@ def _render_review_comment_block(
         lines.append(f"  Reactions: {reactions_summary}")
     comment_id = _as_optional_str(comment.get("id")) or ""
     if comment_id and author == viewer_login:
-        lines.append(f"  🆔 comment_id: {comment_id}")
+        lines.append(f"  ◌ comment_id: {comment_id}")
         lines.append("  ⌨ comment_body: '<comment_body>'")
         lines.append(
             f"  ⏎ Edit comment via gh-llm: `gh-llm pr comment-edit {comment_id} --body '<comment_body>' --pr {ref.number} --repo {ref.owner}/{ref.name}`"
@@ -1154,6 +1248,36 @@ def _is_retryable_gh_error(stderr: str) -> bool:
     return any(pattern in lowered for pattern in retryable_patterns)
 
 
+def _is_check_run_passed(*, status: str, conclusion: str | None) -> bool:
+    if status != "COMPLETED":
+        return False
+    return (conclusion or "").upper() in {"SUCCESS", "NEUTRAL", "SKIPPED"}
+
+
+def _extract_actions_run_and_job_ids(details_url: str | None) -> tuple[int | None, int | None]:
+    if not details_url:
+        return None, None
+    parsed = urlparse(details_url)
+    parts = [segment for segment in parsed.path.split("/") if segment]
+    # Expected shape: /<owner>/<repo>/actions/runs/<run_id>/job/<job_id>
+    run_id: int | None = None
+    job_id: int | None = None
+    for idx, part in enumerate(parts):
+        if part == "runs" and idx + 1 < len(parts):
+            run_id = _parse_positive_int(parts[idx + 1])
+        if part == "job" and idx + 1 < len(parts):
+            job_id = _parse_positive_int(parts[idx + 1])
+    return run_id, job_id
+
+
+def _parse_positive_int(raw: str) -> int | None:
+    try:
+        value = int(raw)
+    except ValueError:
+        return None
+    return value if value > 0 else None
+
+
 def _reference_subject_summary(source: dict[str, object] | None) -> ReferenceSubject | None:
     if source is None:
         return None

src/gh_llm/models.py

@@ -57,6 +57,17 @@ class TimelinePage:
     page_info: PageInfo
 
 
+@dataclass(frozen=True)
+class CheckItem:
+    name: str
+    kind: str
+    status: str
+    passed: bool
+    details_url: str | None = None
+    run_id: int | None = None
+    job_id: int | None = None
+
+
 @dataclass
 class TimelineContext:
     owner: str

src/gh_llm/render.py

@@ -7,7 +7,7 @@
 if TYPE_CHECKING:
     from string.templatelib import Template
 
-    from gh_llm.models import TimelineContext, TimelineEvent, TimelinePage
+    from gh_llm.models import CheckItem, TimelineContext, TimelineEvent, TimelinePage
 
 
 def render_header(context: TimelineContext) -> list[str]:
@@ -89,6 +89,65 @@ def render_pr_actions(context: TimelineContext) -> list[str]:
     ]
 
 
+def render_checks_section(
+    *,
+    context: TimelineContext,
+    checks: list[CheckItem],
+    show_all: bool,
+    is_open: bool,
+) -> list[str]:
+    repo = f"{context.owner}/{context.name}"
+    if not is_open:
+        return [
+            "## Checks",
+            f"Closed PR: checks are hidden by default. ⏎ run `gh-llm pr checks --pr {context.number} --repo {repo} --all`",
+            "",
+        ]
+
+    visible = checks if show_all else [item for item in checks if not item.passed]
+    hidden_count = max(0, len(checks) - len(visible))
+    lines = ["## Checks"]
+    if not visible:
+        if checks:
+            lines.append("All checks passed.")
+        else:
+            lines.append("(no checks found)")
+    else:
+        for idx, item in enumerate(visible, start=1):
+            lines.append(f"{idx}. [{item.status}] {item.name} ({item.kind})")
+            if item.run_id is not None:
+                if item.job_id is not None:
+                    lines.append(
+                        f"   ⏎ details: `gh run view {item.run_id} --job {item.job_id} --repo {repo}`"
+                    )
+                    lines.append(
+                        f"   ⏎ logs: `gh run view {item.run_id} --job {item.job_id} --log --repo {repo}`"
+                    )
+                else:
+                    lines.append(
+                        f"   ⏎ details: `gh run view {item.run_id} --repo {repo}`"
+                    )
+                    lines.append(
+                        f"   ⏎ logs: `gh run view {item.run_id} --log --repo {repo}`"
+                    )
+            elif item.details_url:
+                lines.append(f"   ⏎ details: `{item.details_url}`")
+    if show_all:
+        lines.append(
+            f"⏎ show only non-passed: `gh-llm pr checks --pr {context.number} --repo {repo}`"
+        )
+    elif hidden_count > 0:
+        lines.append(
+            f"{hidden_count} passed checks hidden. ⏎ show all: `gh-llm pr checks --pr {context.number} --repo {repo} --all`"
+        )
+    else:
+        lines.append(
+            f"⏎ show all: `gh-llm pr checks --pr {context.number} --repo {repo} --all`"
+        )
+    lines.append("")
+    return lines
+
+
 def render_hidden_gap(context: TimelineContext, hidden_pages: list[int]) -> list[str]:
     if not hidden_pages:
         return []
@@ -118,7 +177,7 @@ def _render_item(index: int, event: TimelineEvent, context: TimelineContext) ->
         if event.reactions_summary:
             lines.append(f"   Reactions: {event.reactions_summary}")
         if event.editable_comment_id:
-            lines.append(f"   🆔 comment_id: {event.editable_comment_id}")
+            lines.append(f"   ◌ comment_id: {event.editable_comment_id}")
             lines.append("   ⌨ comment_body: '<comment_body>'")
             lines.append(
                 f"   ⏎ Edit comment via gh-llm: `gh-llm pr comment-edit {event.editable_comment_id} --body '<comment_body>' --pr {context.number} --repo {context.owner}/{context.name}`"