server.js

'use strict';

let technoDoc = require('techno-gendoc');
let express = require('express');
let app = express();

technoDoc.generate(require('./api'), 'public');
app.use('/', express.static('index'));
app.use('/chat', express.static('index'));
app.use('/game', express.static('index'));
app.use('/scores', express.static('index'));
app.use('/dist', express.static('dist'));


//**********************************************//


let cookieParser = require('cookie-parser');
let websokify = require('express-ws');
let makeUser = require('./make-user');
let parser = require('body-parser');
websokify(app);

const users = new Map();
const messages = [];


app.use(parser.json());

app.use(cookieParser());


// получение списка всех пользователей
app.get('/api/users', function (req, res) {
	// console.dir(req.cookies);
	const secret = req.cookies.secret;
	if (!secret || !users.has(secret)) {
		// console.log('ooops!');
		return res.status(401).end();
	}
	const r = [...users.values()].map(user => {
		user.secret = undefined;
		return user;
	});
	res.json(r);
});

// создание нового пользователя
app.post('/api/users', function (req, res) {
	const body = req.body;
	if (body.username && body.email) {
		const user = makeUser(body.username, body.email);
		users.set(user.secret, user);
		res.cookie('secret', user.secret, {path: '/', maxAge: 12 * 60 * 60 * 1000});
		return res.status(201).end();
	}
	return res.status(400).json({error: 'Неверные имя пользователя и/или пароль'});
});


const clients = new Set();

app.ws('/ws/messages', function (ws, req) {
	const secret = ws.upgradeReq.cookies.secret;
	if (!secret) {
		console.log('Закрываем соединение, потому что в запросе не было нужной куки');
		ws.close();
		return;
	}

	if (!users.has(secret)) {
		console.log('Закрываем соединение, потому что такого пользователя не существует');
		ws.close();
		return;
	}

	const user = users.get(secret);

	console.log('Add new client');
	clients.add(ws);
	console.log(`NOW ${clients.size} clients exists`);


	ws.on('message', function (msg) {
		try {
			if (!msg) {
				return;
			}
			let message = JSON.parse(msg);
			message.email = user.email;
			message.message = message.text;
			messages.push(message);
			ws.send(JSON.stringify(messages));
		} catch (err) {
			console.error(err);
		}
	});

	ws.on('close', function (evt) {
		console.log('Remove client');
		clients.delete(ws);
		console.log(`NOW ${clients.size} clients exists`);
	});

	try {
		ws.send(JSON.stringify(messages));
	} catch (err) {
		console.error(err);
	}

});


const chatClients = new Set();
app.ws('/ws/echo', function (ws) {
	console.log('Установили соединение');
	chatClients.add(ws);
	ws.on('message', function (msg) {
		console.log('Приняли новое сообщение', msg);
		[...chatClients.values()].forEach(client => client.send(msg));
	});
	ws.on('close', function () {
		console.log('Соединение закрыто');
		chatClients.delete(ws);
	});
});


//**********************************************//

app.listen(process.env.PORT || 3000, () => {
	console.log(`App started on port ${process.env.PORT || 3000}`);
});