Files
screeps/JobHarvester.js

42 lines
1.8 KiB
JavaScript

module.exports = {
begin(creep){
const jobMemory = creep.memory.job;
if(!jobMemory.harvesting) creep.memory.job.harvesting = false;
if(!jobMemory.counter) creep.memory.job.counter = 0;
creep.memory.init = true;
},
tick(creep){
const jobMemory = creep.memory.job;
if(!jobMemory.init) this.begin(creep);
if(jobMemory.harvesting){
const roomSources = creep.room.memory.layout.sources
const source = Game.getObjectById(roomSources[jobMemory.counter%roomSources.length]);
if(creep.pos.isNearTo(source)) creep.harvest(source);
else creep.moveTo(source);
if(!creep.store.getFreeCapacity(RESOURCE_ENERGY)) jobMemory.harvesting = false;
} else{
var target;
switch (jobMemory.counter%3) {
case 0: target = creep.room.controller; break;
case 1: target = creep.pos.findClosestByRange(FIND_MY_STRUCTURES, {filter:{structureType:STRUCTURE_SPAWN}}); break;
case 2: target = creep.pos.findClosestByRange(FIND_MY_CONSTRUCTION_SITES); break;
}
var response;
if(creep.pos.isNearTo(target)) {
switch (jobMemory.counter%3) {
case 0: response = creep.upgradeController(target); break;
case 1: response = creep.transfer(target, RESOURCE_ENERGY); break;
case 2: response = creep.build(target); break;
}
if(response == ERR_FULL) creep.memory.job.counter += 1;
}
else creep.moveTo(target);
if(!creep.store.getUsedCapacity(RESOURCE_ENERGY)) {
jobMemory.harvesting = true;
creep.memory.job.counter += 1;
}
}
}
}