Add delete and update commands for project management

cobot/commands/delete.py

@@ -0,0 +1,160 @@
+from __future__ import annotations
+
+import argparse
+import os
+import shutil
+import subprocess
+from pathlib import Path
+from typing import Optional
+
+from textual.app import App
+
+from cobot.tui import SCREEN_CSS, LogScreen, PickScreen
+
+_PROJECT_DIR = Path(__file__).parent.parent.parent
+
+
+def _stop_docker_containers(write) -> None:
+    write("[cyan][*][/cyan] Stopping Docker containers...")
+    result = subprocess.run(
+        ["docker", "ps", "-a", "--filter", "name=lwc", "--format", "{{.Names}}"],
+        capture_output=True, text=True,
+    )
+    containers = [c for c in result.stdout.strip().splitlines() if c]
+    if not containers:
+        write("[dim]No project containers found.[/dim]")
+        return
+    for name in containers:
+        subprocess.run(["docker", "rm", "-f", name], capture_output=True)
+        write(f"[green][ok][/green] Removed container: {name}")
+
+
+def _remove_docker_images(write) -> None:
+    write("[cyan][*][/cyan] Removing Docker images...")
+    result = subprocess.run(
+        ["docker", "images", "--format", "{{.Repository}}:{{.Tag}}"],
+        capture_output=True, text=True,
+    )
+    project_images = [
+        img for img in result.stdout.strip().splitlines()
+        if "lwc" in img.lower()
+    ]
+    if not project_images:
+        write("[dim]No project images found.[/dim]")
+        return
+    for img in project_images:
+        subprocess.run(["docker", "rmi", "-f", img], capture_output=True)
+        write(f"[green][ok][/green] Removed image: {img}")
+
+
+def _remove_ros2(write) -> None:
+    write("[cyan][*][/cyan] Removing ROS2 Jazzy...")
+    if Path("/opt/ros/jazzy").exists():
+        subprocess.run(["sudo", "rm", "-rf", "/opt/ros/jazzy"])
+        write("[green][ok][/green] Removed /opt/ros/jazzy")
+    else:
+        write("[dim]ROS2 Jazzy not found, skipping.[/dim]")
+
+    source_line = "source /opt/ros/jazzy/setup.bash"
+    for rc_name in [".bashrc", ".zshrc"]:
+        rc = Path.home() / rc_name
+        if not rc.exists():
+            continue
+        content = rc.read_text()
+        if source_line not in content:
+            continue
+        new_content = content.replace(f"\n# ROS2 Jazzy\n{source_line}\n", "\n")
+        new_content = new_content.replace(source_line, "")
+        rc.write_text(new_content)
+        write(f"[green][ok][/green] Cleaned up ~/{rc_name}")
+
+
+def _uninstall_cobot(write) -> None:
+    write("[cyan][*][/cyan] Uninstalling cobot CLI...")
+    result = subprocess.run(
+        ["uv", "tool", "uninstall", "lightweight-cobot"],
+        capture_output=True, text=True,
+    )
+    if result.returncode == 0:
+        write("[green][ok][/green] cobot uninstalled")
+    else:
+        write(f"[yellow]Warning:[/yellow] {result.stderr.strip() or 'could not uninstall cobot'}")
+
+
+def _remove_project_dir(write) -> None:
+    write(f"[cyan][*][/cyan] Removing project directory...")
+    try:
+        shutil.rmtree(_PROJECT_DIR)
+        write(f"[green][ok][/green] Removed {_PROJECT_DIR}")
+    except Exception as exc:
+        write(f"[red]Failed:[/red] {exc}")
+        raise
+
+
+
+def _task_delete(screen: LogScreen, remove_ros: bool) -> None:
+    try:
+        _stop_docker_containers(screen.write)
+        _remove_docker_images(screen.write)
+
+        if remove_ros:
+            _remove_ros2(screen.write)
+
+        _uninstall_cobot(screen.write)
+        _remove_project_dir(screen.write)
+
+        screen.write("\n[green]Project fully removed.[/green]")
+        screen.finish(True)
+
+    except Exception as exc:
+        screen.write(f"\n[red]Error:[/red] {exc}")
+        screen.finish(False)
+
+
+
+class _DeleteApp(App[None]):
+    CSS = SCREEN_CSS
+
+    def on_mount(self) -> None:
+        self.push_screen(
+            PickScreen(
+                "Confirm deletion",
+                "This will permanently delete the project, Docker images and containers. Are you sure?",
+                ["No, cancel", "Yes, delete everything"],
+                "No, cancel",
+            ),
+            self._on_confirm,
+        )
+
+    def _on_confirm(self, choice: Optional[str]) -> None:
+        if choice is None or choice.startswith("No"):
+            self.exit()
+            return
+        self.push_screen(
+            PickScreen(
+                "ROS2 Jazzy",
+                "Also remove ROS2 Jazzy (/opt/ros/jazzy)?",
+                ["No, keep ROS2", "Yes, remove ROS2 Jazzy"],
+                "No, keep ROS2",
+            ),
+            self._on_ros_choice,
+        )
+
+    def _on_ros_choice(self, choice: Optional[str]) -> None:
+        remove_ros = choice is not None and choice.startswith("Yes")
+        self.push_screen(
+            LogScreen("Deleting project", lambda s: _task_delete(s, remove_ros)),
+            lambda _: self.exit(),
+        )
+
+
+def register(subparsers: argparse._SubParsersAction) -> None:
+    p = subparsers.add_parser(
+        "delete",
+        help="Remove the project, Docker images, containers, and optionally ROS2",
+    )
+    p.set_defaults(func=run)
+
+
+def run(args: argparse.Namespace) -> None:
+    _DeleteApp().run()