From 73ee657d7444df14e1d0d2a973af733a9bfa3093 Mon Sep 17 00:00:00 2001
From: Nick O'Leary <nick.oleary@gmail.com>
Date: Thu, 25 Jan 2018 20:26:35 +0000
Subject: [PATCH] Add TLS options to WebSocket client

---
 nodes/core/io/10-mqtt.js               |  1 -
 nodes/core/io/22-websocket.html        | 21 +++++++++++++++++++--
 nodes/core/io/22-websocket.js          | 10 +++++++++-
 nodes/core/locales/en-US/messages.json |  5 +++--
 4 files changed, 31 insertions(+), 6 deletions(-)

diff --git a/nodes/core/io/10-mqtt.js b/nodes/core/io/10-mqtt.js
index 507ddaf04..c1a8318a6 100644
--- a/nodes/core/io/10-mqtt.js
+++ b/nodes/core/io/10-mqtt.js
@@ -184,7 +184,6 @@ module.exports = function(RED) {
         this.connect = function () {
             if (!node.connected && !node.connecting) {
                 node.connecting = true;
-                console.log("going for connect");
                 try {
                     node.client = mqtt.connect(node.brokerurl ,node.options);
                     node.client.setMaxListeners(0);
diff --git a/nodes/core/io/22-websocket.html b/nodes/core/io/22-websocket.html
index 703aa7bf6..99eeea2e4 100644
--- a/nodes/core/io/22-websocket.html
+++ b/nodes/core/io/22-websocket.html
@@ -180,12 +180,24 @@
         category: 'config',
         defaults: {
             path: {value:"",required:true,validate:RED.validators.regex(/^((?!\/debug\/ws).)*$/)},
+            tls: {type:"tls-config",required: false},
             wholemsg: {value:"false"}
         },
         inputs:0,
         outputs:0,
         label: function() {
             return this.path;
+        },
+        oneditprepare: function() {
+            $("#node-config-input-path").on("change keyup paste",function() {
+                $(".node-config-row-tls").toggle(/^wss:/i.test($(this).val()))
+            });
+            $("#node-config-input-path").change();
+        },
+        oneditsave: function() {
+            if (!/^wss:/i.test($("#node-config-input-path").val())) {
+                $("#node-config-input-tls").val("_ADD_");
+            }
         }
     });
 
@@ -235,7 +247,7 @@
         <input id="node-config-input-path" type="text" placeholder="/ws/example">
     </div>
     <div class="form-row">
-        <label for="node-config-input-wholemsg">&nbsp;</label>
+        <label for="node-config-input-wholemsg"></label>
         <select type="text" id="node-config-input-wholemsg" style="width: 70%;">
             <option value="false" data-i18n="websocket.payload"></option>
             <option value="true" data-i18n="websocket.message"></option>
@@ -257,8 +269,13 @@
         <label for="node-config-input-path"><i class="fa fa-bookmark"></i> <span data-i18n="websocket.label.url"></span></label>
         <input id="node-config-input-path" type="text" placeholder="ws://example.com/ws">
     </div>
+    <div class="form-row node-config-row-tls hide">
+        <label for="node-config-input-tls" data-i18n="httpin.tls-config"></label>
+        <input type="text" id="node-config-input-tls">
+    </div>
+
     <div class="form-row">
-        <label for="node-config-input-wholemsg">&nbsp;</label>
+        <label for="node-config-input-wholemsg" data-i18n="websocket.sendrec"></label>
         <select type="text" id="node-config-input-wholemsg" style="width: 70%;">
             <option value="false" data-i18n="websocket.payload"></option>
             <option value="true" data-i18n="websocket.message"></option>
diff --git a/nodes/core/io/22-websocket.js b/nodes/core/io/22-websocket.js
index a210cdcee..0d23ab84c 100644
--- a/nodes/core/io/22-websocket.js
+++ b/nodes/core/io/22-websocket.js
@@ -34,10 +34,18 @@ module.exports = function(RED) {
         // match absolute url
         node.isServer = !/^ws{1,2}:\/\//i.test(node.path);
         node.closing = false;
+        node.tls = n.tls;
 
         function startconn() {    // Connect to remote endpoint
             node.tout = null;
-            var socket = new ws(node.path);
+            var options = {};
+            if (node.tls) {
+                var tlsNode = RED.nodes.getNode(node.tls);
+                if (tlsNode) {
+                    tlsNode.addTLSOptions(options);
+                }
+            }
+            var socket = new ws(node.path,options);
             socket.setMaxListeners(0);
             node.server = socket; // keep for closing
             handleConnection(socket);
diff --git a/nodes/core/locales/en-US/messages.json b/nodes/core/locales/en-US/messages.json
index e373d8cf5..369b49327 100644
--- a/nodes/core/locales/en-US/messages.json
+++ b/nodes/core/locales/en-US/messages.json
@@ -400,8 +400,9 @@
         },
         "listenon": "Listen on",
         "connectto": "Connect to",
-        "payload": "Send/Receive payload",
-        "message": "Send/Receive entire message",
+        "sendrec": "Send/Receive",
+        "payload": "payload",
+        "message": "entire message",
         "tip": {
             "path1": "By default, <code>payload</code> will contain the data to be sent over, or received from a websocket. The listener can be configured to send or receive the entire message object as a JSON formatted string.",
             "path2": "This path will be relative to ",