Mikej81/WebRDP

// app.js
//

var path = require('path')
var config = require('read-config')(path.join(__dirname, 'config.json'))
//  var logger = require('morgan')
var validator = require('validator')
var myutil = require('./util')

var session = require('express-session')({
  secret: 'MySecret',
  name: 'WebRDP',
  resave: true,
  saveUninitialized: false,
  unset: 'destroy'
})

var express = require('express')
var app = express()
var server = require('http').createServer(app)
var io = require('socket.io')(server)
var socket = require('./socket')

// Express
app.use(session)
app.use(myutil.basicAuth)
app.use(express.static(path.join(__dirname, '/client')))

app.get('/', function (req, res, next) {
  res.sendFile(path.join(__dirname, '/client/html/index.html'))
})

app.get('/rdp/host/:host?', function (req, res, next) {
  req.session.host = req.params.host
  res.sendFile(path.join(__dirname, '/client/html/client.html'))
  req.session.rdp = {
    host: (validator.isIP(req.params.host + '') && req.params.host) ||
    (validator.isFQDN(req.params.host) && req.params.host) ||
    (/^(([a-z]|[A-Z]|[0-9]|[!^(){}\-_~])+)?\w$/.test(req.params.host) &&
      req.params.host) || config.rdp.host,
    port: (validator.isInt(req.query.port + '', {min: 1, max: 65535}) &&
      req.query.port) || config.rdp.port,
    header: {
      name: req.query.header || config.header.text,
      background: req.query.headerBackground || config.header.background
    },
    algorithms: config.algorithms,
    allowreplay: validator.isBoolean(req.headers.allowreplay + '') || false,
    log: {
      screencapture: config.logging.logScreenOnMouseclicks || false
    },
    readyTimeout: (validator.isInt(req.query.readyTimeout + '', {min: 1, max: 300000}) &&
      req.query.readyTimeout) || config.rdp.readyTimeout
  }
  if (req.session.rdp.header.name) validator.escape(req.session.rdp.header.name)
  if (req.session.rdp.header.background) validator.escape(req.session.rdp.header.background)
})

// Express error handling
app.use(function (req, res, next) {
  res.status(404).send("Sorry can't find that!")
})

app.use(function (err, req, res, next) {
  console.error(err.stack)
  res.status(500).send('Something broke!')
})

server.listen(4200)

// socket.io
// expose express session with socket.request.session
io.use(function (socket, next) {
  (socket.request.res) ? session(socket.request, socket.request.res, next) : next()
})

// bring up socket
io.on('connection', socket)

module.exports = {server: server, config: config}