more refactoring, added miniums

This commit is contained in:
Tyrel Souza 2016-06-28 16:49:31 -04:00
parent 490877fb9c
commit d7223b6d60

16
main.js
View File

@ -2,6 +2,14 @@ var roleHarvester = require('role.harvester');
var roleUpgrader = require('role.upgrader'); var roleUpgrader = require('role.upgrader');
var roleBuilder = require('role.builder'); var roleBuilder = require('role.builder');
var _defaultBodyParts = [WORK, CARRY, MOVE];
var _roleNames = ["harvester", "builder", "upgrader"];
var _minimums = {
harvester: 4,
builder: 2,
upgrader: 2
};
module.exports.loop = function(){ module.exports.loop = function(){
// Cleanup Creeps // Cleanup Creeps
for (var name in Memory.creeps){ for (var name in Memory.creeps){
@ -12,11 +20,11 @@ module.exports.loop = function(){
} }
// Refactor some similar behavior for all roles // Refactor some similar behavior for all roles
var roleNames = ["harvester", "builder", "upgrader"];
roleNames.forEach(function(roleName){ _roleNames.forEach(function(roleName){
var creepsOfKind = _.filter(Game.creeps, (creep) => creep.memory.role == roleName); var creepsOfKind = _.filter(Game.creeps, (creep) => creep.memory.role == roleName);
if (creepsOfKind.length < 2){ if (creepsOfKind.length < _minimums[roleName]){
var newName = Game.spawns.Spawn1.createCreep([WORK, CARRY, MOVE], undefined, {role: roleName}); var newName = Game.spawns.Spawn1.createCreep(_defaultBodyParts, undefined, {role: roleName});
console.log("Spawning new ", roleName, ": ", newName); console.log("Spawning new ", roleName, ": ", newName);
} }
}); });