Fix draw logic for corr games

[vchess.git] / server / models / Game.js

var db = require("../utils/database");
const UserModel = require("./User");

/*
 * Structure table Games:
 *   id: game id (int)
 *   vid: integer (variant id)
 *   fenStart: varchar (initial position)
 *   fen: varchar (current position)
 *   timeControl: string
 *   score: varchar (result)
 *   scoreMsg: varchar ("Time", "Mutual agreement"...)
 *   created: datetime
 *   drawOffer: char ('w','b' or '' for none)
 *
 * Structure table Players:
 *   gid: ref game id
 *   uid: ref user id
 *   color: character
 *
 * Structure table Moves:
 *   gid: ref game id
 *   squares: varchar (description)
 *   played: datetime
 *   idx: integer
 *
 * Structure table Chats:
 *   gid: game id (int)
 *   msg: varchar
 *   name: varchar
 *   added: datetime
 */

const GameModel =
{
  checkGameInfo: function(g) {
    if (!g.vid.toString().match(/^[0-9]+$/))
      return "Wrong variant ID";
    if (!g.vname.match(/^[a-zA-Z0-9]+$/))
      return "Wrong variant name";
    if (!g.timeControl.match(/^[0-9dhms +]+$/))
      return "Wrong characters in time control";
    if (!g.fen.match(/^[a-zA-Z0-9, /-]*$/))
      return "Bad FEN string";
    if (g.players.length != 2)
      return "Need exactly 2 players";
    if (g.players.some(p => !p.id.toString().match(/^[0-9]+$/)))
      return "Wrong characters in player ID";
    return "";
  },

  create: function(vid, fen, timeControl, players, cb)
  {
    db.serialize(function() {
      let query =
        "INSERT INTO Games"
        + " (vid, fenStart, fen, score, timeControl, created, drawOffer)"
        + " VALUES (" + vid + ",'" + fen + "','" + fen + "','*','"
        + timeControl + "'," + Date.now() + ",'')";
      db.run(query, function(err) {
        if (!!err)
          return cb(err);
        players.forEach((p,idx) => {
          const color = (idx==0 ? "w" : "b");
          query =
            "INSERT INTO Players VALUES " +
            "(" + this.lastID + "," + p.id + ",'" + color + "')";
          db.run(query);
        });
        cb(null, {gid: this.lastID});
      });
    });
  },

  // TODO: queries here could be async, and wait for all to complete
  getOne: function(id, cb)
  {
    db.serialize(function() {
      // TODO: optimize queries?
      let query =
        // NOTE: g.scoreMsg can be NULL
        // (in this case score = "*" and no reason to look at it)
        "SELECT g.id, g.vid, g.fen, g.fenStart, g.timeControl, g.score, " +
          "g.scoreMsg, g.drawOffer, v.name AS vname " +
        "FROM Games g " +
        "JOIN Variants v " +
        "  ON g.vid = v.id " +
        "WHERE g.id = " + id;
      db.get(query, (err,gameInfo) => {
        if (!!err)
          return cb(err);
        query =
          "SELECT p.uid, p.color, u.name " +
          "FROM Players p " +
          "JOIN Users u " +
          "  ON p.uid = u.id " +
          "WHERE p.gid = " + id;
        db.all(query, (err2,players) => {
          if (!!err2)
            return cb(err2);
          query =
            "SELECT squares, played, idx " +
            "FROM Moves " +
            "WHERE gid = " + id;
          db.all(query, (err3,moves) => {
            if (!!err3)
              return cb(err3);
            query =
              "SELECT msg, name, added " +
              "FROM Chats " +
              "WHERE gid = " + id;
            db.all(query, (err4,chats) => {
              if (!!err4)
                return cb(err4);
              const game = Object.assign({},
                gameInfo,
                {
                  players: players,
                  moves: moves,
                  chats: chats,
                }
              );
              return cb(null, game);
            });
          });
        });
      });
    });
  },

  getByUser: function(uid, excluded, cb)
  {
    db.serialize(function() {
      const query =
        "SELECT DISTINCT gid " +
        "FROM Players " +
        "WHERE uid " + (excluded ? "<>" : "=") + " " + uid;
      db.all(query, (err,gameIds) => {
        if (!!err)
          return cb(err);
        gameIds = gameIds || []; //might be empty
        let gameArray = [];
        for (let i=0; i<gameIds.length; i++)
        {
          GameModel.getOne(gameIds[i]["gid"], (err2,game) => {
            if (!!err2)
              return cb(err2);
            gameArray.push(game);
            // Call callback function only when gameArray is complete:
            if (i == gameIds.length - 1)
              return cb(null, gameArray);
          });
        }
      });
    });
  },

  getPlayers: function(id, cb)
  {
    db.serialize(function() {
      const query =
        "SELECT id " +
        "FROM Players " +
        "WHERE gid = " + id;
      db.all(query, (err,players) => {
        return cb(err, players);
      });
    });
  },

  checkGameUpdate: function(obj)
  {
    // Check all that is possible (required) in obj:
    if (!!obj.move)
    {
      if (!obj.move.played.toString().match(/^[0-9]+$/))
        return "Wrong move played time";
      if (!obj.move.idx.toString().match(/^[0-9]+$/))
        return "Wrong move index";
    }
    if (!!obj.drawOffer && !obj.drawOffer.match(/^[wbtn]$/))
      return "Wrong draw offer format";
    if (!!obj.fen && !obj.fen.match(/^[a-zA-Z0-9, /-]*$/))
      return "Wrong FEN string";
    if (!!obj.score && !obj.score.match(/^[012?*\/-]+$/))
      return "Wrong characters in score";
    if (!!obj.scoreMsg && !obj.scoreMsg.match(/^[a-zA-Z ]+$/))
      return "Wrong characters in score message";
    if (!!obj.chat)
      return UserModel.checkNameEmail({name: obj.chat.name});
    return "";
  },

  // obj can have fields move, chat, fen, drawOffer and/or score
  update: function(id, obj)
  {
    db.parallelize(function() {
      let query =
        "UPDATE Games " +
        "SET ";
      let modifs = "";
      if (!!obj.message)
        modifs += "message = message || ' ' || '" + obj.message + "',";
      // NOTE: if drawOffer is set, we should check that it's player's turn
      // A bit overcomplicated. Let's trust the client on that for now...
      if (!!obj.drawOffer)
      {
        if (obj.drawOffer == "n") //Special "None" update
          obj.drawOffer = "";
        modifs += "drawOffer = '" + obj.drawOffer + "',";
      }
      if (!!obj.fen)
        modifs += "fen = '" + obj.fen + "',";
      if (!!obj.score)
        modifs += "score = '" + obj.score + "',";
      if (!!obj.scoreMsg)
        modifs += "scoreMsg = '" + obj.scoreMsg + "',";
      modifs = modifs.slice(0,-1); //remove last comma
      if (modifs.length > 0)
      {
        query += modifs + " WHERE id = " + id;
        db.run(query);
      }
      if (!!obj.move)
      {
        const m = obj.move;
        query =
          "INSERT INTO Moves (gid, squares, played, idx) VALUES " +
          "(" + id + ",?," + m.played + "," + m.idx + ")";
        db.run(query, JSON.stringify(m.squares));
      }
      if (!!obj.chat)
      {
        query =
          "INSERT INTO Chats (gid, msg, name, added) VALUES ("
            + id + ",?,'" + obj.chat.name + "'," + Date.now() + ")";
        db.run(query, obj.chat.msg);
      }
    });
  },

  remove: function(id)
  {
    db.parallelize(function() {
      let query =
        "DELETE FROM Games " +
        "WHERE id = " + id;
      db.run(query);
      query =
        "DELETE FROM Players " +
        "WHERE gid = " + id;
      db.run(query);
      query =
        "DELETE FROM Moves " +
        "WHERE gid = " + id;
      db.run(query);
      query =
        "DELETE FROM Chats " +
        "WHERE gid = " + id;
      db.run(query);
    });
  },

  cleanGamesDb: function()
  {
    const tsNow = Date.now();
    // 86400000 = 24 hours in milliseconds
    const day = 86400000;
    db.serialize(function() {
      let query =
        "SELECT id,score " +
        "FROM Games ";
      db.all(query, (err,games) => {
        games.forEach(g => {
          query =
            "SELECT max(played) AS lastMaj " +
            "FROM Moves " +
            "WHERE gid = " + g.id;
          db.get(query, (err2,updated) => {
            if (!updated && tsNow - g.created > 7*day)
              return GameModel.remove(g.id);
            const lastMaj = updated.lastMaj;
            if (g.score != "*" && tsNow - lastMaj > 7*day ||
              g.score == "*" && tsNow - lastMaj > 91*day)
            {
              GameModel.remove(g.id);
            }
          });
        });
      });
    });
  },
}

module.exports = GameModel;