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 064e2c2..b2d4c7a 100644 (file)
--- a/main.js
+++ b/main.js
-const electron = require('electron')
-const http = require('http')
-const path = require('path')
-const url = require('url')
-const fs = require('fs')
-const app = electron.app
-const BrowserWindow = electron.BrowserWindow
-const ipcMain = electron.ipcMain
-
-let mainWindow
-
-function beginRendererProcess() {
-  mainWindow = new BrowserWindow({show: true, webPreferences: { webSecurity: false, experimentalFeatures: true}})
-  mainWindow.loadURL(url.format({
-    pathname: path.join(__dirname, 'renderer.html'),
-    protocol: 'file:',
-    slashes: true
-  }))
-  mainWindow.webContents.openDevTools()
-}
-
-function captureStream() {
-  mainWindow.webContents.send('defaultStreamRequest', 'slr')
-  let p = new Promise((resolve,reject) => {
-    ipcMain.on('defaultStreamResponse', (event, arg) => {
-      console.log('received defaultStreamResponse')
-      resolve(arg)
-    })
-  })
-
-  return p
-}
-
-
-const routes = {
-  '/': function root(response) {
-    response.write(fs.readFileSync('admin.html'))
-    response.end()
-  },
-  '/admin.js': function js(response) {
-    response.write(fs.readFileSync('admin.js'))
-    response.end()
-  },
-  '/admin.css': function css(response) {
-    response.write(fs.readFileSync('admin.css'))
-    response.end()
-  },
-  '/favicon.ico': function favicon() {
-    return 'strapp'
-    response.end()
-  },
-  '/streamlist': function streamlist(response) {
-    let mediaStream = captureStream().then((stream) => {
-      console.log(stream)
-    })
-
-    //response.write(mediaStream)
-  }
-}
-
-app.on('ready', function() {
-  beginRendererProcess()
-
-
-  http.createServer(function(request, response) {
-    const url = request.url
-    console.log('routing to ' + url);
-    routes[url](response)
-
-  }).listen(5140);
-});
+/**\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