Function main

ci/clean-worktrees.py:267–335

(argv: list[str] | None = None)

def main(argv: list[str] | None = None) -> int:
    parser = argparse.ArgumentParser(
        description="Clean up stale agent worktrees under .claude/worktrees/",
    )
    parser.add_argument(
        "--dry-run",
        action="store_true",
        help="List worktrees and their safety status without removing anything.",
    )
    parser.add_argument(
        "--all",
        action="store_true",
        help="Remove every worktree, including unsafe ones (dangerous).",
    )
    args = parser.parse_args(argv)

    paths = list_worktree_paths()
    if not paths:
        print(f"No worktrees found under {WORKTREES_DIR}")
        return 0

    classifications: list[WorktreeStatus] = []
    for p in paths:
        classifications.append(classify_worktree(p))
    removed = 0
    skipped = 0
    failed = 0
    reclaimed = 0

    for cls in classifications:
        size = directory_size_bytes(cls.path)
        target = cls.safe or args.all
        prefix = "[dry-run] " if args.dry_run else ""
        action = "REMOVE" if target else "SKIP"
        print(
            f"{prefix}{action:6} {cls.path.name:40} "
            f"({format_bytes(size):>10})  {cls.reason}"
        )

        if not target:
            skipped += 1
            continue
        if args.dry_run:
            continue

        result = remove_worktree(cls.path)
        if result.removed:
            removed += 1
            reclaimed += size
            print(f"         removed via {result.method}")
        else:
            failed += 1
            print(f"         FAILED: {result.error}", file=sys.stderr)

    # Always prune admin entries at the end (cheap and harmless).
    if not args.dry_run:
        run_git(["worktree", "prune"], cwd=PROJECT_ROOT)