Added unit banning support to server interface

client/src/skeleton.coffee

@@ -5,9 +5,7 @@ Skeleton::playerJoined = (gameId, playerNumber, playerName, isMe) ->
 
 Skeleton::playerLeft = (gameId, playerNumber) ->
 
-Skeleton::unitBanned = (unitType) ->
-
-Skeleton::unitUnbanned = (unitType) ->
+Skeleton::bannedUnits = (unitTypes) ->
 
 Skeleton::gameStarted = (gameId) ->
 

server/game/management.js

@@ -321,3 +321,26 @@ GameManagement.prototype.myGames = function(userId, callback) {
     }
   });
 }
+
+GameManagement.prototype.setBannedUnits = function(gameId, bannedUnits, userId, callback) {
+  var database = this.database;
+  database.game(gameId, function(result) {
+    if(!result.success) {
+      callback({success: false, reason: result.reason});
+    } else if(result.game.authorId != userId) {
+      callback({success: false, reason: "Not the game author!"});
+    } else if(result.game.state != result.game.STATE_PREGAME) {
+      callback({success: false, reason: "Banned units can only be set during pregame!"});
+    } else {
+      var game = result.game;
+      game.settings.bannedUnits = bannedUnits;
+      database.updateGame(game, function(result) {
+        if(result.success) {
+          callback({success: true});
+        } else {
+          callback({success: false, reason: result.reason});
+        }
+      });
+    }
+  });
+}

server/messenger.js

@@ -21,6 +21,13 @@ Messenger.prototype.sendGameEvents = function(gameId, events) {
   }, "game-" + gameId);
 }
 
+Messenger.prototype.sendBannedUnits = function(gameId, bannedUnits) {
+  this.subscriptions.forSubscribers(function(sub) {
+    if(sub.client.stub.bannedUnits)
+      sub.client.stub.bannedUnits(gameId, bannedUnits);
+  }, "game-" + gameId);
+}
+
 Messenger.prototype.sendPlayerJoined = function(gameId, playerNumber, username, userId) {
   this.subscriptions.forSubscribers(function(sub) {
     var isMe = userId == sub.session.userId;

server/skeleton.js

@@ -579,6 +579,32 @@ Skeleton.prototype.gameRules = function(gameId) {
   }
 }
 
+Skeleton.prototype.setBannedUnits = function(gameId, bannedUnits) {
+  if(!requireArgs([gameId, bannedUnits])) return {success: false, reason: "Missing method arguments!"};
+
+  if(this.sessionId === null)
+    return {success: false, reason: "Not logged in"}
+
+  var requestId = this.client.requestId;
+  var this_ = this;
+  var userId = this.session.userId;
+
+  var mutex = this.server.gameMutex(gameId);
+  mutex.lock(function() {
+    var timer = new utils.Timer("Skeleton.setBannedUnits");
+    this_.server.gameManagement.setBannedUnits(gameId, bannedUnits, userId, function(result) {
+      if(result.success) {
+        this_.server.messenger.sendBannedUnits(gameId, bannedUnits);
+        this_.client.sendResponse(requestId, {success: true});
+        timer.end();
+      } else {
+        this_.client.sendResponse(requestId, {success: false, reason: result.reason});
+      }
+      mutex.release();
+    });
+  });
+}
+
 Skeleton.prototype.gameData = function(gameId) {
   if(!requireArgs([gameId])) return {success: false, reason: "Missing method arguments!"};
   var timer = new utils.Timer("Skeleton.gameData");