xenfra 0.1.8__tar.gz → 0.1.9__tar.gz

{xenfra-0.1.8 → xenfra-0.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: xenfra
-Version: 0.1.8
+Version: 0.1.9
 Summary: A 'Zen Mode' infrastructure engine for Python developers.
 Author: xenfra-cloud
 Author-email: xenfra-cloud <xenfracloud@gmail.com>

{xenfra-0.1.8 → xenfra-0.1.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "xenfra"
-version = "0.1.8"
+version = "0.1.9"
 description = "A 'Zen Mode' infrastructure engine for Python developers."
 readme = "README.md"
 authors = [

{xenfra-0.1.8 → xenfra-0.1.9}/src/xenfra/engine.py

@@ -44,7 +44,15 @@ class InfraEngine:
 
     def _get_connection(self, ip_address: str):
         """Establishes a Fabric connection to the server."""
-        return fabric.Connection(host=ip_address, user="root", connect_kwargs={"password": None}) # Assumes SSH key auth
+        private_key_path = str(Path.home() / ".ssh" / "id_rsa")
+        if not Path(private_key_path).exists():
+            raise DeploymentError("No private SSH key found at ~/.ssh/id_rsa.", stage="Setup")
+
+        return fabric.Connection(
+            host=ip_address,
+            user="root",
+            connect_kwargs={"key_filename": [private_key_path]},
+        )
 
     def get_user_info(self):
         """Retrieves user account information."""
@@ -189,12 +197,33 @@ class InfraEngine:
             while True:
                 droplet.load()
                 if droplet.status == 'active':
-                    logger("   - Droplet is active. Waiting for cloud-init to complete...")
+                    logger("   - Droplet is active. Waiting for SSH to be available...")
                     break
                 time.sleep(10)
-            
+
             ip_address = droplet.ip_address
-            with self._get_connection(ip_address) as conn:
+
+            # Retry SSH connection
+            conn = None
+            max_retries = 12 # 2-minute timeout for SSH
+            for i in range(max_retries):
+                try:
+                    logger(f"   - Attempting SSH connection ({i+1}/{max_retries})...")
+                    conn = self._get_connection(ip_address)
+                    conn.open() # Explicitly open the connection
+                    logger("   - SSH connection established.")
+                    break
+                except Exception as e:
+                    if i < max_retries - 1:
+                        logger(f"   - SSH connection failed. Retrying in 10s...")
+                        time.sleep(10)
+                    else:
+                        raise DeploymentError(f"Failed to establish SSH connection: {e}", stage="Polling")
+
+            if not conn or not conn.is_connected:
+                raise DeploymentError("Could not establish SSH connection.", stage="Polling")
+
+            with conn:
                 for i in range(30): # 5-minute timeout for cloud-init
                     if conn.run("test -f /root/setup_complete", warn=True).ok:
                         logger("   - Cloud-init setup complete.")