routing re-enabled

[henge/kiak.git] / main.js

diff --git a/main.js b/main.js
index 2db2240..6a631af 100644 (file)
--- a/main.js
+++ b/main.js
@@ -1,50 +1,112 @@
+/**
+ * @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 https = require('https')
+const path = require('path')
 const http = require('http')
+const https = require('https')
 const getport = require('get-port')
-const argv = require('minimist')(process.argv.slice(2))
+const mime = require('mime')
+const opts = require('./opts.js')
 
-const https_router_opts = {
-  key: fs.readFileSync('stunnel.key'),
-  cert: fs.readFileSync('stunnel.cert')
-}
+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 = {}
 
-let https_routes = {}
-const https_router = https.createServer(https_router_opts, (request, response) => {
-  let ht_argv = request.url.slice(1).split("?")
-  console.log(ht_argv)
-  if (ht_argv[0] in https_routes) {
-    response.writeHead(200, { 'Content-Type': 'text/plain' })
-    response.write('You are a remote client\r\n')
-    let route = https_routes[ht_argv[0]]
-    response.write('You should connect to ' + route.host + "\r\nOn port: " + route.port + "\r\n")
-    response.end()
+const httpsOpts = (opts['no-tls'] ?
+                  undefined      :
+                  {
+                    key: fs.readFileSync(opts['ca-key']),
+                    cert: fs.readFileSync(opts['ca-cert'])
+                  })
+
+const routeConnection = (request,response) => {
+  const serveFile = (fPath) => {
+    fs.readFile(fPath, { encoding: 'utf8' }, (err, data) => {
+      if (err || data == undefined) {
+       response.write(404)
+       response.end()
+      }
+      else {
+       response.write(200, mime.lookup(fPath))
+       response.write(data)
+       response.end()
+      }
+    })
   }
-  else if (ht_argv[0].indexOf(".") == -1) {
-    https_routes[ht_argv[0]] = 'true'
+  const htArgv = request.url.slice(1).split("?")
+  console.log(htArgv)
+  let routePath = htArgv[0].split(path.sep)
+  let routeName = routePath[0]
+  if (routeName === '')
+    routeName = opts['index']
+  if (routeName in opts['bindings']) {
+    const followBind = (dir, fPaths) => {
+      fs.readdir(dir, (err, file) => {
+       if (err)
+         serveFile(dir)
+       else if (fPaths.length == 0)
+         serveFile(`${dir}/index.html`)
+       else if (fPaths[0] !== '..' || path.normalize(`${dir}${path.sep}..`).includes(path.normalize(opts['bindings'][routeName]))) {
+         let nextFPath = fPaths.shift()
+         if (nextFPath in fPaths)
+           followBind(`${dir}${path.sep}${nextFPath}`, fPaths)
+         else {
+           response.writeHead(404)
+           response.end()
+         }
+       }
+       else {
+         console.log(`SEC: Ignored '..' in URL ${request.url}`)
+         console.log(request)
+       }
+      })
+    }
+    followBind(opts['bindings'][routeName], htArgv[0].split(path.sep).slice(1))
+  }
+  else if (routeName in routes) {
+    const route = routes[routeName]
     response.writeHead(200, { 'Content-Type': 'text/html' })
-    let new_route = {}
-    new_route.host = request.headers['x-forwarded-for'] || request.connection.remoteAddress
+    response.write(`${skelPage[0]}${clientJS}${skelPage[1]}`)
+    response.end()
+    route.socket.send(request.headers['x-forwarded-for'] || request.connection.remoteAddress)
+  }
+  else {
+    routes[htArgv[0]] = true
+    const newRoute = {}
+    newRoute.host = request.headers['x-forwarded-for'] || request.connection.remoteAddress
     getport().then( (port) => {
-      new_route.port = port
-      new_route.httpd = https.createServer(https_router_opts, (request, response) => {
-      }).listen(port)
-      new_route.ws = new ws.Server( { server: new_route.httpd } )
-      new_route.ws.on('connection', (ws) => { console.log("socket connected"); ws.send("CONNECTED") } )
-      new_route.ws.on('message', (msg) => { console.log("Received message" + msg) })
-      console.log("Listening for websocket socket " + new_route.port + " on " + new_route.host)
-      console.log(new_route)
-      https_routes[ht_argv[0]] = new_route
+      newRoute.port = port
+      if (opts['no-tls'])
+       newRoute.httpd = http.createServer()
+      else
+       newRoute.httpd = https.createServer(httpsOpts)
+      newRoute.httpd.listen(newRoute.port)
+      newRoute.ws = new ws.Server( { server: newRoute.httpd } )
+      newRoute.ws.on('connection', (ws) => { console.log("socket connected"); newRoute.socket = ws; ws.send("CONNECTED") } )
+      newRoute.ws.on('message', (msg) => { console.log("Received message" + msg) })
+      console.log("Listening for websocket " + newRoute.host + " on port " + newRoute.port)
+      routes[htArgv[0]] = newRoute
     }).then(() => {
-      let str = String(fs.readFileSync('remote-server.html'))
-      response.write(str.replace("$HOST","www.strapp.io").replace("$PORT",new_route.port))
+      response.writeHead(200, { 'Content-Type': 'text/html' })
+      response.write(skelPage[0] + 'const _strapp_port = \'' + newRoute.port + '\'\n' + hostJS + skelPage[1])
       response.end()
     })
   }
-}).listen(2443)
-
+}
 
+const router = ((opts['no-tls']) ?
+               http.createServer(routeConnection) :
+               https.createServer(httpsOpts, routeConnection))
+      .listen(opts['port'])
 
-if ("electron" in process.versions);
+//TODO: if ("electron" in process.versions) open a local renderwindow, and route to it