Room.js

'use strict';

const Task = require('Task');
const TaskNetwork = require('TaskNetwork');
const ReactionManager = require('economy.reactionManager');

Room.prototype.myOwned = function () {
    return this.controller && this.controller.my;
};


Room.prototype.hasInvaderAndNoTower = function(){
    let towers = this.find(FIND_MY_STRUCTURES, {
        filter: (s) => s.structureType === STRUCTURE_TOWER
    });
    let hostiles = this.find(FIND_HOSTILE_CREEPS, {
        filter: c => c.owner.username === 'Invader'
    });
    return towers.length === 0 && hostiles.length > 0;

};

Room.prototype.run = function () {
    this.cpuUsed = 0;
    let cpu = Game.cpu.getUsed();
    if(!this.myOwned()){
        return;
    }
    if(this.hasInvaderAndNoTower()){
        //suicide if invader and no tower
        _.forEach(_.filter(Game.creeps, (c) => c.memory.home === this.name),
            (c) => c.suicide());
    }
    let cManager = this.getColonyManager();
    cManager.run();
    let hManager = this.getHarvestManager();
    hManager.operateLink();
    let rManager = this.getResourceManager();
    try{
        rManager.run();
    }catch (e) {
        console.log(e.stack);
    }
    TaskNetwork.runTask(this);
    this.cpuUsed += (Game.cpu.getUsed() - cpu);
};

const HarvestManager = require('economy.harvestManager');
const ResourceManager = require('economy.resourceManager');

/**
 *
 * @return {HarvestManager}
 */
Room.prototype.getHarvestManager = function () {
    if (!this.harvestManager) {
        this.harvestManager = new HarvestManager(this);
    }
    return this.harvestManager;

};


/**
 *
 * @return {ColonyManager}
 */
Room.prototype.getColonyManager = function () {
    let ColonyManager = require('economy.colonyManager');
    if (!this.colonyManager) {
        this.colonyManager = new ColonyManager(this);
    }
    return this.colonyManager;

};

Room.prototype.countRole = function (role) {
    if (this.roles === undefined) {
        this.roles = {starter: 0};
        for (let name in Game.creeps) {
            if (!Game.creeps.hasOwnProperty(name) || Game.creeps[name].memory.home !== this.name) continue;
            if (this.roles[Game.creeps[name].memory.role] === undefined) {
                this.roles[Game.creeps[name].memory.role] = 1;
            } else {
                this.roles[Game.creeps[name].memory.role]++;
            }
        }
    }
    if(role === undefined){return _.sum(this.roles);}
    return this.roles[role] || 0;
};

Room.prototype.incrementRole = function(role){
    let c = this.countRole(role);
    this.roles[role] = c + 1;
};


/**
 *
 * @return {RoomBuilder}
 */
Room.prototype.getRoomBuilder = function () {
    let RoomBuilder = require('economy.roomBuilder');
    if (!this.roomBuilder) {
        this.roomBuilder = new RoomBuilder(this);
    }
    return this.roomBuilder;

};


/**
 *
 * @return {ResourceManager}
 */
Room.prototype.getResourceManager = function () {
    if (!this.resourceManager) {
        this.resourceManager = new ResourceManager(this);
    }
    return this.resourceManager;

};

/**
 *
 * @return {ReactionManager}
 */
Room.prototype.getReactionManager = function () {
    if (!this.reactionManager) {
        this.reactionManager = new ReactionManager(this);
    }
    return this.reactionManager;

};

Room.parseRoomName = function (roomName) {
    let pattern = /([NSEW]|[0-9]+)/g;
    let parts = roomName.match(pattern);
    if(parts.length !== 4){
        return;
    }
    return {
        xSect: parts[0],
        x: parseInt(parts[1]),
        ySect: parts[2],
        y: parseInt(parts[3])
    }

};


Room.prototype.getTasks = function () {
    return this.memory.tasks || [];
};

Room.prototype.addTask = function (task) {
    if(!this.memory.tasks) this.memory.tasks = [];
    this.memory.tasks.push(task);
};

Room.prototype.getCreeps = function () {
    return _.filter(Game.creeps, (c) => c.memory.home === this.name);
};