Add a Task system

src/Tasks/Task.ts

@@ -0,0 +1,97 @@
+import {
+  packId, packPos, unpackId, unpackPos,
+} from "../../deps/screeps-packrat/src/packrat";
+
+export enum TaskType {
+  Harvest,
+  Upgrade,
+  Withdraw,
+  Transfer,
+  Build,
+  Repair,
+}
+
+export enum TaskStatus {
+  DONE,
+  IN_PROGRESS,
+}
+
+declare global {
+  interface CreepMemory {
+    task?: string;
+  }
+  interface Creep {
+    readonly isIdle: boolean;
+    task: TaskData | null;
+    _task?: TaskData | null;
+  }
+  interface RoomObject {
+    readonly assignedCreeps: Creep[];
+  }
+}
+
+export interface TaskData {
+  type: TaskType;
+  targetPos: RoomPosition;
+  target: RoomObject & _HasId | null;
+  options: object;
+  data: object;
+}
+
+const packTaskData = (td: TaskData): string =>
+  String.fromCharCode(td.type + 65)
+  + packPos(td.targetPos)
+  + JSON.stringify({ o: td.options, d: td.data, t: td.target?.id });
+
+const unpackTaskData = (s: string): TaskData => {
+  const { o: options, d: data, t: targetId } = JSON.parse(s.substring(3));
+  const target = Game.getObjectById(targetId as Id<RoomObject & _HasId>);
+  return {
+    type: s.charCodeAt(0) - 65,
+    targetPos: unpackPos(s.substring(1, 3)),
+    target,
+    options,
+    data,
+  };
+};
+
+Object.defineProperty(Creep.prototype, "isIdle", {
+  get: function (this: Creep) {
+    return this.memory.task == null;
+  },
+  configurable: true,
+});
+
+Object.defineProperty(Creep.prototype, "task", {
+  get: function (this: Creep) {
+    if (this._task != null) {
+      return this._task;
+    }
+    if (this.memory.task == null) {
+      return null;
+    }
+    this._task = unpackTaskData(this.memory.task);
+    return this._task;
+  },
+  set: function (this: Creep, task: TaskData | null) {
+    this._task = task;
+    if (task != null) {
+      this.memory.task = packTaskData(task);
+    }
+    else {
+      delete this.memory.task;
+    }
+  },
+  configurable: true,
+});
+
+Object.defineProperty(RoomObject.prototype, "assignedCreeps", {
+  get: function (this: RoomObject) {
+    if ("id" in this) {
+      return Object.values(Game.creeps)
+        .filter(x => x.task?.target?.id == this.id);
+    }
+    return [];
+  },
+  configurable: true,
+});