X-Git-Url: https://git.kengrimes.com/?p=henge%2Fkiak.git;a=blobdiff_plain;f=main.js;h=f123a413fd73593d94bbbce8705586395cac3c90;hp=9153f767c221590606b910689b0b9346613edebf;hb=d33866cf8b1a188ad9ab23a934986bfd1b4726cb;hpb=309f9b9b1dbb1fc2811ea603e4ac67e5395397fb

diff --git a/main.js b/main.js
index 9153f76..f123a41 100644
--- a/main.js
+++ b/main.js
@@ -1,50 +1,156 @@
+/**
+* @file      Node entry and main driver
+* @author    Jordan Lavatai, Ken Grimes
+* @version   0.0.1
+* @license   AGPL-3.0
+* @copyright loljk 2017
+* @summary   HTTP(S) Router that uses the first directory in the requested URL
+*            as the route name
+*/
 const fs = require('fs')
 const ws = require('ws')
-const https = require('https')
+const path = require('path')
 const http = require('http')
+const https = require('https')
 const getport = require('get-port')
-const argv = require('minimist')(process.argv.slice(2))
-
-const https_router_opts = {
-  key: fs.readFileSync('stunnel.key'),
-  cert: fs.readFileSync('stunnel.cert')
-}
+const mime = require('mime')
+const opts = require('./opts.js')
 
-let https_routes = {}
-const https_router = https.createServer(https_router_opts, (request, response) => {
-  let ht_argv = request.url.slice(1).split("?")
-  console.log(ht_argv)
-  if (ht_argv[0] in https_routes) {
-    response.writeHead(200, { 'Content-Type': 'text/plain' })
-    response.write('You are a remote client\r\n')
-    let route = https_routes[ht_argv[0]]
-    response.write('You should connect to ' + route.host + "\r\nOn port: " + route.port + "\r\n")
-    response.end()
-  }
-  else if (ht_argv[0].indexOf(".") == -1) {
-    https_routes[ht_argv[0]] = 'true'
-    response.writeHead(200, { 'Content-Type': 'text/html' })
-    let new_route = {}
-    new_route.host = request.headers['x-forwarded-for'] || request.connection.remoteAddress
-    getport().then( (port) => {
-      new_route.port = port
-      new_route.httpd = https.createServer(https_router_opts, (request, response) => {
-      }).listen(port)
-      new_route.ws = new ws.Server( { server: new_route.httpd } )
-      new_route.ws.on('connection', (ws) => { console.log("socket connected"); ws.send("CONNECTED") } )
-      new_route.ws.on('message', (msg) => { console.log("Received message" + msg) })
-      console.log("Listening for websocket socket " + new_route.port + " on " + new_route.host)
-      console.log(new_route)
-      https_routes[ht_argv[0]] = new_route
-    }).then(() => {
-      let str = String(fs.readFileSync('remote-server.html'))
-      response.write(str.replace("$HOST","wss://www.strapp.io").replace("$PORT",new_route.port))
-      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) => {
+    let body = []
+    request.on('error', function(err) {
+      console.error(`error is ${err}`);
+    }).on('data', function(chunk) {
+      console.log(`chunk is ${chunk}`)
+      body.push(chunk);
+    }).on('end', function() {
+      console.log(`body is ${body}`)
     })
-  }
-}).listen(2443)
+    console.log('server handling request')
+    const serveFile = (fPath) => {
+      fs.readFile(fPath, { encoding: 'utf8' }, (err, data) => {
+        if (err || data == undefined) {
+          response.writeHead(404)
+          response.end()
+        }
+        else {
+          response.writeHead(200, { 'Content-Type': mime.lookup(fPath) })
+          response.write(data)
+          response.end()
+        }
+      })
+    }
+    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`)
+      }
+    }
+    /* TODO: Handle reconnecting host */
+    else if (routeName in router.routes) {
+
+      const route = router.routes[routeName]
 
+      /* Client is INIT GET */
+      if (request.headers['x-strapp-type'] == undefined) {
+        console.log('client init GET')
+        response.writeHead(200, { 'Content-Type': 'text/html' })
+        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)
+      }
+      else { /* Client sent offer, waiting for answer */
+        console.log(JSON.parse(request.headers['x-strapp-type']))
+        route.socket.on('message', (hostResponse) => {
+          console.log(hostResponse)
+        })
+      }
 
+    }
+    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()
+        })
+      }
 
-if ("electron" in process.versions);
+    }
+  }
+
+  /**
+  * @summary Boot up the router.  With TLS, we must wait for file reads to sync.
+  */
+  if (!opts['no-tls']) {
+    console.log('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