app.js

var createError = require('http-errors');
var express = require('express');
var path = require('path');
var cookieParser = require('cookie-parser');
var logger = require('morgan');

var WebSocketServer = require('ws').Server;
var messageHandler = require('./bin/messageHandler');
var wss = new WebSocketServer({ port: 3001 });

var indexRouter = require('./routes/index');
var usersRouter = require('./routes/users');

var app = express();

// view engine setup
app.set('views', path.join(__dirname, 'views'));
app.set('view engine', 'ejs');

app.use(logger('dev'));
app.use(express.json());
app.use(express.urlencoded({ extended: false }));
app.use(cookieParser());
app.use(express.static(path.join(__dirname, 'public')));

app.use('/', indexRouter);
app.use('/users', usersRouter);

// catch 404 and forward to error handler
app.use(function(req, res, next) {
  next(createError(404));
});

// error handler
app.use(function(err, req, res, next) {
  // set locals, only providing error in development
  res.locals.message = err.message;
  res.locals.error = req.app.get('env') === 'development' ? err : {};

  // render the error page
  res.status(err.status || 500);
  res.render('error');
});

wss.on('connection', function connection(ws, req) {
  console.log('connection from a client');
  //console.log("id:", req.headers['sec-websocket-key']);
  ws.on('message', function incoming(message) {
      var objMessage = JSON.parse(message);
      messageHandler(ws, objMessage);
  });
  ws.on('close', function (event) {
      messageHandler(ws, {type: 'close'});
  });
});

module.exports = app;