README.md

@@ -13,7 +13,3 @@ $ nixprstatus 345501
 ❌ nixos-24.05
 ```
 
-## TODO
-
-* Support backported commits
-* JSON output

nixprstatus/__main__.py

@@ -1,6 +1,5 @@
 import typer
 import requests
-from typing import Annotated
 from rich.console import Console
 
 app = typer.Typer()
@@ -12,25 +11,10 @@ DEFAULT_HEADERS = {
 BRANCHES = ["nixos-unstable-small", "nixos-unstable", "nixos-24.05"]
 
 
-def commit_in_branch(commit_sha: str, branch: str) -> bool:
-    url = f"https://api.github.com/repos/NixOS/nixpkgs/compare/{branch}...{commit_sha}"
-    commit_response = requests.get(url, headers=DEFAULT_HEADERS)
-    commit_response.raise_for_status()
-    status = commit_response.json().get("status")
-
-    if status in ["identical", "behind"]:
-        return True
-    return False
-
-
 @app.command()
-def pr(
-    pr: str,
-    branch: Annotated[str | None, typer.Option(help="Check specific branch")] = None,
-):
+def pr(pr: str):
     """Get status of pull request"""
     url = f"https://api.github.com/repos/NixOS/nixpkgs/pulls/{pr}"
-    typer.echo(f"branch {branch}")
 
     pr_response = requests.get(url, headers=DEFAULT_HEADERS)
     pr_response.raise_for_status()
@@ -39,24 +23,7 @@ def pr(
 
     console = Console()
 
-    merged = pr_data["merged"]
-
-    if branch and not merged:
-        console.print(f":x: {branch}")
-        return
-
-    commit_sha = pr_data.get("merge_commit_sha")
-
-    # Check only specified branch
-    if branch:
-        in_branch = commit_in_branch(commit_sha, branch)
-        if in_branch:
-            console.print(f":white_check_mark: {branch}", highlight=False)
-        else:
-            console.print(f":x: {branch}", highlight=False)
-        return
-
-    if not merged:
+    if pr_data["merged"] is False:
         console.print(":x: master", highlight=False)
         for branch in BRANCHES:
             console.print(f":x: {branch}", highlight=False)
@@ -64,12 +31,18 @@ def pr(
 
     console.print(":white_check_mark: master", highlight=False)
 
-    for b in BRANCHES:
-        in_branch = commit_in_branch(commit_sha, b)
-        if in_branch:
-            console.print(f":white_check_mark: {b}", highlight=False)
+    commit_sha = pr_data["merge_commit_sha"]
+
+    for branch in BRANCHES:
+        url = f"https://api.github.com/repos/NixOS/nixpkgs/compare/{branch}...{commit_sha}"
+        commit_response = requests.get(url, headers=DEFAULT_HEADERS)
+        commit_response.raise_for_status()
+        status = commit_response.json().get("status")
+
+        if status in ["identical", "behind"]:
+            console.print(f":white_check_mark: {branch}", highlight=False)
         else:
-            console.print(f":x: {b}", highlight=False)
+            console.print(f":x: {branch}", highlight=False)
 
 
 def main():

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nixprstatus"
-version = "0.1.1"
+version = "0.1.0"
 description = "Nixpkgs PR status checker"
 authors = ["Torjus Håkestad <torjus@usit.uio.no>"]
 license = "MIT"