gitwise-cli 0.24.2__py3-none-any.whl

gitwise/optimize.py

@@ -0,0 +1,212 @@
+"""Git repo optimizations: commit-graph, repack, prune."""
+
+import os
+import subprocess
+import time
+from pathlib import Path
+
+from .git import git_dir as get_git_dir
+from .git import require_root
+from .git import run as git_run
+from .git import version as git_version
+from .i18n import t
+from .output import (
+    confirm,
+    debug,
+    ok,
+    print_blank,
+    print_dim,
+    print_header,
+    print_json,
+    print_section,
+    print_status_line,
+    status,
+    warn,
+)
+from .utils.json_envelope import error_envelope, ok_envelope
+
+
+def _gc_is_running(cwd: Path) -> bool:
+    gd = get_git_dir(cwd)
+    if gd is None:
+        return False
+    pid_file = gd / "gc.pid"
+    if not pid_file.exists():
+        return False
+    try:
+        if (time.time() - pid_file.stat().st_mtime) > 86400:
+            return False
+        pid = int(pid_file.read_text().strip().splitlines()[0])
+        os.kill(pid, 0)
+        return True
+    except (ValueError, ProcessLookupError, PermissionError, OSError):
+        return False
+
+
+def _repo_size_kb(cwd: Path) -> int:
+    gd = get_git_dir(cwd)
+    if gd is None:
+        return 0
+    try:
+        r = subprocess.run(
+            ["du", "-sk", str(gd)], capture_output=True, text=True, check=True, timeout=120
+        )
+        return int(r.stdout.split()[0])
+    except (subprocess.SubprocessError, ValueError, IndexError, FileNotFoundError):
+        return sum(f.stat().st_size for f in gd.rglob("*") if f.is_file()) // 1024
+
+
+def _write_commit_graph(cwd: Path) -> bool:
+    args = ["commit-graph", "write", "--reachable"]
+    if git_version() >= (2, 31, 0):
+        args.append("--changed-paths")
+    r = git_run(args, cwd=cwd, check=False)
+    return r.returncode == 0
+
+
+def _repack(cwd: Path) -> bool:
+    r = git_run(["repack", "-A", "-d", "--write-bitmap-index"], cwd=cwd, check=False)
+    if r.returncode != 0:
+        debug(t("repack_fallo_bitmap"))
+        r = git_run(["repack", "-A", "-d"], cwd=cwd, check=False)
+    return r.returncode == 0
+
+
+def _prune(cwd: Path) -> bool:
+    return git_run(["prune"], cwd=cwd, check=False).returncode == 0
+
+
+def _get_steps() -> list[tuple[str, str]]:
+    return [
+        ("commit-graph", t("step_commit_graph")),
+        ("repack", t("step_repack")),
+        ("prune", t("step_prune")),
+    ]
+
+
+def run_optimize(*, dry_run: bool = False, yes: bool = False, as_json: bool = False) -> int:
+    root, err = require_root()
+    if err:
+        return err
+    if root is None:
+        return 1
+    cwd = root
+
+    if _gc_is_running(cwd):
+        warn(t("gc_already_running"))
+        return 1
+
+    steps = _get_steps()
+
+    if dry_run:
+        if as_json:
+            print_json(
+                ok_envelope(
+                    payload={
+                        "dry_run": True,
+                        "applied": False,
+                        "steps": [{"name": n, "desc": d} for n, d in steps],
+                    }
+                )
+            )
+            return 0
+        print_header(t("optimizing", root=str(cwd)))
+        print_blank()
+        for _name, desc in steps:
+            print_status_line("›", desc, t("pending"), ok_flag=False)
+        print_blank()
+        print_dim(t("dry_run_no_exec"))
+        return 0
+
+    if as_json and not yes:
+        print_json(
+            error_envelope(
+                error=t("yes_required_with_json"),
+                code="yes_required",
+                hint=t("yes_required_hint"),
+            )
+        )
+        return 2
+
+    if not as_json:
+        print_header(t("optimizing", root=str(cwd)))
+        print_blank()
+        for _name, desc in steps:
+            print_status_line("›", desc, t("pending"), ok_flag=False)
+        print_blank()
+        if not yes:
+            if not confirm(t("confirm_optimize")):
+                print_dim(t("cancelled"))
+                return 0
+            print_blank()
+
+    size_before = _repo_size_kb(cwd)
+
+    if as_json:
+        graph_ok = _write_commit_graph(cwd)
+        repack_ok = _repack(cwd)
+        prune_ok = _prune(cwd)
+    else:
+        with status(t("status_optimize_commit_graph")):
+            graph_ok = _write_commit_graph(cwd)
+        if graph_ok:
+            ok(t("commit_graph_updated"))
+        else:
+            warn(t("commit_graph_failed"))
+
+        with status(t("status_optimize_repack")):
+            repack_ok = _repack(cwd)
+        if repack_ok:
+            ok(t("repack_complete"))
+        else:
+            warn(t("repack_failed"))
+
+        with status(t("status_optimize_prune")):
+            prune_ok = _prune(cwd)
+        if prune_ok:
+            ok(t("prune_complete"))
+        else:
+            warn(t("prune_not_critical"))
+
+    size_after = _repo_size_kb(cwd)
+    saved = size_before - size_after
+
+    if not graph_ok and not repack_ok:
+        rc = 1
+    elif not prune_ok:
+        rc = 2
+    else:
+        rc = 0
+
+    if as_json:
+        payload: dict[str, object] = {
+            "dry_run": False,
+            "applied": True,
+            "steps": [
+                {"name": "commit-graph", "desc": t("step_commit_graph"), "ok": graph_ok},
+                {"name": "repack", "desc": t("step_repack"), "ok": repack_ok},
+                {"name": "prune", "desc": t("step_prune"), "ok": prune_ok},
+            ],
+            "size_before_kb": size_before,
+            "size_after_kb": size_after,
+            "saved_kb": saved,
+            "rc": rc,
+        }
+        if rc == 0:
+            print_json(ok_envelope(payload=payload))
+        else:
+            print_json(
+                error_envelope(
+                    error=t("optimize_partial_failure"),
+                    code="optimize_partial_failure",
+                    payload=payload,
+                )
+            )
+    else:
+        print_section(t("summary"))
+        if saved > 0:
+            ok(t("space_freed", saved=str(saved), before=str(size_before), after=str(size_after)))
+        else:
+            ok(t("repo_size", size=str(size_after)))
+
+    return rc