and so it begins
This commit is contained in:
Caleb James DeLisle
189
ChainPadSrv.js
Normal file
189
ChainPadSrv.js
Normal file
@@ -0,0 +1,189 @@
|
||||
/*
|
||||
* Copyright 2014 XWiki SAS
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU Affero General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Affero General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
var WebSocket = require('ws');
|
||||
|
||||
var REGISTER = 0;
|
||||
var REGISTER_ACK = 1;
|
||||
var PATCH = 2;
|
||||
var DISCONNECT = 3;
|
||||
var PING = 4;
|
||||
var PONG = 5;
|
||||
|
||||
var parseMessage = function (msg) {
|
||||
var passLen = msg.substring(0,msg.indexOf(':'));
|
||||
msg = msg.substring(passLen.length+1);
|
||||
var pass = msg.substring(0,Number(passLen));
|
||||
msg = msg.substring(pass.length);
|
||||
|
||||
var unameLen = msg.substring(0,msg.indexOf(':'));
|
||||
msg = msg.substring(unameLen.length+1);
|
||||
var userName = msg.substring(0,Number(unameLen));
|
||||
msg = msg.substring(userName.length);
|
||||
|
||||
var channelIdLen = msg.substring(0,msg.indexOf(':'));
|
||||
msg = msg.substring(channelIdLen.length+1);
|
||||
var channelId = msg.substring(0,Number(channelIdLen));
|
||||
msg = msg.substring(channelId.length);
|
||||
|
||||
var contentStrLen = msg.substring(0,msg.indexOf(':'));
|
||||
msg = msg.substring(contentStrLen.length+1);
|
||||
var contentStr = msg.substring(0,Number(contentStrLen));
|
||||
|
||||
return {
|
||||
user: userName,
|
||||
pass: pass,
|
||||
channelId: channelId,
|
||||
content: JSON.parse(contentStr)
|
||||
};
|
||||
};
|
||||
|
||||
// get the password off the message before sending it to other clients.
|
||||
var popPassword = function (msg) {
|
||||
var passLen = msg.substring(0,msg.indexOf(':'));
|
||||
return msg.substring(passLen.length+1 + Number(passLen));
|
||||
};
|
||||
|
||||
var sendMsg = function (msg, socket) {
|
||||
socket.send(msg);
|
||||
};
|
||||
|
||||
var sendChannelMessage = function (ctx, channel, msg, cb) {
|
||||
ctx.store.message(channel.name, msg, function () {
|
||||
channel.forEach(function (user) {
|
||||
try {
|
||||
sendMsg(msg, user.socket);
|
||||
} catch (e) {
|
||||
console.log(e.stack);
|
||||
dropClient(ctx, userPass);
|
||||
}
|
||||
});
|
||||
cb && cb();
|
||||
});
|
||||
};
|
||||
|
||||
var mkMessage = function (user, channel, content) {
|
||||
content = JSON.stringify(content);
|
||||
return user.length + ':' + user +
|
||||
channel.length + ':' + channel +
|
||||
content.length + ':' + content;
|
||||
};
|
||||
|
||||
var dropClient = function (ctx, userpass) {
|
||||
var client = ctx.registeredClients[userpass];
|
||||
if (client.socket.readyState !== WebSocket.CLOSING
|
||||
&& client.socket.readyState !== WebSocket.CLOSED)
|
||||
{
|
||||
try {
|
||||
client.socket.close();
|
||||
} catch (e) {
|
||||
console.log("Failed to disconnect ["+client.userName+"], attempting to terminate");
|
||||
try {
|
||||
client.socket.terminate();
|
||||
} catch (ee) {
|
||||
console.log("Failed to terminate ["+client.userName+"] *shrug*");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
for (var i = 0; i < client.channels.length; i++) {
|
||||
var chanName = client.channels[i];
|
||||
var chan = ctx.channels[chanName];
|
||||
var idx = chan.indexOf(client);
|
||||
if (idx < 0) { throw new Error(); }
|
||||
console.log("Removing ["+client.userName+"] from channel ["+chanName+"]");
|
||||
chan.splice(idx, 1);
|
||||
if (chan.length === 0) {
|
||||
console.log("Removing empty channel ["+chanName+"]");
|
||||
delete ctx.channels[chanName];
|
||||
} else {
|
||||
sendChannelMessage(ctx, chan, mkMessage(client.userName, chanName, [DISCONNECT,0]));
|
||||
}
|
||||
}
|
||||
delete ctx.registeredClients[userpass];
|
||||
};
|
||||
|
||||
var handleMessage = function (ctx, socket, msg) {
|
||||
var parsed = parseMessage(msg);
|
||||
var userPass = parsed.user + ':' + parsed.pass;
|
||||
msg = popPassword(msg);
|
||||
|
||||
if (parsed.content[0] === REGISTER) {
|
||||
console.log("[" + userPass + "] registered");
|
||||
var client = ctx.registeredClients[userPass] = ctx.registeredClients[userPass] || {
|
||||
channels: [parsed.channelId],
|
||||
userName: parsed.user
|
||||
};
|
||||
if (client.socket && client.socket !== socket) { client.socket.close(); }
|
||||
client.socket = socket;
|
||||
|
||||
var chan = ctx.channels[parsed.channelId] = ctx.channels[parsed.channelId] || [];
|
||||
chan.name = parsed.channelId;
|
||||
chan.push(client);
|
||||
|
||||
// we send a register ack right away but then we fallthrough
|
||||
// to let other users know that we were registered.
|
||||
sendMsg(mkMessage('', parsed.channelId, [1,0]), socket);
|
||||
sendChannelMessage(ctx, chan, msg, function () {
|
||||
ctx.store.getMessages(chan.name, function (msg) {
|
||||
sendMsg(msg, socket);
|
||||
});
|
||||
});
|
||||
}
|
||||
|
||||
if (parsed.content[0] === PING) {
|
||||
// 31:xwiki:XWiki.Admin-141475016907510:RWJ5xF2+SL17:[5,1414752676547]
|
||||
// 1:y31:xwiki:XWiki.Admin-141475016907510:RWJ5xF2+SL17:[4,1414752676547]
|
||||
sendMsg(mkMessage(parsed.user, parsed.channelId, [ PONG, parsed.content[1] ]), socket);
|
||||
return;
|
||||
}
|
||||
|
||||
var client = ctx.registeredClients[userPass];
|
||||
if (typeof(client) === 'undefined') { throw new Error('unregistered'); }
|
||||
|
||||
var channel = ctx.channels[parsed.channelId];
|
||||
if (typeof(channel) === 'undefined') { throw new Error('no such channel'); }
|
||||
|
||||
if (channel.indexOf(client) === -1) { throw new Error('client not in channel'); }
|
||||
|
||||
sendChannelMessage(ctx, channel, msg);
|
||||
};
|
||||
|
||||
var create = module.exports.create = function (socketServer, store) {
|
||||
var ctx = {
|
||||
registeredClients: {},
|
||||
channels: {},
|
||||
store: store
|
||||
};
|
||||
|
||||
socketServer.on('connection', function(socket) {
|
||||
socket.on('message', function(message) {
|
||||
try {
|
||||
handleMessage(ctx, socket, message);
|
||||
} catch (e) {
|
||||
console.log(e.stack);
|
||||
socket.close();
|
||||
}
|
||||
});
|
||||
socket.on('close', function (evt) {
|
||||
for (client in ctx.registeredClients) {
|
||||
if (ctx.registeredClients[client].socket === socket) {
|
||||
dropClient(ctx, client);
|
||||
}
|
||||
}
|
||||
});
|
||||
});
|
||||
};
|
||||
Reference in New Issue
Block a user