failover for certs and keys

[henge/kiak.git] / main.js

diff --git a/main.js b/main.js
index 4ae47da..8e01ae3 100644 (file)
--- a/main.js
+++ b/main.js
@@ -1,58 +1,117 @@
-const electron = require('electron')
-const app = electron.app
-const url = require('url')
-const WebSocket = require('ws')
-const http = require('http')
+/**
+ * @file      Node entry and main driver
+ * @author    Jordan Lavatai, Ken Grimes
+ * @version   0.0.1
+ * @license   AGPL-3.0
+ * @copyright jk software 2017
+ * @summary   HTTP(S) Router that uses the first directory in the requested URL
+ *            as the route name
+ */
 const fs = require('fs')
+const ws = require('ws')
+const path = require('path')
+const http = require('http')
+const https = require('https')
+const getport = require('get-port')
+const mime = require('mime')
+const opts = require('./opts.js')
 
-const routes = {
-  '/': function root(response) {
-    response.write(fs.readFileSync('host/host.html'))
-    response.end()
-  },
-  '/admin.js': function js(response) {
-    response.write(fs.readFileSync('host/host.js'))
-    response.end()
-  },
-  '/adapter.js': function adapter(response) {
-    response.write(fs.readFileSync('adapter.js'))
-    response.end()
-  },
-  '/favicon.ico': function favicon(response) {
-    response.write('N/A')
-    response.end()
-  },
-  '/streamlist': function streamlist(response) {
-    getDefaultStream()
-  }
+const skelPage  = fs.readFileSync('./skel.html', { encoding: 'utf8' }).split('<!--STRAPP_SRC-->')
+const clientJS  = fs.readFileSync(opts['client-js'])
+const hostJS    = fs.readFileSync(opts['host-js'])
+const routes    = {}
+const httpsOpts = {}
+if (!opts['no-tls']) {
+  fs.readFile(opts['ca-key'], { encoding: 'utf8' }, (err, data) => {
+    if (err) {
+      console.log(`WARN: Key ${opts['ca-key']} not accessible, tls will fail`)
+      httpsOpts.key = ''
+    }
+    else
+      httpsOpts.key = data
+  })
+  fs.readFile(opts['ca-cert'], { encoding: 'utf8' }, (err, data) => {
+    if (err) {
+      console.log(`WARN: Cert ${opts['ca-cert']} not accessible, tls will fail`)
+      httpsOpts.cert = ''
+    }
+    else
+      httpsOpts.cert = data
+  })
 }
 
-const rooms = {}
-
-
-/* TODO: Swap out url.parse with regex? */
-function determineResponse(request) {
-  const clientIP  = request.connection.remoteAddress
-                  || request.socket.remoteAddress
-                  || request.connection.socket.remoteAddress
-                  || req.headers['x-forwarded-for'].split(',')[0]
-  const parsedURL = url.parse(request.url)
-
-  /* Get the room name */
-  const pathname = parsedURL.pathname
-
-
+const routeConnection = (request,response) => {
+  const serveFile = (fPath) => {
+    fs.readFile(fPath, { encoding: 'utf8' }, (err, data) => {
+      if (err || data == undefined) {
+       response.writeHead(404)
+       response.end()
+      }
+      else {
+       response.writeHead(200, { 'Content-Type': mime.lookup(fPath) })
+       response.write(data)
+       response.end()
+      }
+    })
+  }
+  const htArgv = request.url.slice(1).split("?")
+  let routePath = htArgv[0].split('/')
+  let routeName = routePath[0]
+  if (routeName === '' || routeName === 'index.html')
+    serveFile(opts['index'])
+  else if (routeName in opts['bindings']) {
+    let localPath = path.normalize(opts['bindings'][routeName].concat(path.sep + routePath.slice(1).join(path.sep)))
+    if (localPath.includes(opts['bindings'][routeName])) {
+      fs.readdir(localPath, (err, files) => {
+       if (err)
+         serveFile(localPath)
+       else
+         serveFile(`${localPath}/index.html`)
+      })
+    }
+    else {
+      console.log(`SEC: ${localPath} references files not in route`)
+    }
+  }
+  else if (routeName in routes) {
+    const route = routes[routeName]
+    response.writeHead(200, { 'Content-Type': 'text/html' })
+    response.write(`${skelPage[0]}${clientJS}${skelPage[1]}`)
+    response.end()
+    //TODO: if route.socket == undefined: have server delay this send until host connects
+    //      (this happens when a client connects to an active route with no currently-online host)
+    route.socket.send(request.headers['x-forwarded-for'] || request.connection.remoteAddress)
+  }
+  else {
+    routes[routeName] = true
+    const newRoute = {}
+    newRoute.host = request.headers['x-forwarded-for'] || request.connection.remoteAddress
+    getport().then( (port) => {
+      newRoute.port = port
+      if (opts['no-tls'])
+       newRoute.httpd = http.createServer()
+      else
+       newRoute.httpd = https.createServer(httpsOpts)
+      newRoute.httpd.listen(newRoute.port)
+      newRoute.wsd = new ws.Server( { server: newRoute.httpd } )
+      newRoute.wsd.on('connection', (sock) => {
+       newRoute.socket = sock
+       sock.on('message', (msg) => { console.log(`[${newRoute.host}] ${msg}`) })
+      })
+      console.log(`Listening for websocket ${newRoute.host} on port ${newRoute.port}`)
+      routes[routeName] = newRoute
+    }).then(() => {
+      response.writeHead(200, { 'Content-Type': 'text/html' })
+      response.write(`${skelPage[0]}const _strapp_port = ${newRoute.port}\n${hostJS}\n${skelPage[1]}`)
+      response.end()
+    })
+  }
 }
 
-app.on('ready', function() {
-  console.log('create server')
-  http.createServer(function(request, response) {
-
-    let route = determineResponse(request)
-
-    console.log('routing to ' + route)
-    routes[route](response)
+const router = ((opts['no-tls']) ?
+               http.createServer(routeConnection) :
+               https.createServer(httpsOpts, routeConnection))
+      .listen(opts['port'])
 
-  }).listen(5140);
+//TODO: if ("electron" in process.versions) open a local renderwindow, and route to it
 
-});