clonebox 0.1.6__tar.gz → 0.1.8__tar.gz

{clonebox-0.1.6 → clonebox-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: clonebox
-Version: 0.1.6
+Version: 0.1.8
 Summary: Clone your workstation environment to an isolated VM with selective apps, paths and services
 Author: CloneBox Team
 License: Apache-2.0
@@ -38,6 +38,7 @@ Requires-Dist: ruff>=0.1.0; extra == "dev"
 Dynamic: license-file
 
 # CloneBox 📦
+![img.png](img.png)
 
 ```commandline
 ╔═══════════════════════════════════════════════════════╗

{clonebox-0.1.6 → clonebox-0.1.8}/README.md

@@ -1,4 +1,5 @@
 # CloneBox 📦
+![img.png](img.png)
 
 ```commandline
 ╔═══════════════════════════════════════════════════════╗

{clonebox-0.1.6 → clonebox-0.1.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "clonebox"
-version = "0.1.6"
+version = "0.1.8"
 description = "Clone your workstation environment to an isolated VM with selective apps, paths and services"
 readme = "README.md"
 license = {text = "Apache-2.0"}

{clonebox-0.1.6 → clonebox-0.1.8}/src/clonebox/cli.py

@@ -592,6 +592,28 @@ def generate_clonebox_yaml(
             paths_mapping[host_path] = f"/mnt/workdir{idx}"
             idx += 1
 
+    # Add default user folders (Downloads, Documents)
+    home_dir = Path.home()
+    default_folders = [
+        (home_dir / "Downloads", "/home/ubuntu/Downloads"),
+        (home_dir / "Documents", "/home/ubuntu/Documents"),
+    ]
+    for host_folder, guest_folder in default_folders:
+        if host_folder.exists() and str(host_folder) not in paths_mapping:
+            paths_mapping[str(host_folder)] = guest_folder
+
+    # Detect and add app-specific data directories for running applications
+    # This includes browser profiles, IDE settings, credentials, extensions, etc.
+    app_data_dirs = detector.detect_app_data_dirs(snapshot.applications)
+    app_data_mapping = {}
+    for app_data in app_data_dirs:
+        host_path = app_data["path"]
+        if host_path not in paths_mapping:
+            # Map to same relative path in VM user home
+            rel_path = host_path.replace(str(home_dir), "").lstrip("/")
+            guest_path = f"/home/ubuntu/{rel_path}"
+            app_data_mapping[host_path] = guest_path
+
     # Determine VM name
     if not vm_name:
         if target_path:
@@ -604,10 +626,10 @@ def generate_clonebox_yaml(
     vcpus = max(2, sys_info["cpu_count"] // 2)
 
     # Auto-detect packages from running applications and services
-    suggested_app_packages = detector.suggest_packages_for_apps(snapshot.applications)
-    suggested_service_packages = detector.suggest_packages_for_services(snapshot.running_services)
+    app_packages = detector.suggest_packages_for_apps(snapshot.applications)
+    service_packages = detector.suggest_packages_for_services(snapshot.running_services)
     
-    # Combine with base packages
+    # Combine with base packages (apt only)
     base_packages = [
         "build-essential",
         "git",
@@ -615,10 +637,15 @@ def generate_clonebox_yaml(
         "vim",
     ]
     
-    # Merge all packages and deduplicate
-    all_packages = base_packages + suggested_app_packages + suggested_service_packages
+    # Merge apt packages and deduplicate
+    all_apt_packages = base_packages + app_packages["apt"] + service_packages["apt"]
+    if deduplicate:
+        all_apt_packages = deduplicate_list(all_apt_packages)
+    
+    # Merge snap packages and deduplicate
+    all_snap_packages = app_packages["snap"] + service_packages["snap"]
     if deduplicate:
-        all_packages = deduplicate_list(all_packages)
+        all_snap_packages = deduplicate_list(all_snap_packages)
 
     # Build config
     config = {
@@ -635,17 +662,24 @@ def generate_clonebox_yaml(
             "password": "${VM_PASSWORD}",
         },
         "services": services,
-        "packages": all_packages,
+        "packages": all_apt_packages,
+        "snap_packages": all_snap_packages,
+        "post_commands": [],  # User can add custom commands to run after setup
         "paths": paths_mapping,
+        "app_data_paths": app_data_mapping,  # App-specific config/data directories
         "detected": {
             "running_apps": [
-                {"name": a.name, "cwd": a.working_dir, "memory_mb": round(a.memory_mb)}
+                {"name": a.name, "cwd": a.working_dir or "", "memory_mb": round(a.memory_mb)}
                 for a in snapshot.applications[:10]
             ],
+            "app_data_dirs": [
+                {"path": d["path"], "app": d["app"], "size_mb": d["size_mb"]}
+                for d in app_data_dirs[:15]
+            ],
             "all_paths": {
-                "projects": paths_by_type["project"],
-                "configs": paths_by_type["config"][:5],
-                "data": paths_by_type["data"][:5],
+                "projects": list(paths_by_type["project"]),
+                "configs": list(paths_by_type["config"][:5]),
+                "data": list(paths_by_type["data"][:5]),
             },
         },
     }
@@ -764,15 +798,21 @@ def create_vm_from_config(
     replace: bool = False,
 ) -> str:
     """Create VM from YAML config dict."""
+    # Merge paths and app_data_paths
+    all_paths = config.get("paths", {}).copy()
+    all_paths.update(config.get("app_data_paths", {}))
+    
     vm_config = VMConfig(
         name=config["vm"]["name"],
         ram_mb=config["vm"].get("ram_mb", 4096),
         vcpus=config["vm"].get("vcpus", 4),
         gui=config["vm"].get("gui", True),
         base_image=config["vm"].get("base_image"),
-        paths=config.get("paths", {}),
+        paths=all_paths,
         packages=config.get("packages", []),
+        snap_packages=config.get("snap_packages", []),
         services=config.get("services", []),
+        post_commands=config.get("post_commands", []),
         user_session=user_session,
         network_mode=config["vm"].get("network_mode", "auto"),
         username=config["vm"].get("username", "ubuntu"),
@@ -896,16 +936,27 @@ def cmd_clone(args):
                 password = config['vm'].get('password', 'ubuntu')
                 console.print("\n[bold yellow]⏰ GUI Setup Process:[/]")
                 console.print("  [yellow]•[/] Installing desktop environment (~5-10 minutes)")
+                console.print("  [yellow]•[/] Running health checks on all components")
                 console.print("  [yellow]•[/] Automatic restart after installation")
                 console.print("  [yellow]•[/] GUI login screen will appear")
                 console.print(f"  [yellow]•[/] Login: [cyan]{username}[/] / [cyan]{'*' * len(password)}[/] (from .env)")
                 console.print("\n[dim]💡 Progress will be monitored automatically below[/]")
 
+            # Show health check info
+            console.print("\n[bold]📊 Health Check (inside VM):[/]")
+            console.print("  [cyan]cat /var/log/clonebox-health.log[/]  # View full report")
+            console.print("  [cyan]cat /var/log/clonebox-health-status[/]  # Quick status")
+            console.print("  [cyan]clonebox-health[/]  # Re-run health check")
+
             # Show mount instructions
-            if config.get("paths"):
-                console.print("\n[bold]Inside VM, mount paths with:[/]")
-                for idx, (host, guest) in enumerate(config["paths"].items()):
-                    console.print(f"  [cyan]sudo mount -t 9p -o trans=virtio mount{idx} {guest}[/]")
+            all_paths = config.get("paths", {}).copy()
+            all_paths.update(config.get("app_data_paths", {}))
+            if all_paths:
+                console.print("\n[bold]📁 Mounted paths (automatic):[/]")
+                for idx, (host, guest) in enumerate(list(all_paths.items())[:5]):
+                    console.print(f"  [dim]{host}[/] → [cyan]{guest}[/]")
+                if len(all_paths) > 5:
+                    console.print(f"  [dim]... and {len(all_paths) - 5} more paths[/]")
         except PermissionError as e:
             console.print(f"[red]❌ Permission Error:[/]\n{e}")
             console.print("\n[yellow]💡 Try running with --user flag:[/]")

{clonebox-0.1.6 → clonebox-0.1.8}/src/clonebox/cloner.py

@@ -31,7 +31,9 @@ class VMConfig:
     base_image: Optional[str] = None
     paths: dict = field(default_factory=dict)
     packages: list = field(default_factory=list)
+    snap_packages: list = field(default_factory=list)  # Snap packages to install
     services: list = field(default_factory=list)
+    post_commands: list = field(default_factory=list)  # Commands to run after setup
     user_session: bool = False  # Use qemu:///session instead of qemu:///system
     network_mode: str = "auto"  # auto|default|user
     username: str = "ubuntu"  # VM default username
@@ -474,6 +476,179 @@ class SelectiveVMCloner:
 
         return ET.tostring(root, encoding="unicode")
 
+    def _generate_health_check_script(self, config: VMConfig) -> str:
+        """Generate a health check script that validates all installed components."""
+        import base64
+        
+        # Build package check commands
+        apt_checks = []
+        for pkg in config.packages:
+            apt_checks.append(f'check_apt_package "{pkg}"')
+        
+        snap_checks = []
+        for pkg in config.snap_packages:
+            snap_checks.append(f'check_snap_package "{pkg}"')
+        
+        service_checks = []
+        for svc in config.services:
+            service_checks.append(f'check_service "{svc}"')
+        
+        mount_checks = []
+        for idx, (host_path, guest_path) in enumerate(config.paths.items()):
+            mount_checks.append(f'check_mount "{guest_path}" "mount{idx}"')
+        
+        apt_checks_str = "\n".join(apt_checks) if apt_checks else "echo 'No apt packages to check'"
+        snap_checks_str = "\n".join(snap_checks) if snap_checks else "echo 'No snap packages to check'"
+        service_checks_str = "\n".join(service_checks) if service_checks else "echo 'No services to check'"
+        mount_checks_str = "\n".join(mount_checks) if mount_checks else "echo 'No mounts to check'"
+        
+        script = f'''#!/bin/bash
+# CloneBox Health Check Script
+# Generated automatically - validates all installed components
+
+REPORT_FILE="/var/log/clonebox-health.log"
+PASSED=0
+FAILED=0
+WARNINGS=0
+
+# Colors for output
+RED='\\033[0;31m'
+GREEN='\\033[0;32m'
+YELLOW='\\033[1;33m'
+NC='\\033[0m'
+
+log() {{
+    echo "[$(date '+%Y-%m-%d %H:%M:%S')] $1" | tee -a "$REPORT_FILE"
+}}
+
+check_apt_package() {{
+    local pkg="$1"
+    if dpkg -l "$pkg" 2>/dev/null | grep -q "^ii"; then
+        log "[PASS] APT package '$pkg' is installed"
+        ((PASSED++))
+        return 0
+    else
+        log "[FAIL] APT package '$pkg' is NOT installed"
+        ((FAILED++))
+        return 1
+    fi
+}}
+
+check_snap_package() {{
+    local pkg="$1"
+    if snap list "$pkg" &>/dev/null; then
+        log "[PASS] Snap package '$pkg' is installed"
+        ((PASSED++))
+        return 0
+    else
+        log "[FAIL] Snap package '$pkg' is NOT installed"
+        ((FAILED++))
+        return 1
+    fi
+}}
+
+check_service() {{
+    local svc="$1"
+    if systemctl is-enabled "$svc" &>/dev/null; then
+        if systemctl is-active "$svc" &>/dev/null; then
+            log "[PASS] Service '$svc' is enabled and running"
+            ((PASSED++))
+            return 0
+        else
+            log "[WARN] Service '$svc' is enabled but not running"
+            ((WARNINGS++))
+            return 1
+        fi
+    else
+        log "[INFO] Service '$svc' is not enabled (may be optional)"
+        return 0
+    fi
+}}
+
+check_mount() {{
+    local path="$1"
+    local tag="$2"
+    if mountpoint -q "$path" 2>/dev/null; then
+        log "[PASS] Mount '$path' ($tag) is active"
+        ((PASSED++))
+        return 0
+    elif [ -d "$path" ]; then
+        log "[WARN] Directory '$path' exists but not mounted"
+        ((WARNINGS++))
+        return 1
+    else
+        log "[INFO] Mount point '$path' does not exist yet"
+        return 0
+    fi
+}}
+
+check_gui() {{
+    if systemctl get-default | grep -q graphical; then
+        log "[PASS] System configured for graphical target"
+        ((PASSED++))
+        if systemctl is-active gdm3 &>/dev/null || systemctl is-active gdm &>/dev/null; then
+            log "[PASS] Display manager (GDM) is running"
+            ((PASSED++))
+        else
+            log "[WARN] Display manager not yet running (may start after reboot)"
+            ((WARNINGS++))
+        fi
+    else
+        log "[INFO] System not configured for GUI"
+    fi
+}}
+
+# Start health check
+log "=========================================="
+log "CloneBox Health Check Report"
+log "VM Name: {config.name}"
+log "Date: $(date)"
+log "=========================================="
+
+log ""
+log "--- APT Packages ---"
+{apt_checks_str}
+
+log ""
+log "--- Snap Packages ---"
+{snap_checks_str}
+
+log ""
+log "--- Services ---"
+{service_checks_str}
+
+log ""
+log "--- Mounts ---"
+{mount_checks_str}
+
+log ""
+log "--- GUI Status ---"
+check_gui
+
+log ""
+log "=========================================="
+log "Health Check Summary"
+log "=========================================="
+log "Passed:   $PASSED"
+log "Failed:   $FAILED"
+log "Warnings: $WARNINGS"
+
+if [ $FAILED -eq 0 ]; then
+    log ""
+    log "[SUCCESS] All critical checks passed!"
+    echo "HEALTH_STATUS=OK" > /var/log/clonebox-health-status
+    exit 0
+else
+    log ""
+    log "[ERROR] Some checks failed. Review log for details."
+    echo "HEALTH_STATUS=FAILED" > /var/log/clonebox-health-status
+    exit 1
+fi
+'''
+        # Encode script to base64 for safe embedding in cloud-init
+        encoded = base64.b64encode(script.encode()).decode()
+        return encoded
+
     def _create_cloudinit_iso(self, vm_dir: Path, config: VMConfig) -> Path:
         """Create cloud-init ISO with user-data and meta-data."""
 
@@ -508,7 +683,7 @@ class SelectiveVMCloner:
             "\n".join(f"  - {pkg}" for pkg in all_packages) if all_packages else ""
         )
         
-        # Build runcmd - services and mounts
+        # Build runcmd - services, mounts, snaps, post_commands
         runcmd_lines = []
         
         # Add service enablement
@@ -519,6 +694,12 @@ class SelectiveVMCloner:
         for cmd in mount_commands:
             runcmd_lines.append(cmd)
         
+        # Install snap packages
+        if config.snap_packages:
+            runcmd_lines.append("  - echo 'Installing snap packages...'")
+            for snap_pkg in config.snap_packages:
+                runcmd_lines.append(f"  - snap install {snap_pkg} --classic || snap install {snap_pkg} || true")
+        
         # Add GUI setup if enabled - runs AFTER package installation completes
         if config.gui:
             runcmd_lines.extend([
@@ -526,6 +707,17 @@ class SelectiveVMCloner:
                 "  - systemctl enable gdm3 || systemctl enable gdm || true",
             ])
         
+        # Run user-defined post commands
+        if config.post_commands:
+            runcmd_lines.append("  - echo 'Running post-setup commands...'")
+            for cmd in config.post_commands:
+                runcmd_lines.append(f"  - {cmd}")
+        
+        # Generate health check script
+        health_script = self._generate_health_check_script(config)
+        runcmd_lines.append(f"  - echo '{health_script}' | base64 -d > /usr/local/bin/clonebox-health")
+        runcmd_lines.append("  - chmod +x /usr/local/bin/clonebox-health")
+        runcmd_lines.append("  - /usr/local/bin/clonebox-health >> /var/log/clonebox-health.log 2>&1")
         runcmd_lines.append("  - echo 'CloneBox VM ready!' > /var/log/clonebox-ready")
         
         # Add reboot command at the end if GUI is enabled

{clonebox-0.1.6 → clonebox-0.1.8}/src/clonebox/detector.py

@@ -144,59 +144,209 @@ class SystemDetector:
         "esbuild",
         "tmux",
         "screen",
+        # IDEs and desktop apps
+        "pycharm",
+        "idea",
+        "webstorm",
+        "phpstorm",
+        "goland",
+        "clion",
+        "rider",
+        "datagrip",
+        "sublime",
+        "atom",
+        "slack",
+        "discord",
+        "telegram",
+        "spotify",
+        "vlc",
+        "gimp",
+        "inkscape",
+        "blender",
+        "obs",
+        "postman",
+        "insomnia",
+        "dbeaver",
     ]
 
-    # Map process/service names to Ubuntu packages
+    # Map process/service names to Ubuntu packages or snap packages
+    # Format: "process_name": ("package_name", "install_type") where install_type is "apt" or "snap"
     APP_TO_PACKAGE_MAP = {
-        "python": "python3",
-        "python3": "python3",
-        "pip": "python3-pip",
-        "node": "nodejs",
-        "npm": "npm",
-        "yarn": "yarnpkg",
-        "docker": "docker.io",
-        "dockerd": "docker.io",
-        "docker-compose": "docker-compose",
-        "podman": "podman",
-        "nginx": "nginx",
-        "apache2": "apache2",
-        "httpd": "apache2",
-        "postgres": "postgresql",
-        "postgresql": "postgresql",
-        "mysql": "mysql-server",
-        "mysqld": "mysql-server",
-        "mongod": "mongodb",
-        "mongodb": "mongodb",
-        "redis-server": "redis-server",
-        "redis": "redis-server",
-        "vim": "vim",
-        "nvim": "neovim",
-        "emacs": "emacs",
-        "firefox": "firefox",
-        "chromium": "chromium-browser",
-        "jupyter": "jupyter-notebook",
-        "jupyter-lab": "jupyterlab",
-        "gunicorn": "gunicorn",
-        "uvicorn": "uvicorn",
-        "tmux": "tmux",
-        "screen": "screen",
-        "git": "git",
-        "curl": "curl",
-        "wget": "wget",
-        "ssh": "openssh-client",
-        "sshd": "openssh-server",
-        "go": "golang",
-        "cargo": "cargo",
-        "rustc": "rustc",
-        "java": "default-jdk",
-        "gradle": "gradle",
-        "mvn": "maven",
+        "python": ("python3", "apt"),
+        "python3": ("python3", "apt"),
+        "pip": ("python3-pip", "apt"),
+        "node": ("nodejs", "apt"),
+        "npm": ("npm", "apt"),
+        "yarn": ("yarnpkg", "apt"),
+        "docker": ("docker.io", "apt"),
+        "dockerd": ("docker.io", "apt"),
+        "docker-compose": ("docker-compose", "apt"),
+        "podman": ("podman", "apt"),
+        "nginx": ("nginx", "apt"),
+        "apache2": ("apache2", "apt"),
+        "httpd": ("apache2", "apt"),
+        "postgres": ("postgresql", "apt"),
+        "postgresql": ("postgresql", "apt"),
+        "mysql": ("mysql-server", "apt"),
+        "mysqld": ("mysql-server", "apt"),
+        "mongod": ("mongodb", "apt"),
+        "mongodb": ("mongodb", "apt"),
+        "redis-server": ("redis-server", "apt"),
+        "redis": ("redis-server", "apt"),
+        "vim": ("vim", "apt"),
+        "nvim": ("neovim", "apt"),
+        "emacs": ("emacs", "apt"),
+        "firefox": ("firefox", "apt"),
+        "chromium": ("chromium-browser", "apt"),
+        "jupyter": ("jupyter-notebook", "apt"),
+        "jupyter-lab": ("jupyterlab", "apt"),
+        "gunicorn": ("gunicorn", "apt"),
+        "uvicorn": ("uvicorn", "apt"),
+        "tmux": ("tmux", "apt"),
+        "screen": ("screen", "apt"),
+        "git": ("git", "apt"),
+        "curl": ("curl", "apt"),
+        "wget": ("wget", "apt"),
+        "ssh": ("openssh-client", "apt"),
+        "sshd": ("openssh-server", "apt"),
+        "go": ("golang", "apt"),
+        "cargo": ("cargo", "apt"),
+        "rustc": ("rustc", "apt"),
+        "java": ("default-jdk", "apt"),
+        "gradle": ("gradle", "apt"),
+        "mvn": ("maven", "apt"),
+        # Popular desktop apps (snap packages)
+        "chrome": ("chromium", "snap"),
+        "google-chrome": ("chromium", "snap"),
+        "pycharm": ("pycharm-community", "snap"),
+        "idea": ("intellij-idea-community", "snap"),
+        "code": ("code", "snap"),
+        "vscode": ("code", "snap"),
+        "slack": ("slack", "snap"),
+        "discord": ("discord", "snap"),
+        "spotify": ("spotify", "snap"),
+        "vlc": ("vlc", "apt"),
+        "gimp": ("gimp", "apt"),
+        "inkscape": ("inkscape", "apt"),
+        "blender": ("blender", "apt"),
+        "obs": ("obs-studio", "apt"),
+        "telegram": ("telegram-desktop", "snap"),
+        "postman": ("postman", "snap"),
+        "insomnia": ("insomnia", "snap"),
+        "dbeaver": ("dbeaver-ce", "snap"),
+        "sublime": ("sublime-text", "snap"),
+        "atom": ("atom", "snap"),
+    }
+
+    # Map applications to their config/data directories for complete cloning
+    # These directories contain user settings, extensions, profiles, credentials
+    APP_DATA_DIRS = {
+        # Browsers - profiles, extensions, bookmarks, passwords
+        "chrome": [".config/google-chrome", ".config/chromium"],
+        "chromium": [".config/chromium"],
+        "firefox": [".mozilla/firefox", ".cache/mozilla/firefox"],
+        
+        # IDEs and editors - settings, extensions, projects history
+        "code": [".config/Code", ".vscode", ".vscode-server"],
+        "vscode": [".config/Code", ".vscode", ".vscode-server"],
+        "pycharm": [".config/JetBrains", ".local/share/JetBrains", ".cache/JetBrains"],
+        "idea": [".config/JetBrains", ".local/share/JetBrains"],
+        "webstorm": [".config/JetBrains", ".local/share/JetBrains"],
+        "goland": [".config/JetBrains", ".local/share/JetBrains"],
+        "sublime": [".config/sublime-text", ".config/sublime-text-3"],
+        "atom": [".atom"],
+        "vim": [".vim", ".vimrc", ".config/nvim"],
+        "nvim": [".config/nvim", ".local/share/nvim"],
+        "emacs": [".emacs.d", ".emacs"],
+        "cursor": [".config/Cursor", ".cursor"],
+        
+        # Development tools
+        "docker": [".docker"],
+        "git": [".gitconfig", ".git-credentials", ".config/git"],
+        "npm": [".npm", ".npmrc"],
+        "yarn": [".yarn", ".yarnrc"],
+        "pip": [".pip", ".config/pip"],
+        "cargo": [".cargo"],
+        "rustup": [".rustup"],
+        "go": [".go", "go"],
+        "gradle": [".gradle"],
+        "maven": [".m2"],
+        
+        # Python environments
+        "python": [".pyenv", ".virtualenvs", ".local/share/virtualenvs"],
+        "python3": [".pyenv", ".virtualenvs", ".local/share/virtualenvs"],
+        "conda": [".conda", "anaconda3", "miniconda3"],
+        
+        # Node.js
+        "node": [".nvm", ".node", ".npm"],
+        
+        # Databases
+        "postgres": [".pgpass", ".psqlrc", ".psql_history"],
+        "mysql": [".my.cnf", ".mysql_history"],
+        "mongodb": [".mongorc.js", ".dbshell"],
+        "redis": [".rediscli_history"],
+        
+        # Communication apps
+        "slack": [".config/Slack"],
+        "discord": [".config/discord"],
+        "telegram": [".local/share/TelegramDesktop"],
+        "teams": [".config/Microsoft/Microsoft Teams"],
+        
+        # Other tools
+        "postman": [".config/Postman"],
+        "insomnia": [".config/Insomnia"],
+        "dbeaver": [".local/share/DBeaverData"],
+        "ssh": [".ssh"],
+        "gpg": [".gnupg"],
+        "aws": [".aws"],
+        "gcloud": [".config/gcloud"],
+        "kubectl": [".kube"],
+        "terraform": [".terraform.d"],
+        "ansible": [".ansible"],
+        
+        # General app data
+        "spotify": [".config/spotify"],
+        "vlc": [".config/vlc"],
+        "gimp": [".config/GIMP", ".gimp-2.10"],
+        "obs": [".config/obs-studio"],
     }
 
     def __init__(self):
         self.user = pwd.getpwuid(os.getuid()).pw_name
         self.home = Path.home()
 
+    def detect_app_data_dirs(self, applications: list) -> list:
+        """Detect config/data directories for running applications.
+        
+        Returns list of paths that contain user data needed by running apps.
+        """
+        app_data_paths = []
+        seen_paths = set()
+        
+        for app in applications:
+            app_name = app.name.lower()
+            
+            # Check each known app pattern
+            for pattern, dirs in self.APP_DATA_DIRS.items():
+                if pattern in app_name:
+                    for dir_name in dirs:
+                        full_path = self.home / dir_name
+                        if full_path.exists() and str(full_path) not in seen_paths:
+                            seen_paths.add(str(full_path))
+                            # Calculate size
+                            try:
+                                size = self._get_dir_size(full_path, max_depth=2)
+                            except:
+                                size = 0
+                            app_data_paths.append({
+                                "path": str(full_path),
+                                "app": app.name,
+                                "type": "app_data",
+                                "size_mb": round(size / 1024 / 1024, 1)
+                            })
+        
+        return app_data_paths
+
     def detect_all(self) -> SystemSnapshot:
         """Detect all services, applications and paths."""
         return SystemSnapshot(
@@ -437,29 +587,53 @@ class SystemDetector:
             pass
         return containers
 
-    def suggest_packages_for_apps(self, applications: list) -> list:
-        """Suggest Ubuntu packages based on detected applications."""
-        packages = set()
+    def suggest_packages_for_apps(self, applications: list) -> dict:
+        """Suggest packages based on detected applications.
+        
+        Returns:
+            dict with 'apt' and 'snap' keys containing lists of packages
+        """
+        apt_packages = set()
+        snap_packages = set()
+        
         for app in applications:
             app_name = app.name.lower()
-            # Check if app name matches any known mapping
-            for key, package in self.APP_TO_PACKAGE_MAP.items():
+            for key, (package, install_type) in self.APP_TO_PACKAGE_MAP.items():
                 if key in app_name:
-                    packages.add(package)
+                    if install_type == "snap":
+                        snap_packages.add(package)
+                    else:
+                        apt_packages.add(package)
                     break
-        return sorted(list(packages))
+        
+        return {
+            "apt": sorted(list(apt_packages)),
+            "snap": sorted(list(snap_packages))
+        }
 
-    def suggest_packages_for_services(self, services: list) -> list:
-        """Suggest Ubuntu packages based on detected services."""
-        packages = set()
+    def suggest_packages_for_services(self, services: list) -> dict:
+        """Suggest packages based on detected services.
+        
+        Returns:
+            dict with 'apt' and 'snap' keys containing lists of packages
+        """
+        apt_packages = set()
+        snap_packages = set()
+        
         for service in services:
             service_name = service.name.lower()
-            # Check if service name matches any known mapping
-            for key, package in self.APP_TO_PACKAGE_MAP.items():
+            for key, (package, install_type) in self.APP_TO_PACKAGE_MAP.items():
                 if key in service_name:
-                    packages.add(package)
+                    if install_type == "snap":
+                        snap_packages.add(package)
+                    else:
+                        apt_packages.add(package)
                     break
-        return sorted(list(packages))
+        
+        return {
+            "apt": sorted(list(apt_packages)),
+            "snap": sorted(list(snap_packages))
+        }
 
     def get_system_info(self) -> dict:
         """Get basic system information."""

{clonebox-0.1.6 → clonebox-0.1.8}/src/clonebox.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: clonebox
-Version: 0.1.6
+Version: 0.1.8
 Summary: Clone your workstation environment to an isolated VM with selective apps, paths and services
 Author: CloneBox Team
 License: Apache-2.0
@@ -38,6 +38,7 @@ Requires-Dist: ruff>=0.1.0; extra == "dev"
 Dynamic: license-file
 
 # CloneBox 📦
+![img.png](img.png)
 
 ```commandline
 ╔═══════════════════════════════════════════════════════╗

{clonebox-0.1.6 → clonebox-0.1.8}/tests/test_network.py

@@ -50,10 +50,16 @@ class TestNetworkMode:
     @patch("clonebox.cloner.libvirt")
     def test_resolve_network_mode_auto_user_no_default(self, mock_libvirt):
         """Test auto mode with user session and no default network falls back to user."""
-        import libvirt as real_libvirt
+        # Handle missing libvirt module in test environment
+        try:
+            import libvirt as real_libvirt
+            libvirt_error = real_libvirt.libvirtError
+        except ImportError:
+            class libvirt_error(Exception):
+                pass
 
         mock_conn = MagicMock()
-        mock_conn.networkLookupByName.side_effect = real_libvirt.libvirtError("No network")
+        mock_conn.networkLookupByName.side_effect = libvirt_error("No network")
         mock_libvirt.open.return_value = mock_conn
 
         cloner = SelectiveVMCloner(user_session=True)
@@ -126,10 +132,16 @@ class TestNetworkMode:
     @patch("clonebox.cloner.libvirt")
     def test_default_network_active_not_found(self, mock_libvirt):
         """Test _default_network_active returns False when network not found."""
-        import libvirt as real_libvirt
+        # Handle missing libvirt module in test environment
+        try:
+            import libvirt as real_libvirt
+            libvirt_error = real_libvirt.libvirtError
+        except ImportError:
+            class libvirt_error(Exception):
+                pass
 
         mock_conn = MagicMock()
-        mock_conn.networkLookupByName.side_effect = real_libvirt.libvirtError("Not found")
+        mock_conn.networkLookupByName.side_effect = libvirt_error("Not found")
         mock_libvirt.open.return_value = mock_conn
 
         cloner = SelectiveVMCloner()