unitlab 2.3.14__tar.gz → 2.3.16__tar.gz

{unitlab-2.3.14/src/unitlab.egg-info → unitlab-2.3.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unitlab
-Version: 2.3.14
+Version: 2.3.16
 Home-page: https://github.com/teamunitlab/unitlab-sdk
 Author: Unitlab Inc.
 Author-email: team@unitlab.ai

{unitlab-2.3.14 → unitlab-2.3.16}/setup.py

@@ -2,7 +2,7 @@ from setuptools import find_packages, setup
 
 setup(
     name="unitlab",
-    version="2.3.14",
+    version="2.3.16",
     license="MIT",
     author="Unitlab Inc.",
     author_email="team@unitlab.ai",

{unitlab-2.3.14 → unitlab-2.3.16}/src/unitlab/cloudflare_api_tunnel.py

@@ -231,9 +231,12 @@ class CloudflareAPITunnel:
                 # Create new tunnel
                 print(f"📦 Creating new tunnel: {tunnel_name}")
                 create_url = f"{self.api_base}/accounts/{self.account_id}/tunnels"
+                # Generate secret first
+                tunnel_secret = os.urandom(32).hex()
+                
                 create_data = {
                     "name": tunnel_name,
-                    "tunnel_secret": os.urandom(32).hex()  # Generate random secret
+                    "tunnel_secret": tunnel_secret
                 }
                 
                 create_response = requests.post(create_url, headers=self.headers, json=create_data)
@@ -242,6 +245,9 @@ class CloudflareAPITunnel:
                     existing_tunnel = create_response.json()['result']
                     print(f"✅ Created tunnel: {tunnel_name}")
                     
+                    # Add the secret to the tunnel info (API doesn't return it)
+                    existing_tunnel['tunnel_secret'] = tunnel_secret
+                    
                     # Save credentials for this tunnel
                     self._save_tunnel_credentials(existing_tunnel)
                     
@@ -308,19 +314,32 @@ class CloudflareAPITunnel:
     def _save_tunnel_credentials(self, tunnel_info):
         """
         Save tunnel credentials locally for this device
+        Credentials must be base64 encoded for cloudflared
         """
+        import base64
+        import json
+        
         creds_dir = Path.home() / '.cloudflared'
         creds_dir.mkdir(exist_ok=True)
         
         creds_file = creds_dir / f"{tunnel_info['id']}.json"
         
+        # Get the secret - it should be hex string
+        secret_hex = tunnel_info.get('tunnel_secret') or tunnel_info.get('secret')
+        if secret_hex:
+            # Convert hex to bytes then to base64
+            secret_bytes = bytes.fromhex(secret_hex)
+            secret_b64 = base64.b64encode(secret_bytes).decode('ascii')
+        else:
+            print("⚠️  No tunnel secret found")
+            return None
+        
         credentials = {
             "AccountTag": self.account_id,
-            "TunnelSecret": tunnel_info.get('tunnel_secret') or tunnel_info.get('secret'),
+            "TunnelSecret": secret_b64,  # Must be base64!
             "TunnelID": tunnel_info['id']
         }
         
-        import json
         with open(creds_file, 'w') as f:
             json.dump(credentials, f)
         

{unitlab-2.3.14 → unitlab-2.3.16}/src/unitlab/main.py

@@ -146,13 +146,27 @@ def run_agent(
     if not device_id:
         import uuid
         import platform
+        from pathlib import Path
+        
         # Try environment variable first
         device_id = os.getenv('DEVICE_ID')
         if not device_id:
-            # Generate a unique ID based on hostname and random UUID
-            hostname = platform.node().replace('.', '-').replace(' ', '-')[:20]
-            random_suffix = str(uuid.uuid4())[:8]
-            device_id = f"{hostname}-{random_suffix}"
+            # Try to load saved device ID
+            device_id_file = Path.home() / '.unitlab' / 'device_id'
+            device_id_file.parent.mkdir(exist_ok=True, parents=True)
+            
+            if device_id_file.exists():
+                device_id = device_id_file.read_text().strip()
+                print(f"📌 Using saved device ID: {device_id}")
+            else:
+                # Generate a unique ID based on hostname and random UUID
+                hostname = platform.node().replace('.', '-').replace(' ', '-')[:20]
+                random_suffix = str(uuid.uuid4())[:8]
+                device_id = f"{hostname}-{random_suffix}"
+                
+                # Save for future runs
+                device_id_file.write_text(device_id)
+                print(f"📝 Generated and saved device ID: {device_id}")
           
     
     # Create client and initialize device agent

{unitlab-2.3.14 → unitlab-2.3.16/src/unitlab.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unitlab
-Version: 2.3.14
+Version: 2.3.16
 Home-page: https://github.com/teamunitlab/unitlab-sdk
 Author: Unitlab Inc.
 Author-email: team@unitlab.ai