-
Notifications
You must be signed in to change notification settings - Fork 0
/
role.harvester.js
76 lines (72 loc) · 2.49 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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
var s = require('shared');
var getSource = function(creep) {
var sources = creep.room.find(FIND_DROPPED_RESOURCES, {filter: (resource) => resource.type = RESOURCE_ENERGY});
if (sources.length === 0) {
sources = creep.room.find(FIND_SOURCES, {
filter: (src) => src.energy > 0
});
}
if (sources.length > 0) {
return creep.pos.findClosestByRange(sources);
} else {
console.log('no source here');
return {};
}
};
var getTarget = function getTarget(creep) {
var targets = creep.room.find(FIND_STRUCTURES, {
filter: (structure) => {
return (structure.structureType == STRUCTURE_EXTENSION || structure.structureType == STRUCTURE_SPAWN) &&
structure.energy < structure.energyCapacity;
}
});
if (targets.length === 0) {
targets = creep.room.find(FIND_STRUCTURES, {
filter: (structure) => {
return (structure.structureType == STRUCTURE_CONTAINER || structure.structureType == STRUCTURE_STORAGE) && structure.store[RESOURCE_ENERGY] < structure.storeCapacity;
}
});
}
if (targets.length === 0) {
targets = creep.room.find(FIND_STRUCTURES, {
filter: (structure) => {
return (structure.structureType == STRUCTURE_TOWER) && structure.energy < structure.energyCapacity;
}
});
}
return creep.pos.findClosestByRange(targets);
};
var roleHarvester = {
run: function(creep) {
var err = OK;
var m = creep.memory;
var source = getSource(creep);
if(creep.carry.energy < creep.carryCapacity) {
if (source.resourceType === undefined) {
err = creep.harvest(source);
} else {
err = creep.pickup(source);
}
switch (err) {
case ERR_NOT_IN_RANGE:
creep.moveTo(source, {visualizePathStyle: {stroke: '#ffaa00'}});
break;
case ERR_FULL:
source = null;
break;
case OK:
break;
default:
s.creepErr(creep, err);
break;
}
} else {
let target = getTarget(creep);
if(creep.transfer(target, RESOURCE_ENERGY) == ERR_NOT_IN_RANGE) {
creep.moveTo(target, {visualizePathStyle: {stroke: '#ffffff'}});
}
}
return OK;
}
};
module.exports = roleHarvester;