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