request-vm-on-golem 0.1.24__py3-none-any.whl → 0.1.27__py3-none-any.whl

requestor/cli/commands.py

@@ -1,3 +1,4 @@
+"""CLI interface for VM on Golem."""
 import click
 import asyncio
 from typing import Optional
@@ -5,27 +6,30 @@ from pathlib import Path
 import subprocess
 import aiohttp
 from tabulate import tabulate
+import uvicorn
 
 from ..config import config
 from ..provider.client import ProviderClient
-from ..ssh.manager import SSHKeyManager
-from ..db.sqlite import Database
 from ..errors import RequestorError
 from ..utils.logging import setup_logger
-from ..utils.spinner import step
+from ..utils.spinner import step, Spinner
+from ..services.vm_service import VMService
+from ..services.provider_service import ProviderService
+from ..services.ssh_service import SSHService
+from ..services.database_service import DatabaseService
 
 # Initialize logger
 logger = setup_logger('golem.requestor')
 
-# Initialize components
-db = Database(config.db_path)
+# Initialize services
+db_service = DatabaseService(config.db_path)
 
 
 def async_command(f):
     """Decorator to run async commands."""
     async def wrapper(*args, **kwargs):
         # Initialize database
-        await db.init()
+        await db_service.init()
         return await f(*args, **kwargs)
     return lambda *args, **kwargs: asyncio.run(wrapper(*args, **kwargs))
 
@@ -47,8 +51,9 @@ def vm():
 @click.option('--memory', type=int, help='Minimum memory (GB) required')
 @click.option('--storage', type=int, help='Minimum storage (GB) required')
 @click.option('--country', help='Preferred provider country')
+@click.option('--driver', type=click.Choice(['central', 'golem-base']), default=None, help='Discovery driver to use')
 @async_command
-async def list_providers(cpu: Optional[int], memory: Optional[int], storage: Optional[int], country: Optional[str]):
+async def list_providers(cpu: Optional[int], memory: Optional[int], storage: Optional[int], country: Optional[str], driver: Optional[str]):
     """List available providers matching requirements."""
     try:
         # Log search criteria if any
@@ -65,66 +70,29 @@ async def list_providers(cpu: Optional[int], memory: Optional[int], storage: Opt
         
         logger.process("Querying discovery service")
         
-        params = {
-            k: v for k, v in {
-                'cpu': cpu,
-                'memory': memory,
-                'storage': storage,
-                'country': country
-            }.items() if v is not None
-        }
-
-        async with aiohttp.ClientSession() as session:
-            async with session.get(
-                f"{config.discovery_url}/api/v1/advertisements",
-                params=params
-            ) as response:
-                if not response.ok:
-                    raise RequestorError("Failed to query discovery service")
-                providers = await response.json()
+        # Initialize provider service
+        provider_service = ProviderService()
+        async with provider_service:
+            providers = await provider_service.find_providers(
+                cpu=cpu,
+                memory=memory,
+                storage=storage,
+                country=country,
+                driver=driver
+            )
 
         if not providers:
             logger.warning("No providers found matching criteria")
             return
 
-        # Format provider information
-        headers = ["Provider ID", "IP Address", "Country",
-                   "CPU", "Memory (GB)", "Storage (GB)", "Updated"]
-        rows = []
-        for p in providers:
-            # Get provider IP based on environment
-            provider_ip = 'localhost' if config.environment == "development" else p.get(
-                'ip_address')
-            if not provider_ip and config.environment == "production":
-                logger.warning(f"Provider {p['provider_id']} has no IP address")
-                provider_ip = 'N/A'
-            rows.append([
-                p['provider_id'],
-                provider_ip,
-                p['country'],
-                p['resources']['cpu'],
-                p['resources']['memory'],
-                p['resources']['storage'],
-                p['updated_at']
-            ])
+        # Format provider information using service with colors
+        headers = provider_service.provider_headers
+        rows = await asyncio.gather(*(provider_service.format_provider_row(p, colorize=True) for p in providers))
 
         # Show fancy header
-        click.echo("\n" + "─" * 60)
+        click.echo("\n" + "─" * 80)
         click.echo(click.style(f"  🌍 Available Providers ({len(providers)} total)", fg="blue", bold=True))
-        click.echo("─" * 60)
-
-        # Add color and formatting to columns
-        for row in rows:
-            # Format Provider ID to be more readable
-            row[0] = click.style(row[0], fg="yellow")  # Provider ID
-            
-            # Format resources with icons and colors
-            row[3] = click.style(f"💻 {row[3]}", fg="cyan", bold=True)  # CPU
-            row[4] = click.style(f"🧠 {row[4]}", fg="cyan", bold=True)  # Memory
-            row[5] = click.style(f"💾 {row[5]}", fg="cyan", bold=True)  # Storage
-            
-            # Format location info
-            row[2] = click.style(f"🌍 {row[2]}", fg="green", bold=True)  # Country
+        click.echo("─" * 80)
 
         # Show table with colored headers
         click.echo("\n" + tabulate(
@@ -132,67 +100,13 @@ async def list_providers(cpu: Optional[int], memory: Optional[int], storage: Opt
             headers=[click.style(h, bold=True) for h in headers],
             tablefmt="grid"
         ))
-        click.echo("\n" + "─" * 60)
+        click.echo("\n" + "─" * 80)
 
     except Exception as e:
         logger.error(f"Failed to list providers: {str(e)}")
         raise click.Abort()
 
 
-# Helper functions for VM creation with spinners
-@step("Checking VM name availability")
-async def check_vm_name(db, name):
-    existing_vm = await db.get_vm(name)
-    if existing_vm:
-        raise RequestorError(f"VM with name '{name}' already exists")
-
-@step("Locating provider")
-async def find_provider(discovery_url, provider_id):
-    async with aiohttp.ClientSession() as session:
-        async with session.get(f"{discovery_url}/api/v1/advertisements") as response:
-            if not response.ok:
-                raise RequestorError("Failed to query discovery service")
-            providers = await response.json()
-    
-    provider = next((p for p in providers if p['provider_id'] == provider_id), None)
-    if not provider:
-        raise RequestorError(f"Provider {provider_id} not found")
-    return provider
-
-@step("Verifying resource availability")
-async def verify_resources(provider, cpu, memory, storage):
-    if (cpu > provider['resources']['cpu'] or
-        memory > provider['resources']['memory'] or
-            storage > provider['resources']['storage']):
-        raise RequestorError("Requested resources exceed provider capacity")
-
-@step("Setting up SSH access")
-async def setup_ssh(ssh_manager):
-    return await ssh_manager.get_key_pair()
-
-@step("Deploying VM on provider")
-async def deploy_vm(client, name, cpu, memory, storage, ssh_key):
-    return await client.create_vm(
-        name=name,
-        cpu=cpu,
-        memory=memory,
-        storage=storage,
-        ssh_key=ssh_key
-    )
-
-@step("Configuring VM access")
-async def get_vm_access(client, vm_id):
-    return await client.get_vm_access(vm_id)
-
-@step("Saving VM configuration")
-async def save_vm_config(db, name, provider_ip, vm_id, config):
-    await db.save_vm(
-        name=name,
-        provider_ip=provider_ip,
-        vm_id=vm_id,
-        config=config
-    )
-
 @vm.command(name='create')
 @click.argument('name')
 @click.option('--provider-id', required=True, help='Provider ID to use')
@@ -203,45 +117,50 @@ async def save_vm_config(db, name, provider_ip, vm_id, config):
 async def create_vm(name: str, provider_id: str, cpu: int, memory: int, storage: int):
     """Create a new VM on a specific provider."""
     try:
-        logger.command(f"🚀 Creating VM '{name}'")
-        logger.detail(f"Provider: {provider_id}")
-        logger.detail(f"Resources: {cpu} CPU, {memory}GB RAM, {storage}GB Storage")
-
-        # Check VM name
-        await check_vm_name(db, name)
-
-        # Find and verify provider
-        provider = await find_provider(config.discovery_url, provider_id)
-        await verify_resources(provider, cpu, memory, storage)
-
-        # Setup SSH
-        ssh_manager = SSHKeyManager(config.ssh_key_dir)
-        key_pair = await setup_ssh(ssh_manager)
+        # Show configuration details
+        click.echo("\n" + "─" * 60)
+        click.echo(click.style("  VM Configuration", fg="blue", bold=True))
+        click.echo("─" * 60)
+        click.echo(f"  Provider   : {click.style(provider_id, fg='cyan')}")
+        click.echo(f"  Resources  : {click.style(f'{cpu} CPU, {memory}GB RAM, {storage}GB Storage', fg='cyan')}")
+        click.echo("─" * 60 + "\n")
 
-        # Get provider IP
-        provider_ip = 'localhost' if config.environment == "development" else provider.get('ip_address')
-        if not provider_ip and config.environment == "production":
-            raise RequestorError("Provider IP address not found in advertisement")
-
-        # Create and configure VM
-        provider_url = config.get_provider_url(provider_ip)
-        async with ProviderClient(provider_url) as client:
-            # Create VM and get full VM ID
-            vm = await deploy_vm(client, name, cpu, memory, storage, key_pair.public_key_content)
-            access_info = await get_vm_access(client, vm['id'])
-            
-            # Store the full VM ID from access info
-            full_vm_id = access_info['vm_id']
-            
-            await save_vm_config(
-                db, name, provider_ip, full_vm_id,
-                {
-                    'cpu': cpu,
-                    'memory': memory,
-                    'storage': storage,
-                    'ssh_port': access_info['ssh_port']
-                }
-            )
+        # Now start the deployment with spinner
+        with Spinner("Deploying VM..."):
+            # Initialize services
+            provider_service = ProviderService()
+            async with provider_service:
+                # Verify provider and resources
+                provider = await provider_service.verify_provider(provider_id)
+                if not await provider_service.check_resource_availability(provider_id, cpu, memory, storage):
+                    raise RequestorError("Provider doesn't have enough resources available")
+
+                # Get provider IP
+                provider_ip = 'localhost' if config.environment == "development" else provider.get('ip_address')
+                if not provider_ip and config.environment == "production":
+                    raise RequestorError("Provider IP address not found in advertisement")
+
+                # Setup SSH
+                ssh_service = SSHService(config.ssh_key_dir)
+                key_pair = await ssh_service.get_key_pair()
+
+                # Initialize VM service
+                provider_url = config.get_provider_url(provider_ip)
+                async with ProviderClient(provider_url) as client:
+                    vm_service = VMService(db_service, ssh_service, client)
+                    
+                    # Create VM
+                    vm = await vm_service.create_vm(
+                        name=name,
+                        cpu=cpu,
+                        memory=memory,
+                        storage=storage,
+                        provider_ip=provider_ip,
+                        ssh_key=key_pair.public_key_content
+                    )
+
+                    # Get access info from config
+                    ssh_port = vm['config']['ssh_port']
 
         # Create a visually appealing success message
         click.echo("\n" + "─" * 60)
@@ -259,13 +178,18 @@ async def create_vm(name: str, provider_id: str, cpu: int, memory: int, storage:
         click.echo("\n" + click.style("  Connection Details", fg="blue", bold=True))
         click.echo("  " + "┈" * 25)
         click.echo(f"  🌐 IP Address : {click.style(provider_ip, fg='cyan')}")
-        click.echo(f"  🔌 Port       : {click.style(str(access_info['ssh_port']), fg='cyan')}")
+        click.echo(f"  🔌 Port       : {click.style(str(ssh_port), fg='cyan')}")
         
         # Quick Connect Section
         click.echo("\n" + click.style("  Quick Connect", fg="blue", bold=True))
         click.echo("  " + "┈" * 25)
-        ssh_command = f"ssh -i {key_pair.private_key.absolute()} -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -p {access_info['ssh_port']} ubuntu@{provider_ip}"
-        click.echo(f"  🔑 SSH Command : {click.style(ssh_command, fg='yellow')}")
+        ssh_command = ssh_service.format_ssh_command(
+            host=provider_ip,
+            port=ssh_port,
+            private_key_path=key_pair.private_key.absolute(),
+            colorize=True
+        )
+        click.echo(f"  🔑 SSH Command : {ssh_command}")
         
         click.echo("\n" + "─" * 60)
 
@@ -289,34 +213,34 @@ async def ssh_vm(name: str):
     try:
         logger.command(f"🔌 Connecting to VM '{name}'")
         
-        # Get VM details
+        # Initialize services
+        ssh_service = SSHService(config.ssh_key_dir)
+        
+        # Get VM details using database service
         logger.process("Retrieving VM details")
-        vm = await db.get_vm(name)
+        vm = await db_service.get_vm(name)
         if not vm:
             raise click.BadParameter(f"VM '{name}' not found")
 
         # Get SSH key
         logger.process("Loading SSH credentials")
-        ssh_manager = SSHKeyManager(config.ssh_key_dir)
-        key_pair = await ssh_manager.get_key_pair()
+        key_pair = await ssh_service.get_key_pair()
 
-        # Get VM access info
+        # Get VM access info using service
         logger.process("Fetching connection details")
         provider_url = config.get_provider_url(vm['provider_ip'])
         async with ProviderClient(provider_url) as client:
-            access_info = await client.get_vm_access(vm['vm_id'])
+            vm_service = VMService(db_service, ssh_service, client)
+            vm = await vm_service.get_vm(name)  # Get fresh VM info
+            ssh_port = vm['config']['ssh_port']
 
         # Execute SSH command
-        logger.success(f"Connecting to {vm['provider_ip']}:{access_info['ssh_port']}")
-        cmd = [
-            "ssh",
-            "-i", str(key_pair.private_key.absolute()),
-            "-p", str(access_info['ssh_port']),
-            "-o", "StrictHostKeyChecking=no",
-            "-o", "UserKnownHostsFile=/dev/null",
-            f"ubuntu@{vm['provider_ip']}"
-        ]
-        subprocess.run(cmd)
+        logger.success(f"Connecting to {vm['provider_ip']}:{ssh_port}")
+        ssh_service.connect_to_vm(
+            host=vm['provider_ip'],
+            port=ssh_port,
+            private_key_path=key_pair.private_key.absolute()
+        )
 
     except Exception as e:
         error_msg = str(e)
@@ -338,29 +262,17 @@ async def destroy_vm(name: str):
     try:
         logger.command(f"💥 Destroying VM '{name}'")
 
-        # Get VM details
+        # Get VM details using database service
         logger.process("Retrieving VM details")
-        vm = await db.get_vm(name)
+        vm = await db_service.get_vm(name)
         if not vm:
             raise click.BadParameter(f"VM '{name}' not found")
 
-        try:
-            # Connect to provider using full VM ID
-            logger.process("Requesting VM termination")
-            provider_url = config.get_provider_url(vm['provider_ip'])
-            async with ProviderClient(provider_url) as client:
-                await client.destroy_vm(vm['vm_id'])
-                logger.success("VM terminated on provider")
-        except Exception as e:
-            error_msg = str(e)
-            if "Not Found" in error_msg:
-                logger.warning("VM already removed from provider")
-            else:
-                raise
-
-        # Always remove from database
-        logger.process("Cleaning up VM records")
-        await db.delete_vm(name)
+        # Initialize VM service
+        provider_url = config.get_provider_url(vm['provider_ip'])
+        async with ProviderClient(provider_url) as client:
+            vm_service = VMService(db_service, SSHService(config.ssh_key_dir), client)
+            await vm_service.destroy_vm(name)
         
         # Show fancy success message
         click.echo("\n" + "─" * 60)
@@ -386,7 +298,7 @@ async def destroy_vm(name: str):
 
 
 @vm.command(name='purge')
-@click.option('--force', is_flag=True, help='Force purge even if errors occur')
+@click.option('--force', is_flag=True, help='Force purge even if other errors occur')
 @click.confirmation_option(prompt='Are you sure you want to purge all VMs?')
 @async_command
 async def purge_vms(force: bool):
@@ -394,61 +306,48 @@ async def purge_vms(force: bool):
     try:
         logger.command("🌪️  Purging all VMs")
         
-        # Get all VMs
-        logger.process("Retrieving all VM details")
-        vms = await db.list_vms()
+        vms = await db_service.list_vms()
         if not vms:
             logger.warning("No VMs found to purge")
             return
 
-        # Track results
-        results = {
-            'success': [],
-            'failed': []
-        }
+        results = {'success': [], 'failed': []}
 
-        # Process each VM
         for vm in vms:
             try:
                 logger.process(f"Purging VM '{vm['name']}'")
+                provider_url = config.get_provider_url(vm['provider_ip'])
                 
-                try:
-                    # Try to destroy on provider using full VM ID
-                    provider_url = config.get_provider_url(vm['provider_ip'])
-                    async with ProviderClient(provider_url) as client:
-                        # Get latest VM access info to ensure we have the correct ID
-                        try:
-                            access_info = await client.get_vm_access(vm['vm_id'])
-                            full_vm_id = access_info['vm_id']
-                        except Exception:
-                            # If we can't get access info, use stored VM ID
-                            full_vm_id = vm['vm_id']
-                        
-                        await client.destroy_vm(full_vm_id)
+                async with ProviderClient(provider_url) as client:
+                    vm_service = VMService(db_service, SSHService(config.ssh_key_dir), client)
+                    await vm_service.destroy_vm(vm['name'])
                     results['success'].append((vm['name'], 'Destroyed successfully'))
-                except Exception as e:
-                    error_msg = str(e)
-                    if "Not Found" in error_msg:
-                        results['success'].append((vm['name'], 'Already removed from provider'))
-                    else:
-                        if not force:
-                            raise
-                        results['failed'].append((vm['name'], f"Provider error: {error_msg}"))
-
-                # Always remove from local database
-                await db.delete_vm(vm['name'])
-                
+
+            except (aiohttp.ClientError, asyncio.TimeoutError) as e:
+                await db_service.delete_vm(vm['name'])
+                msg = f"Could not connect to provider ({e}). Removed from local DB. Please destroy manually."
+                results['failed'].append((vm['name'], msg))
+            
             except Exception as e:
-                if not force:
+                if "Cannot connect to host" in str(e):
+                    await db_service.delete_vm(vm['name'])
+                    msg = f"Could not connect to provider ({e}). Removed from local DB. Please destroy manually."
+                    results['failed'].append((vm['name'], msg))
+                elif "not found in multipass" in str(e).lower():
+                    await db_service.delete_vm(vm['name'])
+                    msg = "VM not found on provider. Removed from local DB."
+                    results['success'].append((vm['name'], msg))
+                elif not force:
+                    logger.error(f"Failed to purge VM '{vm['name']}'. Use --force to ignore errors and continue.")
                     raise
-                results['failed'].append((vm['name'], str(e)))
+                else:
+                    results['failed'].append((vm['name'], str(e)))
 
         # Show results
         click.echo("\n" + "─" * 60)
         click.echo(click.style("  🌪️  VM Purge Complete", fg="blue", bold=True))
         click.echo("─" * 60 + "\n")
 
-        # Success section
         if results['success']:
             click.echo(click.style("  ✅ Successfully Purged", fg="green", bold=True))
             click.echo("  " + "┈" * 25)
@@ -456,7 +355,6 @@ async def purge_vms(force: bool):
                 click.echo(f"  • {click.style(name, fg='cyan')}: {click.style(msg, fg='green')}")
             click.echo()
 
-        # Failures section
         if results['failed']:
             click.echo(click.style("  ❌ Failed to Purge", fg="red", bold=True))
             click.echo("  " + "┈" * 25)
@@ -464,7 +362,6 @@ async def purge_vms(force: bool):
                 click.echo(f"  • {click.style(name, fg='cyan')}: {click.style(error, fg='red')}")
             click.echo()
 
-        # Summary
         total = len(results['success']) + len(results['failed'])
         success_rate = (len(results['success']) / total) * 100 if total > 0 else 0
         
@@ -478,10 +375,7 @@ async def purge_vms(force: bool):
         click.echo("\n" + "─" * 60)
 
     except Exception as e:
-        error_msg = str(e)
-        if "database" in error_msg.lower():
-            error_msg = "Failed to access local database"
-        logger.error(f"Purge operation failed: {error_msg}")
+        logger.error(f"Purge operation failed: {str(e)}")
         raise click.Abort()
 
 
@@ -493,18 +387,17 @@ async def start_vm(name: str):
     try:
         logger.command(f"🟢 Starting VM '{name}'")
 
-        # Get VM details
+        # Get VM details using database service
         logger.process("Retrieving VM details")
-        vm = await db.get_vm(name)
+        vm = await db_service.get_vm(name)
         if not vm:
             raise click.BadParameter(f"VM '{name}' not found")
 
-        # Connect to provider
-        logger.process("Powering up VM")
+        # Initialize VM service
         provider_url = config.get_provider_url(vm['provider_ip'])
         async with ProviderClient(provider_url) as client:
-            await client.start_vm(vm['vm_id'])
-            await db.update_vm_status(name, "running")
+            vm_service = VMService(db_service, SSHService(config.ssh_key_dir), client)
+            await vm_service.start_vm(name)
 
         # Show fancy success message
         click.echo("\n" + "─" * 60)
@@ -540,30 +433,29 @@ async def stop_vm(name: str):
     try:
         logger.command(f"🔴 Stopping VM '{name}'")
 
-        # Get VM details
+        # Get VM details using database service
         logger.process("Retrieving VM details")
-        vm = await db.get_vm(name)
+        vm = await db_service.get_vm(name)
         if not vm:
             raise click.BadParameter(f"VM '{name}' not found")
 
-        # Connect to provider
-        logger.process("Shutting down VM")
+        # Initialize VM service
         provider_url = config.get_provider_url(vm['provider_ip'])
         async with ProviderClient(provider_url) as client:
-            await client.stop_vm(vm['vm_id'])
-            await db.update_vm_status(name, "stopped")
+            vm_service = VMService(db_service, SSHService(config.ssh_key_dir), client)
+            await vm_service.stop_vm(name)
 
         # Show fancy success message
         click.echo("\n" + "─" * 60)
         click.echo(click.style("  🔴 VM Stopped Successfully!", fg="yellow", bold=True))
         click.echo("─" * 60 + "\n")
-        
+
         click.echo(click.style("  VM Status", fg="blue", bold=True))
         click.echo("  " + "┈" * 25)
         click.echo(f"  🏷️  Name      : {click.style(name, fg='cyan')}")
         click.echo(f"  💫 Status     : {click.style('stopped', fg='yellow')}")
         click.echo(f"  💾 Resources  : {click.style('preserved', fg='cyan')}")
-        
+
         click.echo("\n" + "─" * 60)
 
     except Exception as e:
@@ -574,6 +466,39 @@ async def stop_vm(name: str):
         raise click.Abort()
 
 
+@cli.group()
+def server():
+    """Server management commands"""
+    pass
+
+
+@server.command(name='api')
+@click.option('--host', default='127.0.0.1', help='Host to bind the API server to.')
+@click.option('--port', default=8000, type=int, help='Port to run the API server on.')
+@click.option('--reload', is_flag=True, help='Enable auto-reload for development.')
+def run_api_server(host: str, port: int, reload: bool):
+    """Run the Requestor API server."""
+    logger.command(f"🚀 Starting Requestor API server on {host}:{port}")
+    if reload:
+        logger.warning("Auto-reload enabled (for development)")
+
+    # Ensure the database directory exists before starting uvicorn
+    try:
+        config.db_path.parent.mkdir(parents=True, exist_ok=True)
+        logger.detail(f"Ensured database directory exists: {config.db_path.parent}")
+    except Exception as e:
+        logger.error(f"Failed to create database directory {config.db_path.parent}: {e}")
+        raise click.Abort()
+
+    uvicorn.run(
+        "requestor.api.main:app",
+        host=host,
+        port=port,
+        reload=reload,
+        log_level="info" # Or adjust as needed
+    )
+
+
 @vm.command(name='list')
 @async_command
 async def list_vms():
@@ -582,41 +507,23 @@ async def list_vms():
         logger.command("📋 Listing your VMs")
         logger.process("Fetching VM details")
         
-        vms = await db.list_vms()
+        # Initialize VM service with temporary client (not needed for listing)
+        ssh_service = SSHService(config.ssh_key_dir)
+        vm_service = VMService(db_service, ssh_service, None)
+        vms = await vm_service.list_vms()
         if not vms:
             logger.warning("No VMs found")
             return
 
-        headers = ["Name", "Status", "IP Address", "SSH Port",
-                   "CPU", "Memory (GB)", "Storage (GB)", "Created"]
-        rows = []
-        for vm in vms:
-            rows.append([
-                vm['name'],
-                vm['status'],
-                vm['provider_ip'],
-                vm['config'].get('ssh_port', 'N/A'),
-                vm['config']['cpu'],
-                vm['config']['memory'],
-                vm['config']['storage'],
-                vm['created_at']
-            ])
+        # Format VM information using service
+        headers = vm_service.vm_headers
+        rows = [vm_service.format_vm_row(vm, colorize=True) for vm in vms]
 
         # Show fancy header
         click.echo("\n" + "─" * 60)
         click.echo(click.style(f"  📋 Your VMs ({len(vms)} total)", fg="blue", bold=True))
         click.echo("─" * 60)
         
-        # Add color to status column (index 1)
-        for i, row in enumerate(rows):
-            status = row[1]
-            if status == "running":
-                rows[i][1] = click.style("● " + status, fg="green", bold=True)
-            elif status == "stopped":
-                rows[i][1] = click.style("● " + status, fg="yellow", bold=True)
-            else:
-                rows[i][1] = click.style("● " + status, fg="red", bold=True)
-        
         # Show table with colored status
         click.echo("\n" + tabulate(
             rows,
@@ -640,3 +547,43 @@ def main():
 
 if __name__ == '__main__':
     main()
+
+
+@vm.command(name='stats')
+@click.argument('name')
+@async_command
+async def vm_stats(name: str):
+    """Display live resource usage statistics for a VM."""
+    try:
+        # Initialize services
+        ssh_service = SSHService(config.ssh_key_dir)
+        vm_service = VMService(db_service, ssh_service)
+
+        # Get VM details
+        vm = await vm_service.get_vm(name)
+        if not vm:
+            raise click.BadParameter(f"VM '{name}' not found")
+
+        # Loop to fetch and display stats continuously
+        while True:
+            stats = await vm_service.get_vm_stats(name)
+            
+            click.clear()
+            click.echo("\n" + "─" * 60)
+            click.echo(click.style(f"  📊 Live Stats for VM: {name} (Press Ctrl+C to exit)", fg="blue", bold=True))
+            click.echo("─" * 60)
+            
+            if 'cpu' in stats and 'usage' in stats['cpu']:
+                click.echo(f"  💻 CPU Usage : {click.style(stats['cpu']['usage'], fg='cyan')}")
+            if 'memory' in stats and 'used' in stats['memory']:
+                click.echo(f"  🧠 Memory    : {click.style(stats['memory']['used'], fg='cyan')} / {click.style(stats['memory']['total'], fg='cyan')}")
+            if 'disk' in stats and 'used' in stats['disk']:
+                click.echo(f"  💾 Disk      : {click.style(stats['disk']['used'], fg='cyan')} / {click.style(stats['disk']['total'], fg='cyan')}")
+            
+            click.echo("─" * 60)
+            
+            await asyncio.sleep(2)  # Update every 2 seconds
+
+    except Exception as e:
+        logger.error(f"Failed to get VM stats: {str(e)}")
+        raise click.Abort()