"use strict"; /* * Implements Gather Model * * Gather States * - Gathering * - Election (Electing leaders) * - Selection (Selecting teams) * - Done * */ var Gatherer = require("./gatherer"); var StateMachine = require("javascript-state-machine"); function Gather (options) { if (!(this instanceof Gather)) { return new Gather(options); } if (options && typeof options.onEvent === 'function') { this.onEvent = options.onEvent; } else { this.onEvent = ()=>{}; } if (options && typeof options.onDone === 'function') { this.onDone = options.onDone; } else { this.onDone = ()=>{}; } this.TEAM_SIZE = 6; this.gatherers = []; this.ELECTION_INTERVAL = 300000; // 5 mins this.electionStartTime = null; this.initState(); } Gather.prototype.alienLeader = function () { return this.gatherers.reduce(function (acc, gatherer) { if (gatherer.team === "alien" && gatherer.leader) acc.push(gatherer); return acc; }, []).pop(); }; Gather.prototype.marineLeader = function () { return this.gatherers.reduce(function (acc, gatherer) { if (gatherer.team === "marine" && gatherer.leader) acc.push(gatherer); return acc; }, []).pop(); }; Gather.prototype.assignMarineLeader = function (id) { this.modifyGatherer({id: id}, function (gatherer) { gatherer.leader = true; gatherer.team = "marine"; return gatherer; }); }; Gather.prototype.assignAlienLeader = function (id) { this.modifyGatherer({id: id}, function (gatherer) { gatherer.leader = true; gatherer.team = "alien"; return gatherer; }); }; Gather.prototype.containsUser = function (user) { return this.gatherers.some(function (gatherer) { return gatherer.id === user.id; }); }; Gather.prototype.addUser = function (user) { if (this.containsUser(user)) { return null; } var gatherer = new Gatherer(user); this.gatherers.push(gatherer); return gatherer; }; Gather.prototype.removeUser = function (user) { this.gatherers = this.gatherers.filter(function (gatherer) { return user.id !== gatherer.id; }); }; Gather.prototype.modifyGatherer = function (user, callback) { this.gatherers.forEach(function (gatherer, index, array) { if (gatherer.id === user.id) { var modifiedUser = callback(gatherer); array[index] = modifiedUser; } }); } Gather.prototype.moveToMarine = function (user) { if (this.marines().length >= this.TEAM_SIZE) return; this.modifyGatherer(user, function (gatherer) { gatherer.team = "marine"; return gatherer; }); }; Gather.prototype.moveToAlien = function (user) { if (this.aliens().length >= this.TEAM_SIZE) return; this.modifyGatherer(user, function (gatherer) { gatherer.team = "alien"; return gatherer; }); }; Gather.prototype.moveToLobby = function (user) { this.gatherers.forEach(function (gatherer, index, array) { if (gatherer.id === user.id) { gatherer.team = "lobby"; array[index] = gatherer; } }); }; Gather.prototype.retrieveGroup = function (team) { return this.gatherers.filter(function (gatherer) { return gatherer.team === team; }); } Gather.prototype.lobby = function () { return this.retrieveGroup("lobby"); }; Gather.prototype.aliens = function () { return this.retrieveGroup("alien"); }; Gather.prototype.marines = function () { return this.retrieveGroup("marine"); }; Gather.prototype.toJson = function () { return { gatherers: this.gatherers, state: this.current, election: { startTime: (this.electionStartTime === null) ? null : this.electionStartTime.toISOString(), interval: this.ELECTION_INTERVAL } } }; Gather.prototype.voteForMap = function (voter, mapId) { this.gatherers.forEach(function (gatherer, index, array) { if (gatherer.id === voter.id) { array[index].mapVote = mapId; } }); }; Gather.prototype.voteForServer = function (voter, serverId) { this.gatherers.forEach(function (gatherer, index, array) { if (gatherer.id === voter.id) { array[index].serverVote = serverId; } }); }; // Returns an array of IDs representing votes for leaders Gather.prototype.leaderVotes = function () { var self = this; return self.gatherers.map(function (gatherer) { return gatherer.leaderVote }).filter(function (leaderId) { return (typeof leaderId === 'number'); }).filter(function (candidate) { return self.gatherers.some(function (gatherer) { return gatherer.id === candidate; }); }); }; Gather.prototype.voteForLeader = function (voter, candidate) { // Find voter and then assign their vote to candidate id this.gatherers.forEach(function (gatherer, index, array) { if (gatherer.id === voter.id) { array[index].voteForLeader(candidate); } }); }; Gather.prototype.getGatherer = function (user) { var matchingGatherer = null; this.gatherers.forEach(function (gatherer) { if (gatherer.id === user.id) matchingGatherer = gatherer; }); return matchingGatherer; }; // Gather States // - Gathering // - Election // - Selection // - Done // gather.current - contains the current state // gather.is(s) - return true if state s is the current state // gather.can(e) - return true if event e can be fired in the current state // gather.cannot(e) - return true if event e cannot be fired in the current state // gather.transitions() - return list of events that are allowed from the current state StateMachine.create({ target: Gather.prototype, events: [ { name: "initState", from: "none", to: "gathering" }, { name: "addGatherer", from: "gathering", to: "election" }, { name: "selectLeader", from: "election", to: "selection" }, { name: "electionTimeout", from: "election", to: "selection" }, { name: "confirmSelection", from: "selection", to: "done" }, { name: "removeGatherer", from: ["gathering", "election", "selection"], to: "gathering" } ], callbacks: { // Callbacks for events onafterevent: function () { this.onEvent.call(this); }, // Gathering State onbeforeaddGatherer: function (event, from, to, user) { this.addUser(user); if (this.gatherers.length !== 12) { return false; } }, // Election State onbeforeselectLeader: function (event, from, to, voter, candidate) { this.voteForLeader(voter, candidate); if (this.leaderVotes().length !== 12) { return false; } }, onenterelection: function () { // Setup timer for elections var self = this; self.electionStartTime = new Date(); setTimeout(function () { if (self.can("electionTimeout")) { self.electionTimeout(); } }, self.ELECTION_INTERVAL); }, onleaveelection: function () { this.electionStartTime = null; }, // Selection State onenterselection: function () { // Remove all leaders and teams this.gatherers.forEach(function (gatherer, index, array) { array[index].leader = false; array[index].team = "lobby"; }); // Assign leaders based on vote // 1st place alien comm // 2nd place marine comm var voteCount = {}; this.gatherers.forEach(function (gatherer) { voteCount[gatherer.id] = 0; }); this.leaderVotes().forEach(function (candidateId) { voteCount[candidateId]++; }); var rank = []; for (var candidate in voteCount) { rank.push({ candidate: candidate, count: voteCount[candidate] }); } rank.sort(function (a, b) { return a.count - b.count; }); this.assignAlienLeader(parseInt(rank.pop().candidate, 0)); this.assignMarineLeader(parseInt(rank.pop().candidate, 0)); }, onbeforeconfirmSelection: function (event, from, to, leader) { return (this.aliens().length === this.TEAM_SIZE && this.marines().length === this.TEAM_SIZE); }, // Remove gatherer event onbeforeremoveGatherer: function (event, from, to, user) { // Cancel transition if no gatherers have been removed let userCount = this.gatherers.length; this.removeUser(user); return (userCount > this.gatherers.length); }, // On enter done onenterdone: function () { this.onDone.call(this); } } }); module.exports = Gather;