-
Notifications
You must be signed in to change notification settings - Fork 0
/
role.harvester.js
56 lines (53 loc) · 1.58 KB
/
role.harvester.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
45
46
47
48
49
50
51
52
53
54
55
56
module.exports = {
// a function to run the logic for this role
/** @param {Creep} creep */
run: function (creep) {
// if creep is bringing energy to a structure but has no energy left
if (
creep.memory.working == true &&
creep.store.getUsedCapacity(RESOURCE_ENERGY) == 0
) {
// switch state
creep.memory.working = false;
}
// if creep is harvesting energy but is full
else if (
creep.memory.working == false &&
creep.store.getFreeCapacity() == 0
) {
// switch state
creep.memory.working = true;
}
// if creep is supposed to transfer energy to a structure
if (creep.memory.working == true) {
// find closest spawn, extension or tower which is not full
var structure = creep.pos.findClosestByPath(FIND_MY_STRUCTURES, {
// the second argument for findClosestByPath is an object which takes
// a property called filter which can be a function
// we use the arrow operator to define it
filter: (s) =>
(s.structureType == STRUCTURE_SPAWN ||
s.structureType == STRUCTURE_EXTENSION ||
s.structureType == STRUCTURE_TOWER) &&
s.energy < s.energyCapacity,
});
if (structure == undefined) {
structure = creep.room.storage;
}
// if we found one
if (structure != undefined) {
// try to transfer energy, if it is not in range
if (creep.transfer(structure, RESOURCE_ENERGY) == ERR_NOT_IN_RANGE) {
// move towards it
creep.moveTo(structure);
}
}
}
// if creep is supposed to harvest energy from source
else {
if (creep.getSalvage() == -1) {
creep.getEnergy(false, true);
}
}
},
};