Refactored into strapp Modules. All in progress with File System current focus
[henge/kiak.git] / main.js
diff --git a/main.js b/main.js
index c7d5bd7..b2d4c7a 100644 (file)
--- a/main.js
+++ b/main.js
-/**
- * @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 http = require('http')
-const https = require('https')
-const getport = require('get-port')
-const mime = require('mime')
-const opts = require('./opts.js')
-
-/* const routeConnection = (request,response) => {
-  const htArgv = request.url.slice(1).split("?")
-  let   routeName = /[^\/]*/.match(htArgv[0])[0]
-  if (routeName === '')
-    routeName = opts['index']
-  if (routeName.indexOf('.') != -1) {
-    if (routeName in fileBuf) {
-      response.writeHead(200, { 'Content-Type': fileBuf[routeName].mime })
-      response.write(fileBuf[routeName].data)
-    }
-    else {
-      if ('404.html' in fileBuf) {
-       response.writeHead(404, fileBuf['404.html'].mime)
-       response.write(fileBuf['404.html'].data)
-      }
-      else
-       response.writeHead(404)
-    }
-    response.end()
-  }
-  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()
-    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) => {
-      newRoute.port = port
-      newRoute.httpd = https.createServer(routerOpts, (request, response) => {
-      }).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(() => {
-      response.writeHead(200, { 'Content-Type': 'text/html' })
-      response.write(skelPage[0] + 'const _strapp_port = \'' + newRoute.port + '\'\n' + hostJS + skelPage[1])
-      response.end()
-    })
-  }
-}
-
-if (!opts['no-tls']) {
-  router.httpd = http.createServer(router.listener)
-}
-else {
-  /* Setup TLS */
-  if (!fs.existsSync(opts['ca-key'])) {
-    console.log(`ERR: Key ${opts['ca-key']} inaccessible (required for tls)`)
-    process.exit()
-  }
-  if (!fs.existsSync(opts['ca-cert'])) {
-    console.log(`ERR: Cert ${opts['ca-cert']} inaccessible (required for tls)`)
-    process.exit()
-  }
-  router.opts = {
-    key: fs.readFileSync(opts['ca-key']),
-    cert: fs.readFileSync(opts['ca-cert'])
-  }
-}
-
-const skelPage = String(fs.readFileSync('skel.html')).split("<!--STRAPP_SRC-->")
-const clientJS = fs.readFileSync(opts['client-js'])
-const hostJS   = fs.readFileSync(opts['host-js'])
-const routes   = {}
-/*
-const router = https.createServer(routerOpts, (request, response) => {
-  const htArgv = request.url.slice(1).split("?")
-  let   routeName = htArgv[0].split('/')[0]
-  if (routeName === '')
-    routeName = 'index.html'
-  if (routeName.indexOf('.') != -1) {
-    if (routeName in fileBuf) {
-      response.writeHead(200, { 'Content-Type': fileBuf[routeName].mime })
-      response.write(fileBuf[routeName].data)
-    }
-    else {
-      if ('404.html' in fileBuf) {
-       response.writeHead(404, fileBuf['404.html'].mime)
-       response.write(fileBuf['404.html'].data)
-      }
-      else
-       response.writeHead(404)
-    }
-    response.end()
-  }
-  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()
-    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) => {
-      newRoute.port = port
-      newRoute.httpd = https.createServer(routerOpts, (request, response) => {
-      }).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(() => {
-      response.writeHead(200, { 'Content-Type': 'text/html' })
-      response.write(skelPage[0] + 'const _strapp_port = \'' + newRoute.port + '\'\n' + hostJS + skelPage[1])
-      response.end()
-    })
-  }
-}).listen(argv['port'])
-*/
-//TODO: if ("electron" in process.versions) open a local renderwindow, and route to it
-
+/**\r
+* @file      Node entry and main driver\r
+* @author    Jordan Lavatai, Ken Grimes\r
+* @version   0.0.1\r
+* @license   AGPL-3.0\r
+* @copyright loljk 2017\r
+* @summ ary   HTTP(S) Router that uses the first directory in the requested URL\r
+*            as the route name\r
+*/\r
+const fs = require('fs')\r
+const ws = require('ws')\r
+const path = require('path')\r
+const http = require('http')\r
+const https = require('https')\r
+const getport = require('get-port')\r
+const mime = require('mime')\r
+const opts = require('./opts.js')\r
+\r
+const router = {\r
+  skelPage:   fs.readFileSync('./skel.html', { encoding: 'utf8' }).split('<!--STRAPP_SRC-->'),\r
+  clientJS:   fs.readFileSync(opts['client-js']),\r
+  hostJS:     fs.readFileSync(opts['host-js']),\r
+  routes:     {},\r
+  httpsOpt:   undefined,\r
+  httpd:      undefined,\r
+  wsProtocol: opts['no-tls'] ? 'ws' : 'wss',\r
+  respond:    (request,response) => {\r
+    console.log('server handling request')\r
+    const serveFile = (fPath) => {\r
+      fs.readFile(fPath, { encoding: 'utf8' }, (err, data) => {\r
+        if (err || data == undefined) {\r
+          response.writeHead(404)\r
+          response.end()\r
+        }\r
+        else {\r
+          response.writeHead(200, { 'Content-Type': mime.lookup(fPath) })\r
+          response.write(data)\r
+          response.end()\r
+        }\r
+      })\r
+    }\r
+    const htArgv = request.url.slice(1).split("?")\r
+    let routePath = htArgv[0].split('/')\r
+    let routeName = routePath[0]\r
+\r
+\r
+    if (routeName === '' || routeName === 'index.html')\r
+    serveFile(opts['index'])\r
+    else if (routeName in opts['bindings']) {\r
+      let localPath = path.normalize(opts['bindings'][routeName].concat(path.sep + routePath.slice(1).join(path.sep)))\r
+      if (localPath.includes(opts['bindings'][routeName])) {\r
+        fs.readdir(localPath, (err, files) => {\r
+          if (err)\r
+          serveFile(localPath)\r
+          else\r
+          serveFile(`${localPath}/index.html`)\r
+        })\r
+      }\r
+      else {\r
+        console.log(`SEC: ${localPath} references files not in route`)\r
+      }\r
+    }\r
+    /* TODO: Handle reconnecting host */\r
+    else if (routeName in router.routes) {\r
+      const route = router.routes[routeName]\r
+      const clients = route['clients']\r
+      const headerData = request.headers['x-strapp-type']\r
+\r
+\r
+\r
+\r
+      /* Client is INIT GET */\r
+      if (headerData === undefined) {\r
+        console.log('client init GET')\r
+        response.writeHead(200, { 'Content-Type': 'text/html' })\r
+        response.write(`${router.skelPage[0]}${router.clientJS}${router.skelPage[1]}`)\r
+        response.end()\r
+        //TODO: if route.socket == undefined: have server delay this send until host connects\r
+        //      (this happens when a client connects to an active route with no currently-online host)\r
+      }\r
+      else if (headerData.localeCompare('ice-candidate-request') === 0) {\r
+        console.log('Server: received ice-candidate-request from Client ') \r
+        let pubKey = request.headers['x-client-pubkey']\r
+        clients.set(pubKey, response)\r
+        pubKey = '{ "pubKey": "'  + pubKey + '" }'\r
+        route.socket.send(pubKey)\r
+      }\r
+      else if (headerData.localeCompare('ice-candidate-submission') === 0) {\r
+        console.log('Server: recieved ice-candidate-submission from Client')\r
+        let data = []\r
+        request.on('data', (chunk) => {\r
+          data.push(chunk)\r
+        }).on('end', () => {\r
+          console.log('Sending ice-candidate-submission to Host')\r
+          data = Buffer.concat(data).toString();\r
+          clients.set(JSON.parse(data)['pubKey'], response)\r
+          route.socket.send(data)\r
+        })\r
+      }\r
+      else if (headerData.localeCompare('client-sdp-offer') === 0){ /* Client sent offer, waiting for answer */\r
+        console.log('Server: Sending client offer to host')\r
+        clients.set(JSON.parse(request.headers['x-client-offer'])['pubKey'], response)\r
+        route.socket.send(request.headers['x-client-offer'])\r
+      } else {\r
+        console.log('Unhandled stuff')\r
+        console.log(request.headers)\r
+      }\r
+\r
+    }\r
+    else {\r
+      router.routes[routeName] = true\r
+      const newRoute = {}\r
+      newRoute.clients = new Map([])\r
+      newRoute.host = request.headers['x-forwarded-for'] || request.connection.remoteAddress\r
+      getport().then( (port) => {\r
+        newRoute.port = port\r
+        if (opts['no-tls'])\r
+        newRoute.httpd = http.createServer()\r
+        else\r
+        newRoute.httpd = https.createServer(router.httpsOpts)\r
+        newRoute.httpd.listen(newRoute.port)\r
+        newRoute.wsd = new ws.Server( { server: newRoute.httpd } )\r
+        newRoute.wsd.on('connection', (sock) => {\r
+          console.log(`${routeName} server has been established`)\r
+          newRoute.socket = sock\r
+\r
+          /* Handle all messages from host */\r
+          sock.on('message', (hostMessage) => {\r
+            hostMessage = JSON.parse(hostMessage)\r
+            response = newRoute.clients.get(hostMessage['clientPubKey'])\r
+\r
+            /* If the host response is a answer */\r
+            if (hostMessage['cmd'].localeCompare('< sdp pubKey') === 0) {\r
+              console.log('Server: Sending host answer to client')\r
+              response.writeHead(200, { 'Content-Type': 'application/json' })\r
+              response.write(JSON.stringify(hostMessage))\r
+              response.end()\r
+            }\r
+            else if (hostMessage['cmd'].localeCompare('< ice pubKey') === 0){\r
+              /* if the host response is an ice candidate */\r
+              console.log('Server: Handling host ICE message')\r
+              let iceState = hostMessage['iceState']\r
+              /* If there are any ice candidates, send them back */\r
+              switch(iceState) {\r
+                case "a":\r
+                  response.writeHead('200', {'x-strapp-type': 'ice-candidate-available'})\r
+                  response.write(JSON.stringify(hostMessage))\r
+                  response.end()\r
+                  break\r
+                case "g":\r
+                  console.log('Server: Host is still gathering candidates, keep trying')\r
+                  response.writeHead('200', {'x-strapp-type': 'ice-state-gathering'})\r
+                  response.write(JSON.stringify(hostMessage))\r
+                  response.end()\r
+                break\r
+                case "c":\r
+                  console.log('Server: Host has completed gathering candidates')\r
+                  response.writeHead('200', {'x-strapp-type': 'ice-state-complete'})\r
+                  response.write(JSON.stringify(hostMessage))\r
+                  response.end()\r
+                  break\r
+                default:\r
+                  console.log('unhandled iceState from host')\r
+                  break\r
+                }\r
+            }\r
+\r
+          })\r
+        })\r
+\r
+        console.log(`Listening for websocket ${newRoute.host} on port ${newRoute.port}`)\r
+        router.routes[routeName] = newRoute\r
+      }).then(() => {\r
+        response.writeHead(200, { 'Content-Type': 'text/html' })\r
+        response.write(`${router.skelPage[0]}` +\r
+          `\tconst _strapp_port = ${newRoute.port}\n` +\r
+          `\tconst _strapp_protocol = '${router.wsProtocol}'\n` +\r
+          `${router.hostJS}\n${router.skelPage[1]}`)\r
+          response.end()\r
+        })\r
+      }\r
+\r
+\r
+    }\r
+  }\r
+\r
+  /**\r
+  * @summary Boot up the router.  With TLS, we must wait for file reads to sync.\r
+  */\r
+  if (!opts['no-tls']) {\r
+    console.log('tls')\r
+    let filesRead = 0\r
+    let key = undefined\r
+    let cert = undefined\r
+    const syncRead = () => {\r
+      if (++filesRead == 2) {\r
+        if (key == undefined)\r
+        console.log(`ERR: Key ${opts['ca-key']} inaccessible, tls will fail`)\r
+        if(cert == undefined)\r
+        console.log(`ERR: Cert ${opts['ca-cert']} inaccessible, tls will fail`)\r
+        else if (key != undefined) {\r
+          router.httpsOpts = { cert: cert, key: key}\r
+          router.httpd = https.createServer(router.httpsOpts, router.respond)\r
+          .listen(opts['port'])\r
+        }\r
+      }\r
+    }\r
+    fs.readFile(opts['ca-key'], { encoding: 'utf8' }, (err, data) => {\r
+      if (!err) key = data\r
+      syncRead()\r
+    })\r
+    fs.readFile(opts['ca-cert'], { encoding: 'utf8' }, (err, data) => {\r
+      if (!err) cert = data\r
+      syncRead()\r
+    })\r
+  }\r
+  else\r
+  router.httpd = http.createServer(router.respond).listen(opts['port'])\r
+\r
+  //TODO: if ("electron" in process.versions) open a local renderwindow, and route to it\r