screeps/role.harvester.ts at main · SystemDisc/screeps · GitHub

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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
const returnEnergy = (myHarvester: Creep, spawn: StructureSpawn) => {
  if (myHarvester.transfer(spawn, RESOURCE_ENERGY) == ERR_NOT_IN_RANGE) {
    myHarvester.moveTo(spawn);
  }
}

const findClosestSpawnForTransfer = (myHarvester: Creep) => {
  return myHarvester.room.find(FIND_MY_SPAWNS).reduce<StructureSpawn | null>((closestSpawn, spawn) => (
    ((closestSpawn && myHarvester.pos.getRangeTo(spawn) < myHarvester.pos.getRangeTo(closestSpawn)) || (!closestSpawn)) &&
    spawn.store[RESOURCE_ENERGY] < spawn.store.getCapacity(RESOURCE_ENERGY) ? spawn : closestSpawn
  ), null);
}

export const harvesterRun = (room: Room) => {
  const myHarvesters = room.find(FIND_MY_CREEPS).filter(creep => creep.name.startsWith('Harvester'));

  const spawn = room.find(FIND_MY_SPAWNS).reduce((highestEnergySpawn, spawn) => (
    spawn.store[RESOURCE_ENERGY] > highestEnergySpawn.store[RESOURCE_ENERGY] ? spawn : highestEnergySpawn
  ));

  if (myHarvesters.length === 0) {
    if (spawn.store[RESOURCE_ENERGY] < 300) {
      spawn.spawnCreep([WORK, CARRY, MOVE], 'Harvester1');
    } else {
      spawn.spawnCreep([WORK, CARRY, CARRY, MOVE, MOVE], 'Harvester1');
    }
  } else if (myHarvesters.length < 10) {
    if (spawn.store[RESOURCE_ENERGY] >= 300) {
      spawn.spawnCreep([WORK, CARRY, CARRY, MOVE, MOVE], `Harvester${Math.random()}`);
    }
  }

  for (const myHarvester of myHarvesters) {
    if (myHarvester.store[RESOURCE_ENERGY] < myHarvester.store.getCapacity(RESOURCE_ENERGY)) {
      const source = room.find(FIND_SOURCES).reduce((closestSource, source) => {
        const upTiles = room.lookAt(source.pos.x, source.pos.y - 1);
        const upRightTiles = room.lookAt(source.pos.x + 1, source.pos.y - 1);
        const rightTiles = room.lookAt(source.pos.x + 1, source.pos.y);
        const downRightTiles = room.lookAt(source.pos.x + 1, source.pos.y + 1);
        const downTiles = room.lookAt(source.pos.x, source.pos.y + 1);
        const downLeftTiles = room.lookAt(source.pos.x - 1, source.pos.y + 1);
        const leftTiles = room.lookAt(source.pos.x - 1, source.pos.y);
        const upLeftTiles = room.lookAt(source.pos.x - 1, source.pos.y - 1);
        const sourceAvailable = (
          (upTiles.length === 1 && upTiles[0].terrain !== 'wall') ||
          (upRightTiles.length === 1 && upRightTiles[0].terrain !== 'wall') ||
          (rightTiles.length === 1 && rightTiles[0].terrain !== 'wall') ||
          (downRightTiles.length === 1 && downRightTiles[0].terrain !== 'wall') ||
          (downTiles.length === 1 && downTiles[0].terrain !== 'wall') ||
          (downLeftTiles.length === 1 && downLeftTiles[0].terrain !== 'wall') ||
          (leftTiles.length === 1 && leftTiles[0].terrain !== 'wall') ||
          (upLeftTiles.length === 1 && upLeftTiles[0].terrain !== 'wall') ||
          myHarvester.pos.getRangeTo(source) === 1
        );
        const closestUpTiles = room.lookAt(closestSource.pos.x, closestSource.pos.y - 1);
        const closestUpRightTiles = room.lookAt(closestSource.pos.x + 1, closestSource.pos.y - 1);
        const closestRightTiles = room.lookAt(closestSource.pos.x + 1, closestSource.pos.y);
        const closestDownRightTiles = room.lookAt(closestSource.pos.x + 1, closestSource.pos.y + 1);
        const closestDownTiles = room.lookAt(closestSource.pos.x, closestSource.pos.y + 1);
        const closestDownLeftTiles = room.lookAt(closestSource.pos.x - 1, closestSource.pos.y + 1);
        const closestLeftTiles = room.lookAt(closestSource.pos.x - 1, closestSource.pos.y);
        const closestUpLeftTiles = room.lookAt(closestSource.pos.x - 1, closestSource.pos.y - 1);
        const closestSourceAvailable = (
          (closestUpTiles.length === 1 && closestUpTiles[0].terrain !== 'wall') ||
          (closestUpRightTiles.length === 1 && closestUpRightTiles[0].terrain !== 'wall') ||
          (closestRightTiles.length === 1 && closestRightTiles[0].terrain !== 'wall') ||
          (closestDownRightTiles.length === 1 && closestDownRightTiles[0].terrain !== 'wall') ||
          (closestDownTiles.length === 1 && closestDownTiles[0].terrain !== 'wall') ||
          (closestDownLeftTiles.length === 1 && closestDownLeftTiles[0].terrain !== 'wall') ||
          (closestLeftTiles.length === 1 && closestLeftTiles[0].terrain !== 'wall') ||
          (closestUpLeftTiles.length === 1 && closestUpLeftTiles[0].terrain !== 'wall') ||
          myHarvester.pos.getRangeTo(closestSource) === 1
        );
        if (
          (myHarvester.pos.getRangeTo(source) < myHarvester.pos.getRangeTo(closestSource) && sourceAvailable && closestSourceAvailable) ||
          sourceAvailable && !closestSourceAvailable
        ) {
          return source;
        }
        return closestSource;
      });
      const spawn = findClosestSpawnForTransfer(myHarvester);
      const constructionSites = room.find(FIND_MY_CONSTRUCTION_SITES);
      if (myHarvester.store[RESOURCE_ENERGY] > 0 && spawn && myHarvester.pos.getRangeTo(spawn) < myHarvester.pos.getRangeTo(source)) {
        returnEnergy(myHarvester, spawn);
      } else if (myHarvester.store[RESOURCE_ENERGY] > 0 && constructionSites[0] && myHarvester.pos.getRangeTo(constructionSites[0]) < myHarvester.pos.getRangeTo(source)) {
        if (myHarvester.build(constructionSites[0]) === ERR_NOT_IN_RANGE) {
          myHarvester.moveTo(constructionSites[0]);
        }
      } else if (myHarvester.store[RESOURCE_ENERGY] > 0 && room.controller && myHarvester.pos.getRangeTo(room.controller) < myHarvester.pos.getRangeTo(source)) {
        if (myHarvester.upgradeController(room.controller) === ERR_NOT_IN_RANGE) {
          myHarvester.moveTo(room.controller);
        }
      } else if (myHarvester.harvest(source) === ERR_NOT_IN_RANGE) {
        myHarvester.moveTo(source);
      }
    } else {
      const spawn = findClosestSpawnForTransfer(myHarvester);
      const constructionSites = room.find(FIND_MY_CONSTRUCTION_SITES);
      if (spawn && myHarvester.transfer(spawn, RESOURCE_ENERGY) === ERR_NOT_IN_RANGE) {
        myHarvester.moveTo(spawn);
      } else if (constructionSites[0] && myHarvester.build(constructionSites[0]) === ERR_NOT_IN_RANGE) {
        myHarvester.moveTo(constructionSites[0]);
      } else if (room.controller) {
        if (myHarvester.upgradeController(room.controller) === ERR_NOT_IN_RANGE) {
          myHarvester.moveTo(room.controller);
        }
      }
    }
  }
};