unitlab 2.3.29__tar.gz → 2.3.32__tar.gz

{unitlab-2.3.29/src/unitlab.egg-info → unitlab-2.3.32}/PKG-INFO

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

{unitlab-2.3.29 → unitlab-2.3.32}/setup.py

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

{unitlab-2.3.29 → unitlab-2.3.32}/src/unitlab/client.py

@@ -286,21 +286,14 @@ class UnitlabClient:
             from .persistent_tunnel import PersistentTunnel
             logger.info("Using Persistent Tunnel with Cloudflare API")
             self.tunnel_manager = PersistentTunnel(device_id=self.device_id)
-            # URLs will be set after tunnel starts
-            self.jupyter_url = self.tunnel_manager.jupyter_url
-            self.ssh_url = self.jupyter_url
+            # Don't call run() here - it has infinite loop. Call start() in setup_tunnels()
+            self.jupyter_url = None
+            self.ssh_url = None
+            
         except ImportError as e:
             logger.warning(f"Could not import PersistentTunnel: {e}")
             # Fallback to easy tunnel
-            try:
-                from .easy_tunnel import EasyTunnelAdapter
-                logger.info("Using Easy Dynamic Tunnel (random URL)")
-                self.tunnel_manager = EasyTunnelAdapter(device_id=self.device_id)
-                self.jupyter_url = None
-                self.ssh_url = None
-            except ImportError:
-                logger.error("No tunnel implementation available!")
-                raise
+            
         
         # Setup signal handlers
         signal.signal(signal.SIGINT, self._handle_shutdown)
@@ -410,6 +403,10 @@ class UnitlabClient:
             if self.tunnel_manager.start():
                 # Store the processes for monitoring
                 self.jupyter_proc = self.tunnel_manager.jupyter_process
+                # Update URLs after tunnel starts successfully
+                self.jupyter_url = self.tunnel_manager.jupyter_url
+                self.ssh_url = self.tunnel_manager.jupyter_url
+                logger.info(f"Tunnel started successfully at {self.jupyter_url}")
                 self.tunnel_proc = self.tunnel_manager.tunnel_process
                 self.jupyter_port = "8888"  # Both use fixed port
                 

{unitlab-2.3.29 → unitlab-2.3.32}/src/unitlab/persistent_tunnel.py

@@ -67,9 +67,42 @@ class PersistentTunnel:
             "Content-Type": "application/json"
         }
     
-    def create_tunnel(self):
-        """Create a new tunnel via API"""
-        print("🔧 Creating tunnel: {}...".format(self.tunnel_name))
+    def get_or_create_tunnel(self):
+        """Get existing tunnel or create a new one"""
+        # First, check if tunnel already exists
+        print("🔍 Checking for existing tunnel: {}...".format(self.tunnel_name))
+        
+        list_url = "https://api.cloudflare.com/client/v4/accounts/{}/cfd_tunnel".format(self.cf_account_id)
+        headers = self._get_headers()
+        
+        # Check if tunnel exists
+        response = requests.get(list_url, headers=headers)
+        if response.status_code == 200:
+            tunnels = response.json().get("result", [])
+            for tunnel in tunnels:
+                if tunnel["name"] == self.tunnel_name:
+                    print("✅ Found existing tunnel: {}".format(tunnel["id"]))
+                    self.tunnel_id = tunnel["id"]
+                    
+                    # For persistent device IDs, always recreate to ensure fresh state
+                    print("🔄 Recreating tunnel for persistent device...")
+                    delete_url = "https://api.cloudflare.com/client/v4/accounts/{}/cfd_tunnel/{}".format(
+                        self.cf_account_id, tunnel["id"]
+                    )
+                    del_resp = requests.delete(delete_url, headers=headers)
+                    if del_resp.status_code in [200, 204]:
+                        print("✅ Deleted old tunnel")
+                        time.sleep(2)
+                    else:
+                        print("⚠️  Could not delete old tunnel, trying to create new one anyway")
+                    break
+        
+        # Create new tunnel
+        return self.create_new_tunnel()
+    
+    def create_new_tunnel(self):
+        """Create a brand new tunnel"""
+        print("🔧 Creating new tunnel: {}...".format(self.tunnel_name))
         
         # Generate random tunnel secret (32 bytes)
         import secrets
@@ -96,8 +129,8 @@ class PersistentTunnel:
                 "TunnelID": self.tunnel_id
             }
             
-            # Save credentials to file
-            cred_file = "/tmp/tunnel-{}.json".format(self.tunnel_id)
+            # Save credentials to file with tunnel name (not ID) for consistency
+            cred_file = "/tmp/tunnel-{}.json".format(self.tunnel_name)
             with open(cred_file, 'w') as f:
                 json.dump(self.tunnel_credentials, f)
             
@@ -143,19 +176,7 @@ class PersistentTunnel:
     
     def create_tunnel_config(self, cred_file):
         """Create tunnel config file"""
-        config = {
-            "ingress": [
-                {
-                    "hostname": "{}.{}".format(self.subdomain, self.domain),
-                    "service": "http://localhost:8888"
-                },
-                {
-                    "service": "http_status:404"
-                }
-            ]
-        }
-        
-        config_file = "/tmp/tunnel-config-{}.yml".format(self.tunnel_id)
+        config_file = "/tmp/tunnel-config-{}.yml".format(self.tunnel_name)
         with open(config_file, 'w') as f:
             f.write("tunnel: {}\n".format(self.tunnel_id))
             f.write("credentials-file: {}\n\n".format(cred_file))
@@ -167,25 +188,75 @@ class PersistentTunnel:
         return config_file
     
     def get_cloudflared_path(self):
-        """Get or download cloudflared"""
+        """Get or download cloudflared for any platform"""
         import shutil
+        import platform
+        
+        # Check if already in system PATH
         if shutil.which("cloudflared"):
             return "cloudflared"
         
-        local_bin = os.path.expanduser("~/.local/bin/cloudflared")
+        # Determine binary location based on OS
+        system = platform.system().lower()
+        machine = platform.machine().lower()
+        
+        if system == "windows":
+            local_bin = os.path.expanduser("~/cloudflared/cloudflared.exe")
+        else:
+            local_bin = os.path.expanduser("~/.local/bin/cloudflared")
+        
+        # Check if already downloaded
         if os.path.exists(local_bin):
             return local_bin
         
-        # Download
-        print("📦 Downloading cloudflared...")
-        import platform
-        system = platform.system().lower()
-        arch = "amd64" if "x86" in platform.machine() else "arm64"
-        url = "https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-{}".format(arch)
+        # Download based on platform
+        print("📦 Downloading cloudflared for {}...".format(system))
+        
+        if system == "linux":
+            # Linux: detect architecture
+            if "arm" in machine or "aarch64" in machine:
+                arch = "arm64"
+            elif "386" in machine or "i686" in machine:
+                arch = "386"
+            else:
+                arch = "amd64"
+            url = "https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-linux-{}".format(arch)
+            
+            os.makedirs(os.path.dirname(local_bin), exist_ok=True)
+            subprocess.run("curl -L {} -o {}".format(url, local_bin), shell=True, capture_output=True)
+            subprocess.run("chmod +x {}".format(local_bin), shell=True)
+            
+        elif system == "darwin":
+            # macOS: supports both Intel and Apple Silicon
+            if "arm" in machine:
+                url = "https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-darwin-arm64.tgz"
+            else:
+                url = "https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-darwin-amd64.tgz"
+            
+            os.makedirs(os.path.dirname(local_bin), exist_ok=True)
+            # Download and extract tar.gz
+            subprocess.run("curl -L {} | tar xz -C {}".format(url, os.path.dirname(local_bin)), shell=True, capture_output=True)
+            subprocess.run("chmod +x {}".format(local_bin), shell=True)
+            
+        elif system == "windows":
+            # Windows: typically amd64
+            if "arm" in machine:
+                arch = "arm64"
+            elif "386" in machine:
+                arch = "386"
+            else:
+                arch = "amd64"
+            url = "https://github.com/cloudflare/cloudflared/releases/latest/download/cloudflared-windows-{}.exe".format(arch)
+            
+            os.makedirs(os.path.dirname(local_bin), exist_ok=True)
+            # Use PowerShell on Windows to download
+            subprocess.run("powershell -Command \"Invoke-WebRequest -Uri {} -OutFile {}\"".format(url, local_bin), shell=True, capture_output=True)
+        
+        else:
+            print("❌ Unsupported platform: {}".format(system))
+            raise Exception("Platform {} not supported".format(system))
         
-        os.makedirs(os.path.dirname(local_bin), exist_ok=True)
-        subprocess.run("curl -L {} -o {}".format(url, local_bin), shell=True, capture_output=True)
-        subprocess.run("chmod +x {}".format(local_bin), shell=True)
+        print("✅ cloudflared downloaded successfully")
         return local_bin
     
     def start_jupyter(self):
@@ -198,7 +269,10 @@ class PersistentTunnel:
             "--no-browser",
             "--ip", "0.0.0.0",
             "--NotebookApp.token=''",
-            "--NotebookApp.password=''"
+            "--NotebookApp.password=''",
+            "--NotebookApp.allow_origin='*'"
+
+            
         ]
         
         self.jupyter_process = subprocess.Popen(
@@ -241,8 +315,8 @@ class PersistentTunnel:
             
             # API credentials are hardcoded, so we're ready to go
             
-            # 1. Create tunnel via API
-            cred_file = self.create_tunnel()
+            # 1. Get existing or create new tunnel via API
+            cred_file = self.get_or_create_tunnel()
             if not cred_file:
                 print("⚠️  Falling back to quick tunnel")
                 return self.start_quick_tunnel()
@@ -316,7 +390,7 @@ class PersistentTunnel:
                 )
                 requests.delete(url, headers=self._get_headers())
                 print("🗑️  Tunnel deleted")
-            except Exception as e:
+            except Exception:
                 pass  # Ignore cleanup errors
     
     def run(self):

{unitlab-2.3.29 → unitlab-2.3.32/src/unitlab.egg-info}/PKG-INFO

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