Small fix

[vchess.git] / server / sockets.js

const url = require('url');

// Node version in Ubuntu 16.04 does not know about URL class
function getJsonFromUrl(url)
{
  const query = url.substr(2); //starts with "/?"
  let result = {};
  query.split("&").forEach((part) => {
    const item = part.split("=");
    result[item[0]] = decodeURIComponent(item[1]);
  });
  return result;
}

module.exports = function(wss) {
  let clients = {}; //associative array sid --> socket
  wss.on("connection", (socket, req) => {
    const query = getJsonFromUrl(req.url);
    const sid = query["sid"];
    // TODO: later, allow duplicate connections (shouldn't be much more complicated)
    if (!!clients[sid])
      return socket.send(JSON.stringify({code:"duplicate"}));
    clients[sid] = socket;
    // Notify room:
    Object.keys(clients).forEach(k => {
      if (k != sid)
        clients[k].send(JSON.stringify({code:"connect",sid:sid}));
    });
    socket.on("message", objtxt => {
      let obj = JSON.parse(objtxt);
      if (!!obj.target && !clients[obj.target])
        return; //receiver not connected, nothing we can do
      switch (obj.code)
      {
        case "pollclients":
          socket.send(JSON.stringify({code:"pollclients",
            sockIds:Object.keys(clients).filter(k => k != sid)}));
          break;
        case "askidentity":
          clients[obj.target].send(
            JSON.stringify({code:"askidentity",from:sid}));
          break;
        case "askchallenge":
          clients[obj.target].send(
            JSON.stringify({code:"askchallenge",from:sid}));
          break;
        case "askgame":
          clients[obj.target].send(
            JSON.stringify({code:"askgame",from:sid}));
          break;
        case "identity":
          clients[obj.target].send(
            JSON.stringify({code:"identity",user:obj.user}));
          break;
        case "refusechallenge":
          clients[obj.target].send(
            JSON.stringify({code:"refusechallenge", cid:obj.cid, from:sid}));
          break;
        case "deletechallenge":
          clients[obj.target].send(
            JSON.stringify({code:"deletechallenge", cid:obj.cid, from:sid}));
          break;
        case "newgame":
          clients[obj.target].send(JSON.stringify(
            {code:"newgame", gameInfo:obj.gameInfo, cid:obj.cid}));
          break;
        case "challenge":
          clients[obj.target].send(JSON.stringify(
            {code:"challenge", chall:obj.chall, from:sid}));
          break;
        case "game":
          clients[obj.target].send(JSON.stringify(
            {code:"game", game:obj.game, from:sid}));
          break;
        case "newchat":
          clients[obj.target].send(JSON.stringify({code:"newchat",msg:obj.msg}));
          break;
        // TODO: WebRTC instead in this case (most demanding?)
        case "newmove":
          clients[obj.target].send(JSON.stringify({code:"newmove",move:obj.move}));
          break;
        case "ping":
          // If this code is reached, then obj.target is connected
          socket.send(JSON.stringify({code:"pong"}));
          break;
        case "lastate":
          const oppId = obj.target;
          obj.oppid = sid; //I'm the opponent of my opponent(s)
          clients[oppId].send(JSON.stringify(obj));
          break;
        case "resign":
          clients[obj.target].send(JSON.stringify({code:"resign"}));
          break;
        case "abort":
          clients[obj.target].send(JSON.stringify({code:"abort",msg:obj.msg}));
          break;
        case "drawoffer":
          clients[obj.target].send(JSON.stringify({code:"drawoffer"}));
          break;
        case "draw":
          clients[obj.target].send(JSON.stringify({code:"draw"}));
          break;
      }
    });
    socket.on("close", () => {
      delete clients[sid];
      // Notify every other connected client
      Object.keys(clients).forEach( k => {
        clients[k].send(JSON.stringify({code:"disconnect",sid:sid}));
      });
    });
  });
}