You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
3.0 KiB
JavaScript
85 lines
3.0 KiB
JavaScript
'use strict'
|
|
let WebSocketServer = require('ws').Server
|
|
const PORT = 8000
|
|
const UNSUPPORTED_DATA = 1007
|
|
const POLICY_VIOLATION = 1008
|
|
const CLOSE_UNSUPPORTED = 1003
|
|
|
|
var run = module.exports.run = function(storage, server) {
|
|
server.on('connection', (socket) => {
|
|
socket.on('message', (data) => {
|
|
try {
|
|
let msg = JSON.parse(data)
|
|
console.log(msg);
|
|
if (msg.hasOwnProperty('key')) {
|
|
for (let master of server.clients) {
|
|
if (master.key === msg.key) {
|
|
socket.close(POLICY_VIOLATION, 'The key already exists')
|
|
console.log('ERROR key exists');
|
|
return
|
|
}
|
|
}
|
|
socket.key = msg.key
|
|
socket.joiningClients = []
|
|
} else if (msg.hasOwnProperty('id')) {
|
|
for (let index in socket.joiningClients) {
|
|
if (index == msg.id) {
|
|
socket.joiningClients[index].send(JSON.stringify({data: msg.data}))
|
|
return
|
|
}
|
|
}
|
|
socket.close(POLICY_VIOLATION, 'Unknown id')
|
|
} else if (msg.hasOwnProperty('join')) {
|
|
for (let master of server.clients) {
|
|
if (master.key === msg.join) {
|
|
console.log('joined');
|
|
socket.master = master
|
|
master.joiningClients.push(socket)
|
|
let id = master.joiningClients.length - 1
|
|
console.log(id);
|
|
master.send(JSON.stringify({id, data: msg.data}))
|
|
return
|
|
}
|
|
}
|
|
console.log('ERROR unknown key');
|
|
socket.close(POLICY_VIOLATION, 'Unknown key')
|
|
} else if (msg.hasOwnProperty('data') && socket.hasOwnProperty('master')) {
|
|
let id = socket.master.joiningClients.indexOf(socket)
|
|
socket.master.send(JSON.stringify({id, data: msg.data}))
|
|
} else {
|
|
socket.close(UNSUPPORTED_DATA, 'Unsupported message format')
|
|
}
|
|
} catch (event) {
|
|
socket.close(CLOSE_UNSUPPORTED, 'Server accepts only JSON')
|
|
}
|
|
})
|
|
|
|
socket.on('close', (event) => {
|
|
console.log('someone has closed');
|
|
// If not master
|
|
if (socket.hasOwnProperty('master')) {
|
|
let masterClients = socket.master.joiningClients
|
|
for (let client of masterClients) {
|
|
if(client.id === socket.id) {
|
|
console.log('close client '+client.key)
|
|
client.close(POLICY_VIOLATION, 'The peer is no longer available')
|
|
//masterClients.splice(masterClients.indexOf(client),1);
|
|
}
|
|
}
|
|
}
|
|
else if (socket.hasOwnProperty('joiningClients')) {
|
|
let firstClient
|
|
let masterClients = socket.joiningClients
|
|
for (let client of masterClients) {
|
|
firstClient = client
|
|
break;
|
|
}
|
|
firstClient.close(POLICY_VIOLATION, 'The master is no longer available')
|
|
//masterClients.splice(masterClients.indexOf(firstClient),1);
|
|
firstClient.joiningClients = masterClients
|
|
console.log('change master from '+socket.key+' to '+firstClient.key)
|
|
socket = firstClient
|
|
}
|
|
})
|
|
})
|
|
} |