clonebox 1.1.21__py3-none-any.whl → 1.1.23__py3-none-any.whl

clonebox/validation/services.py

@@ -0,0 +1,125 @@
+from typing import Dict
+
+from rich.table import Table
+
+
+class ServiceValidationMixin:
+    # Services that should NOT be validated in VM (host-specific)
+    VM_EXCLUDED_SERVICES = {
+        "libvirtd",
+        "virtlogd",
+        "libvirt-guests",
+        "qemu-guest-agent",
+        "bluetooth",
+        "bluez",
+        "upower",
+        "thermald",
+        "tlp",
+        "power-profiles-daemon",
+        "gdm",
+        "gdm3",
+        "sddm",
+        "lightdm",
+        "snap.cups.cups-browsed",
+        "snap.cups.cupsd",
+        "ModemManager",
+        "wpa_supplicant",
+        "accounts-daemon",
+        "colord",
+        "switcheroo-control",
+    }
+
+    def validate_services(self) -> Dict:
+        """Validate services are enabled and running."""
+        setup_in_progress = self._setup_in_progress() is True
+        self.console.print("\n[bold]⚙️  Validating Services...[/]")
+
+        services = self.config.get("services", [])
+        if not services:
+            self.console.print("[dim]No services configured[/]")
+            return self.results["services"]
+
+        total_svcs = len(services)
+        self.console.print(f"[dim]Checking {total_svcs} services via QGA...[/]")
+
+        if "skipped" not in self.results["services"]:
+            self.results["services"]["skipped"] = 0
+
+        svc_table = Table(title="Service Validation", border_style="cyan")
+        svc_table.add_column("Service", style="bold")
+        svc_table.add_column("Enabled", justify="center")
+        svc_table.add_column("Running", justify="center")
+        svc_table.add_column("PID", justify="right", style="dim")
+        svc_table.add_column("Note", style="dim")
+
+        for idx, service in enumerate(services, 1):
+            if idx == 1 or idx % 25 == 0 or idx == total_svcs:
+                self.console.print(f"[dim]   ...services progress: {idx}/{total_svcs}[/]")
+            if service in self.VM_EXCLUDED_SERVICES:
+                svc_table.add_row(service, "[dim]—[/]", "[dim]—[/]", "[dim]—[/]", "host-only")
+                self.results["services"]["skipped"] += 1
+                self.results["services"]["details"].append(
+                    {
+                        "service": service,
+                        "enabled": None,
+                        "running": None,
+                        "skipped": True,
+                        "reason": "host-specific service",
+                    }
+                )
+                continue
+
+            self.results["services"]["total"] += 1
+
+            enabled_cmd = f"systemctl is-enabled {service} 2>/dev/null"
+            enabled_status = self._exec_in_vm(enabled_cmd)
+            is_enabled = enabled_status == "enabled"
+
+            running_cmd = f"systemctl is-active {service} 2>/dev/null"
+            running_status = self._exec_in_vm(running_cmd)
+            is_running = running_status == "active"
+
+            pid_value = ""
+            if is_running:
+                pid_out = self._exec_in_vm(f"systemctl show -p MainPID --value {service} 2>/dev/null")
+                if pid_out is None:
+                    pid_value = "?"
+                else:
+                    pid_value = pid_out.strip() or "?"
+            else:
+                pid_value = "—"
+
+            enabled_icon = (
+                "[green]✅[/]" if is_enabled else ("[yellow]⏳[/]" if setup_in_progress else "[yellow]⚠️[/]")
+            )
+            running_icon = (
+                "[green]✅[/]" if is_running else ("[yellow]⏳[/]" if setup_in_progress else "[red]❌[/]")
+            )
+
+            svc_table.add_row(service, enabled_icon, running_icon, pid_value, "")
+
+            if is_enabled and is_running:
+                self.results["services"]["passed"] += 1
+            elif setup_in_progress:
+                self.results["services"]["skipped"] += 1
+            else:
+                self.results["services"]["failed"] += 1
+
+            self.results["services"]["details"].append(
+                {
+                    "service": service,
+                    "enabled": is_enabled,
+                    "running": is_running,
+                    "pid": None if pid_value in ("", "—", "?") else pid_value,
+                    "skipped": False,
+                }
+            )
+
+        self.console.print(svc_table)
+        skipped = self.results["services"].get("skipped", 0)
+        msg = f"{self.results['services']['passed']}/{self.results['services']['total']} services active"
+        if skipped > 0:
+            msg += f" ({skipped} host-only skipped)"
+        self.console.print(f"[dim]{msg}[/]")
+
+        return self.results["services"]

clonebox/validation/smoke.py

@@ -0,0 +1,177 @@
+from typing import Dict, Optional
+
+from rich.table import Table
+
+
+class SmokeValidationMixin:
+    def validate_smoke_tests(self) -> Dict:
+        setup_in_progress = self._setup_in_progress() is True
+        packages = self.config.get("packages", [])
+        snap_packages = self.config.get("snap_packages", [])
+        copy_paths = self.config.get("copy_paths", None)
+        if not isinstance(copy_paths, dict) or not copy_paths:
+            copy_paths = self.config.get("app_data_paths", {})
+        vm_user = self.config.get("vm", {}).get("username", "ubuntu")
+
+        expected = []
+
+        if "firefox" in packages:
+            expected.append("firefox")
+
+        for snap_pkg in snap_packages:
+            if snap_pkg in {"pycharm-community", "chromium", "firefox", "code"}:
+                expected.append(snap_pkg)
+
+        for _, guest_path in copy_paths.items():
+            if guest_path == "/home/ubuntu/.config/google-chrome":
+                expected.append("google-chrome")
+                break
+
+        if "docker" in (self.config.get("services", []) or []) or "docker.io" in packages:
+            expected.append("docker")
+
+        expected = sorted(set(expected))
+        if not expected:
+            return self.results["smoke"]
+
+        def _installed(app: str) -> Optional[bool]:
+            if app in {"pycharm-community", "chromium", "firefox", "code"}:
+                out = self._exec_in_vm(f"snap list {app} >/dev/null 2>&1 && echo yes || echo no", timeout=10)
+                return None if out is None else out.strip() == "yes"
+
+            if app == "google-chrome":
+                out = self._exec_in_vm(
+                    "(command -v google-chrome >/dev/null 2>&1 || command -v google-chrome-stable >/dev/null 2>&1) && echo yes || echo no",
+                    timeout=10,
+                )
+                return None if out is None else out.strip() == "yes"
+
+            if app == "docker":
+                out = self._exec_in_vm("command -v docker >/dev/null 2>&1 && echo yes || echo no", timeout=10)
+                return None if out is None else out.strip() == "yes"
+
+            if app == "firefox":
+                out = self._exec_in_vm("command -v firefox >/dev/null 2>&1 && echo yes || echo no", timeout=10)
+                return None if out is None else out.strip() == "yes"
+
+            out = self._exec_in_vm(f"command -v {app} >/dev/null 2>&1 && echo yes || echo no", timeout=10)
+            return None if out is None else out.strip() == "yes"
+
+        def _run_test(app: str) -> Optional[bool]:
+            uid_out = self._exec_in_vm(f"id -u {vm_user} 2>/dev/null || true", timeout=10)
+            vm_uid = (uid_out or "").strip()
+            if not vm_uid.isdigit():
+                vm_uid = "1000"
+
+            runtime_dir = f"/run/user/{vm_uid}"
+            self._exec_in_vm(
+                f"mkdir -p {runtime_dir} && chown {vm_uid}:{vm_uid} {runtime_dir} && chmod 700 {runtime_dir}",
+                timeout=10,
+            )
+
+            user_env = (
+                f"sudo -u {vm_user} env HOME=/home/{vm_user} USER={vm_user} LOGNAME={vm_user} XDG_RUNTIME_DIR={runtime_dir}"
+            )
+
+            if app == "pycharm-community":
+                out = self._exec_in_vm(
+                    "/snap/pycharm-community/current/jbr/bin/java -version >/dev/null 2>&1 && echo yes || echo no",
+                    timeout=20,
+                )
+                return None if out is None else out.strip() == "yes"
+
+            if app == "chromium":
+                out = self._exec_in_vm(
+                    f"{user_env} timeout 20 chromium --headless=new --no-sandbox --disable-gpu --dump-dom about:blank >/dev/null 2>&1 && echo yes || echo no",
+                    timeout=30,
+                )
+                return None if out is None else out.strip() == "yes"
+
+            if app == "firefox":
+                out = self._exec_in_vm(
+                    f"{user_env} timeout 20 firefox --headless --version >/dev/null 2>&1 && echo yes || echo no",
+                    timeout=30,
+                )
+                return None if out is None else out.strip() == "yes"
+
+            if app == "google-chrome":
+                out = self._exec_in_vm(
+                    f"{user_env} timeout 20 google-chrome --headless=new --no-sandbox --disable-gpu --dump-dom about:blank >/dev/null 2>&1 && echo yes || echo no",
+                    timeout=30,
+                )
+                return None if out is None else out.strip() == "yes"
+
+            if app == "docker":
+                out = self._exec_in_vm("timeout 20 docker info >/dev/null 2>&1 && echo yes || echo no", timeout=30)
+                return None if out is None else out.strip() == "yes"
+
+            out = self._exec_in_vm(f"timeout 20 {app} --version >/dev/null 2>&1 && echo yes || echo no", timeout=30)
+            return None if out is None else out.strip() == "yes"
+
+        self.console.print("\n[bold]🧪 Smoke Tests (installed ≠ works)...[/]")
+        table = Table(title="Smoke Tests", border_style="cyan")
+        table.add_column("App", style="bold")
+        table.add_column("Installed", justify="center")
+        table.add_column("Launch", justify="center")
+        table.add_column("Note", style="dim")
+
+        for app in expected:
+            self.results["smoke"]["total"] += 1
+            installed = _installed(app)
+            launched: Optional[bool] = None
+            note = ""
+            pending = False
+
+            if installed is True:
+                launched = _run_test(app)
+                if launched is None:
+                    note = "test failed to execute"
+                elif launched is False and setup_in_progress:
+                    pending = True
+                    note = note or "setup in progress"
+            elif installed is False:
+                if setup_in_progress:
+                    pending = True
+                    note = "setup in progress"
+                else:
+                    note = "not installed"
+            else:
+                note = "install status unknown"
+
+            installed_icon = (
+                "[green]✅[/]"
+                if installed is True
+                else ("[yellow]⏳[/]" if pending else "[red]❌[/]")
+                if installed is False
+                else "[dim]?[/]"
+            )
+            launch_icon = (
+                "[green]✅[/]"
+                if launched is True
+                else ("[yellow]⏳[/]" if pending else "[red]❌[/]")
+                if launched is False
+                else ("[dim]—[/]" if installed is not True else "[dim]?[/]")
+            )
+
+            table.add_row(app, installed_icon, launch_icon, note)
+
+            passed = installed is True and launched is True
+            if pending:
+                self.results["smoke"]["skipped"] += 1
+            elif passed:
+                self.results["smoke"]["passed"] += 1
+            else:
+                self.results["smoke"]["failed"] += 1
+
+            self.results["smoke"]["details"].append(
+                {
+                    "app": app,
+                    "installed": installed,
+                    "launched": launched,
+                    "note": note,
+                    "pending": pending,
+                }
+            )
+
+        self.console.print(table)
+        return self.results["smoke"]

clonebox/validation/validator.py

@@ -0,0 +1,24 @@
+from clonebox.validation.apps import AppValidationMixin
+from clonebox.validation.core import VMValidatorCore
+from clonebox.validation.disk import DiskValidationMixin
+from clonebox.validation.mounts import MountValidationMixin
+from clonebox.validation.overall import OverallValidationMixin
+from clonebox.validation.packages import PackageValidationMixin
+from clonebox.validation.services import ServiceValidationMixin
+from clonebox.validation.smoke import SmokeValidationMixin
+
+
+class VMValidator(
+    VMValidatorCore,
+    MountValidationMixin,
+    PackageValidationMixin,
+    ServiceValidationMixin,
+    AppValidationMixin,
+    SmokeValidationMixin,
+    DiskValidationMixin,
+    OverallValidationMixin,
+):
+    pass
+
+
+__all__ = ["VMValidator"]