xenfra 0.1.8__py3-none-any.whl → 0.2.0__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/templates/cloud-init.sh.j2

@@ -27,18 +27,18 @@ dpkg --configure -a || true
 # -----------------------------------------------
 
 # 1. System Updates
-echo "🔄 [1/6] Refreshing Package Lists..." >> $LOG
+echo "🔄 [1/5] Refreshing Package Lists..." >> $LOG
 apt-get update
 apt-get install -y python3-pip git curl
 
 # 2. Install Docker & Compose
-echo "🐳 [2/6] Installing Docker..." >> $LOG
+echo "🐳 [2/5] Installing Docker..." >> $LOG
 apt-get install -y docker.io || (curl -fsSL https://get.docker.com | sh)
-echo "🎶 [3/6] Installing Docker Compose..." >> $LOG
+echo "🎶 [3/5] Installing Docker Compose..." >> $LOG
 apt-get install -y docker-compose-v2
 
 # --- DOCKERIZED DEPLOYMENT ---
-echo "📦 [4/6] Installing Caddy..." >> $LOG
+echo "📦 [4/5] Installing Caddy..." >> $LOG
 apt-get install -y debian-keyring debian-archive-keyring apt-transport-https
 curl -LsSf 'https://dl.cloudsmith.io/public/caddy/stable/gpg.key' | gpg --dearmor -o /usr/share/keyrings/caddy-stable-archive-keyring.gpg
 curl -LsSf 'https://dl.cloudsmith.io/public/caddy/stable/debian.deb.txt' | tee /etc/apt/sources.list.d/caddy-stable.list
@@ -56,15 +56,11 @@ cat << EOF > /etc/caddy/Caddyfile
 EOF
 {% endif %}
 
-echo "🚀 [5/6] Starting application with Docker Compose..." >> $LOG
-cd /root/app
-docker-compose -f /root/app/docker-compose.yml up -d
-
 {% if domain %}
-echo "🔒 [6/6] Starting Caddy..." >> $LOG
+echo "🚀 [5/5] Starting Caddy..." >> $LOG
 systemctl restart caddy
 {% else %}
-echo "✅ [6/6] Skipping Caddy start (no domain specified)." >> $LOG
+echo "✅ [5/5] Skipping Caddy start (no domain specified)." >> $LOG
 {% endif %}
 
 # Finish

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: xenfra
-Version: 0.1.8
+Version: 0.2.0
 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.2.0.dist-info}/RECORD

@@ -10,16 +10,16 @@ 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
 xenfra/security.py,sha256=w7aIFnHDHOH2zDjYD_LnJnSE01NeGeoAh_V8hC1HipY,2003
 xenfra/templates/Dockerfile.j2,sha256=apWts895OOoUYwj_fOa6OiylFB5m8zFEYvJ1Nki32YM,664
-xenfra/templates/cloud-init.sh.j2,sha256=Ln7XO4rwaHgauevMKfkxAgBFNLHPnWv-TnJl3NJbU8c,2359
+xenfra/templates/cloud-init.sh.j2,sha256=QCWG8hL1V05bAQ7BQ70QfuhIvS4tnsL8ZTCVtyi9F0A,2222
 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.2.0.dist-info/WHEEL,sha256=ZyFSCYkV2BrxH6-HRVRg3R9Fo7MALzer9KiPYqNxSbo,79
+xenfra-0.2.0.dist-info/entry_points.txt,sha256=O6JNPm-inoMzmW29WPYl6jHHlWCrcs8ShHEo-CXvERs,49
+xenfra-0.2.0.dist-info/METADATA,sha256=uQUHeQQz1bAm_cWLisVLjWoeRKBrEZXHfw1BBZYZTlY,4128
+xenfra-0.2.0.dist-info/RECORD,,