assertion-cli 0.1.0__py3-none-any.whl

git.py

@@ -0,0 +1,189 @@
+import subprocess
+from pathlib import Path
+from typing import NoReturn, Sequence
+
+import typer
+
+
+def exit_with_error(message: str) -> NoReturn:
+    typer.echo(message, err=True)
+    raise typer.Exit(code=1)
+
+
+def run_git_command(repo_root: Path, args: Sequence[str]) -> str:
+    completed = subprocess.run(
+        ["git", *args],
+        cwd=repo_root,
+        capture_output=True,
+        text=True,
+        check=False,
+    )
+    if completed.returncode != 0:
+        message = completed.stderr.strip() or completed.stdout.strip() or "git failed"
+        raise RuntimeError(message)
+    return completed.stdout.strip()
+
+
+def find_git_root(start_path: Path) -> Path:
+    completed = subprocess.run(
+        ["git", "rev-parse", "--show-toplevel"],
+        cwd=start_path,
+        capture_output=True,
+        text=True,
+        check=False,
+    )
+    if completed.returncode != 0:
+        exit_with_error("Current directory is not inside a git repository.")
+    return Path(completed.stdout.strip())
+
+
+def require_head_pushed(repo_root: Path) -> None:
+    try:
+        remote_refs = run_git_command(
+            repo_root, ["for-each-ref", "--format=%(refname:short)", "refs/remotes"]
+        )
+    except RuntimeError as exc:
+        exit_with_error(f"Failed to inspect remote refs: {exc}")
+
+    refs = [
+        ref for ref in remote_refs.splitlines() if ref and not ref.endswith("/HEAD")
+    ]
+    if not refs:
+        exit_with_error(
+            "Current HEAD commit is not present on any remote-tracking branch."
+        )
+
+    for ref in refs:
+        completed = subprocess.run(
+            ["git", "merge-base", "--is-ancestor", "HEAD", ref],
+            cwd=repo_root,
+            capture_output=True,
+            text=True,
+            check=False,
+        )
+        if completed.returncode == 0:
+            return
+        if completed.returncode != 1:
+            message = (
+                completed.stderr.strip() or completed.stdout.strip() or "git failed"
+            )
+            exit_with_error(f"Failed to verify remote commit state: {message}")
+
+    exit_with_error("Current HEAD commit is not present on any remote-tracking branch.")
+
+
+def get_head_sha(repo_root: Path) -> str:
+    try:
+        return run_git_command(repo_root, ["rev-parse", "HEAD"])
+    except RuntimeError as exc:
+        exit_with_error(f"Failed to get HEAD SHA: {exc}")
+
+
+def get_head_branch(repo_root: Path) -> str | None:
+    """Return the current branch name, or None if HEAD is detached."""
+    try:
+        name = run_git_command(repo_root, ["rev-parse", "--abbrev-ref", "HEAD"])
+    except RuntimeError:
+        return None
+    return name if name and name != "HEAD" else None
+
+
+def get_origin_github_repo(repo_root: Path) -> str | None:
+    """Return the current repo's GitHub `owner/name` from origin, or None if not parseable.
+
+    Accepts the common remote URL forms:
+      git@github.com:owner/name(.git)?
+      https://github.com/owner/name(.git)?
+      ssh://git@github.com/owner/name(.git)?
+    """
+    try:
+        url = run_git_command(repo_root, ["remote", "get-url", "origin"]).strip()
+    except RuntimeError:
+        return None
+    if not url:
+        return None
+
+    if url.startswith("git@github.com:"):
+        path = url[len("git@github.com:") :]
+    elif url.startswith("ssh://git@github.com/"):
+        path = url[len("ssh://git@github.com/") :]
+    elif url.startswith("https://github.com/"):
+        path = url[len("https://github.com/") :]
+    elif url.startswith("http://github.com/"):
+        path = url[len("http://github.com/") :]
+    else:
+        return None
+
+    path = path.rstrip("/")
+    if path.endswith(".git"):
+        path = path[: -len(".git")]
+    if path.count("/") != 1 or not all(path.split("/")):
+        return None
+    return path
+
+
+def _build_untracked_diff(repo_root: Path, rel_path: str) -> str:
+    completed = subprocess.run(
+        [
+            "git",
+            "diff",
+            "--no-index",
+            "--src-prefix=a/",
+            "--dst-prefix=b/",
+            "--",
+            "/dev/null",
+            rel_path,
+        ],
+        cwd=repo_root,
+        capture_output=True,
+        text=True,
+        check=False,
+    )
+    if completed.returncode not in (0, 1):
+        message = completed.stderr.strip() or completed.stdout.strip() or "git failed"
+        raise RuntimeError(message)
+    return completed.stdout.strip()
+
+
+# Paths the assertion-cli owns end-to-end. Excluded from the diff bundle so
+# reviewers don't flag our own bootstrap files as "unrelated changes" — they're
+# generated/refreshed by `asrt init` and `asrt checkpoint` and have nothing to
+# do with the customer's feature work. CLAUDE.md / AGENTS.md are intentionally
+# NOT in this list: those are customer-owned files we only patch a marked
+# block into, so the customer's other edits to them should still flow through.
+_ASSERTION_EXCLUDED_PATHSPECS = [
+    ":(exclude).assertion",
+    ":(exclude).claude/skills/assertion-cli",
+    ":(exclude).agents/skills/assertion-cli",
+]
+
+
+def get_uncommitted_diff(repo_root: Path) -> str:
+    try:
+        tracked = run_git_command(
+            repo_root, ["diff", "--", *_ASSERTION_EXCLUDED_PATHSPECS]
+        )
+        staged = run_git_command(
+            repo_root, ["diff", "--cached", "--", *_ASSERTION_EXCLUDED_PATHSPECS]
+        )
+
+        untracked_output = run_git_command(
+            repo_root,
+            [
+                "ls-files",
+                "--others",
+                "--exclude-standard",
+                "--",
+                *_ASSERTION_EXCLUDED_PATHSPECS,
+            ],
+        )
+        untracked_diffs = [
+            _build_untracked_diff(repo_root, rel_path)
+            for rel_path in untracked_output.splitlines()
+            if rel_path
+        ]
+
+        parts = [p for p in [tracked, staged, "\n".join(untracked_diffs)] if p]
+        return "\n".join(parts)
+    except RuntimeError as exc:
+        exit_with_error(f"Failed to collect git diff: {exc}")

link.py

@@ -0,0 +1,21 @@
+from pathlib import Path
+
+from git import exit_with_error
+
+LINK_FILE_NAME = "link"
+
+
+def save_link(assertion_dir: Path, url: str) -> None:
+    (assertion_dir / LINK_FILE_NAME).write_text(url + "\n", encoding="utf-8")
+
+
+def load_link(assertion_dir: Path) -> str:
+    link_path = assertion_dir / LINK_FILE_NAME
+    if not link_path.exists():
+        exit_with_error(
+            "No session link found. Run `asrt verify` first to generate a link."
+        )
+    content = link_path.read_text(encoding="utf-8").strip()
+    if not content:
+        exit_with_error("Session link file is empty. Run `asrt verify` to regenerate.")
+    return content