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 }