xenfra 0.1.8__py3-none-any.whl → 0.1.9__py3-none-any.whl

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.")

{xenfra-0.1.8.dist-info → xenfra-0.1.9.dist-info}/METADATA

@@ -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.dist-info → xenfra-0.1.9.dist-info}/RECORD

@@ -10,7 +10,7 @@ xenfra/db/models.py,sha256=nRqbG9cbcX8LWz2Jr4inKysAq47yALcxNl3_UYEpSoQ,1305
 xenfra/db/session.py,sha256=Q03RIiPlYBtPD-qVHlTqQtacnOPTZnOy6_f9p6lABPU,510
 xenfra/dependencies.py,sha256=pQ7bFNZHC24chMxRa_69mu2fIqcoc8DG9MD1POSGgjA,1211
 xenfra/dockerizer.py,sha256=5qzgqw2MZrBwI4IFzXPJZ6k3F3IRdFnX-KGIpv5cjXc,3126
-xenfra/engine.py,sha256=fzdBIW1CmOQtTETjlaMtltt0cgH8r4HN7ImH-s2Wnzk,11793
+xenfra/engine.py,sha256=ZQOyThord9mEhthBxHSL7_8o_d-DsgyN_4VJ_WHCz-U,12954
 xenfra/mcp_client.py,sha256=IxkTC3NPHvPJn-2dl5BUHQ3d-A9MBiKP9fIftzhmCuA,5756
 xenfra/models.py,sha256=9Dvr1_kb29LJIR0a-uwx_tsCWt3GZeko6FKQtYY8F4g,1852
 xenfra/recipes.py,sha256=q5ilpDzGuUM6GyvlW97pOpPS9nzQAJGI4pgzSU_URik,862
@@ -19,7 +19,7 @@ xenfra/templates/Dockerfile.j2,sha256=apWts895OOoUYwj_fOa6OiylFB5m8zFEYvJ1Nki32Y
 xenfra/templates/cloud-init.sh.j2,sha256=Ln7XO4rwaHgauevMKfkxAgBFNLHPnWv-TnJl3NJbU8c,2359
 xenfra/templates/docker-compose.yml.j2,sha256=zKUT2cd_FrxXvRxE-vAAjuQk3-nLNQjRe-StkhAWRQA,860
 xenfra/utils.py,sha256=aGXjJm-pwVCHuCn5UBdrxRcYvM8aJwHQ1kihl7gcxiM,2387
-xenfra-0.1.8.dist-info/WHEEL,sha256=ZyFSCYkV2BrxH6-HRVRg3R9Fo7MALzer9KiPYqNxSbo,79
-xenfra-0.1.8.dist-info/entry_points.txt,sha256=O6JNPm-inoMzmW29WPYl6jHHlWCrcs8ShHEo-CXvERs,49
-xenfra-0.1.8.dist-info/METADATA,sha256=hebQ0mFii9qF8Gs7PeVgzbBizBZ8qQ7GuFA0FycJTT4,4128
-xenfra-0.1.8.dist-info/RECORD,,
+xenfra-0.1.9.dist-info/WHEEL,sha256=ZyFSCYkV2BrxH6-HRVRg3R9Fo7MALzer9KiPYqNxSbo,79
+xenfra-0.1.9.dist-info/entry_points.txt,sha256=O6JNPm-inoMzmW29WPYl6jHHlWCrcs8ShHEo-CXvERs,49
+xenfra-0.1.9.dist-info/METADATA,sha256=qjXKY6vbu_LM2QfJtAo_kC3tL_8V4UBly4dVJ1mn9nA,4128
+xenfra-0.1.9.dist-info/RECORD,,