Merge pull request #104 from consideRatio/pr/small-refactor

refactor: a few smaller details

Author: consideRatio

jupyter_remote_desktop_proxy/setup_websockify.py

@@ -7,18 +7,12 @@
 
 
 def setup_websockify():
-    # make a secure temporary directory for sockets
-    # This is only readable, writeable & searchable by our uid
-    sockets_dir = tempfile.mkdtemp()
-    sockets_path = os.path.join(sockets_dir, 'vnc-socket')
     vncserver = which('vncserver')
-
     if not vncserver:
         raise RuntimeError(
             "vncserver executable not found, please install a VNC server"
         )
 
-    # TigerVNC provides the option to connect a Unix socket. TurboVNC does not.
     # TurboVNC and TigerVNC share the same origin and both use a Perl script
     # as the executable vncserver. We can determine if vncserver is TigerVNC
     # by searching tigervnc string in the Perl script.
@@ -34,11 +28,17 @@ def setup_websockify():
         is_tigervnc = "tigervnc" in vncserver_file.read().casefold()
 
     if is_tigervnc:
+        # Make a secure temporary directory for sockets that is only readable,
+        # writeable, and searchable by our uid - TigerVNC can listen to a Unix
+        # socket!
+        sockets_dir = tempfile.mkdtemp()
+        sockets_path = os.path.join(sockets_dir, 'vnc-socket')
+
+        websockify_args = ['--unix-target', sockets_path]
         vnc_args = [vncserver, '-rfbunixpath', sockets_path]
-        socket_args = ['--unix-target', sockets_path]
     else:
-        vnc_args = [vncserver, '-rfbport', '{port}']
-        socket_args = []
+        websockify_args = []
+        vnc_args = [vncserver, '-localhost', '-rfbport', '{port}']
 
     if not os.path.exists(os.path.expanduser('~/.vnc/xstartup')):
         vnc_args.extend(['-xstartup', os.path.join(HERE, 'share/xstartup')])
@@ -47,23 +47,22 @@ def setup_websockify():
         vnc_args
         + [
             '-verbose',
+            '-fg',
             '-geometry',
             '1680x1050',
             '-SecurityTypes',
             'None',
-            '-fg',
         ]
     )
 
     return {
         'command': [
             'websockify',
-            '-v',
-            '--heartbeat',
-            '30',
+            '--verbose',
+            '--heartbeat=30',
             '{port}',
         ]
-        + socket_args
+        + websockify_args
         + ['--', '/bin/sh', '-c', f'cd {os.getcwd()} && {vnc_command}'],
         'timeout': 30,
         'new_browser_window': True,