Configurer une route WebSocket avec des flags, l'authentification et des paramètres

Total.js 5 · total5@0.0.19 websocketrealtimesecurity Exécutée avec Total.js 5

Une route SOCKET utilise les mêmes éléments qu'une route HTTP. Par défaut, les messages sont des objets JSON. @text fait des messages entrants de simples chaînes et @binary des Buffer. Dans une route de texte, send(message) encode toujours en JSON ce qu'il envoie, donc une chaîne arrive entre guillemets ; passez true en deuxième argument, send(message, true), pour envoyer la chaîne telle quelle. Un + devant SOCKET n'accepte que les utilisateurs approuvés par AUTH() (sinon la connexion est refusée), et les segments {name} sont disponibles dans self.params. @csrf vérifie un jeton CSRF avant d'accepter la connexion : envoyez-le comme paramètre csrf de la query string (/secure/?csrf=...), et voyez la recette CSRF pour secret_csrf et $.csrf(). Un middleware se rattache avec #name comme sur n'importe quelle route.

config

secret_csrf : change-this-secret-in-production

definitions/auth.js

AUTH(function($) {
	if ($.query.token === 'letmein')
		$.success({ id: '1', name: 'Louis' });
	else
		$.invalid();
});

controllers/sockets.js

exports.install = function() {
	ROUTE('SOCKET /text/ @text', text);
	ROUTE('SOCKET /bin/ @binary', binary);
	ROUTE('+SOCKET /private/', private_socket);
	ROUTE('SOCKET /room/{name}/', room);
	ROUTE('SOCKET /secure/ @csrf', secure);
	ROUTE('GET /where/', where);
	ROUTE('GET /csrf/', csrf);
};

function text() {
	this.on('message', function(client, message) {
		client.send(typeof(message) + ':' + message.toUpperCase(), true);
	});
}

function binary() {
	this.on('message', function(client, message) {
		client.send(Buffer.from(Buffer.isBuffer(message) ? 'buffer:' + message.length : 'other'));
	});
}

function private_socket() {
	this.on('open', function(client) {
		client.send({ hello: client.user.name });
	});
}

function room() {
	var self = this;
	self.on('open', function(client) {
		client.send({ room: self.params.name });
	});
}

function secure() {
	this.on('open', function(client) {
		client.send({ welcome: true });
	});
}

function where($) {
	$.json({ url: $.hostname('/bin/') });
}

function csrf($) {
	$.json({ csrf: $.csrf() });
}