Implements Upgrader Handler Method For Cleanup

This commit is contained in:
2025-07-12 19:24:31 -03:00
parent c7eb577a63
commit 56e626e575
3 changed files with 34 additions and 3 deletions

View File

@@ -5,7 +5,16 @@ import { getNextEmptySpot, getPositionWithDelta, setSpotStatus, SourceSpotStatus
class UpgraderHandler extends RoleHandler { class UpgraderHandler extends RoleHandler {
public static destroy(creepMemory: CreepMemory, state: GameState): void {} public static destroy(creepMemory: CreepMemory, state: GameState): void {
if (creepMemory.destination?.type === "source") {
this.releaseSourceSpot(creepMemory.destination, state);
delete creepMemory.destination; // Clear destination after releasing the spot
}
if (creepMemory.previousDestination?.type === "source") {
this.releaseSourceSpot(creepMemory.previousDestination, state);
delete creepMemory.previousDestination; // Clear previous destination after releasing the spot
}
}
public static run(creep: Creep, state: GameState): GameState { public static run(creep: Creep, state: GameState): GameState {
this.validateCreepMemory(creep, state); this.validateCreepMemory(creep, state);
@@ -148,6 +157,24 @@ class UpgraderHandler extends RoleHandler {
return sources as Source[]; return sources as Source[];
} }
private static releaseSourceSpot(destination: SourceDestination, state: GameState) {
if (!destination || destination.type !== "source") {
return; // Not a source destination, nothing to release
}
const sourceState = state.sourcesStates[destination.id];
if (!sourceState) {
console.log(`Source state not found for ID: ${destination.id}`);
return;
}
setSpotStatus(
sourceState.spots,
destination.sourceSpot,
SourceSpotStatus.EMPTY
);
}
} }

View File

@@ -16,9 +16,13 @@ class SpawnHandler {
for(const name in Memory.creeps) { for(const name in Memory.creeps) {
if (!Game.creeps[name]) { if (!Game.creeps[name]) {
console.log(`Creep ${name} is dead, cleaning up memory.`);
const roleDefinition = CreepRoles[Memory.creeps[name].role as CreepRole]; const roleDefinition = CreepRoles[Memory.creeps[name].role as CreepRole];
roleDefinition.handler.destroy(Memory.creeps[name], state); roleDefinition.handler.destroy(Memory.creeps[name], state);
delete Memory.creeps[name]; // Clean up memory for dead creeps delete Memory.creeps[name]; // Clean up memory for dead creeps
continue; // Skip to the next creep
} }
const creep = Game.creeps[name]; const creep = Game.creeps[name];

View File

@@ -22,10 +22,10 @@ export type GameConfig = {
* @type {GameConfig} * @type {GameConfig}
*/ */
export const DEFAULT_GAME_CONFIG: GameConfig = { export const DEFAULT_GAME_CONFIG: GameConfig = {
maxCreeps: 100, maxCreeps: 10,
minCreepsPerRole: { minCreepsPerRole: {
harvester: 3, harvester: 3,
upgrader: 20, upgrader: 7,
builder: 0 builder: 0
} }
}; };