Add plain install command

Author: davegaeddert

plain/plain/cli/core.py

@@ -12,6 +12,7 @@
 from .docs import docs
 from .formatting import PlainContext
 from .help import help_cmd
+from .install import install
 from .preflight import preflight_checks
 from .registry import cli_registry
 from .scaffold import create
@@ -38,6 +39,7 @@ def plain_cli():
 plain_cli.add_command(setting)
 plain_cli.add_command(shell)
 plain_cli.add_command(run)
+plain_cli.add_command(install)
 plain_cli.add_command(upgrade)
 plain_cli.add_command(help_cmd)
 

plain/plain/cli/install.py

@@ -0,0 +1,69 @@
+import subprocess
+import sys
+
+import click
+
+from .agent import prompt_agent
+
+
+@click.command()
+@click.argument("packages", nargs=-1, required=True)
+@click.option(
+    "--agent-command",
+    envvar="PLAIN_AGENT_COMMAND",
+    help="Run command with generated prompt",
+)
+def install(packages: tuple[str, ...], agent_command: str | None = None) -> None:
+    """Install Plain packages with the help of an agent."""
+    # Validate all package names
+    invalid_packages = [pkg for pkg in packages if not pkg.startswith("plain")]
+    if invalid_packages:
+        raise click.UsageError(
+            f"The following packages do not start with 'plain': {', '.join(invalid_packages)}\n"
+            "This command is only for Plain framework packages."
+        )
+
+    # Install all packages first
+    if len(packages) == 1:
+        click.secho(f"Installing {packages[0]}...", bold=True, err=True)
+    else:
+        click.secho(f"Installing {len(packages)} packages...", bold=True, err=True)
+        for pkg in packages:
+            click.secho(f"  - {pkg}", err=True)
+        click.echo(err=True)
+
+    install_cmd = ["uv", "add"] + list(packages)
+    result = subprocess.run(install_cmd, check=False, stderr=sys.stderr)
+
+    if result.returncode != 0:
+        raise click.ClickException("Failed to install packages")
+
+    click.echo(err=True)
+    if len(packages) == 1:
+        click.secho(f"✓ {packages[0]} installed successfully", fg="green", err=True)
+    else:
+        click.secho(f"✓ {len(packages)} packages installed successfully", fg="green", err=True)
+    click.echo(err=True)
+
+    # Build the prompt for the agent to complete setup
+    lines = [
+        f"Complete the setup for the following Plain packages that were just installed: {', '.join(packages)}",
+        "",
+        "## Instructions",
+        "",
+        "For each package:",
+        "1. Run `uv run plain docs <package>` and follow any setup instructions",
+        "2. If the docs point out that it is a --dev tool, move it to dev dependencies: `uv remove <package> && uv add <package> --dev`",
+        "",
+        "DO NOT commit any changes",
+        "",
+        "Report back with:",
+        "- Whether the setup completed successfully",
+        "- Any manual steps that still need to be completed",
+        "- Any issues or errors encountered",
+    ]
+
+    prompt = "\n".join(lines)
+    success = prompt_agent(prompt, agent_command)
+    if not success:
+        raise click.Abort()

plain/plain/cli/upgrade.py

@@ -23,7 +23,7 @@
 def upgrade(
     packages: tuple[str, ...], diff: bool, agent_command: str | None = None
 ) -> None:
-    """Generate an upgrade prompt for plain packages."""
+    """Upgrade Plain packages with the help of an agent."""
     if not packages:
         click.secho("Getting installed packages...", bold=True, err=True)
         packages = tuple(sorted(get_installed_plain_packages()))