Add https support

Cyclonecode · Cyclonecode · commit bc4ebd63e915 · 2019-08-17T22:23:29.000+02:00
diff --git a/.eslintrc.js b/.eslintrc.js
@@ -173,7 +173,7 @@ module.exports = {
                 "allowForLoopAfterthoughts": true
             }
         ],
-        "no-process-env": "error",
+        "no-process-env": "off",
         "no-process-exit": "off",
         "no-proto": "error",
         "no-prototype-builtins": "error",
@@ -234,7 +234,7 @@ module.exports = {
         "prefer-const": "off",
         "prefer-destructuring": "off",
         "prefer-numeric-literals": "error",
-        "prefer-object-spread": "error",
+        "prefer-object-spread": "off",
         "prefer-promise-reject-errors": "error",
         "prefer-reflect": "error",
         "prefer-rest-params": "error",
@@ -294,4 +294,4 @@ module.exports = {
             "never"
         ]
     }
-};
+};
diff --git a/README.md b/README.md
@@ -112,6 +112,20 @@ Start the server at a specific address and port:
 
 By default the server is running on port **2048** and the websocket on port **8080**.
 
+## SSL
+
+If you like to run the server over https you could use the `ssl`, `key` and `cert` arguments:
+
+    npm run start -- --port 2048 --host example.com --websocket 8080 --ssl=1 --key=./server.key --cert=./server.crt
+
+Another way is to set the following variables in your environment:
+
+    SSL_ENABLED=1
+    SSL_KEY=./server.key
+    SSL_CERTIFICATE=./server.crt
+
+Notice that the ssl key and certificate files need to be readable by the application.
+
 # Testing
 
 You can then test the behavior by running queries against your database:
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -5,6 +5,7 @@
   "main": "server.js",
   "dependencies": {
     "cross-env": "^5.2.0",
+    "https": "^1.0.0",
     "lodash": "^4.17.14",
     "minimist": "^1.2.0",
     "mysql": "^2.17.1",
diff --git a/scripts/server.js b/scripts/server.js
@@ -1,6 +1,5 @@
 #!/usr/bin/env node
 const WebSocketServer = require('websocket').server
-const http = require('http')
 const net = require('net')
 const fs = require('fs')
 
@@ -9,11 +8,26 @@ let connections = []
 
 // parse any arguments
 const argv = require('minimist')(process.argv.slice(2))
+const http = require(process.env.SSL_ENABLED || argv.ssl ? 'https' : 'http')
 
 const SERVER_PORT = argv.port ? parseInt(argv.port) : 2048
 const WEBSOCKET_PORT = argv.websocket ? parseInt(argv.websocket) : 8080
 const SERVER_ADDR = argv.host ? argv.host.replace(/['"]+/g, '') : '127.0.0.1'
 
+const config = Object.assign({}, argv, process.env)
+
+let credentials = {}
+
+if (config.SSL_ENABLED || config.ssl) {
+  const privateKey = fs.readFileSync(config.SSL_KEY || config.key, 'utf8')
+  const certificate = fs.readFileSync(config.SSL_CERTIFICATE || config.cert, 'utf8')
+
+  credentials = {
+    key: privateKey,
+    cert: certificate
+  }
+}
+
 // create a listening socket
 net.createServer((sock) => {
   sock.on('data', (data) => {
@@ -31,7 +45,7 @@ net.createServer((sock) => {
 }).listen(SERVER_PORT)
 
 // create a http server
-const server = http.createServer((request, response) => {
+const server = http.createServer(credentials,(request, response) => {
   console.log((new Date()) + ' Received request for ' + request.url)
   response.writeHead(404)
   response.end()
@@ -56,7 +70,8 @@ function originIsAllowed (origin) {
 }
 
 function createTemplate () {
-  var output = '<!DOCTYPE html>\n' +
+  const protocol = config.SSL_ENABLED || argv.ssl ? 'wss' : 'ws'
+  const output = '<!DOCTYPE html>\n' +
         '<html>\n' +
         '    <head>\n' +
         '        <title>Example of a user defined function (UDF) in MySQL</title>\n' +
@@ -68,7 +83,7 @@ function createTemplate () {
         '                crossorigin="anonymous"></script>\n' +
         '        <script>\n' +
         '            $(document).ready(function() {\n' +
-        '               var ws = new WebSocket(\'ws://' + SERVER_ADDR + ':' + WEBSOCKET_PORT + '\', \'echo-protocol\');\n' +
+        '               var ws = new WebSocket(\'' + protocol + '://' + SERVER_ADDR + ':' + WEBSOCKET_PORT + '\', \'echo-protocol\');\n' +
         '               ws.onmessage = function(event) {\n' +
         '                   $(\'body\').append(event.data + "<br />");\n' +
         '               };\n' +
