@screeps/engine
Version:
This is a module for Screeps standalone server. See [main repository](https://github.com/screeps/screeps) for more info.
41 lines (34 loc) • 1.34 kB
JavaScript
var _ = require('lodash'),
utils = require('../../utils'),
driver = utils.getDriver(),
C = driver.constants;
module.exports = function(roomSpawns, roomExtensions, roomController, bulk) {
var spawns = roomSpawns;
if(spawns.length > C.CONTROLLER_STRUCTURES.spawn[roomController.level|0]) {
spawns.sort(utils.comparatorDistance(roomController));
spawns = _.take(spawns, C.CONTROLLER_STRUCTURES.spawn[roomController.level|0]);
roomSpawns.forEach(i => i._off = !_.contains(spawns, i));
}
else {
roomSpawns.forEach(i => i._off = false);
}
roomSpawns.forEach(i => {
if(i._off !== i.off) {
bulk.update(i._id, {off: i._off});
}
});
var extensions = roomExtensions;
if(extensions.length > C.CONTROLLER_STRUCTURES.extension[roomController.level|0]) {
extensions.sort(utils.comparatorDistance(roomController));
extensions = _.take(extensions, C.CONTROLLER_STRUCTURES.extension[roomController.level|0]);
roomExtensions.forEach(i => i._off = !_.contains(extensions, i));
}
else {
roomExtensions.forEach(i => i._off = false);
}
roomExtensions.forEach(i => {
if(i._off !== i.off) {
bulk.update(i._id, {off: i._off});
}
});
};