Initial Working Creep Handler

src/main.ts

@@ -1,5 +1,4 @@
 import SpawnHandler from "spawnHandler";
-import SpawnStorage from "spawnStorage";
 
 declare global {
     /*
@@ -33,22 +32,8 @@ declare global {
 
 
 export const loop = () => {
-    const spawnStorage = new SpawnStorage();
-    Memory.spawnHandlers = Memory.spawnHandlers || {};
-
-    // Check if spawn still exists
-    const activeSpawns = Object.keys(Game.spawns);
-
-    spawnStorage.clearDeadHandlers(activeSpawns);
-
-    for (const spawnName of activeSpawns) {
-        // Create a handler for each spawn
-        var currentHandler = spawnStorage.getHandler(spawnName);
-        if (!currentHandler) {
-            currentHandler = spawnStorage.addHandler(spawnName, new SpawnHandler(Game.spawns[spawnName]));
-        }
-
+    for (const spawnName of Object.keys(Game.spawns)) {
         // Run the handler
-        currentHandler.run();
+        new SpawnHandler(Game.spawns[spawnName]).run();
     }
 };

src/roleHandlers/harvesterHandler.ts

@@ -1,12 +1,10 @@
-import { CreepRole } from "types/creeps";
-import { registerRoleHandler, RoleHandler } from "./roleHandler.interface";
+import { RoleHandler } from "./roleHandler.interface";
 
 
 
-class HarvesterHandler implements RoleHandler {
-    public readonly mappedRole: CreepRole = "harvester";
-
-    public run(creep: Creep): void {
+class HarvesterHandler extends RoleHandler {
+    public static run(creep: Creep): void {
+        console.log(`Running HarvesterHandler for creep: ${creep.name}`);
         if (creep.store.getFreeCapacity() > 0) {
             const sources = creep.room.find(FIND_SOURCES);
             if (creep.harvest(sources[0]) === ERR_NOT_IN_RANGE) {
@@ -22,7 +20,6 @@ class HarvesterHandler implements RoleHandler {
 }
 
 
-registerRoleHandler("harvester", HarvesterHandler);
 
 
 

src/roleHandlers/index.ts

@@ -0,0 +1,8 @@
+import { RoleHandler } from "./roleHandler.interface";
+
+import HarvesterHandler from "./harvesterHandler";
+
+
+
+
+export { RoleHandler, HarvesterHandler };

src/roleHandlers/roleHandler.interface.ts

@@ -1,19 +1,3 @@
-import { CreepRole } from "types/creeps";
-
-
-export interface RoleHandler {
-    mappedRole: CreepRole;
-
-    run(creep: Creep): void;
-}
-
-
-export const ImplementationRegistry: { [roleName: string]: RoleHandler } = {};
-
-
-export const registerRoleHandler = (roleName: string, handler: typeof RoleHandler) => {
-    if (ImplementationRegistry[roleName]) {
-        console.warn(`Role handler for ${roleName} is already registered. Overwriting.`);
-    }
-    ImplementationRegistry[roleName] = handler;
+export abstract class RoleHandler {
+    static run(creep: Creep): void {};
 }

src/spawnHandler.ts

@@ -5,23 +5,23 @@ import { get_role_const as get_role_cost } from "utils/funcs/get_role_const";
 class SpawnHandler {
     constructor(private spawn: StructureSpawn) {}
 
+    public get spawnName(): string {
+        return this.spawn.name;
+    }
+
     public run(): void {
         this.validateSpawnState();
 
-            for(var name in Game.creeps) {
-                var creep = Game.creeps[name];
+            for(const name in Game.creeps) {
+                const creep = Game.creeps[name];
 
-                if(creep.store.getFreeCapacity() > 0) {
-                    var sources = creep.room.find(FIND_SOURCES);
-                    if(creep.harvest(sources[0]) == ERR_NOT_IN_RANGE) {
-                        creep.moveTo(sources[0]);
-                    }
-                }
-                else {
-                    if(creep.transfer(Game.spawns['Spawn1'], RESOURCE_ENERGY) == ERR_NOT_IN_RANGE) {
-                        creep.moveTo(Game.spawns['Spawn1']);
-                    }
+                const roleDefinition = CreepRoles[creep.memory.role as CreepRole];
+                if (!roleDefinition) {
+                    console.warn(`Creep ${creep.name} has an unknown role: ${creep.memory.role}`);
+                    continue;
                 }
+
+                roleDefinition.handler.run(creep);
             }
     }
 
@@ -87,6 +87,10 @@ class SpawnHandler {
             }
             const roleType = role as CreepRole;
             requisition[roleType] = DEFAULT_GAME_CONFIG.minCreepsPerRole[roleType] - (creepCounts[role] || 0);
+
+            if (requisition[roleType] < 0) {
+                requisition[roleType] = 0; // Ensure we don't have negative requisitions
+            }
         }
 
         return requisition;

src/spawnStorage.ts

@@ -1,60 +0,0 @@
-import SpawnHandler from "spawnHandler";
-
-class SpawnStorage {
-    constructor() {
-        // Initialize the spawn storage
-        if (!Memory.spawnHandlers) {
-            Memory.spawnHandlers = {};
-        }
-    }
-
-    get allHandlers(): { [name: string]: SpawnHandler } {
-        const handlers: { [name: string]: SpawnHandler } = {};
-        for (const spawnName in Memory.spawnHandlers) {
-            const instance = this.getHandler(spawnName);
-            if (!instance) {
-                console.log(`No handler found for spawn: ${spawnName}`);
-                continue;
-            }
-            handlers[spawnName] = instance;
-        }
-        return handlers;
-    }
-
-    // Method to get a spawn handler by name
-    getHandler(spawnName: string): SpawnHandler | null {
-        if (!Memory.spawnHandlers[spawnName]) {
-            console.log(`No handler found for spawn: ${spawnName}`);
-            return null;
-        }
-        const data = JSON.parse(Memory.spawnHandlers[spawnName]) as SpawnHandler;
-        return Object.assign(new SpawnHandler(Game.spawns[spawnName]), data);
-    }
-
-    // Method to add or update a spawn handler
-    addHandler(spawnName: string, handler: SpawnHandler): SpawnHandler {
-        Memory.spawnHandlers[spawnName] = JSON.stringify(handler);
-        return handler;
-    }
-
-    // Method to remove a spawn handler
-    removeHandler(spawnName: string) {
-        delete Memory.spawnHandlers[spawnName];
-    }
-
-    checkHandlerExists(spawnName: string): boolean {
-        return !!Memory.spawnHandlers[spawnName];
-    }
-
-    clearDeadHandlers(activeSpawns: string[]) {
-        for (const spawnName of Object.keys(this.allHandlers)) {
-            if (!(spawnName in activeSpawns)) {
-                continue;
-            }
-            this.removeHandler(spawnName);
-        }
-    }
-}
-
-
-export default SpawnStorage;

src/types/creeps.ts

@@ -1,6 +1,9 @@
+import { HarvesterHandler, RoleHandler } from "roleHandlers";
+
 export type RoleDefinition = {
     name: string;
     body: BodyPartConstant[];
+    handler: RoleHandler;
     priority: number;
 };
 
@@ -8,16 +11,19 @@ export const CreepRoles = {
     harvester: {
         name: "harvester",
         body: [WORK, CARRY, MOVE],
+        handler: HarvesterHandler,
         priority: 1
     },
     upgrader: {
         name: "upgrader",
         body: [WORK, CARRY, MOVE],
+        handler: HarvesterHandler,
         priority: 2
     },
     builder: {
         name: "builder",
         body: [WORK, CARRY, MOVE],
+        handler: HarvesterHandler,
         priority: 3
     }
 } satisfies Record<string, RoleDefinition>;

src/types/gameConfig.ts

@@ -22,10 +22,10 @@ export type GameConfig = {
  * @type {GameConfig}
  */
 export const DEFAULT_GAME_CONFIG: GameConfig = {
-    maxCreeps: 50,
+    maxCreeps: 15,
     minCreepsPerRole: {
         harvester: 5,
-        upgrader: 5,
-        builder: 5
+        upgrader: 0,
+        builder: 0
     }
 };