Commit | Line | Data |
---|---|---|
1d184b4c BA |
1 | //const url = require('url'); |
2 | const Variants = require("./variants"); | |
3 | ||
4 | function getJsonFromUrl(url) { | |
5 | var query = url.substr(2); //starts with "/?" | |
6 | var result = {}; | |
7 | query.split("&").forEach(function(part) { | |
8 | var item = part.split("="); | |
9 | result[item[0]] = decodeURIComponent(item[1]); | |
10 | }); | |
11 | return result; | |
12 | } | |
13 | ||
14 | module.exports = function(wss) { | |
15 | ||
16 | let clients = { "index": {} }; | |
17 | let games = {}; //pending games (player sid) | |
18 | for (const v of Variants) | |
19 | clients[v.name] = {}; | |
20 | ||
30ff6e04 | 21 | // (resign, newgame, newmove). See https://github.com/websockets/ws/blob/master/lib/websocket.js around line 313 |
a68d899d BA |
22 | // TODO: awaiting newmove, resign, (+newgame?) :: in memory structure (use Redis ?) |
23 | let newmoves = {}; | |
24 | let newresign = {}; | |
25 | for (const v of Variants) | |
26 | { | |
27 | newmoves[v.name] = {}; | |
28 | newresign[v.name] = {}; | |
29 | } | |
30ff6e04 | 30 | |
1d184b4c BA |
31 | wss.on("connection", (socket, req) => { |
32 | //const params = new URL("http://localhost" + req.url).searchParams; | |
33 | var query = getJsonFromUrl(req.url); | |
34 | const sid = query["sid"]; //params.get("sid"); | |
35 | const page = query["page"]; //params.get("page"); | |
36 | clients[page][sid] = socket; | |
37 | if (page == "index") | |
38 | { | |
39 | // Send counting info | |
40 | const countings = {}; | |
41 | for (const v of Variants) | |
42 | countings[v.name] = Object.keys(clients[v.name]).length; | |
43 | socket.send(JSON.stringify({code:"counts",counts:countings})); | |
44 | } | |
45 | else | |
46 | { | |
47 | // Send to every client connected on index an update message for counts | |
48 | Object.keys(clients["index"]).forEach( k => { | |
49 | clients["index"][k].send(JSON.stringify({code:"increase",vname:page})); | |
50 | }); | |
51 | // Also notify potential opponents: hit all clients which check if sid corresponds | |
52 | Object.keys(clients[page]).forEach( k => { | |
53 | clients[page][k].send(JSON.stringify({code:"connect",id:sid})); | |
54 | }); | |
a68d899d BA |
55 | if (!!newmoves[page][sid]) |
56 | { | |
57 | socket.send(JSON.stringify({code:"newmove",move:newmoves[page][sid]})); | |
58 | delete newmoves[page][sid]; | |
59 | } | |
60 | if (!!newresign[page][sid]) | |
61 | { | |
62 | socket.send(JSON.stringify({code:"resign"})); | |
63 | delete newresign[page][sid]; | |
64 | } | |
1d184b4c BA |
65 | socket.on("message", objtxt => { |
66 | let obj = JSON.parse(objtxt); | |
67 | switch (obj.code) | |
68 | { | |
69 | case "newmove": | |
001344b9 | 70 | if (!!clients[page][obj.oppid]) // && clients[page][obj.oppid].readyState == WebSocket.OPEN) |
a68d899d BA |
71 | clients[page][obj.oppid].send(JSON.stringify({code:"newmove",move:obj.move})); |
72 | else | |
73 | newmoves[page][obj.oppid] = obj.move; | |
1d184b4c BA |
74 | break; |
75 | case "ping": | |
001344b9 | 76 | if (!!clients[page][obj.oppid]) // && clients[page][obj.oppid].readyState == WebSocket.OPEN) |
1d184b4c BA |
77 | socket.send(JSON.stringify({code:"pong"})); |
78 | break; | |
79 | case "newgame": | |
80 | if (!!games[page]) | |
81 | { | |
82 | // Start a new game | |
83 | const oppId = games[page]["id"]; | |
84 | const fen = games[page]["fen"]; | |
85 | delete games[page]; | |
86 | const mycolor = Math.random() < 0.5 ? 'w' : 'b'; | |
87 | socket.send(JSON.stringify({code:"newgame",fen:fen,oppid:oppId,color:mycolor})); | |
88 | clients[page][oppId].send(JSON.stringify({code:"newgame",fen:fen,oppid:sid,color:mycolor=="w"?"b":"w"})); | |
89 | } | |
90 | else | |
91 | games[page] = {id:sid, fen:obj.fen}; //wait for opponent | |
92 | break; | |
93 | case "resign": | |
001344b9 | 94 | if (!!clients[page][obj.oppid]) // && clients[page][obj.oppid].readyState == WebSocket.OPEN) |
a68d899d BA |
95 | clients[page][obj.oppid].send(JSON.stringify({code:"resign"})); |
96 | else | |
97 | newresign[page][obj.oppid] = true; | |
1d184b4c BA |
98 | break; |
99 | } | |
100 | }); | |
101 | } | |
102 | socket.on("close", () => { | |
103 | delete clients[page][sid]; | |
104 | // Remove potential pending game | |
105 | if (!!games[page] && games[page]["id"] == sid) | |
106 | delete games[page]; | |
107 | if (page != "index") | |
108 | { | |
109 | // Send to every client connected on index an update message for counts | |
110 | Object.keys(clients["index"]).forEach( k => { | |
111 | clients["index"][k].send(JSON.stringify({code:"decrease",vname:page})); | |
112 | }); | |
113 | } | |
114 | // Also notify potential opponents: hit all clients which check if sid corresponds | |
115 | Object.keys(clients[page]).forEach( k => { | |
116 | clients[page][k].send(JSON.stringify({code:"disconnect",id:sid})); | |
117 | }); | |
118 | }); | |
119 | }); | |
120 | } |