* @author Jordan Lavatai, Ken Grimes
* @version 0.0.1
* @license AGPL-3.0
- * @copyright jk software 2017
+ * @copyright loljk 2017
* @summary HTTP(S) Router that uses the first directory in the requested URL
* as the route name
*/
const mime = require('mime')
const opts = require('./opts.js')
-const skelPage = fs.readFileSync('./skel.html', { encoding: 'utf8' }).split('<!--STRAPP_SRC-->')
-const clientJS = fs.readFileSync(opts['client-js'])
-const hostJS = fs.readFileSync(opts['host-js'])
-const routes = {}
-
-const httpsOpts = (opts['no-tls'] ?
- undefined :
- {
- key: fs.readFileSync(opts['ca-key']),
- cert: fs.readFileSync(opts['ca-cert'])
- })
-
-const routeConnection = (request,response) => {
- const serveFile = (fPath) => {
- fs.readFile(fPath, { encoding: 'utf8' }, (err, data) => {
- if (err || data == undefined) {
- response.write(404)
- response.end()
- }
- else {
- response.write(200, mime.lookup(fPath))
- response.write(data)
- response.end()
- }
- })
- }
- const htArgv = request.url.slice(1).split("?")
- console.log(htArgv)
- let routePath = htArgv[0].split(path.sep)
- let routeName = routePath[0]
- if (routeName === '')
- routeName = opts['index']
- if (routeName in opts['bindings']) {
- const followBind = (dir, fPaths) => {
- fs.readdir(dir, (err, file) => {
- if (err)
- serveFile(dir)
- else if (fPaths.length == 0)
- serveFile(`${dir}/index.html`)
- else if (fPaths[0] !== '..' || path.normalize(`${dir}${path.sep}..`).includes(path.normalize(opts['bindings'][routeName]))) {
- let nextFPath = fPaths.shift()
- if (nextFPath in fPaths)
- followBind(`${dir}${path.sep}${nextFPath}`, fPaths)
- else {
- response.writeHead(404)
- response.end()
- }
+const router = {
+ skelPage: fs.readFileSync('./skel.html', { encoding: 'utf8' }).split('<!--STRAPP_SRC-->'),
+ clientJS: fs.readFileSync(opts['client-js']),
+ hostJS: fs.readFileSync(opts['host-js']),
+ routes: {},
+ httpsOpt: undefined,
+ httpd: undefined,
+ wsProtocol: opts['no-tls'] ? 'ws' : 'wss',
+ respond: (request,response) => {
+ const serveFile = (fPath) => {
+ fs.readFile(fPath, { encoding: 'utf8' }, (err, data) => {
+ if (err || data == undefined) {
+ response.writeHead(404)
+ response.end()
}
else {
- console.log(`SEC: Ignored '..' in URL ${request.url}`)
- console.log(request)
+ response.writeHead(200, { 'Content-Type': mime.lookup(fPath) })
+ response.write(data)
+ response.end()
}
})
}
- followBind(opts['bindings'][routeName], htArgv[0].split(path.sep).slice(1))
- }
- 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
- if (opts['no-tls'])
- newRoute.httpd = http.createServer()
- else
- newRoute.httpd = https.createServer(httpsOpts)
- newRoute.httpd.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(() => {
+ const htArgv = request.url.slice(1).split("?")
+ let routePath = htArgv[0].split('/')
+ let routeName = routePath[0]
+ if (routeName === '' || routeName === 'index.html')
+ serveFile(opts['index'])
+ else if (routeName in opts['bindings']) {
+ let localPath = path.normalize(opts['bindings'][routeName].concat(path.sep + routePath.slice(1).join(path.sep)))
+ if (localPath.includes(opts['bindings'][routeName])) {
+ fs.readdir(localPath, (err, files) => {
+ if (err)
+ serveFile(localPath)
+ else
+ serveFile(`${localPath}/index.html`)
+ })
+ }
+ else {
+ console.log(`SEC: ${localPath} references files not in route`)
+ }
+ }
+ else if (routeName in router.routes) {
+ const route = router.routes[routeName]
response.writeHead(200, { 'Content-Type': 'text/html' })
- response.write(skelPage[0] + 'const _strapp_port = \'' + newRoute.port + '\'\n' + hostJS + skelPage[1])
+ response.write(`${router.skelPage[0]}${router.clientJS}${router.skelPage[1]}`)
response.end()
- })
+ //TODO: if route.socket == undefined: have server delay this send until host connects
+ // (this happens when a client connects to an active route with no currently-online host)
+ route.socket.send(request.headers['x-forwarded-for'] || request.connection.remoteAddress)
+ }
+ else {
+ router.routes[routeName] = true
+ const newRoute = {}
+ newRoute.host = request.headers['x-forwarded-for'] || request.connection.remoteAddress
+ getport().then( (port) => {
+ newRoute.port = port
+ if (opts['no-tls'])
+ newRoute.httpd = http.createServer()
+ else
+ newRoute.httpd = https.createServer(router.httpsOpts)
+ newRoute.httpd.listen(newRoute.port)
+ newRoute.wsd = new ws.Server( { server: newRoute.httpd } )
+ newRoute.wsd.on('connection', (sock) => {
+ newRoute.socket = sock
+ sock.on('message', (msg) => { console.log(`[${newRoute.host}] ${msg}`) })
+ })
+ console.log(`Listening for websocket ${newRoute.host} on port ${newRoute.port}`)
+ router.routes[routeName] = newRoute
+ }).then(() => {
+ response.writeHead(200, { 'Content-Type': 'text/html' })
+ response.write(`${router.skelPage[0]}` +
+ `\tconst _strapp_port = ${newRoute.port}\n` +
+ `\tconst _strapp_protocol = '${router.wsProtocol}'\n` +
+ `${router.hostJS}\n${router.skelPage[1]}`)
+ response.end()
+ })
+ }
+
}
}
-const router = ((opts['no-tls']) ?
- http.createServer(routeConnection) :
- https.createServer(httpsOpts, routeConnection))
- .listen(opts['port'])
+/**
+ * @summary Boot up the router. With TLS, we must wait for file reads to sync.
+ */
+if (!opts['no-tls']) {
+ let filesRead = 0
+ let key = undefined
+ let cert = undefined
+ const syncRead = () => {
+ if (++filesRead == 2) {
+ if (key == undefined)
+ console.log(`ERR: Key ${opts['ca-key']} inaccessible, tls will fail`)
+ if(cert == undefined)
+ console.log(`ERR: Cert ${opts['ca-cert']} inaccessible, tls will fail`)
+ else if (key != undefined) {
+ router.httpsOpts = { cert: cert, key: key}
+ router.httpd = https.createServer(router.httpsOpts, router.respond)
+ .listen(opts['port'])
+ }
+ }
+ }
+ fs.readFile(opts['ca-key'], { encoding: 'utf8' }, (err, data) => {
+ if (!err) key = data
+ syncRead()
+ })
+ fs.readFile(opts['ca-cert'], { encoding: 'utf8' }, (err, data) => {
+ if (!err) cert = data
+ syncRead()
+ })
+}
+else
+ router.httpd = http.createServer(router.respond).listen(opts['port'])
//TODO: if ("electron" in process.versions) open a local renderwindow, and route to it