creep.behaviour.collapseWorker.js

let mod = {};
module.exports = mod;
mod.name = 'collapseWorker';
mod.run = function(creep) {
    if( creep.action == null || creep.action.name === 'idle' ) {
        this.nextAction(creep);
    }

    // Do some work
    if( creep.action && creep.target ) {
        creep.action.step(creep);
    } else {
        logError('Creep without action/activity!\nCreep: ' + creep.name + '\ndata: ' + JSON.stringify(creep.data));
    }
};
mod.nextAction = function(creep){
    if( creep.pos.roomName !== creep.data.homeRoom ) {
        if( DEBUG && TRACE ) trace('Behaviour', {actionName:'travelling', behaviourName:this.name, creepName:creep.name, assigned: true, Behaviour:'nextAction', Action:'assign'});
        Creep.action.travelling.assignRoom(creep, creep.data.homeRoom);
        return true;
    }

    if( !creep.room.collapsed ) {
        return Creep.action.recycling.assign(creep);
    }

    let priority;
    if( creep.sum < (creep.carryCapacity*0.5) ) {
        priority = [
            Creep.action.picking,
            Creep.action.withdrawing,
            Creep.action.uncharging,
            Creep.action.harvesting,
            Creep.action.dismantling,
            Creep.action.reallocating,
            Creep.action.idle];
    }
    else {
        if( creep.room.situation.invasion && creep.room.controller && creep.room.controller.level > 2 ){
            priority = [
                Creep.action.feeding,
                Creep.action.fueling,
                Creep.action.repairing,
                Creep.action.idle];
        } else {
            priority = [
                Creep.action.feeding,
                Creep.action.fueling,
                Creep.action.charging,
                Creep.action.repairing,
                Creep.action.building,
                Creep.action.fortifying,
                Creep.action.upgrading,
                Creep.action.storing,
                Creep.action.picking,
                Creep.action.idle];
        }
        if( creep.room.controller && creep.room.controller.ticksToDowngrade < 500 ) { // urgent upgrading
            priority.unshift(Creep.action.upgrading);
        }
    }
    for(var iAction = 0; iAction < priority.length; iAction++) {
        var action = priority[iAction];
        const valid = action.isValidAction(creep);
        if( DEBUG && TRACE ) trace('Action', {actionName:action.name, behaviourName:this.name, creepName:creep.name, valid, Action:'isValidAction'});
        if( !valid ) continue;

        const addable = action.isAddableAction(creep);
        if( DEBUG && TRACE ) trace('Action', {actionName:action.name, behaviourName:this.name, creepName:creep.name, addable, Action:'isAddableAction'});
        if( !addable ) continue;

        const assigned = action.assign(creep);
        if( assigned ) {
            if( DEBUG && TRACE ) trace(assigned ? 'Behaviour' : 'Action', {actionName:action.name, behaviourName:this.name, reepName:creep.name, assigned, Behaviour:'nextAction', Action:'assign'});
            return true;
        }
    }
    return false;
};
mod.strategies = {
    defaultStrategy: {
        name: `default-${mod.name}`,
        canWithdrawEnergy: function(creep, target) {
            return function(amount) {
                return amount > 0;
            };
        },
    },
};
mod.selectStrategies = function(actionName) {
    return [mod.strategies.defaultStrategy, mod.strategies[actionName]];
};