From 9a9645829a1575a155998c15932849e3d1ddedc0 Mon Sep 17 00:00:00 2001
From: Pawel Raczyk <pawel.raczyk@vectorsynergy.com>
Date: Wed, 10 Apr 2019 10:32:15 +0200
Subject: [PATCH] setting token in light version

---
 vnc_lite.html | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/vnc_lite.html b/vnc_lite.html
index 426d79b3..387430e3 100644
--- a/vnc_lite.html
+++ b/vnc_lite.html
@@ -78,6 +78,7 @@
     <script type="module" crossorigin="anonymous">
         // RFB holds the API to connect and communicate with a VNC server
         import RFB from './core/rfb.js';
+        import * as WebUtil from './app/webutil.js';
 
         let rfb;
         let desktopName;
@@ -152,12 +153,23 @@
         const host = readQueryVariable('host', window.location.hostname);
         let port = readQueryVariable('port', window.location.port);
         const password = readQueryVariable('password', '');
-        const path = readQueryVariable('path', 'websockify');
-
+        let path = readQueryVariable('path', 'websockify');
         // | | |         | | |
         // | | | Connect | | |
         // v v v         v v v
 
+        // If a token variable is passed in, set the parameter in a cookie.
+        // This is used by nova-novncproxy.
+
+        let token = WebUtil.getConfigVar('token', null);
+
+        if (token) {
+            // if token is already present in the path we should use it
+            path = WebUtil.injectParamIfMissing(path, "token", token);
+
+            WebUtil.createCookie('token', token, 1)
+        }
+
         status("Connecting");
 
         // Build the websocket URL used to connect