ai-cli-toolkit 0.2.0__py3-none-any.whl

ai_cli/update.py

@@ -0,0 +1,200 @@
+"""Tool install/update commands."""
+
+from __future__ import annotations
+
+import argparse
+import subprocess
+import sys
+from typing import Optional
+
+from ai_cli.config import ensure_config, get_tool_config
+from ai_cli.tools import load_registry
+
+
+def _run_shell(command: str) -> tuple[int, str]:
+    """Run a shell command and return (exit_code, combined_output)."""
+    result = subprocess.run(
+        ["bash", "-lc", command],
+        check=False,
+        capture_output=True,
+        text=True,
+    )
+    output = (result.stdout or "") + (result.stderr or "")
+    return result.returncode, output.strip()
+
+
+def available_targets() -> list[str]:
+    """Return updatable tool names sorted by registry order."""
+    return list(load_registry().keys())
+
+
+def _regenerate_completions() -> None:
+    """Regenerate shell completions to pick up newly installed tool flags."""
+    try:
+        from ai_cli.completion_gen import generate
+        print("\nRegenerating shell completions...")
+        generate(shell="all")
+    except Exception as exc:
+        print(f"Warning: could not regenerate completions: {exc}", file=sys.stderr)
+
+
+def update_tool(tool_name: str, dry_run: bool = False, method: Optional[str] = None,
+                regen_completions: bool = True) -> int:
+    """Install or update one tool using its ToolSpec install command."""
+    registry = load_registry()
+    spec = registry.get(tool_name)
+    if spec is None:
+        print(f"Unknown tool: {tool_name}", file=sys.stderr)
+        return 1
+
+    # Resolve method (explicit, auto-detected, or default)
+    effective_method = method
+    if not effective_method and spec.install_methods:
+        effective_method = spec.detect_best_method()
+
+    command = (spec.get_install_command(method) or "").strip()
+    if not command:
+        if method:
+            available = ", ".join(spec.install_methods.keys()) or "(none)"
+            print(
+                f"Unknown install method '{method}' for {tool_name}. "
+                f"Available: {available}",
+                file=sys.stderr,
+            )
+        else:
+            print(f"No install/update command configured for {tool_name}.", file=sys.stderr)
+        return 1
+
+    config = ensure_config()
+    tool_cfg = get_tool_config(config, tool_name)
+
+    installed_before = spec.detect_installed(tool_cfg.get("binary", ""))
+    version_before = spec.get_version(tool_cfg.get("binary", "")) if installed_before else None
+
+    status = "update" if installed_before else "install"
+    method_label = f" via {effective_method}" if effective_method else ""
+    print(f"{tool_name}: running {status}{method_label}")
+    print(f"  $ {command}")
+
+    if dry_run:
+        return 0
+
+    code, output = _run_shell(command)
+    if output:
+        print(output)
+    if code != 0:
+        print(f"{tool_name}: command failed with exit code {code}", file=sys.stderr)
+        return code
+
+    installed_after = spec.detect_installed(tool_cfg.get("binary", ""))
+    version_after = spec.get_version(tool_cfg.get("binary", "")) if installed_after else None
+
+    if installed_after:
+        before = version_before or "unknown"
+        after = version_after or "unknown"
+        print(f"{tool_name}: done ({before} -> {after})")
+        if regen_completions:
+            _regenerate_completions()
+        return 0
+
+    print(f"{tool_name}: command succeeded but binary still not found", file=sys.stderr)
+    return 1
+
+
+def update_many(tool_names: list[str], dry_run: bool = False, method: Optional[str] = None) -> int:
+    """Update multiple tools and return a combined exit status."""
+    if not tool_names:
+        print("No tools selected for update.", file=sys.stderr)
+        return 1
+
+    failed: list[str] = []
+    succeeded = 0
+    for name in tool_names:
+        print()
+        rc = update_tool(name, dry_run=dry_run, method=method, regen_completions=False)
+        if rc != 0:
+            failed.append(name)
+        else:
+            succeeded += 1
+
+    if succeeded and not dry_run:
+        _regenerate_completions()
+
+    if failed:
+        print(f"\nFailed: {', '.join(failed)}", file=sys.stderr)
+        return 1
+    return 0
+
+
+def main(argv: Optional[list[str]] = None) -> int:
+    """CLI entrypoint for `ai-cli update`."""
+    parser = argparse.ArgumentParser(
+        prog="ai-cli update",
+        description="Install or update wrapped CLI tools.",
+    )
+    parser.add_argument(
+        "tool",
+        nargs="?",
+        choices=available_targets(),
+        help="Tool to update.",
+    )
+    parser.add_argument(
+        "--all",
+        action="store_true",
+        help="Update all tools.",
+    )
+    parser.add_argument(
+        "--dry-run",
+        action="store_true",
+        help="Show commands without running them.",
+    )
+    parser.add_argument(
+        "--list",
+        action="store_true",
+        help="Show configured update commands.",
+    )
+    parser.add_argument(
+        "--method", "-m",
+        help="Install method to use (e.g. npm, brew, macports, curl). "
+             "Use --list-methods to see available methods per tool.",
+    )
+    parser.add_argument(
+        "--list-methods",
+        action="store_true",
+        help="Show available install methods per tool.",
+    )
+    args = parser.parse_args(argv)
+
+    registry = load_registry()
+
+    if args.list_methods:
+        for name, spec in registry.items():
+            best = spec.detect_best_method()
+            default = spec.install_command or "(none)"
+            print(f"{name}:")
+            print(f"  default: {default}")
+            for method, cmd in spec.install_methods.items():
+                marker = " <- auto-detected" if method == best else ""
+                print(f"  {method}: {cmd}{marker}")
+            print()
+        return 0
+
+    if args.list:
+        for name, spec in registry.items():
+            cmd = spec.install_command or "(none)"
+            print(f"{name:<8} {cmd}")
+        return 0
+
+    if args.all:
+        return update_many(list(registry.keys()), dry_run=args.dry_run, method=args.method)
+
+    target = args.tool
+    if not target:
+        print("Specify a tool or use --all.", file=sys.stderr)
+        return 1
+
+    return update_tool(target, dry_run=args.dry_run, method=args.method)
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

ai_cli_toolkit-0.2.0.dist-info/METADATA

@@ -0,0 +1,17 @@
+Metadata-Version: 2.4
+Name: ai-cli-toolkit
+Version: 0.2.0
+Summary: Unified AI CLI wrapper for Claude, Codex, Copilot, and Gemini
+Author-email: example-git <admin@xo.vg>
+Requires-Python: >=3.12
+License-File: LICENSE
+Requires-Dist: mitmproxy>=12.1.2
+Requires-Dist: shtab>=1.7
+Requires-Dist: bcrypt>=4.1
+Provides-Extra: dev
+Requires-Dist: mypy>=1.11; extra == "dev"
+Requires-Dist: pre-commit>=3.8; extra == "dev"
+Requires-Dist: pytest>=8.3; extra == "dev"
+Requires-Dist: pytest-cov>=5.0; extra == "dev"
+Requires-Dist: ruff>=0.6.9; extra == "dev"
+Dynamic: license-file

ai_cli_toolkit-0.2.0.dist-info/RECORD

@@ -0,0 +1,30 @@
+ai_cli/__init__.py,sha256=ykhVL4ma8vVE6Hup8HCC-OyZP42kEARImKWBas80qnE,100
+ai_cli/__main__.py,sha256=SDqipGzOAkPYh-3mARY3EVWLI8115tPkmfUJoVDmZNU,127
+ai_cli/ca.py,sha256=IW9Lb8aRfRilHWAHCreW36WVw7bYhjwmsIETiQqt_pM,5376
+ai_cli/completion_gen.py,sha256=y365D8bz1za13JkuEYIXRdGRw29nDJXBAbOop09Lc2k,19558
+ai_cli/config.py,sha256=BZ_KLYoBccnsmJDBlWPcdX2PLvpB_p5NWubMpgRjm1Y,5670
+ai_cli/credentials.py,sha256=dgCkO8w6v_7NO0GRA-JRXlJ1EKE6aFdJLkb4QnLvWxA,10751
+ai_cli/detached_cleanup.py,sha256=a2gGjIqNttXBqhVwMy6HU8SYpwZT3nMddABsb5pObQE,3751
+ai_cli/housekeeping.py,sha256=6XzejdCJVNLFm4mS4hG9U4TP96gUBv01hhLKiwxqiV4,1752
+ai_cli/instructions.py,sha256=oK8FG6QirtvKoKYGYDcebeqzzUx7LfkpfDg_WsfU2QU,10396
+ai_cli/log.py,sha256=3QD6Hjd_RNhFt8BNUSHWGNortncAXvsiWDFAoC5lnok,1654
+ai_cli/main.py,sha256=l5NLGmRZq-DWmVTFiIpCkHmhTjw13aETRu4iy-z7qA8,58708
+ai_cli/main_helpers.py,sha256=5LncZWe_yNRnpL1a3c2bUu3zGaZi_NALdNh3t43y0dQ,18634
+ai_cli/prompt_editor_launcher.py,sha256=i7nJpB-2-Bd7rN-fjCjWuGANbVuyp2wUSMdS1b6sLAk,10393
+ai_cli/proxy.py,sha256=SzwMXYJmBmsR2sY2SCVTAirmFncj_eobPruYhodSA0o,21397
+ai_cli/remote.py,sha256=5qBsZBGcwX8gKCBbuY8YlksrMiZ3UAtQRUCWErIWNM8,26860
+ai_cli/remote_package.py,sha256=28_lrExkMwaCC5yYkSAq8h88Ig88piYQsrKiZ1VdlZ8,41412
+ai_cli/session.py,sha256=yCGQJtSDFK9BbLvsTnC8NM7OMOXoQqEX-WP1jn0CE0o,46533
+ai_cli/session_store.py,sha256=AhUUvIUsQPaIemAwneTBWbhmWOyFUDJvcBSAKxpikmk,7262
+ai_cli/traffic.py,sha256=_LwK7bU_GfF0uklraMaIwCk8AuhHJmWrr2urdvKhn-4,51471
+ai_cli/traffic_db.py,sha256=6Lk3TsaK6TZP99V8cOV9W3f4x-WNuN6F3o_kAdwXdj4,3643
+ai_cli/tui.py,sha256=6Y7NFCHcl8-DGC19ayM1FpbbEO4-xnYBmk-fgXOtay8,17036
+ai_cli/update.py,sha256=DFiTXe950JCIlr-bDJ0KID6ma7cZ4bO41FLVTiKoZCs,6285
+ai_cli/bin/ai-mux-linux-x86_64,sha256=8k4xnWKvvqWUctcaxzv6qMECRJtkFTUhVnSnl0M4rvc,553416
+ai_cli/bin/remote-tty-wrapper,sha256=F5pUlgxJrj-1MKrNHTnIPyb5xLpfsbs6b8KgTjAeA9g,4532
+ai_cli_toolkit-0.2.0.dist-info/licenses/LICENSE,sha256=2dtru61yPdgtJsQHAI0JVBsXTcOy-hmvxMsHcQWAEeM,1068
+ai_cli_toolkit-0.2.0.dist-info/METADATA,sha256=nj2y5DS63qQudd4psytJfKWrhdq1tFW67pHXWMhogjQ,565
+ai_cli_toolkit-0.2.0.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
+ai_cli_toolkit-0.2.0.dist-info/entry_points.txt,sha256=XI2ZZPLLsG49qNeVt86Dfxrw8EJxI5yz1z9yKE22WUc,48
+ai_cli_toolkit-0.2.0.dist-info/top_level.txt,sha256=szfcz2SFm71Iu5jtJQtR3rEDLlOIOIaUPzWFUtAHuhA,7
+ai_cli_toolkit-0.2.0.dist-info/RECORD,,

ai_cli_toolkit-0.2.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

ai_cli_toolkit-0.2.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+ai-cli = ai_cli.main:main_cli

ai_cli_toolkit-0.2.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 example-git
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

ai_cli_toolkit-0.2.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+ai_cli