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

{request_vm_on_golem-0.1.26.dist-info → request_vm_on_golem-0.1.27.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: request-vm-on-golem
-Version: 0.1.26
+Version: 0.1.27
 Summary: VM on Golem Requestor CLI - Create and manage virtual machines on the Golem Network
 Keywords: golem,vm,cloud,decentralized,cli
 Author: Phillip Jensen

{request_vm_on_golem-0.1.26.dist-info → request_vm_on_golem-0.1.27.dist-info}/RECORD

@@ -1,7 +1,7 @@
 requestor/__init__.py,sha256=OqSUAh1uZBMx7GW0MoSMg967PVdmT8XdPJx3QYjwkak,116
 requestor/api/main.py,sha256=7utCzFNbh5Ol-vsBWeSwT4lXeHD7zdA-GFZuS3rHMWc,2180
 requestor/cli/__init__.py,sha256=e3E4oEGxmGj-STPtFkQwg_qIWhR0JAiAQdw3G1hXciU,37
-requestor/cli/commands.py,sha256=x-O7QJmNniTMRtzQ8K0w8IvBAls3mZQ9VnpmwpA30ZM,23388
+requestor/cli/commands.py,sha256=1EtntPcyZqv9rpFAFsL94IG05o3v3xFVSzqGdyPBodI,23351
 requestor/config.py,sha256=KfAvoAL6jktxTOaJyua5JWGHj7HBHFSrMYSLXl7mRE0,3371
 requestor/db/__init__.py,sha256=Gm5DfWls6uvCZZ3HGGnyRHswbUQdeA5OGN8yPwH0hc8,88
 requestor/db/sqlite.py,sha256=l5pWbx2qlHuar1N_a0B9tVnmumLJY1w5rp3yZ7jmsC0,4146
@@ -11,14 +11,14 @@ requestor/provider/client.py,sha256=OUP7CoOCCtKD6DB9eqFkOXK6A2BLFdM4DWSkoulJQxg,
 requestor/run.py,sha256=3w9oTitu8ZESwU5ss5rB3OldqZPfwiY-OPKY7EwIDBo,1609
 requestor/services/__init__.py,sha256=1qSn_6RMn0KB0A7LCnY2IW6_tC3HBQsdfkFeV-h94eM,172
 requestor/services/database_service.py,sha256=GlSrzzzd7PSYQJNup00sxkB-B2PMr1__04K8k5QSWvs,2996
-requestor/services/provider_service.py,sha256=CxDGD6OaneOHbexv16dFlRJZYXOyJ0lulF-nKhIi-wQ,9849
-requestor/services/ssh_service.py,sha256=bhbZ6g_-e7hrK6tytGfWFuNen_G8Ph4H1Rn-TvphKfY,4010
-requestor/services/vm_service.py,sha256=bbw-q3ZiAyHJX62WuYbUjQYPwCrXxzPd_qSDqm2fXq4,6479
+requestor/services/provider_service.py,sha256=iejw8Q-ziK3Ny0cEAD1EHejUsAqf9BwJTa7jFmei0_8,9773
+requestor/services/ssh_service.py,sha256=tcOCtk2SlB9Uuv-P2ghR22e7BJ9kigQh5b4zSGdPFns,4280
+requestor/services/vm_service.py,sha256=yHvGtfzoWw_wAC7MwgZEudbK8ElxB_2fCuG5Xa-F1KE,6820
 requestor/ssh/__init__.py,sha256=hNgSqJ5s1_AwwxVRyFjUqh_LTBpI4Hmzq0F-f_wXN9g,119
-requestor/ssh/manager.py,sha256=h-93AXFJqzGo2lNTG2u-q4ivU9cCFeNDhYN55skPLBo,6566
+requestor/ssh/manager.py,sha256=XhZjz7_BRPnmpu-zxqnGHLCq0b2JZ8Xr8zc1OlMNDkc,9355
 requestor/utils/logging.py,sha256=oFNpO8pJboYM8Wp7g3HOU4HFyBTKypVdY15lUiz1a4I,3721
 requestor/utils/spinner.py,sha256=PUHJdTD9jpUHur__01_qxXy87WFfNmjQbD_sLG-KlGo,2459
-request_vm_on_golem-0.1.26.dist-info/METADATA,sha256=S1Vl70hOO3rqmyJtHmYtFkEcBGwmHxz6wZl48O5LV10,9128
-request_vm_on_golem-0.1.26.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-request_vm_on_golem-0.1.26.dist-info/entry_points.txt,sha256=Z-skRNpJ8aZcIl_En9mEm1ygkp9FKy0bzQoL3zO52-0,44
-request_vm_on_golem-0.1.26.dist-info/RECORD,,
+request_vm_on_golem-0.1.27.dist-info/METADATA,sha256=TyjqzhArp0Stqj6YtUV_Pv82TjV0fYQkUnrrqholjv4,9128
+request_vm_on_golem-0.1.27.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+request_vm_on_golem-0.1.27.dist-info/entry_points.txt,sha256=Z-skRNpJ8aZcIl_En9mEm1ygkp9FKy0bzQoL3zO52-0,44
+request_vm_on_golem-0.1.27.dist-info/RECORD,,

requestor/cli/commands.py

@@ -298,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):
@@ -306,53 +306,48 @@ async def purge_vms(force: bool):
     try:
         logger.command("🌪️  Purging all VMs")
         
-        # Get all VMs using database service
-        logger.process("Retrieving all VM details")
         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']}'")
-                
-                # 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)
-                    try:
-                        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'))
-                            # Still need to clean up database
-                            await db_service.delete_vm(vm['name'])
-                        else:
-                            if not force:
-                                raise
-                            results['failed'].append((vm['name'], f"Provider error: {error_msg}"))
-                
+                    await vm_service.destroy_vm(vm['name'])
+                    results['success'].append((vm['name'], 'Destroyed successfully'))
+
+            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)
@@ -360,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)
@@ -368,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
         
@@ -382,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()
 
 

requestor/services/provider_service.py

@@ -84,7 +84,7 @@ class ProviderService:
                 provider = {
                     'provider_id': annotations.get('golem_provider_id'),
                     'provider_name': annotations.get('golem_provider_name'),
-                    'ip_address': '127.0.0.1' if 'DEVMODE' in annotations.get('golem_provider_name', '') else annotations.get('golem_ip_address'),
+                    'ip_address': annotations.get('golem_ip_address'),
                     'country': annotations.get('golem_country'),
                     'resources': {
                         'cpu': int(annotations.get('golem_cpu', 0)),

requestor/services/ssh_service.py

@@ -19,6 +19,13 @@ class SSHService:
         except Exception as e:
             raise SSHError(f"Failed to get SSH key pair: {str(e)}")
 
+    def get_key_pair_sync(self):
+        """Get or create SSH key pair synchronously."""
+        try:
+            return self.ssh_manager.get_key_pair_sync()
+        except Exception as e:
+            raise SSHError(f"Failed to get SSH key pair: {str(e)}")
+
     def connect_to_vm(
         self,
         host: str,
@@ -36,7 +43,7 @@ class SSHService:
                 "-o", "UserKnownHostsFile=/dev/null",
                 f"{username}@{host}"
             ]
-            subprocess.run(cmd)
+            subprocess.run(cmd, check=True)
         except Exception as e:
             raise SSHError(f"Failed to establish SSH connection: {str(e)}")
 

requestor/services/vm_service.py

@@ -149,6 +149,13 @@ class VMService:
         """Format VM information for display."""
         from click import style
 
+        key_pair = self.ssh_service.get_key_pair_sync()
+        connect_command = self.ssh_service.format_ssh_command(
+            host=vm['provider_ip'],
+            port=vm['config'].get('ssh_port', 'N/A'),
+            private_key_path=key_pair.private_key.absolute()
+        )
+
         row = [
             vm['name'],
             vm['status'],
@@ -157,6 +164,7 @@ class VMService:
             vm['config']['cpu'],
             vm['config']['memory'],
             vm['config']['storage'],
+            connect_command,
             vm['created_at']
         ]
 
@@ -188,6 +196,7 @@ class VMService:
             "CPU",
             "Memory (GB)",
             "Storage (GB)",
+            "Connect Command",
             "Created"
         ]
 

requestor/ssh/manager.py

@@ -93,6 +93,36 @@ class SSHKeyManager:
             public_key_content=golem_pub_key.read_text().strip()
         )
 
+    def get_key_pair_sync(self) -> KeyPair:
+        """Get the SSH key pair to use (synchronous version)."""
+        logger.debug("Checking for system SSH key at %s", self.system_key_path)
+        system_pub_key = self.system_key_path.parent / 'id_rsa.pub'
+        
+        if self.system_key_path.exists() and system_pub_key.exists():
+            logger.info("Using existing system SSH key")
+            try:
+                return KeyPair(
+                    private_key=self.system_key_path,
+                    public_key=system_pub_key,
+                    private_key_content=self.system_key_path.read_text().strip(),
+                    public_key_content=system_pub_key.read_text().strip()
+                )
+            except (PermissionError, OSError) as e:
+                logger.warning("Could not read system SSH key: %s", e)
+        
+        logger.debug("Using Golem SSH key at %s", self.golem_key_path)
+        if not self.golem_key_path.exists():
+            logger.info("No existing Golem SSH key found, generating new key pair")
+            self._generate_key_pair_sync()
+        
+        golem_pub_key = Path(str(self.golem_key_path) + '.pub')
+        return KeyPair(
+            private_key=self.golem_key_path,
+            public_key=golem_pub_key,
+            private_key_content=self.golem_key_path.read_text().strip(),
+            public_key_content=golem_pub_key.read_text().strip()
+        )
+
     async def get_public_key_content(self) -> str:
         """Get the content of the public key file."""
         key_pair = await self.get_key_pair()
@@ -155,6 +185,38 @@ class SSHKeyManager:
             logger.error("Failed to generate key pair: %s", str(e))
             raise
 
+    def _generate_key_pair_sync(self):
+        """Generate a new RSA key pair for Golem VMs (synchronous version)."""
+        logger.debug("Generating new RSA key pair")
+        try:
+            private_key = rsa.generate_private_key(
+                public_exponent=65537,
+                key_size=2048,
+                backend=default_backend()
+            )
+            private_pem = private_key.private_bytes(
+                encoding=serialization.Encoding.PEM,
+                format=serialization.PrivateFormat.PKCS8,
+                encryption_algorithm=serialization.NoEncryption()
+            )
+            self.golem_key_path.write_bytes(private_pem)
+            if os.name == 'posix':
+                os.chmod(self.golem_key_path, 0o600)
+
+            public_key = private_key.public_key()
+            public_pem = public_key.public_bytes(
+                encoding=serialization.Encoding.OpenSSH,
+                format=serialization.PublicFormat.OpenSSH
+            )
+            pub_key_path = Path(str(self.golem_key_path) + '.pub')
+            pub_key_path.write_bytes(public_pem)
+            if os.name == 'posix':
+                os.chmod(pub_key_path, 0o644)
+            logger.info("Successfully generated and saved SSH key pair")
+        except Exception as e:
+            logger.error("Failed to generate key pair: %s", str(e))
+            raise
+
     async def get_private_key_content(self, force_golem_key: bool = False) -> Optional[str]:
         """Get the content of the private key file."""
         key_pair = await self.get_key_pair(force_golem_key)