-
Notifications
You must be signed in to change notification settings - Fork 0
/
role.upgrader.js
44 lines (39 loc) · 1.47 KB
/
role.upgrader.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
var roleUpgrader = {
/** @param {Creep} creep **/
run: function(creep) {
if(creep.memory.upgrading && creep.carry.energy == 0) {
creep.memory.upgrading = false;
//creep.say('harvesting');
}
if(!creep.memory.upgrading && creep.carry.energy == creep.carryCapacity) {
creep.memory.upgrading = true;
//creep.say('upgrading');
}
if(creep.memory.upgrading) {
if(creep.upgradeController(creep.room.controller) == ERR_NOT_IN_RANGE) {
creep.moveTo(creep.room.controller);
}
}
else {
var sources = creep.room.find(FIND_STRUCTURES, {
filter: (structure) => {
return (structure.structureType == STRUCTURE_EXTENSION ||
structure.structureType == STRUCTURE_SPAWN);
}
});
var source = sources[0];
var result = undefined;
if (source && creep.carryCapacity <= source.energy) {
result = source.transferEnergy(creep, creep.carryCapacity);
} else if (Memory.info.transporters < 1) {
var sources = creep.room.find(FIND_SOURCES);
source = sources[0];
result = creep.harvest(source)
}
if (result == ERR_NOT_IN_RANGE) {
creep.moveTo(source);
}
}
}
};
module.exports = roleUpgrader;