zai-cli 0.1.0__py3-none-any.whl

zai/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

zai/__main__.py

@@ -0,0 +1,4 @@
+from zai.main import app
+
+if __name__ == "__main__":
+    app()

zai/cli/__init__.py

@@ -0,0 +1 @@
+"""Typer command modules for the zai CLI."""

zai/cli/common.py

@@ -0,0 +1,16 @@
+import difflib
+
+
+def closest_command(value: str, choices, cutoff: float = 0.72) -> str | None:
+    """Return one unambiguous close command match."""
+    normalized = value.lower().strip()
+    candidates = sorted({str(choice).lower() for choice in choices})
+    matches = difflib.get_close_matches(normalized, candidates, n=2, cutoff=cutoff)
+    if not matches:
+        return None
+    if len(matches) > 1:
+        first = difflib.SequenceMatcher(None, normalized, matches[0]).ratio()
+        second = difflib.SequenceMatcher(None, normalized, matches[1]).ratio()
+        if first - second < 0.08:
+            return None
+    return matches[0]

zai/cli/integrations.py

@@ -0,0 +1,319 @@
+import os
+import subprocess
+import sys
+from pathlib import Path
+from typing import Optional
+
+import typer
+from rich.console import Console
+from rich.prompt import Confirm
+from rich.table import Table
+
+from ..core.hooks import VALID_EVENTS, add_hook, load_hooks, remove_hook
+from ..mcp.manager import (
+    KNOWN_SERVERS,
+    add_server,
+    build_server_command,
+    list_servers,
+    remove_server,
+)
+from ..plugins import loader as plugin_loader
+from ..plugins.loader import (
+    disable_plugin,
+    enable_plugin,
+    local_plugin_path,
+    revoke_plugin_trust,
+    scaffold_plugin,
+    trust_plugin,
+)
+
+console = Console()
+
+
+def hook(
+    action: str = typer.Argument("list", help="list, add, remove"),
+    event: Optional[str] = typer.Argument(None),
+    command: Optional[str] = typer.Argument(None),
+    hook_id: Optional[int] = typer.Option(None, "--id"),
+):
+    """Manage hooks: run commands on events."""
+    if action == "list":
+        hooks = load_hooks()
+        if not hooks:
+            console.print("[dim]No hooks configured.[/dim]")
+            return
+        table = Table(title="Hooks")
+        table.add_column("ID", style="dim")
+        table.add_column("Event", style="cyan")
+        table.add_column("Command")
+        for item in hooks:
+            table.add_row(str(item["id"]), item["event"], item["command"])
+        console.print(table)
+    elif action == "add" and event and command:
+        if event not in VALID_EVENTS:
+            console.print(f"[red]Invalid event. Choose: {', '.join(VALID_EVENTS)}[/red]")
+            return
+        created = add_hook(event, command)
+        console.print(f"[green]Hook added (id={created['id']})[/green]")
+    elif action == "remove" and hook_id:
+        if remove_hook(hook_id):
+            console.print(f"[green]Hook {hook_id} removed.[/green]")
+        else:
+            console.print(f"[red]Hook {hook_id} not found.[/red]")
+    else:
+        console.print(
+            "[red]Usage: zai hook list | zai hook add <event> <command> | "
+            "zai hook remove --id <n>[/red]"
+        )
+        console.print(f"Events: {', '.join(VALID_EVENTS)}")
+
+
+def mcp(
+    action: str = typer.Argument("list", help="list, add, remove, available"),
+    name: Optional[str] = typer.Argument(None),
+):
+    """Manage MCP servers (optional integrations)."""
+    if action == "available":
+        table = Table(title="Available MCP Servers")
+        table.add_column("Name", style="cyan")
+        table.add_column("Description")
+        table.add_column("Requires", style="dim")
+        for server_name, data in KNOWN_SERVERS.items():
+            table.add_row(
+                server_name,
+                data["description"],
+                data.get("env_key") or "nothing",
+            )
+        console.print(table)
+    elif action == "list":
+        servers = list_servers()
+        if not servers:
+            console.print("[dim]No MCP servers connected. Try: zai mcp available[/dim]")
+            return
+        for server in servers:
+            console.print(
+                f"[green]+[/green] [cyan]{server['name']}[/cyan] — "
+                f"{server['description']}"
+            )
+    elif action == "add" and name:
+        result = add_server(name)
+        if result is None:
+            console.print(f"[red]Unknown MCP server: {name}[/red]")
+            console.print("Run: zai mcp available")
+        elif result.get("already"):
+            console.print(f"[yellow]{name} already in config.[/yellow]")
+        else:
+            console.print(f"[green]MCP server '{name}' saved![/green]")
+            if result.get("env_key"):
+                console.print(
+                    f"[yellow]Set {result['env_key']} in ~/.zai/.env then restart zai[/yellow]"
+                )
+            else:
+                console.print("[dim]Restart zai to connect (needs: npx)[/dim]")
+    elif action == "connect" and name:
+        from ..mcp.client import connect
+
+        configured = next(
+            (server for server in list_servers() if server.get("name") == name),
+            None,
+        )
+        server_info = configured or (
+            {"name": name, **KNOWN_SERVERS[name]}
+            if name in KNOWN_SERVERS else None
+        )
+        if not server_info:
+            console.print(f"[red]Unknown MCP server: {name}[/red]")
+            return
+        try:
+            command, env = build_server_command(server_info, os.getcwd())
+        except ValueError as error:
+            console.print(f"[red]{error}[/red]")
+            return
+        console.print(f"[dim]Connecting to {name}...[/dim]")
+        if not connect(name, command, env):
+            console.print(
+                "[red]Failed. Check the server command, runtime, and credentials.[/red]"
+            )
+    elif action == "tools":
+        from ..mcp.client import get_all_tools
+
+        tools = get_all_tools()
+        if not tools:
+            console.print("[dim]No MCP servers connected. Try: zai mcp connect <name>[/dim]")
+        else:
+            for server, tool_list in tools.items():
+                console.print(f"\n[cyan]{server}[/cyan]")
+                for tool in tool_list:
+                    console.print(
+                        f"  [dim]{tool['name']}[/dim] — "
+                        f"{tool.get('description', '')[:60]}"
+                    )
+    elif action == "remove" and name:
+        if remove_server(name):
+            from ..mcp.client import disconnect
+
+            disconnect(name)
+            console.print(f"[green]MCP server '{name}' removed.[/green]")
+        else:
+            console.print(f"[red]{name} not found.[/red]")
+    else:
+        console.print(
+            "[red]Usage: zai mcp list | zai mcp available | "
+            "zai mcp add <name> | zai mcp connect <name> | "
+            "zai mcp tools | zai mcp remove <name>[/red]"
+        )
+
+
+def plugin(
+    action: str = typer.Argument(
+        "list",
+        help="list, new, install, remove, enable, disable, trust, untrust",
+    ),
+    name: Optional[str] = typer.Argument(None, help="Plugin name"),
+):
+    """Manage plugins: extend zai with custom tools, skills, and commands."""
+    if action == "list":
+        plugin_loader.load_all()
+    plugins = plugin_loader.get_loaded()
+    errors = plugin_loader.get_errors()
+
+    if action == "list":
+        if not plugins and not errors:
+            console.print("[dim]No plugins loaded.[/dim]")
+            console.print(
+                "[dim]Drop a .py file in ~/.zai/plugins/ or run: "
+                "zai plugin new <name>[/dim]"
+            )
+            return
+        if plugins:
+            table = Table(title="Loaded Plugins")
+            table.add_column("Name", style="cyan")
+            table.add_column("Description")
+            table.add_column("Tools", justify="right")
+            table.add_column("Skills", justify="right")
+            table.add_column("Commands", justify="right")
+            table.add_column("Version", style="dim")
+            for plugin_name, loaded_plugin in plugins.items():
+                table.add_row(
+                    plugin_name,
+                    loaded_plugin.description,
+                    str(len(loaded_plugin.get_tools())),
+                    str(len(loaded_plugin.get_skills())),
+                    str(len(loaded_plugin.get_commands())),
+                    loaded_plugin.version,
+                )
+            console.print(table)
+        if errors:
+            console.print("\n[red]Load errors:[/red]")
+            for filename, error in errors.items():
+                console.print(f"  [dim]{filename}[/dim]: [red]{error}[/red]")
+    elif action == "new":
+        if not name:
+            console.print("[red]Usage: zai plugin new <name>[/red]")
+            return
+        path = scaffold_plugin(name)
+        console.print(f"[green]Plugin scaffolded:[/green] {path}")
+        console.print("[dim]Edit the file, then restart zai to load it.[/dim]")
+    elif action == "install":
+        if not name:
+            console.print("[red]Usage: zai plugin install <name>[/red]")
+            return
+        package = name if name.startswith("zai-plugin-") else f"zai-plugin-{name}"
+        if not Confirm.ask(
+            f"Install external plugin package '{package}'?",
+            default=False,
+        ):
+            console.print("[yellow]Plugin installation cancelled.[/yellow]")
+            return
+        console.print(f"[cyan]Installing {package}...[/cyan]")
+        result = subprocess.run(
+            [sys.executable, "-m", "pip", "install", package],
+            capture_output=True,
+            text=True,
+        )
+        if result.returncode == 0:
+            console.print(
+                "[green]Installed.[/green] Review the package, then run: "
+                f"[cyan]zai plugin trust {name}[/cyan]"
+            )
+        else:
+            console.print(f"[red]Install failed:[/red]\n{result.stderr.strip()}")
+    elif action == "remove":
+        if not name:
+            console.print("[red]Usage: zai plugin remove <name>[/red]")
+            return
+        plugin_file = local_plugin_path(name)
+        if plugin_file and plugin_file.exists():
+            plugin_file.unlink()
+            revoke_plugin_trust(name)
+            console.print(f"[green]Removed:[/green] {plugin_file}")
+        elif plugin_file:
+            package = name if name.startswith("zai-plugin-") else f"zai-plugin-{name}"
+            result = subprocess.run(
+                [sys.executable, "-m", "pip", "uninstall", package, "-y"],
+                capture_output=True,
+                text=True,
+            )
+            if result.returncode == 0:
+                revoke_plugin_trust(name)
+                console.print(f"[green]Uninstalled {package}[/green]")
+            else:
+                console.print(
+                    f"[red]Plugin '{name}' not found "
+                    "(checked ~/.zai/plugins/ and pip)[/red]"
+                )
+        else:
+            console.print("[red]Invalid plugin name.[/red]")
+    elif action == "disable":
+        if not name:
+            console.print("[red]Usage: zai plugin disable <name>[/red]")
+            return
+        disable_plugin(name)
+        console.print(f"[yellow]Plugin '{name}' disabled.[/yellow]")
+    elif action == "enable":
+        if not name:
+            console.print("[red]Usage: zai plugin enable <name>[/red]")
+            return
+        if not Confirm.ask(
+            f"Enable plugin '{name}' with its declared permissions?",
+            default=False,
+        ):
+            console.print("[yellow]Plugin enable cancelled.[/yellow]")
+            return
+        if enable_plugin(name):
+            console.print(
+                f"[green]Plugin '{name}' enabled. Restart zai to load it.[/green]"
+            )
+        else:
+            console.print(f"[yellow]Plugin '{name}' was not disabled.[/yellow]")
+    elif action == "trust":
+        if not name:
+            console.print("[red]Usage: zai plugin trust <name>[/red]")
+            return
+        if trust_plugin(name):
+            console.print(
+                f"[green]Trusted current code for '{name}'.[/green] "
+                "Restart zai to load it."
+            )
+        else:
+            console.print(f"[red]Plugin '{name}' not found.[/red]")
+    elif action == "untrust":
+        if not name:
+            console.print("[red]Usage: zai plugin untrust <name>[/red]")
+            return
+        if revoke_plugin_trust(name):
+            console.print(f"[yellow]Trust revoked for '{name}'.[/yellow]")
+        else:
+            console.print(f"[yellow]Plugin '{name}' was not trusted.[/yellow]")
+    else:
+        console.print(
+            "[red]Usage: zai plugin list | new <name> | install <name> | "
+            "remove <name> | disable <name> | enable <name> | "
+            "trust <name> | untrust <name>[/red]"
+        )
+
+
+def register_integration_commands(app: typer.Typer) -> None:
+    app.command()(hook)
+    app.command()(mcp)
+    app.command("plugin")(plugin)