Better RoomRunner

src/RoomInspector.ts

@@ -0,0 +1,59 @@
+import { checkPositionWalkable } from "utils/funcs/checkPosition";
+import {
+    createSourcePositionMatrix,
+    forEachMatrixSpot,
+    getPositionWithDelta,
+    PositionSpotStatus,
+    setSpotStatus
+} from "utils/positions";
+
+class RoomInspector {
+    public static inspectState(room: Room, state: GameState): GameState {
+        if (!this.stateWasInitialized(state)) {
+            state = this.initializeState(room, state);
+        }
+        return state;
+    }
+
+    private static stateWasInitialized(state: GameState): boolean {
+        return !!state.sourcesStates;
+    }
+
+    private static initializeState(room: Room, state: GameState): GameState {
+        state.sourcesStates = {};
+        state.maxHarvesters = 0;
+
+        for (const source of room.find(FIND_SOURCES)) {
+            this.configureSourceState(source, state);
+        }
+
+        return state;
+    }
+
+    private static configureSourceState(source: Source, state: GameState): void {
+        const sourceId = source.id.toString();
+
+        if (!state.sourcesStates[sourceId]) {
+            state.sourcesStates[sourceId] = {
+                id: sourceId,
+                pos: source.pos,
+                spots: createSourcePositionMatrix()
+            };
+        }
+
+        forEachMatrixSpot(state.sourcesStates[sourceId].spots, (delta, status) => {
+            if (status !== PositionSpotStatus.UNKNOWN) {
+                return; // Skip known spots
+            }
+            const pos = getPositionWithDelta(source.pos, delta);
+            if (checkPositionWalkable(pos)) {
+                setSpotStatus(state.sourcesStates[sourceId].spots, delta, PositionSpotStatus.EMPTY);
+                state.maxHarvesters += 1;
+            } else {
+                setSpotStatus(state.sourcesStates[sourceId].spots, delta, PositionSpotStatus.INVALID);
+            }
+        });
+    }
+}
+
+export default RoomInspector;