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 jk software 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 skelPage  = fs.readFileSync('./skel.html', { encoding: 'utf8' }).split('<!--STRAPP_SRC-->')
  20 const clientJS  = fs.readFileSync(opts['client-js'])
  21 const hostJS    = fs.readFileSync(opts['host-js'])
  22 const routes    = {}
  23 const httpsOpts = {}
  24 if (!opts['no-tls']) {
  25   fs.readFile(opts['ca-key'], { encoding: 'utf8' }, (err, data) => {
  26     if (err) {
  27       console.log(`WARN: Key ${opts['ca-key']} not accessible, tls will fail`)
  28       httpsOpts.key = ''
  29     }
  30     else
  31       httpsOpts.key = data
  32   })
  33   fs.readFile(opts['ca-cert'], { encoding: 'utf8' }, (err, data) => {
  34     if (err) {
  35       console.log(`WARN: Cert ${opts['ca-cert']} not accessible, tls will fail`)
  36       httpsOpts.cert = ''
  37     }
  38     else
  39       httpsOpts.cert = data
  40   })
  41 }
  42
  43 const routeConnection = (request,response) => {
  44   const serveFile = (fPath) => {
  45     fs.readFile(fPath, { encoding: 'utf8' }, (err, data) => {
  46       if (err || data == undefined) {
  47         response.writeHead(404)
  48         response.end()
  49       }
  50       else {
  51         response.writeHead(200, { 'Content-Type': mime.lookup(fPath) })
  52         response.write(data)
  53         response.end()
  54       }
  55     })
  56   }
  57   const htArgv = request.url.slice(1).split("?")
  58   let routePath = htArgv[0].split('/')
  59   let routeName = routePath[0]
  60   if (routeName === '' || routeName === 'index.html')
  61     serveFile(opts['index'])
  62   else if (routeName in opts['bindings']) {
  63     let localPath = path.normalize(opts['bindings'][routeName].concat(path.sep + routePath.slice(1).join(path.sep)))
  64     if (localPath.includes(opts['bindings'][routeName])) {
  65       fs.readdir(localPath, (err, files) => {
  66         if (err)
  67           serveFile(localPath)
  68         else
  69           serveFile(`${localPath}/index.html`)
  70       })
  71     }
  72     else {
  73       console.log(`SEC: ${localPath} references files not in route`)
  74     }
  75   }
  76   else if (routeName in routes) {
  77     const route = routes[routeName]
  78     response.writeHead(200, { 'Content-Type': 'text/html' })
  79     response.write(`${skelPage[0]}${clientJS}${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     route.socket.send(request.headers['x-forwarded-for'] || request.connection.remoteAddress)
  84   }
  85   else {
  86     routes[routeName] = true
  87     const newRoute = {}
  88     newRoute.host = request.headers['x-forwarded-for'] || request.connection.remoteAddress
  89     getport().then( (port) => {
  90       newRoute.port = port
  91       if (opts['no-tls'])
  92         newRoute.httpd = http.createServer()
  93       else
  94         newRoute.httpd = https.createServer(httpsOpts)
  95       newRoute.httpd.listen(newRoute.port)
  96       newRoute.wsd = new ws.Server( { server: newRoute.httpd } )
  97       newRoute.wsd.on('connection', (sock) => {
  98         newRoute.socket = sock
  99         sock.on('message', (msg) => { console.log(`[${newRoute.host}] ${msg}`) })
 100       })
 101       console.log(`Listening for websocket ${newRoute.host} on port ${newRoute.port}`)
 102       routes[routeName] = newRoute
 103     }).then(() => {
 104       response.writeHead(200, { 'Content-Type': 'text/html' })
 105       response.write(`${skelPage[0]}const _strapp_port = ${newRoute.port}\n${hostJS}\n${skelPage[1]}`)
 106       response.end()
 107     })
 108   }
 109 }
 110
 111 const router = ((opts['no-tls']) ?
 112                 http.createServer(routeConnection) :
 113                 https.createServer(httpsOpts, routeConnection))
 114       .listen(opts['port'])
 115
 116 //TODO: if ("electron" in process.versions) open a local renderwindow, and route to it
 117