role.harvester.js

/**
 * Harvester is the simplest crip role that tries to get some 
 * energy directly from the source. It's small and versatile -
 * can supply energy to many diffent targets. It will get energy from 
 * a random energy source. Not very efficient. 
 * 
 */

var roleHarvester = {

    /** @param {Creep} creep **/
    run: function (creep) {

        if (creep.changeRoomsIfNeeded())
            return;

        if (creep.store.getFreeCapacity() > 0) {
            //still some capacity left, try to fill

            if (creep.pickDroppedEnergyIfAny()) {
                return;
            }

            let sources = creep.room.find(FIND_SOURCES);

            //if there is no dropped then just try to harvest normal sources
            if (creep.memory.assignment === undefined || creep.memory.assignment == 0) {
                //pick random source
                var r = Math.round(Math.random() * (sources.length - 1));
                creep.memory.assignment = sources[r].id;
            }

            var my_source = Game.getObjectById(creep.memory.assignment);

            if (creep.harvest(my_source) == ERR_NOT_IN_RANGE) {
                creep.moveTo(my_source, { visualizePathStyle: { stroke: '#FF0000' } });
                return;
            }


        }
        else {


            var targets = creep.room.find(FIND_STRUCTURES, {
                filter: (structure) => {
                    return (structure.structureType == STRUCTURE_EXTENSION ||
                        structure.structureType == STRUCTURE_STORAGE ||
                        structure.structureType == STRUCTURE_TOWER ||
                        structure.structureType == STRUCTURE_SPAWN)
                        &&
                        structure.store.getFreeCapacity(RESOURCE_ENERGY) > 0
                }
            });

            targets = targets.sort((a, b) =>
                (a.store.getCapacity() - b.store.getCapacity()));

            if (targets.length > 0) {

                let best_targets = _.filter(targets, (t) => t.store.getCapacity() == targets[0].store.getCapacity());
                my_target = creep.pos.findClosestByPath(best_targets);;

                if (creep.transfer(my_target, RESOURCE_ENERGY) == ERR_NOT_IN_RANGE) {
                    creep.moveTo(my_target, { visualizePathStyle: { stroke: '#95C8D8' } });
                }
            } else {
                //stores dont need energy -> upgrade controllers

                if (creep.upgradeController(creep.room.controller) == ERR_NOT_IN_RANGE) {
                    creep.moveTo(creep.room.controller);
                }

            }
        }
    }
};

module.exports = roleHarvester;