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 b24e9f3..b2d4c7a 100644 (file)
--- a/main.js
+++ b/main.js
-const fs = require('fs')
-const ws = require('ws')
-const https = require('https')
-const getport = require('get-port')
-const mime = require('mime')
-
-const argv = require('minimist')(process.argv.slice(2), {
-  string:  [ 'ca-cert', 'ca-key', 'config', 'client-js', 'host-js', 'electron', 'port' ],
-  boolean: [ 'remote-host' ],
-  alias:   { c: 'config',
-            j: 'client-js',
-            J: 'host-js',
-            C: 'ca-cert',
-            K: 'ca-key',
-            e: 'electron',
-            r: 'remote-host',
-            p: 'port'
-          },
-  default: { config:        undefined,
-            'client-js':   'client.js',
-            'host-js':     'host.js',
-            'ca-cert':     'stunnel.cert',
-            'ca-key':      'stunnel.key',
-            'remote-host': true,
-            'port':        2443
-          },
-  stopEarly: true,
-  unknown: (opt) => {
-    console.log(process.argv.join(" ") + '\nUnknown operator: ' + opt + `
-Usage: strapp [OPTION]...  
-Route https connections from a hardware port to a remote host, and initiate
-peer-to-peer connection with clients.
-
-CONFIG
-  -c, --config=path         Configuration file to use (/etc/strapp.conf)
-                            - overridden by command line opts
-  -j, --client-js=path      Path to the client Strapp code (./strapp-client.js)
-  -J, --host-js=path        Path to the host Strapp code (./strapp-host.js)
-  -C, --ca-cert=path        Accessible location of the CA Cert (./stunnel.cert)
-  -K, --ca-key=path         Accessible location of the CA Key (./stunnel.key)
-  -p, --port=number         The local port to bind HTTPS listener to (2443)
-
-ROUTING
-  -e, --electron=route      Route to the local electron user (nil)
-                            - enables optional electron dependency
-  -d, --dedicated=route     Route all incoming connections to this route (nil)
-                            - used in conjunction with '-e=my_route'
-
-(c)2017 jk software
-`)
-    process.exit()
-  }
-})
-
-//TODO: if (argv['config'] !== undefined), read and apply to argv without overwrite
-if (argv['www-path'] == undefined)
-  argv['www-path'] = 'www'
-else if (String(argv['www-path']).endsWith('/'))
-  argv['www-path'] = argv['www-path'].slice(0,-1)
-
-const routerOpts = {
-  key: fs.readFileSync(argv['ca-key']),
-  cert: fs.readFileSync(argv['ca-cert'])
-}
-const skelPage = String(fs.readFileSync('skel.html')).split("<!--STRAPP-->")
-const clientJS = fs.readFileSync(argv['client-js'])
-const hostJS   = fs.readFileSync(argv['host-js'])
-const routes   = {}
-const fileBuf  = {}
-let   fbSize   = 0
-fs.readdirSync(argv['www-path']).forEach((file) => {
-  if (fbSize++ < 50) {
-    fileBuf[file] = { mime: mime.lookup(argv['www-path'] + '/' + file),
-                     data: fs.readFileSync(argv['www-path'] + '/' + file)
-                   }
-  }
-})
-
-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_host = \'www.strapp.io\'\n\tconst _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