main.js

   1 /**
   2 * @file      Node entry and main driver
   3 * @author    Jordan Lavatai, Ken Grimes
   4 * @version   0.0.1
   5 * @license   AGPL-3.0
   6 * @copyright loljk 2017
   7 * @summary   HTTP(S) Router that uses the first directory in the requested URL
   8 *            as the route name
   9 */
  10 const fs = require('fs')
  11 const ws = require('ws')
  12 const path = require('path')
  13 const http = require('http')
  14 const https = require('https')
  15 const getport = require('get-port')
  16 const mime = require('mime')
  17 const opts = require('./opts.js')
  18
  19 const router = {
  20   skelPage:   fs.readFileSync('./skel.html', { encoding: 'utf8' }).split('<!--STRAPP_SRC-->'),
  21   clientJS:   fs.readFileSync(opts['client-js']),
  22   hostJS:     fs.readFileSync(opts['host-js']),
  23   routes:     {},
  24   httpsOpt:   undefined,
  25   httpd:      undefined,
  26   wsProtocol: opts['no-tls'] ? 'ws' : 'wss',
  27   respond:    (request,response) => {
  28     let body = []
  29     request.on('error', function(err) {
  30       console.error(`error is ${err}`);
  31     }).on('data', function(chunk) {
  32       console.log(`chunk is ${chunk}`)
  33       body.push(chunk);
  34     }).on('end', function() {
  35       console.log(`body is ${body}`)
  36     })
  37     console.log('server handling request')
  38     const serveFile = (fPath) => {
  39       fs.readFile(fPath, { encoding: 'utf8' }, (err, data) => {
  40         if (err || data == undefined) {
  41           response.writeHead(404)
  42           response.end()
  43         }
  44         else {
  45           response.writeHead(200, { 'Content-Type': mime.lookup(fPath) })
  46           response.write(data)
  47           response.end()
  48         }
  49       })
  50     }
  51     const htArgv = request.url.slice(1).split("?")
  52     let routePath = htArgv[0].split('/')
  53     let routeName = routePath[0]
  54     if (routeName === '' || routeName === 'index.html')
  55       serveFile(opts['index'])
  56     else if (routeName in opts['bindings']) {
  57       let localPath = path.normalize(opts['bindings'][routeName].concat(path.sep + routePath.slice(1).join(path.sep)))
  58       if (localPath.includes(opts['bindings'][routeName])) {
  59         fs.readdir(localPath, (err, files) => {
  60           if (err)
  61           serveFile(localPath)
  62           else
  63           serveFile(`${localPath}/index.html`)
  64         })
  65       }
  66       else {
  67         console.log(`SEC: ${localPath} references files not in route`)
  68       }
  69     }
  70     /* TODO: Handle reconnecting host */
  71     else if (routeName in router.routes) {
  72
  73       const route = router.routes[routeName]
  74
  75       /* Client is INIT GET */
  76       if (request.headers['x-strapp-type'] !== 'o') {
  77         console.log('client init GET')
  78         response.writeHead(200, { 'Content-Type': 'text/html' })
  79         response.write(`${router.skelPage[0]}${router.clientJS}${router.skelPage[1]}`)
  80         response.end()
  81         //TODO: if route.socket == undefined: have server delay this send until host connects
  82         //      (this happens when a client connects to an active route with no currently-online host)
  83       }
  84       else { /* Client sent offer, waiting for answer */
  85         console.log('client offer/answer GET')
  86         //route.socket.send(JSON.parse(body.join('')))
  87         route.socket.on('message', (hostResponse) => {
  88           console.log(hostResponse)
  89         })
  90       }
  91
  92     }
  93     else {
  94       router.routes[routeName] = true
  95       const newRoute = {}
  96       newRoute.host = request.headers['x-forwarded-for'] || request.connection.remoteAddress
  97       getport().then( (port) => {
  98         newRoute.port = port
  99         if (opts['no-tls'])
 100         newRoute.httpd = http.createServer()
 101         else
 102         newRoute.httpd = https.createServer(router.httpsOpts)
 103         newRoute.httpd.listen(newRoute.port)
 104         newRoute.wsd = new ws.Server( { server: newRoute.httpd } )
 105         newRoute.wsd.on('connection', (sock) => {
 106           newRoute.socket = sock
 107           sock.on('message', (msg) => { console.log(`[${newRoute.host}] ${msg}`) })
 108         })
 109         console.log(`Listening for websocket ${newRoute.host} on port ${newRoute.port}`)
 110         router.routes[routeName] = newRoute
 111       }).then(() => {
 112         response.writeHead(200, { 'Content-Type': 'text/html' })
 113         response.write(`${router.skelPage[0]}` +
 114           `\tconst _strapp_port = ${newRoute.port}\n` +
 115           `\tconst _strapp_protocol = '${router.wsProtocol}'\n` +
 116           `${router.hostJS}\n${router.skelPage[1]}`)
 117           response.end()
 118         })
 119       }
 120
 121     }
 122   }
 123
 124   /**
 125   * @summary Boot up the router.  With TLS, we must wait for file reads to sync.
 126   */
 127   if (!opts['no-tls']) {
 128     console.log('tls')
 129     let filesRead = 0
 130     let key = undefined
 131     let cert = undefined
 132     const syncRead = () => {
 133       if (++filesRead == 2) {
 134         if (key == undefined)
 135         console.log(`ERR: Key ${opts['ca-key']} inaccessible, tls will fail`)
 136         if(cert == undefined)
 137         console.log(`ERR: Cert ${opts['ca-cert']} inaccessible, tls will fail`)
 138         else if (key != undefined) {
 139           router.httpsOpts = { cert: cert, key: key}
 140           router.httpd = https.createServer(router.httpsOpts, router.respond)
 141           .listen(opts['port'])
 142         }
 143       }
 144     }
 145     fs.readFile(opts['ca-key'], { encoding: 'utf8' }, (err, data) => {
 146       if (!err) key = data
 147       syncRead()
 148     })
 149     fs.readFile(opts['ca-cert'], { encoding: 'utf8' }, (err, data) => {
 150       if (!err) cert = data
 151       syncRead()
 152     })
 153   }
 154   else
 155   router.httpd = http.createServer(router.respond).listen(opts['port'])
 156
 157   //TODO: if ("electron" in process.versions) open a local renderwindow, and route to it