autospawn init

Maxim-Mazurok · May 30, 2020 · 5aff552 · 5aff552
1 parent cead4d3
commit 5aff552
Show file tree

Hide file tree

Showing 9 changed files with 86 additions and 1 deletion.
diff --git a/build/helpers.creep.js b/build/helpers.creep.js
@@ -34,6 +34,9 @@ class HelpersCreep {
         }
         return result;
     }
+    static bodyCost(body) {
+        return body.reduce((cost, part) => cost + BODYPART_COST[part], 0);
+    }
 }
 exports.HelpersCreep = HelpersCreep;
 exports.TRANSFER_PATH = { stroke: '#ffffff' };

diff --git a/build/helpers.find.js b/build/helpers.find.js
@@ -71,5 +71,8 @@ class HelpersFind {
     static findLinksWithEnergy(room) {
         return HelpersFind.findStructuresByType(room, STRUCTURE_LINK).filter((link) => link.energy > 0);
     }
+    static findAllMyCreepsInRoom(room) {
+        return Object.values(Game.creeps).filter((creep) => creep.my && creep.room.name === room.name);
+    }
 }
 exports.HelpersFind = HelpersFind;
diff --git a/build/main.js b/build/main.js
@@ -11,6 +11,13 @@ const roleBuilder = require('./builder');
 const roleEnergizer = require('./energizer');
 const roleUptownHarvester = require('./uptown.harvester');
 const roomsConfig = [
+    {
+        roomName: 'E48N17',
+        autoSpawn: {
+            enabled: true,
+            maxCreeps: 3,
+        },
+    },
     {
         roomName: 'E48N17',
         claim: [

diff --git a/build/rooms.js b/build/rooms.js
@@ -59,6 +59,28 @@ class Rooms {
                     });
             }
         }
+        if (config.autoSpawn &&
+            config.autoSpawn.enabled &&
+            helpers_1.HelpersFind.findAllMyCreepsInRoom(room).length <
+                config.autoSpawn.maxCreeps) {
+            const totalEnergy = helpers_1.HelpersFind.getRoomTotalEnergyForSpawningAvailable(room);
+            const bodyParts = [];
+            const bodyPartsOrder = [MOVE, WORK, CARRY];
+            let lastBodyPartIndex = bodyParts.length - 1;
+            while (helpers_1.HelpersCreep.bodyCost(bodyParts) < totalEnergy) {
+                lastBodyPartIndex =
+                    lastBodyPartIndex === bodyParts.length - 1
+                        ? 0
+                        : lastBodyPartIndex + 1;
+                if (helpers_1.HelpersCreep.bodyCost([
+                    ...bodyParts,
+                    bodyPartsOrder[lastBodyPartIndex],
+                ]) <= totalEnergy) {
+                    bodyParts.push(bodyPartsOrder[lastBodyPartIndex]);
+                }
+            }
+            spawn.spawnCreep(bodyPartsOrder, Math.random().toString());
+        }
     }
 }
 exports.Rooms = Rooms;
diff --git a/src/helpers.creep.ts b/src/helpers.creep.ts
@@ -48,6 +48,10 @@ export class HelpersCreep {
     }
     return result;
   }
+
+  static bodyCost(body: BodyPartConstant[]) {
+    return body.reduce((cost, part) => cost + BODYPART_COST[part], 0);
+  }
 }
 
 export const TRANSFER_PATH = {stroke: '#ffffff'};

diff --git a/src/helpers.find.ts b/src/helpers.find.ts
@@ -136,4 +136,10 @@ export class HelpersFind {
       STRUCTURE_LINK
     ).filter((link: StructureLink) => link.energy > 0);
   }
+
+  static findAllMyCreepsInRoom(room: Room): Creep[] {
+    return Object.values(Game.creeps).filter(
+      (creep: Creep) => creep.my && creep.room.name === room.name
+    );
+  }
 }
diff --git a/src/main.ts b/src/main.ts
@@ -13,6 +13,13 @@ const roleEnergizer = require('./energizer');
 const roleUptownHarvester = require('./uptown.harvester');
 
 const roomsConfig: RoomsConfig = [
+  {
+    roomName: 'E48N17',
+    autoSpawn: {
+      enabled: true,
+      maxCreeps: 3,
+    },
+  },
   {
     roomName: 'E48N17',
     claim: [

diff --git a/src/rooms.ts b/src/rooms.ts
@@ -2,7 +2,7 @@ import {Towers} from './towers';
 import {Market} from './market';
 import {filter} from 'lodash';
 import {CreepRole} from './enums';
-import {HelpersFind} from './helpers';
+import {HelpersFind, HelpersCreep} from './helpers';
 import {RoomsConfig, RoomConfig} from './ts';
 
 export class Rooms {
@@ -75,5 +75,34 @@ export class Rooms {
           });
       }
     }
+
+    if (
+      config.autoSpawn &&
+      config.autoSpawn.enabled &&
+      HelpersFind.findAllMyCreepsInRoom(room).length <
+        config.autoSpawn.maxCreeps
+    ) {
+      const totalEnergy = HelpersFind.getRoomTotalEnergyForSpawningAvailable(
+        room
+      );
+      const bodyParts: BodyPartConstant[] = [];
+      const bodyPartsOrder = [MOVE, WORK, CARRY];
+      let lastBodyPartIndex = bodyParts.length - 1;
+      while (HelpersCreep.bodyCost(bodyParts) < totalEnergy) {
+        lastBodyPartIndex =
+          lastBodyPartIndex === bodyParts.length - 1
+            ? 0
+            : lastBodyPartIndex + 1;
+        if (
+          HelpersCreep.bodyCost([
+            ...bodyParts,
+            bodyPartsOrder[lastBodyPartIndex],
+          ]) <= totalEnergy
+        ) {
+          bodyParts.push(bodyPartsOrder[lastBodyPartIndex]);
+        }
+      }
+      spawn.spawnCreep(bodyPartsOrder, Math.random().toString());
+    }
   }
 }
diff --git a/src/ts.ts b/src/ts.ts
@@ -22,6 +22,10 @@ export interface RoomConfig {
   skills?: {
     [index in CreepRole]?: BodyPartConstant[];
   };
+  autoSpawn?: {
+    enabled: boolean;
+    maxCreeps: number;
+  };
 }
 
 export interface EnergySourcesConfig {