doit-only-socket/server.js at master · ramainen/doit-only-socket · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
var app = require('express')();
var server = require('http').Server(app);
var io = require('socket.io')(server);
var url = require('url');
var bodyParser = require('body-parser')

server.listen(process.env.PORT || 1337);

var sockets_array=[];

app.get('/', function (req, res) {
  res.sendFile(__dirname + '/index.html');
});
app.get('/favicon.ico', function (req, res) {
  res.sendFile(__dirname + '/favicon.ico');
});

app.use(bodyParser.urlencoded({
  extended: true
}));

app.get('/stat/all', function (req, res) {
	res.writeHead(200);
	res.end("" + Object.keys(io.sockets.adapter.rooms).length + "");
})

app.get('/emit', function (req, res) {
	var query = url.parse(req.url, true).query;
	res.writeHead(200);
	if((typeof (query.id) != 'undefined') && (typeof (query.message) != 'undefined') /*&& (typeof(sockets_array[query.id]) != 'undefined')*/){

		res.end("OK");
		var query_message = query.message;
		var query_id = query.id;

		query.id = undefined;
		query.message = undefined;
		if ((typeof (query._type) != 'undefined') && (typeof (query._data) != 'undefined') && (query._type == 'string')){
			query = query._data;
		}
		io.to(query_id).emit(query_message, query);
		/*for(key in sockets_array[query_id]){
			sockets_array[query_id][key].emit(query_message, query);
		}*/


	}else{
		res.end("ERROR");
	}
});

//console.log('start')
app.post('/emit', function (req, res) {
	res.writeHead(200);
//		console.log('emit')
	if((typeof (req.param('id')) != 'undefined') && (typeof (req.param('message')) != 'undefined')) {
		var clients = [];
		if(typeof(req.param('id'))=='object'){
			clients = req.param('id');
		}else{
			clients = [req.param('id')];
		}
		var all_ok = false;
//		console.log('emit')
		for (client_key in clients){

			var query_message = req.param('message');
			var query_id = clients[client_key];
			if((typeof(io.sockets.adapter.rooms[query_id])!='undefined') && (Object.keys(io.sockets.adapter.rooms[query_id]).length != 0)){
				all_ok=true;
			}
			if(typeof(req.param('data'))=='undefined'){
				query = {};
			}else{
				query = JSON.parse(req.param('data'));
			}
			io.to(clients[client_key]).emit(query_message, query);

		}
		if(all_ok){
			res.end("OK");
		}else{
			res.end("ERROR");
		}
	}else{
		res.end("ERROR");
	}

});


io.on('connection', function (socket) {
	socket.on('register', function (data) {
		//connect
		if(typeof(data.userid)!='undefined'){
			if(typeof(socket._userid)=='undefined'){
				socket._userid = [];
			}
			socket._userid.push(data.userid);
			socket.join(data.userid);
		}

	});
	socket.on('disconnect', function() {
		if(typeof(socket._userid) != 'undefined'){
			for(key in socket._userid){
				//Socket.io fix
				if(Object.keys(io.sockets.adapter.rooms[socket._userid[key]]).length == 0){
					delete io.sockets.adapter.rooms[socket._userid[key]];
				}
			}
		}
		delete io.sockets.adapter.rooms[socket.id];
	});
});