@screeps/engine
Version:
This is a module for Screeps standalone server. See [main repository](https://github.com/screeps/screeps) for more info.
431 lines (363 loc) • 19.5 kB
JavaScript
;
var q = require('q'),
_ = require('lodash'),
movement = require('./processor/intents/movement'),
utils = require('./utils'),
driver = utils.getDriver(),
C = driver.constants,
config = require('./config');
var roomsQueue,
usersQueue,
lastRoomsStatsSaveTime = 0,
currentHistoryPromise = q.when();
function processRoom(roomId, _ref) {
var intents = _ref.intents,
objects = _ref.objects,
users = _ref.users,
terrain = _ref.terrain,
gameTime = _ref.gameTime,
roomInfo = _ref.roomInfo,
flags = _ref.flags;
return q.when().then(function () {
var bulk = driver.bulkObjectsWrite(),
userBulk = driver.bulkUsersWrite(),
flagsBulk = driver.bulkFlagsWrite(),
oldObjects = {},
roomController,
hasNewbieWalls = false,
stats = driver.getRoomStatsUpdater(roomId),
objectsToHistory = {},
roomSpawns = [],
roomExtensions = [],
oldRoomInfo = _.clone(roomInfo);
roomInfo.active = false;
var terrainItem = terrain[_.findKey(terrain)];
if (terrainItem.terrain) {
terrain = terrainItem.terrain;
}
_.forEach(objects, function (object) {
if (!object) {
return;
}
if (object.type == 'creep') {
object._actionLog = object.actionLog;
object._ticksToLive = object.ageTime - gameTime;
object.actionLog = {
attacked: null,
healed: null,
attack: null,
rangedAttack: null,
rangedMassAttack: null,
rangedHeal: null,
harvest: null,
heal: null,
repair: null,
build: null,
say: null,
upgradeController: null,
reserveController: null
};
}
if (object.type == 'link') {
object._actionLog = object.actionLog;
object.actionLog = {
transferEnergy: null
};
}
if (object.type == 'lab') {
object._actionLog = object.actionLog;
object.actionLog = {
runReaction: null
};
}
if (object.type == 'tower') {
object._actionLog = object.actionLog;
object.actionLog = {
attack: null,
heal: null,
repair: null
};
}
if (object.type == 'controller') {
roomController = object;
}
if (object.type == 'observer') {
object.observeRoom = null;
}
if (object.user && object.user != '3' && !object.userNotActive && object.type != 'flag') {
roomInfo.active = true;
}
if (object.type == 'powerBank' && gameTime > object.decayTime - 500) {
roomInfo.active = true;
}
if (object.type == 'energy') {
roomInfo.active = true;
}
if (object.type == 'nuke') {
roomInfo.active = true;
}
if (object.type == 'portal') {
roomInfo.active = true;
}
if (object.type == 'constructedWall' && object.decayTime && object.user) {
hasNewbieWalls = true;
}
if (object.type == 'extension') {
roomExtensions.push(object);
}
if (object.type == 'spawn') {
roomSpawns.push(object);
}
driver.config.emit('processObject', object, objects, terrain, gameTime, roomInfo, bulk, userBulk);
});
if (roomSpawns.length || roomExtensions.length) {
require('./processor/intents/_calc_spawns')(roomSpawns, roomExtensions, roomController, bulk);
}
movement.init(objects, terrain);
if (intents) {
_.forEach(intents.users, function (userIntents, userId) {
if (userIntents.objectsManual) {
userIntents.objects = userIntents.objects || {};
_.extend(userIntents.objects, userIntents.objectsManual);
}
for (var objectId in userIntents.objects) {
var objectIntents = userIntents.objects[objectId],
object = objects[objectId];
if (objectId == 'room') {
require('./processor/intents/room/intents')(userId, objectIntents, objects, terrain, bulk, userBulk, roomController, flags, flagsBulk);
continue;
}
if (!object || object._skip || object.user && object.user != userId) continue;
if (object.type == 'creep') require('./processor/intents/creeps/intents')(object, objectIntents, objects, terrain, bulk, userBulk, roomController, stats, gameTime, roomInfo, users);
if (object.type == 'link') require('./processor/intents/links/intents')(object, objectIntents, objects, terrain, bulk, userBulk, roomController, stats);
if (object.type == 'tower') require('./processor/intents/towers/intents')(object, objectIntents, objects, terrain, bulk, userBulk, roomController, stats, gameTime, roomInfo);
if (object.type == 'lab') require('./processor/intents/labs/intents')(object, objectIntents, objects, terrain, bulk, userBulk, roomController, stats);
if (object.type == 'spawn') require('./processor/intents/spawns/intents')(object, objectIntents, objects, terrain, bulk, userBulk, roomController, stats, gameTime);
if (object.type == 'constructionSite') {}
if (object.type == 'rampart') {
if (objectIntents.setPublic) {
require('./processor/intents/ramparts/set-public')(object, objectIntents.setPublic, objects, terrain, bulk, userBulk, roomController);
}
}
if (object.type == 'terminal') {
if (objectIntents.send) {
require('./processor/intents/terminal/send')(object, objectIntents.send, objects, terrain, bulk, userBulk, roomController);
}
}
if (object.type == 'nuker') {
if (objectIntents.launchNuke) {
require('./processor/intents/nukers/launch-nuke')(object, objectIntents.launchNuke, objects, terrain, bulk, userBulk, roomController, stats, gameTime, roomInfo);
}
}
if (object.type == 'observer') {
if (objectIntents.observeRoom) {
object.observeRoom = objectIntents.observeRoom.roomName;
}
}
if (object.type == 'powerSpawn') {
require('./processor/intents/power-spawns/intents')(object, objectIntents, objects, terrain, bulk, userBulk, roomController, stats);
}
if (object.type == 'extension' || object.type == 'storage' || object.type == 'powerSpawn' || object.type == 'terminal' || object.type == 'container') {
if (objectIntents.transfer) {
require('./processor/intents/extensions/transfer')(object, objectIntents.transfer, objects, terrain, bulk, userBulk, roomController);
}
}
if (object.type == 'controller') {
if (objectIntents.unclaim) {
require('./processor/intents/controllers/unclaim')(object, objectIntents.unclaim, objects, terrain, bulk, userBulk, gameTime, roomInfo, users);
}
if (objectIntents.activateSafeMode) {
require('./processor/intents/controllers/activateSafeMode')(object, objectIntents.activateSafeMode, objects, terrain, bulk, userBulk, gameTime, roomInfo);
}
}
if (objectIntents.notifyWhenAttacked && (C.CONSTRUCTION_COST[object.type] || object.type == 'creep')) {
bulk.update(object, { notifyWhenAttacked: !!objectIntents.notifyWhenAttacked.enabled });
}
driver.config.emit('processObjectIntents', object, userId, objectIntents, objects, terrain, gameTime, roomInfo, bulk, userBulk);
}
});
}
movement.check(roomController && roomController.safeMode > gameTime ? roomController.user : false);
var energyAvailable = _(objects).filter(function (i) {
return !i.off && (i.type == 'spawn' || i.type == 'extension');
}).sum('energy');
var mapView = {
w: [],
r: [],
pb: [],
p: [],
s: [],
c: [],
m: [],
k: []
};
var resultPromises = [];
var userVisibility = {};
_.forEach(objects, function (object) {
if (!object || object._skip) {
return;
}
if (object.type == 'energy') require('./processor/intents/energy/tick')(object, objects, terrain, bulk, userBulk, roomController);
if (object.type == 'source') require('./processor/intents/sources/tick')(object, objects, terrain, bulk, userBulk, roomController, gameTime);
if (object.type == 'mineral') require('./processor/intents/minerals/tick')(object, objects, terrain, bulk, userBulk, roomController, gameTime);
if (object.type == 'creep') require('./processor/intents/creeps/tick')(object, objects, terrain, bulk, userBulk, roomController, stats, gameTime);
if (object.type == 'spawn') require('./processor/intents/spawns/tick')(object, objects, terrain, bulk, userBulk, roomController, stats, energyAvailable);
if (object.type == 'rampart') require('./processor/intents/ramparts/tick')(object, objects, terrain, bulk, userBulk, roomController, gameTime);
if (object.type == 'extension') require('./processor/intents/extensions/tick')(object, objects, terrain, bulk, userBulk, roomController, gameTime);
if (object.type == 'road') require('./processor/intents/roads/tick')(object, objects, terrain, bulk, userBulk, roomController, gameTime);
if (object.type == 'constructionSite') require('./processor/intents/construction-sites/tick')(object, objects, terrain, bulk, userBulk, roomController);
if (object.type == 'keeperLair') require('./processor/intents/keeper-lairs/tick')(object, objects, terrain, bulk, userBulk, roomController, gameTime);
if (object.type == 'portal') require('./processor/intents/portals/tick')(object, objects, terrain, bulk, userBulk, roomController, gameTime);
if (object.type == 'constructedWall') require('./processor/intents/constructedWalls/tick')(object, objects, terrain, bulk, userBulk, roomController, gameTime);
if (object.type == 'link') require('./processor/intents/links/tick')(object, objects, terrain, bulk, userBulk, roomController);
if (object.type == 'extractor') require('./processor/intents/extractors/tick')(object, objects, terrain, bulk, userBulk, roomController);
if (object.type == 'tower') require('./processor/intents/towers/tick')(object, objects, terrain, bulk, userBulk, roomController);
if (object.type == 'controller') require('./processor/intents/controllers/tick')(object, objects, terrain, bulk, userBulk, roomController, gameTime, roomInfo, users);
if (object.type == 'lab') require('./processor/intents/labs/tick')(object, objects, terrain, bulk, userBulk, roomController, gameTime);
if (object.type == 'container') require('./processor/intents/containers/tick')(object, objects, terrain, bulk, userBulk, roomController, gameTime);
if (object.type == 'terminal') require('./processor/intents/terminal/tick')(object, objects, terrain, bulk, userBulk, roomController, gameTime);
if (object.type == 'nuke') {
require('./processor/intents/nukes/tick')(object, objects, terrain, bulk, userBulk, roomController, stats, gameTime, roomInfo);
}
if (object.type == 'observer') {
bulk.update(object, { observeRoom: object.observeRoom });
//resultPromises.push(core.setUserRoomVisibility(object.user, object.observeRoom));
}
if (object.type == 'storage') {
if (roomController) {
var energyCapacity = roomController.level > 0 && roomController.user == object.user && C.CONTROLLER_STRUCTURES.storage[roomController.level] > 0 ? C.STORAGE_CAPACITY : 0;
if (energyCapacity != object.energyCapacity) {
bulk.update(object, { energyCapacity: energyCapacity });
}
}
}
if (object.type == 'powerBank') {
if (gameTime >= object.decayTime - 1) {
bulk.remove(object._id);
delete objects[object._id];
}
}
if (object.type != 'flag') {
objectsToHistory[object._id] = object;
if (object.type == 'creep') {
objectsToHistory[object._id] = JSON.parse(JSON.stringify(object));
objectsToHistory[object._id]._id = "" + object._id;
delete objectsToHistory[object._id]._actionLog;
delete objectsToHistory[object._id]._ticksToLive;
if (object.actionLog.say && !object.actionLog.say.isPublic) {
delete objectsToHistory[object._id].actionLog.say;
}
}
}
if (object.user) {
//userVisibility[object.user] = true;
if (object.type != 'constructionSite' && !object.newbieWall && (object.type != 'rampart' || !object.isPublic)) {
mapView[object.user] = mapView[object.user] || [];
mapView[object.user].push([object.x, object.y]);
}
} else if (object.type == 'constructedWall') {
mapView.w.push([object.x, object.y]);
} else if (object.type == 'road') {
mapView.r.push([object.x, object.y]);
} else if (object.type == 'powerBank') {
mapView.pb.push([object.x, object.y]);
} else if (object.type == 'portal') {
mapView.p.push([object.x, object.y]);
} else if (object.type == 'source') {
mapView.s.push([object.x, object.y]);
} else if (object.type == 'mineral') {
mapView.m.push([object.x, object.y]);
} else if (object.type == 'controller') {
mapView.c.push([object.x, object.y]);
} else if (object.type == 'keeperLair') {
mapView.k.push([object.x, object.y]);
} else if (object.type == 'energy' && object.resourceType == 'power') {
mapView.pb.push([object.x, object.y]);
}
});
/*for(var user in userVisibility) {
resultPromises.push(core.setUserRoomVisibility(user, roomId));
}*/
driver.config.emit('processRoom', roomId, roomInfo);
driver.config.emit('processorLoopStage', 'saveRoom', roomId);
resultPromises.push(driver.mapViewSave(roomId, mapView));
resultPromises.push(bulk.execute());
resultPromises.push(userBulk.execute());
resultPromises.push(flagsBulk.execute());
if (!_.isEqual(roomInfo, oldRoomInfo)) {
resultPromises.push(driver.saveRoomInfo(roomId, roomInfo));
}
if (roomInfo.active) {
saveRoomHistory(roomId, objectsToHistory, gameTime);
}
if (Date.now() > lastRoomsStatsSaveTime + 30 * 1000) {
driver.roomsStatsSave();
lastRoomsStatsSaveTime = Date.now();
}
return q.all(resultPromises);
});
}
function saveRoomHistory(roomId, objects, gameTime) {
return currentHistoryPromise.then(function () {
var promise = q.when();
if (!(gameTime % driver.config.historyChunkSize)) {
var baseTime = Math.floor((gameTime - 1) / driver.config.historyChunkSize) * driver.config.historyChunkSize;
promise = driver.history.upload(roomId, baseTime);
}
var data = JSON.stringify(objects);
currentHistoryPromise = promise.then(function () {
return driver.history.saveTick(roomId, gameTime, data);
});
return currentHistoryPromise;
});
}
driver.connect('processor').then(function () {
return driver.queue.create('rooms', 'read');
}).then(function (_roomsQueue) {
roomsQueue = _roomsQueue;
function loop() {
var roomId;
driver.config.emit('processorLoopStage', 'start');
roomsQueue.fetch().then(function (_roomId) {
driver.config.emit('processorLoopStage', 'getRoomData', _roomId);
roomId = _roomId;
return q.all([driver.getRoomIntents(_roomId), driver.getRoomObjects(_roomId), driver.getRoomTerrain(_roomId), driver.getGameTime(), driver.getRoomInfo(_roomId), driver.getRoomFlags(_roomId)]);
}).then(function (result) {
driver.config.emit('processorLoopStage', 'processRoom', roomId);
processRoom(roomId, {
intents: result[0],
objects: result[1].objects,
users: result[1].users,
terrain: result[2],
gameTime: result[3],
roomInfo: result[4],
flags: result[5]
}).catch(function (error) {
return console.log('Error processing room ' + roomId + ':', _.isObject(error) ? error.stack || error : error);
}).then(function () {
return driver.clearRoomIntents(roomId);
}).then(function () {
return roomsQueue.markDone(roomId);
});
}).catch(function (error) {
return console.error('Error in processor loop:', _.isObject(error) && error.stack || error);
}).then(function () {
driver.config.emit('processorLoopStage', 'finish', roomId);
setTimeout(loop, 0);
});
}
loop();
}).catch(function (error) {
console.log('Error connecting to driver:', error);
});
if (typeof self == 'undefined') {
setInterval(function () {
var rejections = q.getUnhandledReasons();
rejections.forEach(function (i) {
return console.error('Unhandled rejection:', i);
});
q.resetUnhandledRejections();
}, 1000);
}
//# sourceMappingURL=sourcemaps/processor.js.map