First draft of game balancing solution after reconnect
[vchess.git] / sockets.js
1 const url = require('url');
2 const Variants = require("./variants");
3
4 module.exports = function(wss) {
5
6 let clients = { "index": {} };
7 let games = {}; //pending games (player sid)
8 for (const v of Variants)
9 clients[v.name] = {};
10
11 wss.on("connection", (socket, req) => {
12 const params = new URL("http://localhost" + req.url).searchParams;
13 const sid = params.get("sid");
14 const page = params.get("page");
15 clients[page][sid] = socket;
16 if (page == "index")
17 {
18 // Send counting info
19 const countings = {};
20 for (const v of Variants)
21 countings[v.name] = Object.keys(clients[v.name]).length;
22 socket.send(JSON.stringify({code:"counts",counts:countings}));
23 }
24 else
25 {
26 // Send to every client connected on index an update message for counts
27 Object.keys(clients["index"]).forEach( k => {
28 clients["index"][k].send(JSON.stringify({code:"increase",vname:page}));
29 });
30 // Also notify potential opponents: hit all clients which check if sid corresponds
31 Object.keys(clients[page]).forEach( k => {
32 clients[page][k].send(JSON.stringify({code:"connect",id:sid}));
33 });
34 socket.on("message", objtxt => {
35 let obj = JSON.parse(objtxt);
36 switch (obj.code)
37 {
38 case "newmove":
39 if (!!clients[page][obj.oppid])
40 clients[page][obj.oppid].send(JSON.stringify({code:"newmove",move:obj.move}));
41 break;
42 case "ping":
43 if (!!clients[page][obj.oppid])
44 socket.send(JSON.stringify({code:"pong"}));
45 break;
46 case "lastate":
47 if (!!clients[page][obj.oppid])
48 clients[page][obj.oppid].send(objtxt);
49 break;
50 case "newgame":
51 if (!!games[page])
52 {
53 // Start a new game
54 const oppId = games[page]["id"];
55 const fen = games[page]["fen"];
56 delete games[page];
57 const mycolor = Math.random() < 0.5 ? 'w' : 'b';
58 socket.send(JSON.stringify({code:"newgame",fen:fen,oppid:oppId,color:mycolor}));
59 if (!!clients[page][oppId])
60 clients[page][oppId].send(JSON.stringify({code:"newgame",fen:fen,oppid:sid,color:mycolor=="w"?"b":"w"}));
61 }
62 else
63 games[page] = {id:sid, fen:obj.fen}; //wait for opponent
64 break;
65 case "resign":
66 if (!!clients[page][obj.oppid])
67 clients[page][obj.oppid].send(JSON.stringify({code:"resign"}));
68 break;
69 }
70 });
71 }
72 socket.on("close", () => {
73 delete clients[page][sid];
74 // Remove potential pending game
75 if (!!games[page] && games[page]["id"] == sid)
76 delete games[page];
77 if (page != "index")
78 {
79 // Send to every client connected on index an update message for counts
80 Object.keys(clients["index"]).forEach( k => {
81 clients["index"][k].send(JSON.stringify({code:"decrease",vname:page}));
82 });
83 }
84 // Also notify potential opponents: hit all clients which check if sid corresponds
85 Object.keys(clients[page]).forEach( k => {
86 clients[page][k].send(JSON.stringify({code:"disconnect",id:sid}));
87 });
88 });
89 });
90 }