@supalosa/chronodivide-bot 0.3.1 → 0.5.1

package/dist/bot/logic/mission/missionController.js

@@ -1,51 +1,233 @@
-// Meta-controller for forming and controlling squads.
+// Meta-controller for forming and controlling missions.
+// Missions are groups of zero or more units that aim to accomplish a particular goal.
+import { Vector2 } from "@chronodivide/game-api";
+import { isDisbandMission, isGrabCombatants, isReleaseUnits, isRequestSpecificUnits, isRequestUnits, } from "./mission.js";
 import { createMissionFactories } from "./missionFactories.js";
+import { ActionBatcher } from "./actionBatcher.js";
+import { countBy, isSelectableCombatant } from "../common/utils.js";
+// `missingUnitTypes` priority decays by this much every update loop.
+const MISSING_UNIT_TYPE_REQUEST_DECAY_MULT_RATE = 0.75;
+const MISSING_UNIT_TYPE_REQUEST_DECAY_FLAT_RATE = 1;
 export class MissionController {
     constructor(logger) {
         this.logger = logger;
         this.missions = [];
+        // A mapping of unit IDs to the missions they are assigned to. This may contain units that are dead, but
+        // is periodically cleaned in the update loop.
+        this.unitIdToMission = new Map();
+        // A mapping of unit types to the highest priority requested for a mission.
+        // This decays over time if requests are not 'refreshed' by mission.
+        this.requestedUnitTypes = new Map();
+        // Tracks missions to be externally disbanded the next time the mission update loop occurs.
         this.forceDisbandedMissions = [];
         this.missionFactories = createMissionFactories();
     }
-    onAiUpdate(gameApi, playerData, matchAwareness, squadController) {
+    updateUnitIds(gameApi) {
+        // Check for units in multiple missions, this shouldn't happen.
+        this.unitIdToMission = new Map();
+        this.missions.forEach((mission) => {
+            const toRemove = [];
+            mission.getUnitIds().forEach((unitId) => {
+                if (this.unitIdToMission.has(unitId)) {
+                    this.logger(`WARNING: unit ${unitId} is in multiple missions, please debug.`);
+                }
+                else if (!gameApi.getGameObjectData(unitId)) {
+                    // say, if a unit was killed
+                    toRemove.push(unitId);
+                }
+                else {
+                    this.unitIdToMission.set(unitId, mission);
+                }
+            });
+            toRemove.forEach((unitId) => mission.removeUnit(unitId));
+        });
+    }
+    onAiUpdate(gameApi, actionsApi, playerData, matchAwareness) {
         // Remove inactive missions.
         this.missions = this.missions.filter((missions) => missions.isActive());
+        this.updateUnitIds(gameApi);
+        // Batch actions to reduce spamming of actions for larger armies.
+        const actionBatcher = new ActionBatcher();
         // Poll missions for requested actions.
         const missionActions = this.missions.map((mission) => ({
             mission,
-            action: mission.onAiUpdate(gameApi, playerData, matchAwareness),
+            action: mission.onAiUpdate(gameApi, actionsApi, playerData, matchAwareness, actionBatcher),
         }));
         // Handle disbands and merges.
-        const isDisband = (a) => a.type == "disband";
         const disbandedMissions = new Map();
         const disbandedMissionsArray = [];
         this.forceDisbandedMissions.forEach((name) => disbandedMissions.set(name, null));
         this.forceDisbandedMissions = [];
-        missionActions
-            .filter((a) => isDisband(a.action))
-            .forEach((a) => {
+        missionActions.filter(isDisbandMission).forEach((a) => {
+            this.logger(`Mission ${a.mission.getUniqueName()} disbanding as requested.`);
+            a.mission.getUnitIds().forEach((unitId) => {
+                this.unitIdToMission.delete(unitId);
+                actionsApi.setUnitDebugText(unitId, undefined);
+            });
             disbandedMissions.set(a.mission.getUniqueName(), a.action.reason);
         });
-        // Remove disbanded and merged squads.
+        // Handle unit requests.
+        // Release units
+        missionActions.filter(isReleaseUnits).forEach((a) => {
+            a.action.unitIds.forEach((unitId) => {
+                if (this.unitIdToMission.get(unitId)?.getUniqueName() === a.mission.getUniqueName()) {
+                    this.removeUnitFromMission(a.mission, unitId, actionsApi);
+                }
+            });
+        });
+        // Request specific units by ID
+        const unitIdToHighestRequest = missionActions.filter(isRequestSpecificUnits).reduce((prev, missionWithAction) => {
+            const { unitIds } = missionWithAction.action;
+            unitIds.forEach((unitId) => {
+                if (prev.hasOwnProperty(unitId)) {
+                    if (prev[unitId].action.priority > prev[unitId].action.priority) {
+                        prev[unitId] = missionWithAction;
+                    }
+                }
+                else {
+                    prev[unitId] = missionWithAction;
+                }
+            });
+            return prev;
+        }, {});
+        // Map of Mission ID to Unit Type to Count.
+        const newMissionAssignments = Object.entries(unitIdToHighestRequest)
+            .flatMap(([id, request]) => {
+            const unitId = Number.parseInt(id);
+            const unit = gameApi.getGameObjectData(unitId);
+            const { mission: requestingMission } = request;
+            const missionName = requestingMission.getUniqueName();
+            if (!unit) {
+                this.logger(`mission ${missionName} requested non-existent unit ${unitId}`);
+                return [];
+            }
+            if (!this.unitIdToMission.has(unitId)) {
+                this.addUnitToMission(requestingMission, unit, actionsApi);
+                return [{ unitName: unit?.name, mission: requestingMission.getUniqueName() }];
+            }
+            return [];
+        })
+            .reduce((acc, curr) => {
+            if (!acc[curr.mission]) {
+                acc[curr.mission] = {};
+            }
+            if (!acc[curr.mission][curr.unitName]) {
+                acc[curr.mission][curr.unitName] = 0;
+            }
+            acc[curr.mission][curr.unitName] = acc[curr.mission][curr.unitName] + 1;
+            return acc;
+        }, {});
+        Object.entries(newMissionAssignments).forEach(([mission, assignments]) => {
+            this.logger(`Mission ${mission} received: ${Object.entries(assignments)
+                .map(([unitType, count]) => unitType + " x " + count)
+                .join(", ")}`);
+        });
+        // Request units by type - store the highest priority mission for each unit type.
+        const unitTypeToHighestRequest = missionActions.filter(isRequestUnits).reduce((prev, missionWithAction) => {
+            const { unitNames } = missionWithAction.action;
+            unitNames.forEach((unitName) => {
+                if (prev.hasOwnProperty(unitName)) {
+                    if (prev[unitName].action.priority > prev[unitName].action.priority) {
+                        prev[unitName] = missionWithAction;
+                    }
+                }
+                else {
+                    prev[unitName] = missionWithAction;
+                }
+            });
+            return prev;
+        }, {});
+        // Request combat-capable units in an area
+        const grabRequests = missionActions.filter(isGrabCombatants);
+        // Find un-assigned units and distribute them among all the requesting missions.
+        const unitIds = gameApi.getVisibleUnits(playerData.name, "self");
+        // List of units that are unassigned or not in a locked mission.
+        const freeUnits = unitIds
+            .map((unitId) => gameApi.getGameObjectData(unitId))
+            .filter((unit) => !!unit)
+            .map((unit) => ({
+            unit,
+            mission: this.unitIdToMission.get(unit.id),
+        }))
+            .filter((unitWithMission) => !unitWithMission.mission || unitWithMission.mission.isUnitsLocked() === false);
+        // Sort free units so that unassigned units get chosen before assigned (but unlocked) units.
+        freeUnits.sort((u1, u2) => (u1.mission?.getPriority() ?? 0) - (u2.mission?.getPriority() ?? 0));
+        const newAssignmentsByType = freeUnits
+            .flatMap(({ unit: freeUnit, mission: donatingMission }) => {
+            if (unitTypeToHighestRequest.hasOwnProperty(freeUnit.name)) {
+                const { mission: requestingMission } = unitTypeToHighestRequest[freeUnit.name];
+                if (donatingMission) {
+                    if (donatingMission === requestingMission ||
+                        donatingMission.getPriority() > requestingMission.getPriority()) {
+                        return [];
+                    }
+                    this.removeUnitFromMission(donatingMission, freeUnit.id, actionsApi);
+                }
+                this.logger(`granting unit ${freeUnit.id}#${freeUnit.name} to mission ${requestingMission.getUniqueName()}`);
+                this.addUnitToMission(requestingMission, freeUnit, actionsApi);
+                delete unitTypeToHighestRequest[freeUnit.name];
+                return [
+                    { unitName: freeUnit.name, missionName: requestingMission.getUniqueName(), method: "type" },
+                ];
+            }
+            else if (grabRequests.length > 0) {
+                const grantedMission = grabRequests.find((request) => {
+                    const canGrabUnit = isSelectableCombatant(freeUnit);
+                    return (canGrabUnit &&
+                        request.action.point.distanceTo(new Vector2(freeUnit.tile.rx, freeUnit.tile.ry)) <=
+                            request.action.radius);
+                });
+                if (grantedMission) {
+                    if (donatingMission) {
+                        if (donatingMission === grantedMission.mission ||
+                            donatingMission.getPriority() > grantedMission.mission.getPriority()) {
+                            return [];
+                        }
+                        this.removeUnitFromMission(donatingMission, freeUnit.id, actionsApi);
+                    }
+                    this.addUnitToMission(grantedMission.mission, freeUnit, actionsApi);
+                    return [
+                        {
+                            unitName: freeUnit.name,
+                            missionName: grantedMission.mission.getUniqueName(),
+                            method: "grab",
+                        },
+                    ];
+                }
+            }
+            return [];
+        })
+            .reduce((acc, curr) => {
+            if (!acc[curr.missionName]) {
+                acc[curr.missionName] = {};
+            }
+            if (!acc[curr.missionName][curr.unitName]) {
+                acc[curr.missionName][curr.unitName] = { grab: 0, type: 0 };
+            }
+            acc[curr.missionName][curr.unitName][curr.method] =
+                acc[curr.missionName][curr.unitName][curr.method] + 1;
+            return acc;
+        }, {});
+        Object.entries(newAssignmentsByType).forEach(([mission, assignments]) => {
+            this.logger(`Mission ${mission} received: ${Object.entries(assignments)
+                .flatMap(([unitType, methodToCount]) => Object.entries(methodToCount)
+                .filter(([, count]) => count > 0)
+                .map(([method, count]) => unitType + " x " + count + " (by " + method + ")"))
+                .join(", ")}`);
+        });
+        this.updateRequestedUnitTypes(unitTypeToHighestRequest);
+        // Send all actions that can be batched together.
+        actionBatcher.resolve(actionsApi);
+        // Remove disbanded and merged missions.
         this.missions
             .filter((missions) => disbandedMissions.has(missions.getUniqueName()))
             .forEach((disbandedMission) => {
             const reason = disbandedMissions.get(disbandedMission.getUniqueName());
-            this.logger(`mission disbanded: ${disbandedMission.getUniqueName()}, reason: ${reason}, hasSquad: ${!!disbandedMission.getSquad}`);
+            this.logger(`mission disbanded: ${disbandedMission.getUniqueName()}, reason: ${reason}`);
             disbandedMissionsArray.push({ mission: disbandedMission, reason });
             disbandedMission.endMission(disbandedMissions.get(disbandedMission.getUniqueName()));
-            disbandedMission.getSquad()?.setMission(null);
         });
         this.missions = this.missions.filter((missions) => !disbandedMissions.has(missions.getUniqueName()));
-        // Register new squads
-        const isNewSquad = (a) => a.type == "registerSquad";
-        missionActions
-            .filter((a) => isNewSquad(a.action))
-            .forEach((a) => {
-            const action = a.action;
-            squadController.registerSquad(action.squad);
-            this.logger(`registered a squad: ${action.squad.getName()}`);
-        });
         // Create dynamic missions.
         this.missionFactories.forEach((missionFactory) => {
             missionFactory.maybeCreateMissions(gameApi, playerData, matchAwareness, this, this.logger);
@@ -54,6 +236,43 @@ export class MissionController {
             });
         });
     }
+    updateRequestedUnitTypes(missingUnitTypeToHighestRequest) {
+        // Decay the priority over time.
+        const currentUnitTypes = Array.from(this.requestedUnitTypes.keys());
+        for (const unitType of currentUnitTypes) {
+            const newPriority = this.requestedUnitTypes.get(unitType) * MISSING_UNIT_TYPE_REQUEST_DECAY_MULT_RATE -
+                MISSING_UNIT_TYPE_REQUEST_DECAY_FLAT_RATE;
+            if (newPriority > 0.5) {
+                this.requestedUnitTypes.set(unitType, newPriority);
+            }
+            else {
+                this.requestedUnitTypes.delete(unitType);
+            }
+        }
+        // Add the new missing units to the priority set, if the request is higher than the existing value.
+        Object.entries(missingUnitTypeToHighestRequest).forEach(([unitType, request]) => {
+            const currentPriority = this.requestedUnitTypes.get(unitType);
+            this.requestedUnitTypes.set(unitType, currentPriority ? Math.max(currentPriority, request.action.priority) : request.action.priority);
+        });
+    }
+    /**
+     * Returns the set of units that have been requested for production by the missions.
+     *
+     * @returns A map of unit type to the highest priority for that unit type.
+     */
+    getRequestedUnitTypes() {
+        return this.requestedUnitTypes;
+    }
+    addUnitToMission(mission, unit, actionsApi) {
+        mission.addUnit(unit.id);
+        this.unitIdToMission.set(unit.id, mission);
+        actionsApi.setUnitDebugText(unit.id, mission.getUniqueName() + "_" + unit.id);
+    }
+    removeUnitFromMission(mission, unitId, actionsApi) {
+        mission.removeUnit(unitId);
+        this.unitIdToMission.delete(unitId);
+        actionsApi.setUnitDebugText(unitId, undefined);
+    }
     /**
      * Attempts to add a mission to the active set.
      * @param mission
@@ -74,8 +293,30 @@ export class MissionController {
     disbandMission(missionName) {
         this.forceDisbandedMissions.push(missionName);
     }
-    logDebugOutput() {
-        this.logger(`Missions (${this.missions.length}): ${this.missions.map((m) => m.getUniqueName()).join(", ")}`);
+    // return text to display for global debug
+    getGlobalDebugText(gameApi) {
+        const unitsInMission = (unitIds) => countBy(unitIds, (unitId) => gameApi.getGameObjectData(unitId)?.name);
+        let globalDebugText = "";
+        this.missions.forEach((mission) => {
+            this.logger(`Mission ${mission.getUniqueName()}: ${Object.entries(unitsInMission(mission.getUnitIds()))
+                .map(([unitName, count]) => `${unitName} x ${count}`)
+                .join(", ")}`);
+            const missionDebugText = mission.getGlobalDebugText();
+            if (missionDebugText) {
+                globalDebugText += mission.getUniqueName() + ": " + missionDebugText + "\n";
+            }
+        });
+        return globalDebugText;
+    }
+    updateDebugText(actionsApi) {
+        this.missions.forEach((mission) => {
+            mission
+                .getUnitIds()
+                .forEach((unitId) => actionsApi.setUnitDebugText(unitId, `${unitId}: ${mission.getUniqueName()}`));
+        });
+    }
+    getMissions() {
+        return this.missions;
     }
 }
 //# sourceMappingURL=missionController.js.map

package/dist/bot/logic/mission/missionController.js.map

@@ -1 +1 @@
-{"version":3,"file":"missionController.js","sourceRoot":"","sources":["../../../../src/bot/logic/mission/missionController.ts"],"names":[],"mappings":"AAAA,sDAAsD;AAMtD,OAAO,EAAkB,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAE/E,MAAM,OAAO,iBAAiB;IAM1B,YAAoB,MAAsD;QAAtD,WAAM,GAAN,MAAM,CAAgD;QAJlE,aAAQ,GAAmB,EAAE,CAAC;QAE9B,2BAAsB,GAAa,EAAE,CAAC;QAG1C,IAAI,CAAC,gBAAgB,GAAG,sBAAsB,EAAE,CAAC;IACrD,CAAC;IAEM,UAAU,CACb,OAAgB,EAChB,UAAsB,EACtB,cAA8B,EAC9B,eAAgC;QAEhC,4BAA4B;QAC5B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;QAExE,uCAAuC;QACvC,MAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACnD,OAAO;YACP,MAAM,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,UAAU,EAAE,cAAc,CAAC;SAClE,CAAC,CAAC,CAAC;QAEJ,8BAA8B;QAC9B,MAAM,SAAS,GAAG,CAAC,CAAgB,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,SAAS,CAAC;QAC5D,MAAM,iBAAiB,GAAqB,IAAI,GAAG,EAAE,CAAC;QACtD,MAAM,sBAAsB,GAAwC,EAAE,CAAC;QACvE,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;QACjF,IAAI,CAAC,sBAAsB,GAAG,EAAE,CAAC;QACjC,cAAc;aACT,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;aAClC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACX,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,EAAE,EAAG,CAAC,CAAC,MAA+B,CAAC,MAAM,CAAC,CAAC;QAChG,CAAC,CAAC,CAAC;QAEP,sCAAsC;QACtC,IAAI,CAAC,QAAQ;aACR,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC;aACrE,OAAO,CAAC,CAAC,gBAAgB,EAAE,EAAE;YAC1B,MAAM,MAAM,GAAG,iBAAiB,CAAC,GAAG,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC;YACvE,IAAI,CAAC,MAAM,CACP,sBAAsB,gBAAgB,CAAC,aAAa,EAAE,aAAa,MAAM,eAAe,CAAC,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CACxH,CAAC;YACF,sBAAsB,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,CAAC,CAAC;YACnE,gBAAgB,CAAC,UAAU,CAAC,iBAAiB,CAAC,GAAG,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;YACrF,gBAAgB,CAAC,QAAQ,EAAE,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC;QAClD,CAAC,CAAC,CAAC;QACP,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;QAErG,sBAAsB;QACtB,MAAM,UAAU,GAAG,CAAC,CAAgB,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,eAAe,CAAC;QACnE,cAAc;aACT,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;aACnC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YACX,MAAM,MAAM,GAAG,CAAC,CAAC,MAAoC,CAAC;YACtD,eAAe,CAAC,aAAa,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5C,IAAI,CAAC,MAAM,CAAC,uBAAuB,MAAM,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEP,2BAA2B;QAC3B,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,cAAc,EAAE,EAAE;YAC7C,cAAc,CAAC,mBAAmB,CAAC,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YAC3F,sBAAsB,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE;gBACnD,cAAc,CAAC,eAAe,CAAC,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5G,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;IACP,CAAC;IAED;;;;OAIG;IACI,UAAU,CAAI,OAAyB;QAC1C,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,EAAE,KAAK,OAAO,CAAC,aAAa,EAAE,CAAC,EAAE;YAC1E,kCAAkC;YAClC,OAAO,IAAI,CAAC;SACf;QACD,IAAI,CAAC,MAAM,CAAC,kBAAkB,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;QACzD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5B,OAAO,OAAO,CAAC;IACnB,CAAC;IAED;;OAEG;IACI,cAAc,CAAC,WAAmB;QACrC,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAClD,CAAC;IAEM,cAAc;QACjB,IAAI,CAAC,MAAM,CAAC,aAAa,IAAI,CAAC,QAAQ,CAAC,MAAM,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACjH,CAAC;CACJ"}
+{"version":3,"file":"missionController.js","sourceRoot":"","sources":["../../../../src/bot/logic/mission/missionController.ts"],"names":[],"mappings":"AAAA,wDAAwD;AACxD,sFAAsF;AAEtF,OAAO,EAAyE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AACxH,OAAO,EASH,gBAAgB,EAChB,gBAAgB,EAChB,cAAc,EACd,sBAAsB,EACtB,cAAc,GACjB,MAAM,cAAc,CAAC;AAEtB,OAAO,EAAkB,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAC/E,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAGpE,qEAAqE;AACrE,MAAM,yCAAyC,GAAG,IAAI,CAAC;AACvD,MAAM,yCAAyC,GAAG,CAAC,CAAC;AAEpD,MAAM,OAAO,iBAAiB;IAe1B,YAAoB,MAAsD;QAAtD,WAAM,GAAN,MAAM,CAAgD;QAblE,aAAQ,GAAmB,EAAE,CAAC;QAEtC,wGAAwG;QACxG,8CAA8C;QACtC,oBAAe,GAA8B,IAAI,GAAG,EAAE,CAAC;QAE/D,2EAA2E;QAC3E,oEAAoE;QAC5D,uBAAkB,GAAwB,IAAI,GAAG,EAAE,CAAC;QAE5D,2FAA2F;QACnF,2BAAsB,GAAa,EAAE,CAAC;QAG1C,IAAI,CAAC,gBAAgB,GAAG,sBAAsB,EAAE,CAAC;IACrD,CAAC;IAEO,aAAa,CAAC,OAAgB;QAClC,+DAA+D;QAC/D,IAAI,CAAC,eAAe,GAAG,IAAI,GAAG,EAAE,CAAC;QACjC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC9B,MAAM,QAAQ,GAAa,EAAE,CAAC;YAC9B,OAAO,CAAC,UAAU,EAAE,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBACpC,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE;oBAClC,IAAI,CAAC,MAAM,CAAC,iBAAiB,MAAM,yCAAyC,CAAC,CAAC;iBACjF;qBAAM,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE;oBAC3C,4BAA4B;oBAC5B,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;iBACzB;qBAAM;oBACH,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;iBAC7C;YACL,CAAC,CAAC,CAAC;YACH,QAAQ,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;IACP,CAAC;IAEM,UAAU,CACb,OAAgB,EAChB,UAAsB,EACtB,UAAsB,EACtB,cAA8B;QAE9B,4BAA4B;QAC5B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;QAExE,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;QAE5B,iEAAiE;QACjE,MAAM,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;QAE1C,uCAAuC;QACvC,MAAM,cAAc,GAA6B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YAC7E,OAAO;YACP,MAAM,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,cAAc,EAAE,aAAa,CAAC;SAC7F,CAAC,CAAC,CAAC;QAEJ,8BAA8B;QAC9B,MAAM,iBAAiB,GAAqB,IAAI,GAAG,EAAE,CAAC;QACtD,MAAM,sBAAsB,GAA6C,EAAE,CAAC;QAC5E,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,CAAC;QACjF,IAAI,CAAC,sBAAsB,GAAG,EAAE,CAAC;QACjC,cAAc,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YAClD,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,aAAa,EAAE,2BAA2B,CAAC,CAAC;YAC7E,CAAC,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBACtC,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;gBACpC,UAAU,CAAC,gBAAgB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;YACnD,CAAC,CAAC,CAAC;YACH,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,EAAE,EAAG,CAAC,CAAC,MAA+B,CAAC,MAAM,CAAC,CAAC;QAChG,CAAC,CAAC,CAAC;QAEH,wBAAwB;QAExB,gBAAgB;QAChB,cAAc,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;YAChD,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBAChC,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,aAAa,EAAE,KAAK,CAAC,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE;oBACjF,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;iBAC7D;YACL,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;QAEH,+BAA+B;QAC/B,MAAM,sBAAsB,GAAG,cAAc,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,MAAM,CAC/E,CAAC,IAAI,EAAE,iBAAiB,EAAE,EAAE;YACxB,MAAM,EAAE,OAAO,EAAE,GAAG,iBAAiB,CAAC,MAAM,CAAC;YAC7C,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBACvB,IAAI,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,EAAE;oBAC7B,IAAI,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE;wBAC7D,IAAI,CAAC,MAAM,CAAC,GAAG,iBAAiB,CAAC;qBACpC;iBACJ;qBAAM;oBACH,IAAI,CAAC,MAAM,CAAC,GAAG,iBAAiB,CAAC;iBACpC;YACL,CAAC,CAAC,CAAC;YACH,OAAO,IAAI,CAAC;QAChB,CAAC,EACD,EAA0E,CAC7E,CAAC;QAEF,2CAA2C;QAC3C,MAAM,qBAAqB,GAAG,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC;aAC/D,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,OAAO,CAAC,EAAE,EAAE;YACvB,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YACnC,MAAM,IAAI,GAAG,OAAO,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,EAAE,OAAO,EAAE,iBAAiB,EAAE,GAAG,OAAO,CAAC;YAC/C,MAAM,WAAW,GAAG,iBAAiB,CAAC,aAAa,EAAE,CAAC;YACtD,IAAI,CAAC,IAAI,EAAE;gBACP,IAAI,CAAC,MAAM,CAAC,WAAW,WAAW,gCAAgC,MAAM,EAAE,CAAC,CAAC;gBAC5E,OAAO,EAAE,CAAC;aACb;YACD,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE;gBACnC,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC;gBAC3D,OAAO,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,iBAAiB,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;aACjF;YACD,OAAO,EAAE,CAAC;QACd,CAAC,CAAC;aACD,MAAM,CACH,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;YACV,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE;gBACpB,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;aAC1B;YACD,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE;gBACnC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;aACxC;YACD,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;YACxE,OAAO,GAAG,CAAC;QACf,CAAC,EACD,EAA4C,CAC/C,CAAC;QACN,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,EAAE;YACrE,IAAI,CAAC,MAAM,CACP,WAAW,OAAO,cAAc,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;iBACtD,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,QAAQ,GAAG,KAAK,GAAG,KAAK,CAAC;iBACpD,IAAI,CAAC,IAAI,CAAC,EAAE,CACpB,CAAC;QACN,CAAC,CAAC,CAAC;QAEH,iFAAiF;QACjF,MAAM,wBAAwB,GAAG,cAAc,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,MAAM,CACzE,CAAC,IAAI,EAAE,iBAAiB,EAAE,EAAE;YACxB,MAAM,EAAE,SAAS,EAAE,GAAG,iBAAiB,CAAC,MAAM,CAAC;YAC/C,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAC3B,IAAI,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE;oBAC/B,IAAI,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,QAAQ,EAAE;wBACjE,IAAI,CAAC,QAAQ,CAAC,GAAG,iBAAiB,CAAC;qBACtC;iBACJ;qBAAM;oBACH,IAAI,CAAC,QAAQ,CAAC,GAAG,iBAAiB,CAAC;iBACtC;YACL,CAAC,CAAC,CAAC;YACH,OAAO,IAAI,CAAC;QAChB,CAAC,EACD,EAAkE,CACrE,CAAC;QAEF,0CAA0C;QAC1C,MAAM,YAAY,GAAG,cAAc,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAE7D,gFAAgF;QAChF,MAAM,OAAO,GAAG,OAAO,CAAC,eAAe,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAKjE,gEAAgE;QAChE,MAAM,SAAS,GAAsB,OAAO;aACvC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;aAClD,MAAM,CAAC,CAAC,IAAI,EAA0B,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;aAChD,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACZ,IAAI;YACJ,OAAO,EAAE,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;SAC7C,CAAC,CAAC;aACF,MAAM,CAAC,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC,eAAe,CAAC,OAAO,IAAI,eAAe,CAAC,OAAO,CAAC,aAAa,EAAE,KAAK,KAAK,CAAC,CAAC;QAEhH,4FAA4F;QAC5F,SAAS,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC;QAGhG,MAAM,oBAAoB,GAAG,SAAS;aACjC,OAAO,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,eAAe,EAAE,EAAE,EAAE;YACtD,IAAI,wBAAwB,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;gBACxD,MAAM,EAAE,OAAO,EAAE,iBAAiB,EAAE,GAAG,wBAAwB,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;gBAC/E,IAAI,eAAe,EAAE;oBACjB,IACI,eAAe,KAAK,iBAAiB;wBACrC,eAAe,CAAC,WAAW,EAAE,GAAG,iBAAiB,CAAC,WAAW,EAAE,EACjE;wBACE,OAAO,EAAE,CAAC;qBACb;oBACD,IAAI,CAAC,qBAAqB,CAAC,eAAe,EAAE,QAAQ,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;iBACxE;gBACD,IAAI,CAAC,MAAM,CACP,iBAAiB,QAAQ,CAAC,EAAE,IAAI,QAAQ,CAAC,IAAI,eAAe,iBAAiB,CAAC,aAAa,EAAE,EAAE,CAClG,CAAC;gBACF,IAAI,CAAC,gBAAgB,CAAC,iBAAiB,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;gBAC/D,OAAO,wBAAwB,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;gBAC/C,OAAO;oBACH,EAAE,QAAQ,EAAE,QAAQ,CAAC,IAAI,EAAE,WAAW,EAAE,iBAAiB,CAAC,aAAa,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE;iBACtE,CAAC;aAC7B;iBAAM,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE;gBAChC,MAAM,cAAc,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE;oBACjD,MAAM,WAAW,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;oBACpD,OAAO,CACH,WAAW;wBACX,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;4BAC5E,OAAO,CAAC,MAAM,CAAC,MAAM,CAC5B,CAAC;gBACN,CAAC,CAAC,CAAC;gBACH,IAAI,cAAc,EAAE;oBAChB,IAAI,eAAe,EAAE;wBACjB,IACI,eAAe,KAAK,cAAc,CAAC,OAAO;4BAC1C,eAAe,CAAC,WAAW,EAAE,GAAG,cAAc,CAAC,OAAO,CAAC,WAAW,EAAE,EACtE;4BACE,OAAO,EAAE,CAAC;yBACb;wBACD,IAAI,CAAC,qBAAqB,CAAC,eAAe,EAAE,QAAQ,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;qBACxE;oBACD,IAAI,CAAC,gBAAgB,CAAC,cAAc,CAAC,OAAO,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;oBACpE,OAAO;wBACH;4BACI,QAAQ,EAAE,QAAQ,CAAC,IAAI;4BACvB,WAAW,EAAE,cAAc,CAAC,OAAO,CAAC,aAAa,EAAE;4BACnD,MAAM,EAAE,MAAM;yBACjB;qBACoB,CAAC;iBAC7B;aACJ;YACD,OAAO,EAAE,CAAC;QACd,CAAC,CAAC;aACD,MAAM,CACH,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE;YACV,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE;gBACxB,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC;aAC9B;YACD,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE;gBACvC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;aAC/D;YACD,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;gBAC7C,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC1D,OAAO,GAAG,CAAC;QACf,CAAC,EACD,EAAqE,CACxE,CAAC;QACN,MAAM,CAAC,OAAO,CAAC,oBAAoB,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EAAE,WAAW,CAAC,EAAE,EAAE;YACpE,IAAI,CAAC,MAAM,CACP,WAAW,OAAO,cAAc,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC;iBACtD,OAAO,CAAC,CAAC,CAAC,QAAQ,EAAE,aAAa,CAAC,EAAE,EAAE,CACnC,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC;iBACxB,MAAM,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,CAAC;iBAChC,GAAG,CAAC,CAAC,CAAC,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,QAAQ,GAAG,KAAK,GAAG,KAAK,GAAG,OAAO,GAAG,MAAM,GAAG,GAAG,CAAC,CACnF;iBACA,IAAI,CAAC,IAAI,CAAC,EAAE,CACpB,CAAC;QACN,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,wBAAwB,CAAC,wBAAwB,CAAC,CAAC;QAExD,iDAAiD;QACjD,aAAa,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAElC,wCAAwC;QACxC,IAAI,CAAC,QAAQ;aACR,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC;aACrE,OAAO,CAAC,CAAC,gBAAgB,EAAE,EAAE;YAC1B,MAAM,MAAM,GAAG,iBAAiB,CAAC,GAAG,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC;YACvE,IAAI,CAAC,MAAM,CAAC,sBAAsB,gBAAgB,CAAC,aAAa,EAAE,aAAa,MAAM,EAAE,CAAC,CAAC;YACzF,sBAAsB,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,gBAAgB,EAAE,MAAM,EAAE,CAAC,CAAC;YACnE,gBAAgB,CAAC,UAAU,CAAC,iBAAiB,CAAC,GAAG,CAAC,gBAAgB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;QACzF,CAAC,CAAC,CAAC;QACP,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;QAErG,2BAA2B;QAC3B,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,cAAc,EAAE,EAAE;YAC7C,cAAc,CAAC,mBAAmB,CAAC,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YAC3F,sBAAsB,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE;gBACnD,cAAc,CAAC,eAAe,CAAC,OAAO,EAAE,UAAU,EAAE,cAAc,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5G,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;IACP,CAAC;IAEO,wBAAwB,CAC5B,+BAA6F;QAE7F,gCAAgC;QAChC,MAAM,gBAAgB,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,IAAI,EAAE,CAAC,CAAC;QACpE,KAAK,MAAM,QAAQ,IAAI,gBAAgB,EAAE;YACrC,MAAM,WAAW,GACb,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAE,GAAG,yCAAyC;gBAClF,yCAAyC,CAAC;YAC9C,IAAI,WAAW,GAAG,GAAG,EAAE;gBACnB,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;aACtD;iBAAM;gBACH,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;aAC5C;SACJ;QACD,mGAAmG;QACnG,MAAM,CAAC,OAAO,CAAC,+BAA+B,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,EAAE;YAC5E,MAAM,eAAe,GAAG,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC9D,IAAI,CAAC,kBAAkB,CAAC,GAAG,CACvB,QAAQ,EACR,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,eAAe,EAAE,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CACjG,CAAC;QACN,CAAC,CAAC,CAAC;IACP,CAAC;IAED;;;;OAIG;IACI,qBAAqB;QACxB,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACnC,CAAC;IAEO,gBAAgB,CAAC,OAAqB,EAAE,IAAoB,EAAE,UAAsB;QACxF,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACzB,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC3C,UAAU,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,EAAE,OAAO,CAAC,aAAa,EAAE,GAAG,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC;IAClF,CAAC;IAEO,qBAAqB,CAAC,OAAqB,EAAE,MAAc,EAAE,UAAsB;QACvF,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAC3B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACpC,UAAU,CAAC,gBAAgB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IACnD,CAAC;IAED;;;;OAIG;IACI,UAAU,CAAC,OAAqB;QACnC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,EAAE,KAAK,OAAO,CAAC,aAAa,EAAE,CAAC,EAAE;YAC1E,kCAAkC;YAClC,OAAO,IAAI,CAAC;SACf;QACD,IAAI,CAAC,MAAM,CAAC,kBAAkB,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;QACzD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5B,OAAO,OAAO,CAAC;IACnB,CAAC;IAED;;OAEG;IACI,cAAc,CAAC,WAAmB;QACrC,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAClD,CAAC;IAED,0CAA0C;IACnC,kBAAkB,CAAC,OAAgB;QACtC,MAAM,cAAc,GAAG,CAAC,OAAiB,EAAE,EAAE,CACzC,OAAO,CAAC,OAAO,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC;QAE1E,IAAI,eAAe,GAAG,EAAE,CAAC;QAEzB,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC9B,IAAI,CAAC,MAAM,CACP,WAAW,OAAO,CAAC,aAAa,EAAE,KAAK,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;iBACtF,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,GAAG,QAAQ,MAAM,KAAK,EAAE,CAAC;iBACpD,IAAI,CAAC,IAAI,CAAC,EAAE,CACpB,CAAC;YACF,MAAM,gBAAgB,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;YACtD,IAAI,gBAAgB,EAAE;gBAClB,eAAe,IAAI,OAAO,CAAC,aAAa,EAAE,GAAG,IAAI,GAAG,gBAAgB,GAAG,IAAI,CAAC;aAC/E;QACL,CAAC,CAAC,CAAC;QACH,OAAO,eAAe,CAAC;IAC3B,CAAC;IAEM,eAAe,CAAC,UAAsB;QACzC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC9B,OAAO;iBACF,UAAU,EAAE;iBACZ,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,UAAU,CAAC,gBAAgB,CAAC,MAAM,EAAE,GAAG,MAAM,KAAK,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC;QAC3G,CAAC,CAAC,CAAC;IACP,CAAC;IAEM,WAAW;QACd,OAAO,IAAI,CAAC,QAAQ,CAAC;IACzB,CAAC;CACJ"}

package/dist/bot/logic/mission/missions/attackMission.js

@@ -1,47 +1,128 @@
-import { ObjectType, Vector2 } from "@chronodivide/game-api";
-import { CombatSquad } from "../../squad/behaviours/combatSquad.js";
-import { Mission, disbandMission, noop } from "../mission.js";
-import { Squad } from "../../squad/squad.js";
+import { ObjectType, SideType, Vector2 } from "@chronodivide/game-api";
+import { CombatSquad } from "./squads/combatSquad.js";
+import { Mission, disbandMission, noop, requestUnits } from "../mission.js";
 import { RetreatMission } from "./retreatMission.js";
-import { maxBy } from "../../common/utils.js";
+import { countBy, isOwnedByNeutral, maxBy } from "../../common/utils.js";
+import { getSovietComposition } from "../../composition/sovietCompositions.js";
+import { getAlliedCompositions } from "../../composition/alliedCompositions.js";
+import { manageMoveMicro } from "./squads/common.js";
 export var AttackFailReason;
 (function (AttackFailReason) {
     AttackFailReason[AttackFailReason["NoTargets"] = 0] = "NoTargets";
     AttackFailReason[AttackFailReason["DefenceTooStrong"] = 1] = "DefenceTooStrong";
 })(AttackFailReason || (AttackFailReason = {}));
-const NO_TARGET_IDLE_TIMEOUT_TICKS = 60;
+var AttackMissionState;
+(function (AttackMissionState) {
+    AttackMissionState[AttackMissionState["Preparing"] = 0] = "Preparing";
+    AttackMissionState[AttackMissionState["Attacking"] = 1] = "Attacking";
+    AttackMissionState[AttackMissionState["Retreating"] = 2] = "Retreating";
+})(AttackMissionState || (AttackMissionState = {}));
+const NO_TARGET_RETARGET_TICKS = 450;
+const NO_TARGET_IDLE_TIMEOUT_TICKS = 900;
+function calculateTargetComposition(gameApi, playerData, matchAwareness) {
+    if (!playerData.country) {
+        throw new Error(`player ${playerData.name} has no country`);
+    }
+    else if (playerData.country.side === SideType.Nod) {
+        return getSovietComposition(gameApi, playerData, matchAwareness);
+    }
+    else {
+        return getAlliedCompositions(gameApi, playerData, matchAwareness);
+    }
+}
+const ATTACK_MISSION_PRIORITY_RAMP = 1.01;
+const ATTACK_MISSION_MAX_PRIORITY = 50;
 /**
  * A mission that tries to attack a certain area.
  */
 export class AttackMission extends Mission {
-    constructor(uniqueName, priority, rallyArea, attackArea, radius, logger) {
-        super(uniqueName, priority, logger);
-        this.rallyArea = rallyArea;
+    constructor(uniqueName, priority, rallyArea, attackArea, radius, composition, logger) {
+        super(uniqueName, logger);
+        this.priority = priority;
         this.attackArea = attackArea;
         this.radius = radius;
+        this.composition = composition;
         this.lastTargetSeenAt = 0;
+        this.hasPickedNewTarget = false;
+        this.state = AttackMissionState.Preparing;
+        this.squad = new CombatSquad(rallyArea, attackArea, radius);
+    }
+    _onAiUpdate(gameApi, actionsApi, playerData, matchAwareness, actionBatcher) {
+        switch (this.state) {
+            case AttackMissionState.Preparing:
+                return this.handlePreparingState(gameApi, actionsApi, playerData, matchAwareness, actionBatcher);
+            case AttackMissionState.Attacking:
+                return this.handleAttackingState(gameApi, actionsApi, playerData, matchAwareness, actionBatcher);
+            case AttackMissionState.Retreating:
+                return this.handleRetreatingState(gameApi, actionsApi, playerData, matchAwareness, actionBatcher);
+        }
     }
-    onAiUpdate(gameApi, playerData, matchAwareness) {
-        if (this.getSquad() === null) {
-            return this.setSquad(new Squad(this.getUniqueName(), new CombatSquad(this.rallyArea, this.attackArea, this.radius), this));
+    handlePreparingState(gameApi, actionsApi, playerData, matchAwareness, actionBatcher) {
+        const currentComposition = countBy(this.getUnits(gameApi), (unit) => unit.name);
+        const missingUnits = Object.entries(this.composition).filter(([unitType, targetAmount]) => {
+            return !currentComposition[unitType] || currentComposition[unitType] < targetAmount;
+        });
+        if (missingUnits.length > 0) {
+            this.priority = Math.min(this.priority * ATTACK_MISSION_PRIORITY_RAMP, ATTACK_MISSION_MAX_PRIORITY);
+            return requestUnits(missingUnits.map(([unitName]) => unitName), this.priority);
         }
         else {
-            // Dispatch missions.
-            if (!matchAwareness.shouldAttack()) {
-                return disbandMission(AttackFailReason.DefenceTooStrong);
-            }
-            const foundTargets = matchAwareness.getHostilesNearPoint2d(this.attackArea, this.radius);
-            if (foundTargets.length > 0) {
-                this.lastTargetSeenAt = gameApi.getCurrentTick();
-            }
-            else if (gameApi.getCurrentTick() > this.lastTargetSeenAt + NO_TARGET_IDLE_TIMEOUT_TICKS) {
-                return disbandMission(AttackFailReason.NoTargets);
+            this.priority = ATTACK_MISSION_INITIAL_PRIORITY;
+            this.state = AttackMissionState.Attacking;
+            return noop();
+        }
+    }
+    handleAttackingState(gameApi, actionsApi, playerData, matchAwareness, actionBatcher) {
+        if (this.getUnitIds().length === 0) {
+            // TODO: disband directly (we no longer retreat when losing)
+            this.state = AttackMissionState.Retreating;
+            return noop();
+        }
+        const foundTargets = matchAwareness
+            .getHostilesNearPoint2d(this.attackArea, this.radius)
+            .map((unit) => gameApi.getUnitData(unit.unitId))
+            .filter((unit) => !isOwnedByNeutral(unit));
+        const update = this.squad.onAiUpdate(gameApi, actionsApi, actionBatcher, playerData, this, matchAwareness, this.logger);
+        if (update.type !== "noop") {
+            return update;
+        }
+        if (foundTargets.length > 0) {
+            this.lastTargetSeenAt = gameApi.getCurrentTick();
+            this.hasPickedNewTarget = false;
+        }
+        else if (gameApi.getCurrentTick() > this.lastTargetSeenAt + NO_TARGET_IDLE_TIMEOUT_TICKS) {
+            return disbandMission(AttackFailReason.NoTargets);
+        }
+        else if (!this.hasPickedNewTarget &&
+            gameApi.getCurrentTick() > this.lastTargetSeenAt + NO_TARGET_RETARGET_TICKS) {
+            const newTarget = generateTarget(gameApi, playerData, matchAwareness);
+            if (newTarget) {
+                this.squad.setAttackArea(newTarget);
+                this.hasPickedNewTarget = true;
             }
         }
         return noop();
     }
+    handleRetreatingState(gameApi, actionsApi, playerData, matchAwareness, actionBatcher) {
+        this.getUnits(gameApi).forEach((unitId) => {
+            actionBatcher.push(manageMoveMicro(unitId, matchAwareness.getMainRallyPoint()));
+        });
+        return disbandMission();
+    }
+    getGlobalDebugText() {
+        return this.squad.getGlobalDebugText() ?? "<none>";
+    }
+    getState() {
+        return this.state;
+    }
+    // This mission can give up its units while preparing.
+    isUnitsLocked() {
+        return this.state !== AttackMissionState.Preparing;
+    }
+    getPriority() {
+        return this.priority;
+    }
 }
-const ATTACK_COOLDOWN_TICKS = 120;
 // Calculates the weight for initiating an attack on the position of a unit or building.
 // This is separate from unit micro; the squad will be ordered to attack in the vicinity of the point.
 const getTargetWeight = (unitData, tryFocusHarvester) => {
@@ -55,49 +136,75 @@ const getTargetWeight = (unitData, tryFocusHarvester) => {
         return unitData.maxHitPoints;
     }
 };
+function generateTarget(gameApi, playerData, matchAwareness, includeBaseLocations = false) {
+    // Randomly decide between harvester and base.
+    try {
+        const tryFocusHarvester = gameApi.generateRandomInt(0, 1) === 0;
+        const enemyUnits = gameApi
+            .getVisibleUnits(playerData.name, "enemy")
+            .map((unitId) => gameApi.getUnitData(unitId))
+            .filter((u) => !!u && gameApi.getPlayerData(u.owner).isCombatant);
+        const maxUnit = maxBy(enemyUnits, (u) => getTargetWeight(u, tryFocusHarvester));
+        if (maxUnit) {
+            return new Vector2(maxUnit.tile.rx, maxUnit.tile.ry);
+        }
+        if (includeBaseLocations) {
+            const mapApi = gameApi.mapApi;
+            const enemyPlayers = gameApi
+                .getPlayers()
+                .map(gameApi.getPlayerData)
+                .filter((otherPlayer) => !gameApi.areAlliedPlayers(playerData.name, otherPlayer.name));
+            const unexploredEnemyLocations = enemyPlayers.filter((otherPlayer) => {
+                const tile = mapApi.getTile(otherPlayer.startLocation.x, otherPlayer.startLocation.y);
+                if (!tile) {
+                    return false;
+                }
+                return !mapApi.isVisibleTile(tile, playerData.name);
+            });
+            if (unexploredEnemyLocations.length > 0) {
+                const idx = gameApi.generateRandomInt(0, unexploredEnemyLocations.length - 1);
+                return unexploredEnemyLocations[idx].startLocation;
+            }
+        }
+    }
+    catch (err) {
+        // There's a crash here when accessing a building that got destroyed. Will catch and ignore or now.
+        return null;
+    }
+    return null;
+}
+// Number of ticks between attacking visible targets.
+const VISIBLE_TARGET_ATTACK_COOLDOWN_TICKS = 120;
+// Number of ticks between attacking "bases" (enemy starting locations).
+const BASE_ATTACK_COOLDOWN_TICKS = 1800;
+const ATTACK_MISSION_INITIAL_PRIORITY = 1;
 export class AttackMissionFactory {
-    constructor(lastAttackAt = -ATTACK_COOLDOWN_TICKS) {
+    constructor(lastAttackAt = -VISIBLE_TARGET_ATTACK_COOLDOWN_TICKS) {
         this.lastAttackAt = lastAttackAt;
     }
     getName() {
         return "AttackMissionFactory";
     }
-    generateTarget(gameApi, playerData, matchAwareness) {
-        // Randomly decide between harvester and base.
-        try {
-            const tryFocusHarvester = gameApi.generateRandomInt(0, 1) === 0;
-            const enemyUnits = gameApi
-                .getVisibleUnits(playerData.name, "hostile")
-                .map((unitId) => gameApi.getUnitData(unitId))
-                .filter((u) => !!u && gameApi.getPlayerData(u.owner).isCombatant);
-            const maxUnit = maxBy(enemyUnits, (u) => getTargetWeight(u, tryFocusHarvester));
-            if (maxUnit) {
-                return new Vector2(maxUnit.tile.rx, maxUnit.tile.ry);
-            }
-        }
-        catch (err) {
-            // There's a crash here when accessing a building that got destroyed. Will catch and ignore or now.
-            return null;
-        }
-        return null;
-    }
     maybeCreateMissions(gameApi, playerData, matchAwareness, missionController, logger) {
-        if (!matchAwareness.shouldAttack()) {
+        if (gameApi.getCurrentTick() < this.lastAttackAt + VISIBLE_TARGET_ATTACK_COOLDOWN_TICKS) {
             return;
         }
-        if (gameApi.getCurrentTick() < this.lastAttackAt + ATTACK_COOLDOWN_TICKS) {
+        // can only have one attack 'preparing' at once.
+        if (missionController
+            .getMissions()
+            .some((mission) => mission instanceof AttackMission && mission.getState() === AttackMissionState.Preparing)) {
             return;
         }
-        const attackRadius = 15;
-        const attackArea = this.generateTarget(gameApi, playerData, matchAwareness);
+        const attackRadius = 10;
+        const includeEnemyBases = gameApi.getCurrentTick() > this.lastAttackAt + BASE_ATTACK_COOLDOWN_TICKS;
+        const attackArea = generateTarget(gameApi, playerData, matchAwareness, includeEnemyBases);
         if (!attackArea) {
-            // Nothing to attack.
             return;
         }
-        // TODO: not using a fixed value here. But performance slows to a crawl when this is unique.
-        const squadName = "globalAttack";
-        const tryAttack = missionController.addMission(new AttackMission(squadName, 100, matchAwareness.getMainRallyPoint(), attackArea, attackRadius, logger).then((reason, squad) => {
-            missionController.addMission(new RetreatMission("retreat-from-" + squadName + gameApi.getCurrentTick(), 100, matchAwareness.getMainRallyPoint(), squad?.getUnitIds() ?? [], logger));
+        const squadName = "attack_" + gameApi.getCurrentTick();
+        const composition = calculateTargetComposition(gameApi, playerData, matchAwareness);
+        const tryAttack = missionController.addMission(new AttackMission(squadName, ATTACK_MISSION_INITIAL_PRIORITY, matchAwareness.getMainRallyPoint(), attackArea, attackRadius, composition, logger).then((unitIds, reason) => {
+            missionController.addMission(new RetreatMission("retreat-from-" + squadName + gameApi.getCurrentTick(), matchAwareness.getMainRallyPoint(), unitIds, logger));
         }));
         if (tryAttack) {
             this.lastAttackAt = gameApi.getCurrentTick();