compartment 3.1__tar.gz → 3.2__tar.gz

{compartment-3.1 → compartment-3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: compartment
-Version: 3.1
+Version: 3.2
 Summary: Compartment - high-security, fully offline, encrypted vector memory for AI agents (MCP)
 Project-URL: Homepage, https://github.com/MaxFreedomPollard/Compartment
 Project-URL: Source, https://github.com/MaxFreedomPollard/Compartment

{compartment-3.1 → compartment-3.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "compartment"
-version = "3.1"
+version = "3.2"
 description = "Compartment - high-security, fully offline, encrypted vector memory for AI agents (MCP)"
 readme = "README.md"
 requires-python = ">=3.11"

{compartment-3.1 → compartment-3.2}/src/compartment/__init__.py

@@ -1,6 +1,6 @@
 """Compartment - high-security, fully offline, encrypted vector memory for AI agents."""
 
-__version__ = "3.1"
+__version__ = "3.2"
 
 from . import offline_guard as _og
 

{compartment-3.1 → compartment-3.2}/src/compartment/cli.py

@@ -249,6 +249,106 @@ def _start_status_bar_app(vault: str) -> None:
               "`compartment menubar` yourself")
 
 
+def _install_kind() -> tuple[str, list[str]]:
+    """How this copy was installed, and the command that upgrades it."""
+    exe = Path(sys.argv[0]).resolve()
+    if "uv/tools" in str(exe) or shutil.which("uv") and "uv/tools" in str(
+            Path(shutil.which("compartment") or "").resolve()):
+        return "uv", ["uv", "tool", "upgrade", "compartment"]
+    return "pip", [sys.executable, "-m", "pip", "install", "--upgrade",
+                   "compartment"]
+
+
+def cmd_update(args) -> None:
+    """Upgrade Compartment in place, then put the app back up."""
+    kind, cmd = _install_kind()
+    if args.source:
+        target = "git+https://github.com/MaxFreedomPollard/Compartment@main"
+        cmd = (["uv", "tool", "install", "--force", target] if kind == "uv"
+               else [sys.executable, "-m", "pip", "install", "--upgrade",
+                     "--force-reinstall", target])
+    print(f"Compartment {__version__} - updating ({kind} install)")
+    print("  " + " ".join(cmd))
+    try:
+        res = subprocess.run(cmd, timeout=1800)
+    except (OSError, subprocess.SubprocessError) as exc:
+        _die(f"update failed to run: {exc}")
+    if res.returncode != 0:
+        _die(f"update failed (exit {res.returncode})")
+    ver = "?"
+    try:
+        out = subprocess.run([shutil.which("compartment") or "compartment",
+                              "--version"], capture_output=True, text=True,
+                             timeout=120)
+        ver = (out.stdout or "").strip() or "?"
+    except (OSError, subprocess.SubprocessError):
+        pass
+    print(f"  updated: now {ver}")
+    print("  your vault and settings are untouched")
+    if not args.no_app and sys.platform in ("darwin", "win32"):
+        _restart_status_bar_app(args.vault)
+
+
+def _restart_status_bar_app(vault: str) -> None:
+    """Stop the running status bar app and start the new build."""
+    try:
+        app = _tray_app()
+        app.quit_running()
+    except Exception:                                    # noqa: BLE001
+        pass
+    _start_status_bar_app(vault)
+
+
+def cmd_uninstall(args) -> None:
+    """Remove Compartment from this machine. The vault is KEPT unless asked."""
+    print(f"Compartment {__version__} - uninstalling")
+    if sys.platform in ("darwin", "win32"):
+        try:
+            app = _tray_app()
+            try:
+                app.quit_running()
+                print("  status bar app stopped")
+            except Exception:                            # noqa: BLE001
+                pass
+            print(f"  start at login: {app.set_login(False)}")
+        except Exception as exc:                         # noqa: BLE001
+            print(f"  status bar app: nothing to remove ({exc})")
+    try:
+        if claude_hooks.is_installed():
+            claude_hooks.uninstall()
+            print("  Claude Code capture hook removed")
+    except Exception as exc:                             # noqa: BLE001
+        print(f"  capture hook: {exc}")
+    from .home import LEGACY_NAME, NAME
+    for name in (NAME, LEGACY_NAME):
+        try:
+            r = subprocess.run(["claude", "mcp", "remove", name],
+                               capture_output=True, text=True, timeout=120)
+            if r.returncode == 0:
+                print(f"  MCP registration removed: {name}")
+        except (OSError, subprocess.SubprocessError):
+            pass
+    if args.purge:
+        # only on an explicit --purge: this is the only copy of everything
+        # the agent has ever learned, and it is not recoverable.
+        for f in (args.vault, args.vault + ".config.json",
+                  args.vault + ".flock"):
+            try:
+                os.remove(f)
+                print(f"  deleted {f}")
+            except OSError:
+                pass
+        print("  vault deleted (--purge)")
+    else:
+        print(f"  vault KEPT at {args.vault}")
+        print("  delete it yourself, or re-run with --purge, once you are sure")
+    kind, _ = _install_kind()
+    final = ("uv tool uninstall compartment" if kind == "uv"
+             else f"{sys.executable} -m pip uninstall compartment")
+    print("\nOne step left, which cannot remove itself while it is running:")
+    print(f"  {final}")
+
+
 def _ask_yn(q: str) -> bool:
     if not sys.stdin.isatty():
         return False
@@ -1376,6 +1476,20 @@ def main(argv: list[str] | None = None) -> None:
                    help="only packs signed by the project are trusted by default. Name an author's 64 hex character public key to trust it deliberately; repeatable")
     p.set_defaults(fn=cmd_pack_export)
 
+    p = sub.add_parser("update", help="upgrade Compartment in place")
+    p.add_argument("--source", action="store_true",
+                   help="update from the GitHub main branch instead of PyPI")
+    p.add_argument("--no-app", action="store_true",
+                   help="do not restart the status bar app afterwards")
+    p.set_defaults(fn=cmd_update)
+
+    p = sub.add_parser("uninstall",
+                       help="remove Compartment from this machine "
+                            "(the vault is kept unless --purge)")
+    p.add_argument("--purge", action="store_true",
+                   help="ALSO delete the vault and its config, permanently")
+    p.set_defaults(fn=cmd_uninstall)
+
     p = sub.add_parser("integrate",
                        help="one-command wiring into claude, hermes, or openclaw")
     p.add_argument("target", choices=["claude", "hermes", "openclaw"])

{compartment-3.1 → compartment-3.2}/src/compartment/menubar.py

@@ -413,6 +413,28 @@ def set_login(enabled: bool) -> str:
     return login_status()
 
 
+def quit_running() -> bool:
+    """Stop a running status bar app, so an update or uninstall does not leave
+    the old build sitting in the menu bar with its binary already gone."""
+    import signal
+    killed = False
+    try:
+        out = subprocess.run(["pgrep", "-f", "compartment.*menubar"],
+                             capture_output=True, text=True, timeout=15).stdout
+    except (OSError, subprocess.SubprocessError):
+        return False
+    me = os.getpid()
+    for pid in [int(x) for x in out.split() if x.isdigit()]:
+        if pid == me:
+            continue
+        try:
+            os.kill(pid, signal.SIGTERM)
+            killed = True
+        except OSError:
+            pass
+    return killed
+
+
 # ----------------------------------------------------------------- self test
 
 def self_check(vault: str | None = None) -> int:

{compartment-3.1 → compartment-3.2}/src/compartment/systray.py

@@ -486,5 +486,17 @@ def run(vault: str | None = None, show: bool = False,
     return 0
 
 
-__all__ = ["run", "self_check", "login_status", "set_login", "panel_rows",
+def quit_running() -> bool:
+    """Stop a running tray app before an update or uninstall replaces it."""
+    import subprocess
+    try:
+        subprocess.run(["taskkill", "/F", "/IM", "compartment.exe"],
+                       capture_output=True, timeout=30)
+        return True
+    except (OSError, subprocess.SubprocessError):
+        return False
+
+
+__all__ = ["run", "self_check", "login_status", "set_login", "quit_running",
+           "panel_rows",
            "icon_path"]

{compartment-3.1 → compartment-3.2}/src/compartment.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: compartment
-Version: 3.1
+Version: 3.2
 Summary: Compartment - high-security, fully offline, encrypted vector memory for AI agents (MCP)
 Project-URL: Homepage, https://github.com/MaxFreedomPollard/Compartment
 Project-URL: Source, https://github.com/MaxFreedomPollard/Compartment