rm keys
[henge/kiak.git] / main.js
diff --git a/main.js b/main.js
index 4ae47da..b24e9f3 100644 (file)
--- a/main.js
+++ b/main.js
-const electron = require('electron')
-const app = electron.app
-const url = require('url')
-const WebSocket = require('ws')
-const http = require('http')
 const fs = require('fs')
+const ws = require('ws')
+const https = require('https')
+const getport = require('get-port')
+const mime = require('mime')
 
-const routes = {
-  '/': function root(response) {
-    response.write(fs.readFileSync('host/host.html'))
-    response.end()
-  },
-  '/admin.js': function js(response) {
-    response.write(fs.readFileSync('host/host.js'))
-    response.end()
-  },
-  '/adapter.js': function adapter(response) {
-    response.write(fs.readFileSync('adapter.js'))
-    response.end()
-  },
-  '/favicon.ico': function favicon(response) {
-    response.write('N/A')
-    response.end()
-  },
-  '/streamlist': function streamlist(response) {
-    getDefaultStream()
-  }
-}
+const argv = require('minimist')(process.argv.slice(2), {
+  string:  [ 'ca-cert', 'ca-key', 'config', 'client-js', 'host-js', 'electron', 'port' ],
+  boolean: [ 'remote-host' ],
+  alias:   { c: 'config',
+            j: 'client-js',
+            J: 'host-js',
+            C: 'ca-cert',
+            K: 'ca-key',
+            e: 'electron',
+            r: 'remote-host',
+            p: 'port'
+          },
+  default: { config:        undefined,
+            'client-js':   'client.js',
+            'host-js':     'host.js',
+            'ca-cert':     'stunnel.cert',
+            'ca-key':      'stunnel.key',
+            'remote-host': true,
+            'port':        2443
+          },
+  stopEarly: true,
+  unknown: (opt) => {
+    console.log(process.argv.join(" ") + '\nUnknown operator: ' + opt + `
+Usage: strapp [OPTION]...  
+Route https connections from a hardware port to a remote host, and initiate
+peer-to-peer connection with clients.
 
-const rooms = {}
+CONFIG
+  -c, --config=path         Configuration file to use (/etc/strapp.conf)
+                            - overridden by command line opts
+  -j, --client-js=path      Path to the client Strapp code (./strapp-client.js)
+  -J, --host-js=path        Path to the host Strapp code (./strapp-host.js)
+  -C, --ca-cert=path        Accessible location of the CA Cert (./stunnel.cert)
+  -K, --ca-key=path         Accessible location of the CA Key (./stunnel.key)
+  -p, --port=number         The local port to bind HTTPS listener to (2443)
 
+ROUTING
+  -e, --electron=route      Route to the local electron user (nil)
+                            - enables optional electron dependency
+  -d, --dedicated=route     Route all incoming connections to this route (nil)
+                            - used in conjunction with '-e=my_route'
 
-/* TODO: Swap out url.parse with regex? */
-function determineResponse(request) {
-  const clientIP  = request.connection.remoteAddress
-                  || request.socket.remoteAddress
-                  || request.connection.socket.remoteAddress
-                  || req.headers['x-forwarded-for'].split(',')[0]
-  const parsedURL = url.parse(request.url)
-
-  /* Get the room name */
-  const pathname = parsedURL.pathname
+(c)2017 jk software
+`)
+    process.exit()
+  }
+})
 
+//TODO: if (argv['config'] !== undefined), read and apply to argv without overwrite
+if (argv['www-path'] == undefined)
+  argv['www-path'] = 'www'
+else if (String(argv['www-path']).endsWith('/'))
+  argv['www-path'] = argv['www-path'].slice(0,-1)
 
+const routerOpts = {
+  key: fs.readFileSync(argv['ca-key']),
+  cert: fs.readFileSync(argv['ca-cert'])
 }
+const skelPage = String(fs.readFileSync('skel.html')).split("<!--STRAPP-->")
+const clientJS = fs.readFileSync(argv['client-js'])
+const hostJS   = fs.readFileSync(argv['host-js'])
+const routes   = {}
+const fileBuf  = {}
+let   fbSize   = 0
+fs.readdirSync(argv['www-path']).forEach((file) => {
+  if (fbSize++ < 50) {
+    fileBuf[file] = { mime: mime.lookup(argv['www-path'] + '/' + file),
+                     data: fs.readFileSync(argv['www-path'] + '/' + file)
+                   }
+  }
+})
 
-app.on('ready', function() {
-  console.log('create server')
-  http.createServer(function(request, response) {
-
-    let route = determineResponse(request)
-
-    console.log('routing to ' + route)
-    routes[route](response)
+const router = https.createServer(routerOpts, (request, response) => {
+  const htArgv = request.url.slice(1).split("?")
+  let   routeName = htArgv[0].split('/')[0]
+  if (routeName === '')
+    routeName = 'index.html'
+  if (routeName.indexOf('.') != -1) {
+    if (routeName in fileBuf) {
+      response.writeHead(200, { 'Content-Type': fileBuf[routeName].mime })
+      response.write(fileBuf[routeName].data)
+    }
+    else {
+      if ('404.html' in fileBuf) {
+       response.writeHead(404, fileBuf['404.html'].mime)
+       response.write(fileBuf['404.html'].data)
+      }
+      else
+       response.writeHead(404)
+    }
+    response.end()
+  }
+  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
+      newRoute.httpd = https.createServer(routerOpts, (request, response) => {
+      }).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(() => {
+      response.writeHead(200, { 'Content-Type': 'text/html' })
+      response.write(skelPage[0] + 'const _strapp_host = \'www.strapp.io\'\n\tconst _strapp_port = \'' + newRoute.port + '\'\n' + hostJS + skelPage[1])
+      response.end()
+    })
+  }
+}).listen(argv['port'])
 
-  }).listen(5140);
+//TODO: if ("electron" in process.versions) open a local renderwindow, and route to it
 
-});