game.js

var Moniker = require("moniker");

/* Roles are:
 * - villager
 * - mafia
 * - cop
 * - doctor
 * - jester
 * - bodyguard
 *
 *
 * Not added yet:
 * - vigilante
 */

var canVote = {
  "villager": false,
  "mafia": true,
  "cop": true,
  "doctor": true,
  "jester": false,
  "bodyguard": true
}; // role does some sort of action at night

var deathQuotes = require("./deathQuotes");

function pluralize(name) {
  if(name[name.length - 1].toLowerCase() === "s")
    return name + "'";
  else
    return name + "'s";
}

function getDeathQuote(name) {
  var quote = deathQuotes[Math.floor(Math.random() * deathQuotes.length)];
  quote = quote.replace("%s", name).replace("%p", pluralize(name));
  return quote;
}

function Player(id, socket, name) {
  this.id = id;
  this.socket = socket;
  this.name = name;

  this.role = null;
  this.votingFor = null;
  this.dead = false;
  this.lynched = false; // Lynched is also dead, but if lynched, role is visible
  this.leftGame = false;
}

function Game(io) {
  this.io = io;
  this.players = [];
  this.room = Moniker.choose(); // essentially the game id
  this.closed = false; // closed to new joins (so that the moderator knows how many people there are)
  this.started = false; // game actually started (roles dealt out, etc)
  this.dayNumber = 1; // increases at end of night
  this.night = false; // whether it is night or not

  this.lastAction = ""; // description of the last thing to happen (lynching, night death)

  this.lastUpdate = new Date(); // used to garbage collect inactive games

  this.settings = {
    roles: {
      "mafia": 1,
      "cop": 0,
      "doctor": 0,
      "jester": 0,
      "bodyguard": 0
    }
  }; // settings for the game

  this.moderator = null; // First player to join
}

Game.prototype.addPlayer = function(id, socket, name) {
  this.players.push(new Player(id, socket, name));

  if(this.moderator === null)
    this.moderator = id;

  socket.emit("gameState", this.gameState());
  this.sendPlayerListUpdate(null);
}

Game.prototype.reconnectPlayer = function(id, socket, name) {
  for(var i = 0; i < this.players.length; i++) {
    if(this.players[i].id === id) {
      this.players[i].socket = socket;
      this.players[i].name = name;

      socket.emit("gameState", this.gameState());
      this.sendPlayerListUpdate(null);

      return true;
    }
  }
  return false;
}

Game.prototype.leavePlayer = function(id) {
  var p = this.playerById(id);
  p.leftGame = true;

  this.lastAction = p.name + " left the game.";
  this.sendGameStateUpdate();

  if(!p.dead && this.started) {
    // Kill the player
    p.dead = true;
    this.sendPlayerListUpdate(null);
  } else if(!this.started) {
    // Straight up remove the player
    for(var i = 0; i < this.players.length; i++) {
      if(this.players[i].id == id) {
        this.players.splice(i, 1);
        break;
      }
    }
    this.sendPlayerListUpdate(null);
  }

  p.id += "leave";
}

Game.prototype.playerById = function(id) {
  for(var i = 0; i < this.players.length; i++) {
    // Check for full id or shortened id
    if(this.players[i].id === id || this.players[i].id.split("-")[0] === id) {
      return this.players[i];
    }
  }
  return null;
}

// Determines the number of votes for a specific player from a specific role
Game.prototype.votesForBy = function(id, role) {
  var count = 0;
  for(var i = 0; i < this.players.length; i++) {
    if(this.players[i].votingFor === id && this.players[i].role === role)
      count++;
  }

  return count;
}

// Makes a list of players for a specific player
Game.prototype.playerListFor = function(id) {
  var p = this.playerById(id);
  if(p === null) return [];

  var list = [];
  for(var i = 0; i < this.players.length; i++) {
    var obj = {
      id: this.players[i].id.split("-")[0], // shortened id for identification, but can't impersonate
      name: this.players[i].name,
      role: this.players[i].role === p.role || this.players[i].lynched ? this.players[i].role : null,
      votes: this.votesForBy(this.players[i].id, p.role),
      mod: this.players[i].id === this.moderator,
      dead: this.players[i].dead,
      lynched: this.players[i].lynched
    };
    list.push(obj);
  }

  return list;
}

// Gets the current game state, suitable for sending to all players
Game.prototype.gameState = function() {
  var obj = {
    room: this.room,
    closed: this.closed,
    started: this.started,
    dayNumber: this.dayNumber,
    night: this.night,
    settings: this.settings,
    lastAction: this.lastAction
  };
  return obj;
}

// Sends all players the current game state
Game.prototype.sendGameStateUpdate = function() {
  _this = this;

  this.players.forEach(function(p) {
    if(p.leftGame) return;

    p.socket.emit("gameState", _this.gameState());
  });
}

// Sends all players of a specific role the updated player list
// If role is null, sends it to everyone
Game.prototype.sendPlayerListUpdate = function(role) {
  _this = this;

  this.players.forEach(function(p) {
    if(p.role === role || role === null) {
      if(p.leftGame) return;

      p.socket.emit("playerList", _this.playerListFor(p.id));
    }
  });
}

Game.prototype.dealRoles = function() {
  var roleList = [];
  for(var role in this.settings.roles) {
    if(this.settings.roles.hasOwnProperty(role)) {
      for(var i = 0; i < this.settings.roles[role]; i++) {
        roleList.push(role);
      }
    }
  }

  // Fill remaining roles with villager
  while(roleList.length < this.players.length) {
    roleList.push("villager");
  }

  // Shuffle the roles
  for(var i = roleList.length - 1; i >= 0; i--) {
    var j = Math.floor(Math.random() * (i + 1));
    var t = roleList[i];
    roleList[i] = roleList[j];
    roleList[j] = t;
  }

  // Deal one role out to each player
  this.players.forEach(function(p, i) {
    p.role = roleList[i];
  });
}

// Resets some values and starts the night
Game.prototype.startNight = function() {
  this.night = true;

  this.sendGameStateUpdate();
}

// Checks whether all multi-roles agree on what's happening
Game.prototype.isNightOver = function() {
  var targets = {};
  for(var i = 0; i < this.players.length; i++) {
    var p = this.players[i];
    if(p.dead)
      continue;

    if(p.votingFor === null && canVote[p.role])
      return false;

    if(targets[p.role] === undefined || targets[p.role] === p.votingFor) {
      targets[p.role] = p.votingFor;
    } else {
      return false;
    }
  }
  return targets;
}

// Figures out what happened and ends the night
Game.prototype.endNight = function() {
  _this = this;

  var targets = this.isNightOver();

  var deadPeople = [];

  if(targets["mafia"] === targets["bodyguard"]) {
    // Kill off one of the mafia
    for(var i = 0; i < this.players.length; i++) {
      var p = this.players[i];
      if(p.role === "mafia") {
        p.dead = true;
        deadPeople.push(p);
        targets["mafia"] = null;
        break;
      }
    };
  }

  if(targets["mafia"] === targets["doctor"]) {
    targets["mafia"] = null;
  }

  if(targets["mafia"]) {
    var p = this.playerById(targets["mafia"]);
    p.dead = true;
    deadPeople.push(p);
  }

  // Cop info
  if(targets["cop"]) {
    var copTarget = this.playerById(targets["cop"]);
    var guilty = copTarget.role === "mafia" || copTarget.role === "doctor" || copTarget.role === "bodyguard";
    this.players.forEach(function(p) {
      if(p.role === "cop") {
        p.socket.emit("copInfo", copTarget.name + " is " + (guilty ? "guilty" : "innocent") + ".");
      }
    });
  }

  if(deadPeople.length == 0) {
    this.lastAction = "Nobody died last night.";
  } else {
    this.lastAction = "";
    deadPeople.forEach(function(p) {
      _this.lastAction += getDeathQuote(p.name) + " ";
    });
  }

  this.players.forEach(function(p) {
    p.votingFor = null;
  });

  this.night = false;
  this.dayNumber += 1;

  this.sendGameStateUpdate();
  this.sendPlayerListUpdate(null);
}

Game.prototype.gameAction = function(id, action, data) {
  this.lastUpdate = new Date();

  if(action === "close") { // Closes the game
    if(id === this.moderator && !this.closed) {
      this.closed = true;
      this.sendGameStateUpdate();
    }
  } else if(action === "settings") {
    if(id === this.moderator && this.closed && !this.started) {
      this.started = true;
      this.settings = data;
      // TODO: Make sure settings are valid; right now, trust that the client is a nice client

      this.dealRoles();

      this.sendGameStateUpdate();
      this.sendPlayerListUpdate(null);
    }
  } else if(action === "night") {
    if(id === this.moderator && this.started && !this.night) {
      this.startNight();
    }
  } else if(action === "lynch") {
    if(id === this.moderator && this.started && !this.night) {
      var target = this.playerById(data);
      if(target != null) {
        target.dead = true;
        target.lynched = true;

        this.lastAction = target.name + " was lynched. They were a " + target.role + ".";

        this.sendGameStateUpdate();
        this.sendPlayerListUpdate(null);
      }
    }
  } else if(action === "vote") {
    if(this.started && this.night) {
      var self = this.playerById(id);
      var target = this.playerById(data);

      // TODO: More validation
      if(self && target && self.id !== target.id && self.role !== target.role && !self.dead && !target.dead && canVote[self.role]) {
        self.votingFor = target.id;

        this.sendPlayerListUpdate(self.role);

        // Check if night is over
        if(this.isNightOver()) {
          // End the night!
          this.endNight();
        }
      }
    }
  }
}

module.exports = Game;