unitlab 2.3.34__py3-none-any.whl → 2.3.36__py3-none-any.whl

unitlab/client.py

@@ -406,8 +406,7 @@ class UnitlabClient:
                 else:
                     
                     self.ssh_url = self.jupyter_url
-                
-               
+
                 if hasattr(self.tunnel_manager, 'tunnel_url') and self.tunnel_manager.tunnel_url:
                     self.jupyter_url = self.tunnel_manager.tunnel_url
                     if not hasattr(self.tunnel_manager, 'ssh_url'):

unitlab/main.py

@@ -132,7 +132,7 @@ def run_agent(
         handlers=[logging.StreamHandler()]
     )
 
-    server_url = 'http://localhost:8000/'
+    server_url = 'https://api-dev/unitlab-ai.com/'
     
     if not device_id:
         import uuid

unitlab/persistent_tunnel.py

@@ -208,6 +208,57 @@ class PersistentTunnel:
         
         return True
     
+    def create_access_application(self):
+        """Create Cloudflare Access application for SSH with bypass policy"""
+        print("🔧 Creating Access application for SSH...")
+        
+        # Create Access application
+        app_url = "https://api.cloudflare.com/client/v4/zones/{}/access/apps".format(self.cf_zone_id)
+        headers = self._get_headers()
+        
+        app_data = {
+            "name": "SSH-{}".format(self.device_id),
+            "domain": "{}.{}".format(self.ssh_subdomain, self.domain),
+            "type": "ssh",
+            "session_duration": "24h",
+            "auto_redirect_to_identity": False
+        }
+        
+        app_response = requests.post(app_url, headers=headers, json=app_data)
+        
+        if app_response.status_code in [200, 201]:
+            app_id = app_response.json()["result"]["id"]
+            print("✅ Access application created: {}".format(app_id))
+            
+            # Create bypass policy (no authentication required)
+            policy_url = "https://api.cloudflare.com/client/v4/zones/{}/access/apps/{}/policies".format(
+                self.cf_zone_id, app_id
+            )
+            
+            policy_data = {
+                "name": "Public Access",
+                "decision": "bypass",
+                "include": [
+                    {"everyone": {}}
+                ],
+                "precedence": 1
+            }
+            
+            policy_response = requests.post(policy_url, headers=headers, json=policy_data)
+            
+            if policy_response.status_code in [200, 201]:
+                print("✅ Bypass policy created - SSH is publicly accessible")
+                return True
+            else:
+                print("⚠️  Could not create bypass policy: {}".format(policy_response.text[:200]))
+                return False
+        elif "already exists" in app_response.text:
+            print("⚠️  Access application already exists")
+            return True
+        else:
+            print("⚠️  Could not create Access application: {}".format(app_response.text[:200]))
+            return False
+    
     def create_tunnel_config(self, cred_file):
         """Create tunnel config file"""
         config_file = "/tmp/tunnel-config-{}.yml".format(self.tunnel_name)
@@ -400,10 +451,13 @@ class PersistentTunnel:
             # 2. Create DNS record
             self.create_dns_record()
             
-            # 3. Create config
+            # 3. Create Access application for SSH
+            self.create_access_application()
+            
+            # 4. Create config
             config_file = self.create_tunnel_config(cred_file)
             
-            # 4. Start services
+            # 5. Start services
             self.start_jupyter()
             self.start_api()
             self.start_tunnel(config_file)
@@ -433,34 +487,7 @@ class PersistentTunnel:
             self.stop()
             return False
     
-    def start_quick_tunnel(self):
-        """Fallback to quick tunnel"""
-        print("🔧 Using quick tunnel (temporary URL)...")
-        
-        # Start Jupyter first
-        self.start_jupyter()
-        
-        # Start quick tunnel
-        cloudflared = self.get_cloudflared_path()
-        cmd = [cloudflared, "tunnel", "--url", "http://localhost:8888"]
-        
-        self.tunnel_process = subprocess.Popen(
-            cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True
-        )
-        
-        # Get URL from output
-        for _ in range(30):
-            line = self.tunnel_process.stdout.readline()
-            if "trycloudflare.com" in line:
-                import re
-                match = re.search(r'https://[a-zA-Z0-9-]+\.trycloudflare\.com', line)
-                if match:
-                    self.jupyter_url = match.group(0)
-                    print("✅ Quick tunnel: {}".format(self.jupyter_url))
-                    return True
-            time.sleep(0.5)
-        
-        return False
+
     
     def stop(self):
         """Stop everything"""
@@ -469,16 +496,16 @@ class PersistentTunnel:
         if self.tunnel_process:
             self.tunnel_process.terminate()
         
-        # Optionally delete tunnel when stopping
-        if self.tunnel_id:
-            try:
-                url = "https://api.cloudflare.com/client/v4/accounts/{}/cfd_tunnel/{}".format(
-                    self.cf_account_id, self.tunnel_id
-                )
-                requests.delete(url, headers=self._get_headers())
-                print("🗑️  Tunnel deleted")
-            except Exception:
-                pass  # Ignore cleanup errors
+    #     # Optionally delete tunnel when stopping
+    #     if self.tunnel_id:
+    #         try:
+    #             url = "https://api.cloudflare.com/client/v4/accounts/{}/cfd_tunnel/{}".format(
+    #                 self.cf_account_id, self.tunnel_id
+    #             )
+    #             requests.delete(url, headers=self._get_headers())
+    #             print("🗑️  Tunnel deleted")
+    #         except Exception:
+    #             pass  # Ignore cleanup errors
     
     def run(self):
         """Run and keep alive"""

{unitlab-2.3.34.dist-info → unitlab-2.3.36.dist-info}/METADATA

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

unitlab-2.3.36.dist-info/RECORD

@@ -0,0 +1,13 @@
+unitlab/__init__.py,sha256=Wtk5kQ_MTlxtd3mxJIn2qHVK5URrVcasMMPjD3BtrVM,214
+unitlab/__main__.py,sha256=6Hs2PV7EYc5Tid4g4OtcLXhqVHiNYTGzSBdoOnW2HXA,29
+unitlab/client.py,sha256=wqREtDuYc5ixeloPEGm0hp1sdUtB59sB1bIJjBcO1y0,25983
+unitlab/exceptions.py,sha256=68Tr6LreEzjQ3Vns8HAaWdtewtkNUJOvPazbf6NSnXU,950
+unitlab/main.py,sha256=kazASmzhPaAcf9hsPZdewcry_vplsrRLfziPxKlPT70,4425
+unitlab/persistent_tunnel.py,sha256=MRbfdxfP8ceY-25zbVD-29W3U1rs7XcFHZDHoqBlkSU,22401
+unitlab/utils.py,sha256=9gPRu-d6pbhSoVdll1GXe4eoz_uFYOSbYArFDQdlUZs,1922
+unitlab-2.3.36.dist-info/licenses/LICENSE.md,sha256=Gn7RRvByorAcAaM-WbyUpsgi5ED1-bKFFshbWfYYz2Y,1069
+unitlab-2.3.36.dist-info/METADATA,sha256=JgMYM-Z3mqgMcEEqCgS1AOwIOGFIBPXrzAu3mCAvJsY,1046
+unitlab-2.3.36.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+unitlab-2.3.36.dist-info/entry_points.txt,sha256=ig-PjKEqSCj3UTdyANgEi4tsAU84DyXdaOJ02NHX4bY,45
+unitlab-2.3.36.dist-info/top_level.txt,sha256=Al4ZlTYE3fTJK2o6YLCDMH5_DjuQkffRBMxgmWbKaqQ,8
+unitlab-2.3.36.dist-info/RECORD,,

unitlab-2.3.34.dist-info/RECORD

@@ -1,13 +0,0 @@
-unitlab/__init__.py,sha256=Wtk5kQ_MTlxtd3mxJIn2qHVK5URrVcasMMPjD3BtrVM,214
-unitlab/__main__.py,sha256=6Hs2PV7EYc5Tid4g4OtcLXhqVHiNYTGzSBdoOnW2HXA,29
-unitlab/client.py,sha256=LMZ7HFjRxzPV2IrCXlP2GETlY0vWvAxP0RrjSXOn_Jk,26015
-unitlab/exceptions.py,sha256=68Tr6LreEzjQ3Vns8HAaWdtewtkNUJOvPazbf6NSnXU,950
-unitlab/main.py,sha256=EbQNO-Z5drNQjDXJp_sIs5a3WgPoqxaXxpdFGzMWm6k,4416
-unitlab/persistent_tunnel.py,sha256=XDJo2PPq4EjEtI4vT68LIGUUq7WV4m0bnXYKPfY51cY,21180
-unitlab/utils.py,sha256=9gPRu-d6pbhSoVdll1GXe4eoz_uFYOSbYArFDQdlUZs,1922
-unitlab-2.3.34.dist-info/licenses/LICENSE.md,sha256=Gn7RRvByorAcAaM-WbyUpsgi5ED1-bKFFshbWfYYz2Y,1069
-unitlab-2.3.34.dist-info/METADATA,sha256=ctnyV_ZT0vuOzGMoH0N09v0nalsR79WQCGdi_D-ncw0,1046
-unitlab-2.3.34.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-unitlab-2.3.34.dist-info/entry_points.txt,sha256=ig-PjKEqSCj3UTdyANgEi4tsAU84DyXdaOJ02NHX4bY,45
-unitlab-2.3.34.dist-info/top_level.txt,sha256=Al4ZlTYE3fTJK2o6YLCDMH5_DjuQkffRBMxgmWbKaqQ,8
-unitlab-2.3.34.dist-info/RECORD,,