pi-pr-status 0.1.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Bruno Garcia
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,80 @@
+# pi-pr-status
+
+A [Pi](https://github.com/badlogic/pi) extension that shows your current PR's status right in the footer — so you always know which PR you're working on, whether CI is green, and if there are review comments to address.
+
+## What it shows
+
+When your current git branch has an open pull request, the footer displays:
+
+```
+🟢 PR #42 · ✅ 5 checks passed · https://github.com/owner/repo/pull/42
+```
+
+**CI failures?**
+```
+🟢 PR #42 · ❌ 2/5 checks failed · https://github.com/owner/repo/pull/42
+```
+
+**Checks still running?**
+```
+🟢 PR #42 · ⏳ 3/5 checks pending · https://github.com/owner/repo/pull/42
+```
+
+**Unresolved review comments?**
+```
+🟢 PR #42 · ✅ 5 checks passed · 💬 3 unresolved · https://github.com/owner/repo/pull/42
+```
+
+**PR state icons:**
+
+| Icon | State |
+|------|-------|
+| 🟢 | Open |
+| 🟣 | Merged |
+| 🔴 | Closed |
+
+When there's no PR for the current branch, nothing is shown.
+
+## Requirements
+
+- [Pi](https://github.com/badlogic/pi) coding agent
+- [GitHub CLI](https://cli.github.com/) (`gh`) — authenticated with `gh auth login`
+
+## Install
+
+```bash
+pi install npm:pi-pr-status
+```
+
+Or try it without installing:
+
+```bash
+pi -e npm:pi-pr-status
+```
+
+You can also install from git:
+
+```bash
+pi install git:github.com/bruno-garcia/pi-pr-status
+```
+
+## How it works
+
+1. Detects the current git branch
+2. Runs `gh pr view` to find the associated pull request
+3. Parses CI status check results (pass / fail / pending)
+4. Queries unresolved review threads via the GitHub GraphQL API
+5. Displays everything in the pi footer status bar
+
+The extension polls every 30 seconds to pick up CI and review changes. Repo metadata is cached so only two API calls are made per poll (one for PR + checks, one GraphQL query for review threads). When no PR exists for the branch, no API calls are made after the initial check.
+
+## Development
+
+```bash
+npm install
+npm test
+```
+
+## License
+
+MIT

package/extensions/pr-status.ts

@@ -0,0 +1,66 @@
+/**
+ * PR Status Extension
+ *
+ * Shows the current branch's PR URL, CI check status, and unresolved
+ * review comment count in the pi footer status bar. Polls every 30 seconds.
+ */
+
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { getBranch, getRepoInfo, getPrForBranch, formatStatus, type PrInfo, type RepoInfo } from "../lib/github.ts";
+
+const POLL_INTERVAL = 30_000;
+const STATUS_KEY = "pr-status";
+
+export default function (pi: ExtensionAPI) {
+	let timer: ReturnType<typeof setInterval> | undefined;
+	let lastBranch: string | undefined;
+	let lastPr: PrInfo | undefined;
+	let cachedRepo: RepoInfo | undefined;
+
+	function update(cwd: string, ui: { setStatus: (key: string, value: string | undefined) => void }) {
+		const branch = getBranch(cwd);
+
+		if (branch !== lastBranch) {
+			lastBranch = branch;
+			lastPr = undefined;
+		}
+
+		if (!branch || branch === "HEAD") {
+			lastPr = undefined;
+			ui.setStatus(STATUS_KEY, undefined);
+			return;
+		}
+
+		if (!cachedRepo) {
+			cachedRepo = getRepoInfo(cwd);
+		}
+
+		const pr = getPrForBranch(cwd, cachedRepo);
+		lastPr = pr ?? undefined;
+
+		if (lastPr) {
+			ui.setStatus(STATUS_KEY, formatStatus(lastPr));
+		} else {
+			ui.setStatus(STATUS_KEY, undefined);
+		}
+	}
+
+	pi.on("session_start", async (_event, ctx) => {
+		update(ctx.cwd, ctx.ui);
+		timer = setInterval(() => update(ctx.cwd, ctx.ui), POLL_INTERVAL);
+	});
+
+	pi.on("session_switch", async (_event, ctx) => {
+		lastBranch = undefined;
+		lastPr = undefined;
+		cachedRepo = undefined;
+		update(ctx.cwd, ctx.ui);
+	});
+
+	pi.on("session_shutdown", async () => {
+		if (timer) {
+			clearInterval(timer);
+			timer = undefined;
+		}
+	});
+}

package/lib/github.ts

@@ -0,0 +1,159 @@
+/**
+ * Pure logic for querying GitHub PR status via the `gh` CLI.
+ * Separated from the pi extension API for testability.
+ */
+
+import { execSync } from "node:child_process";
+
+export interface CheckStatus {
+	total: number;
+	pass: number;
+	fail: number;
+	pending: number;
+}
+
+export interface PrInfo {
+	number: number;
+	title: string;
+	url: string;
+	state: string;
+	checks: CheckStatus;
+	unresolvedThreads: number;
+}
+
+export interface RepoInfo {
+	owner: string;
+	name: string;
+}
+
+export function getBranch(cwd: string): string | undefined {
+	try {
+		return execSync("git rev-parse --abbrev-ref HEAD", {
+			cwd,
+			encoding: "utf-8",
+			timeout: 3000,
+			stdio: ["pipe", "pipe", "pipe"],
+		}).trim();
+	} catch {
+		return undefined;
+	}
+}
+
+export function getRepoInfo(cwd: string): RepoInfo | undefined {
+	try {
+		const json = execSync("gh repo view --json owner,name", {
+			cwd,
+			encoding: "utf-8",
+			timeout: 5000,
+			stdio: ["pipe", "pipe", "pipe"],
+		}).trim();
+		const repo = JSON.parse(json);
+		return repo.owner?.login && repo.name ? { owner: repo.owner.login, name: repo.name } : undefined;
+	} catch {
+		return undefined;
+	}
+}
+
+export function parseChecks(statusCheckRollup: unknown[]): CheckStatus {
+	const checks: CheckStatus = { total: 0, pass: 0, fail: 0, pending: 0 };
+	for (const check of statusCheckRollup) {
+		const c = check as Record<string, string>;
+		checks.total++;
+		const conclusion = (c.conclusion || "").toUpperCase();
+		const status = (c.status || "").toUpperCase();
+		if (conclusion === "SUCCESS" || conclusion === "NEUTRAL" || conclusion === "SKIPPED") {
+			checks.pass++;
+		} else if (
+			conclusion === "FAILURE" ||
+			conclusion === "TIMED_OUT" ||
+			conclusion === "CANCELLED" ||
+			conclusion === "ACTION_REQUIRED"
+		) {
+			checks.fail++;
+		} else if (
+			status === "IN_PROGRESS" ||
+			status === "QUEUED" ||
+			status === "PENDING" ||
+			status === "WAITING" ||
+			!conclusion
+		) {
+			checks.pending++;
+		}
+	}
+	return checks;
+}
+
+export function countUnresolvedThreads(threads: { isResolved: boolean }[]): number {
+	return threads.filter((t) => !t.isResolved).length;
+}
+
+export function getPrForBranch(cwd: string, repo?: RepoInfo): PrInfo | undefined {
+	try {
+		const json = execSync("gh pr view --json number,title,url,state,statusCheckRollup", {
+			cwd,
+			encoding: "utf-8",
+			timeout: 10_000,
+			stdio: ["pipe", "pipe", "pipe"],
+		}).trim();
+		if (!json) return undefined;
+		const pr = JSON.parse(json);
+		if (!pr.number || !pr.url) return undefined;
+
+		const checks = Array.isArray(pr.statusCheckRollup) ? parseChecks(pr.statusCheckRollup) : { total: 0, pass: 0, fail: 0, pending: 0 };
+
+		let unresolvedThreads = 0;
+		if (repo) {
+			try {
+				const gql = execSync(
+					`gh api graphql -f query='{ repository(owner: "${repo.owner}", name: "${repo.name}") { pullRequest(number: ${pr.number}) { reviewThreads(first: 100) { nodes { isResolved } } } } }'`,
+					{
+						cwd,
+						encoding: "utf-8",
+						timeout: 10_000,
+						stdio: ["pipe", "pipe", "pipe"],
+					},
+				).trim();
+				const data = JSON.parse(gql);
+				const threads = data?.data?.repository?.pullRequest?.reviewThreads?.nodes;
+				if (Array.isArray(threads)) {
+					unresolvedThreads = countUnresolvedThreads(threads);
+				}
+			} catch {
+				// GraphQL failed — show PR without thread count
+			}
+		}
+
+		return {
+			number: pr.number,
+			title: pr.title,
+			url: pr.url,
+			state: pr.state,
+			checks,
+			unresolvedThreads,
+		};
+	} catch {
+		return undefined;
+	}
+}
+
+export function formatStatus(pr: PrInfo): string {
+	const stateIcon = pr.state === "MERGED" ? "🟣" : pr.state === "CLOSED" ? "🔴" : "🟢";
+	const parts: string[] = [`${stateIcon} PR #${pr.number}`];
+
+	if (pr.checks.total > 0) {
+		if (pr.checks.fail > 0) {
+			parts.push(`❌ ${pr.checks.fail}/${pr.checks.total} checks failed`);
+		} else if (pr.checks.pending > 0) {
+			parts.push(`⏳ ${pr.checks.pending}/${pr.checks.total} checks pending`);
+		} else {
+			parts.push(`✅ ${pr.checks.total} checks passed`);
+		}
+	}
+
+	if (pr.unresolvedThreads > 0) {
+		parts.push(`💬 ${pr.unresolvedThreads} unresolved`);
+	}
+
+	parts.push(pr.url);
+	return parts.join(" · ");
+}

package/package.json

@@ -0,0 +1,33 @@
+{
+  "name": "pi-pr-status",
+  "version": "0.1.0",
+  "description": "A Pi extension that shows the current PR link, CI check status, and unresolved review comments in the footer status bar",
+  "author": "Bruno Garcia",
+  "license": "MIT",
+  "homepage": "https://github.com/bruno-garcia/pi-pr-status#readme",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/bruno-garcia/pi-pr-status"
+  },
+  "keywords": ["pi-package", "github", "pull-request", "ci", "pi-extension"],
+  "files": [
+    "extensions/pr-status.ts",
+    "lib/github.ts",
+    "README.md",
+    "LICENSE"
+  ],
+  "peerDependencies": {
+    "@mariozechner/pi-coding-agent": "*"
+  },
+  "devDependencies": {
+    "@mariozechner/pi-coding-agent": "^0.52.12",
+    "vitest": "^3.0.0"
+  },
+  "pi": {
+    "extensions": ["./extensions"]
+  },
+  "scripts": {
+    "test": "vitest run",
+    "test:watch": "vitest"
+  }
+}