@screeps/engine
Version:
This is a module for Screeps standalone server. See [main repository](https://github.com/screeps/screeps) for more info.
457 lines (387 loc) • 20.3 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, {intents, objects, users, terrain, gameTime, roomInfo, flags}) {
return q.when().then(() => {
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, (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, (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((i) => !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, (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});
}
}
}
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(() => {
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(() => driver.history.saveTick(roomId, gameTime, data));
return currentHistoryPromise;
});
}
driver.connect('processor').then(() => driver.queue.create('rooms', 'read'))
.then(_roomsQueue => {
roomsQueue = _roomsQueue;
function loop() {
var roomId;
driver.config.emit('processorLoopStage','start');
roomsQueue.fetch()
.then((_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((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((error) => console.log('Error processing room '+roomId+':', _.isObject(error) ? (error.stack || error) : error))
.then(() => {
return driver.clearRoomIntents(roomId);
})
.then(() => roomsQueue.markDone(roomId));
})
.catch((error) => console.error('Error in processor loop:', _.isObject(error) && error.stack || error))
.then(() => {
driver.config.emit('processorLoopStage','finish', roomId);
setTimeout(loop, 0)
});
}
loop();
}).catch((error) => {
console.log('Error connecting to driver:', error);
});
if(typeof self == 'undefined') {
setInterval(() => {
var rejections = q.getUnhandledReasons();
rejections.forEach((i) => console.error('Unhandled rejection:', i));
q.resetUnhandledRejections();
}, 1000);
}