warscript 0.0.1-dev.7278154 → 0.0.1-dev.72f8823

package/engine/behavior.lua

@@ -18,39 +18,41 @@ local mutableLuaMap = ____lua_2Dmaps.mutableLuaMap
 local ____lua_2Dsets = require("utility.lua-sets")
 local mutableLuaSet = ____lua_2Dsets.mutableLuaSet
 local safeCall = warpack.safeCall
-local firstBehaviorByObject = {}
-local lastBehaviorByObject = {}
+local firstBehaviorByObjectByPriority = {[0] = {}, [2] = {}, [1] = {}}
+local lastBehaviorByObjectByPriority = {[0] = {}, [2] = {}, [1] = {}}
 local function invokeBehaviorOnPeriod(behavior, ...)
     behavior.onPeriod(behavior, ...)
 end
 local function reduceBehaviors(behaviorConstructor, object, operation, initial, consumerOrKey, ...)
     local accumulator = initial
-    local behavior = firstBehaviorByObject[object]
-    if behavior ~= nil then
-        if type(consumerOrKey) == "function" then
-            repeat
-                do
-                    if __TS__InstanceOf(behavior, behaviorConstructor) then
-                        local isSuccessful, result = safeCall(consumerOrKey, behavior, ...)
-                        if isSuccessful then
-                            accumulator = operation(accumulator, result)
+    for priority = 0, 2 do
+        local behavior = firstBehaviorByObjectByPriority[priority][object]
+        if behavior ~= nil then
+            if type(consumerOrKey) == "function" then
+                repeat
+                    do
+                        if __TS__InstanceOf(behavior, behaviorConstructor) then
+                            local isSuccessful, result = safeCall(consumerOrKey, behavior, ...)
+                            if isSuccessful then
+                                accumulator = operation(accumulator, result)
+                            end
                         end
+                        behavior = behavior[1]
                     end
-                    behavior = behavior[1]
-                end
-            until not (behavior ~= nil)
-        else
-            repeat
-                do
-                    if __TS__InstanceOf(behavior, behaviorConstructor) then
-                        local isSuccessful, result = safeCall(behavior[consumerOrKey], behavior, ...)
-                        if isSuccessful then
-                            accumulator = operation(accumulator, result)
+                until not (behavior ~= nil)
+            else
+                repeat
+                    do
+                        if __TS__InstanceOf(behavior, behaviorConstructor) then
+                            local isSuccessful, result = safeCall(behavior[consumerOrKey], behavior, ...)
+                            if isSuccessful then
+                                accumulator = operation(accumulator, result)
+                            end
                         end
+                        behavior = behavior[1]
                     end
-                    behavior = behavior[1]
-                end
-            until not (behavior ~= nil)
+                until not (behavior ~= nil)
+            end
         end
     end
     return accumulator
@@ -62,12 +64,17 @@ ____exports.Behavior = __TS__Class()
 local Behavior = ____exports.Behavior
 Behavior.name = "Behavior"
 __TS__ClassExtends(Behavior, AbstractDestroyable)
-function Behavior.prototype.____constructor(self, object)
+function Behavior.prototype.____constructor(self, object, priority)
+    if priority == nil then
+        priority = 1
+    end
     AbstractDestroyable.prototype.____constructor(self)
     self.object = object
+    self.priority = priority
+    local lastBehaviorByObject = lastBehaviorByObjectByPriority[priority]
     local lastBehavior = lastBehaviorByObject[object]
     if lastBehavior == nil then
-        firstBehaviorByObject[object] = self
+        firstBehaviorByObjectByPriority[priority][object] = self
         lastBehaviorByObject[object] = self
     else
         self[0] = lastBehavior
@@ -99,12 +106,12 @@ function Behavior.prototype.onDestroy(self)
     if previousBehavior ~= nil then
         previousBehavior[1] = nextBehavior
     else
-        firstBehaviorByObject[self.object] = nextBehavior
+        firstBehaviorByObjectByPriority[self.priority][self.object] = nextBehavior
     end
     if nextBehavior ~= nil then
         nextBehavior[0] = previousBehavior
     else
-        lastBehaviorByObject[self.object] = previousBehavior
+        lastBehaviorByObjectByPriority[self.priority][self.object] = previousBehavior
     end
     return AbstractDestroyable.prototype.onDestroy(self)
 end
@@ -127,6 +134,15 @@ function Behavior.prototype.registerEvent(self, event, listener)
     end
     behaviors:add(self)
 end
+function Behavior.prototype.deregisterEvent(self, event)
+    local behaviors = behaviorsByEvent[event]
+    if behaviors ~= nil and behaviors:remove(self) then
+        eventsByBehavior[self][event] = nil
+        listenerByBehaviorByEvent[event][self] = nil
+        return true
+    end
+    return false
+end
 function Behavior.prototype.onPeriod(self, ...)
 end
 function Behavior.prototype.startPeriodicAction(self, interval, ...)
@@ -152,78 +168,94 @@ function Behavior.prototype.stopPeriodicAction(self)
 end
 function Behavior.count(self, object, limit)
     local behaviorsCount = 0
-    local behavior = firstBehaviorByObject[object]
-    while behavior ~= nil and (limit == nil or behaviorsCount < limit) do
-        if __TS__InstanceOf(behavior, self) then
-            behaviorsCount = behaviorsCount + 1
+    for priority = 0, 2 do
+        local behavior = firstBehaviorByObjectByPriority[priority][object]
+        while behavior ~= nil and (limit == nil or behaviorsCount < limit) do
+            if __TS__InstanceOf(behavior, self) then
+                behaviorsCount = behaviorsCount + 1
+            end
+            behavior = behavior[1]
         end
-        behavior = behavior[1]
     end
     return behaviorsCount
 end
 function Behavior.getFirst(self, object, countOrPredicate, ...)
-    local behavior = firstBehaviorByObject[object]
     if type(countOrPredicate) ~= "number" then
-        while behavior ~= nil do
-            if __TS__InstanceOf(behavior, self) and (countOrPredicate == nil or countOrPredicate(behavior, ...)) then
-                return behavior
+        for priority = 0, 2 do
+            local behavior = firstBehaviorByObjectByPriority[priority][object]
+            while behavior ~= nil do
+                if __TS__InstanceOf(behavior, self) and (countOrPredicate == nil or countOrPredicate(behavior, ...)) then
+                    return behavior
+                end
+                behavior = behavior[1]
             end
-            behavior = behavior[1]
         end
         return nil
     end
     local behaviors = {}
     local behaviorsCount = 0
-    while behavior ~= nil and behaviorsCount < countOrPredicate do
-        if __TS__InstanceOf(behavior, self) then
-            behaviorsCount = behaviorsCount + 1
-            behaviors[behaviorsCount] = behavior
+    for priority = 0, 2 do
+        local behavior = firstBehaviorByObjectByPriority[priority][object]
+        while behavior ~= nil and behaviorsCount < countOrPredicate do
+            if __TS__InstanceOf(behavior, self) then
+                behaviorsCount = behaviorsCount + 1
+                behaviors[behaviorsCount] = behavior
+            end
+            behavior = behavior[1]
         end
-        behavior = behavior[1]
     end
     return behaviors
 end
 function Behavior.getLast(self, object)
-    local behavior = lastBehaviorByObject[object]
-    while behavior ~= nil do
-        if __TS__InstanceOf(behavior, self) then
-            return behavior
+    for priority = 2, 0, -1 do
+        local behavior = lastBehaviorByObjectByPriority[priority][object]
+        while behavior ~= nil do
+            if __TS__InstanceOf(behavior, self) then
+                return behavior
+            end
+            behavior = behavior[0]
         end
-        behavior = behavior[0]
     end
     return nil
 end
 function Behavior.getAll(self, object, predicate, ...)
     local behaviors = {}
     local behaviorsCount = 0
-    local behavior = firstBehaviorByObject[object]
-    while behavior ~= nil do
-        if __TS__InstanceOf(behavior, self) and (predicate == nil or predicate(behavior, ...)) then
-            behaviorsCount = behaviorsCount + 1
-            behaviors[behaviorsCount] = behavior
+    for priority = 0, 2 do
+        local behavior = firstBehaviorByObjectByPriority[priority][object]
+        while behavior ~= nil do
+            if __TS__InstanceOf(behavior, self) and (predicate == nil or predicate(behavior, ...)) then
+                behaviorsCount = behaviorsCount + 1
+                behaviors[behaviorsCount] = behavior
+            end
+            behavior = behavior[1]
         end
-        behavior = behavior[1]
     end
     return behaviors
 end
 function Behavior.forFirst(self, object, count, consumerOrKey, ...)
     local behaviorsCount = 0
-    local behavior = firstBehaviorByObject[object]
     if type(consumerOrKey) == "function" then
-        while behavior ~= nil and behaviorsCount < count do
-            if __TS__InstanceOf(behavior, self) then
-                safeCall(consumerOrKey, behavior, ...)
-                behaviorsCount = behaviorsCount + 1
+        for priority = 0, 2 do
+            local behavior = firstBehaviorByObjectByPriority[priority][object]
+            while behavior ~= nil and behaviorsCount < count do
+                if __TS__InstanceOf(behavior, self) then
+                    safeCall(consumerOrKey, behavior, ...)
+                    behaviorsCount = behaviorsCount + 1
+                end
+                behavior = behavior[1]
             end
-            behavior = behavior[1]
         end
     else
-        while behavior ~= nil and behaviorsCount < count do
-            if __TS__InstanceOf(behavior, self) then
-                safeCall(behavior[consumerOrKey], behavior, ...)
-                behaviorsCount = behaviorsCount + 1
+        for priority = 0, 2 do
+            local behavior = firstBehaviorByObjectByPriority[priority][object]
+            while behavior ~= nil and behaviorsCount < count do
+                if __TS__InstanceOf(behavior, self) then
+                    safeCall(behavior[consumerOrKey], behavior, ...)
+                    behaviorsCount = behaviorsCount + 1
+                end
+                behavior = behavior[1]
             end
-            behavior = behavior[1]
         end
     end
     return behaviorsCount

package/engine/behaviour/ability/remove-buffs.d.ts

@@ -14,3 +14,12 @@ export declare class RemoveBuffsSelfAbilityBehavior extends AbilityBehavior {
     constructor(ability: Ability, polarity?: AbilityDependentValue<BuffPolarity> | undefined, resistanceType?: AbilityDependentValue<BuffResistanceType> | undefined, includeExpirationTimers?: AbilityDependentValue<boolean> | undefined, includeAuras?: AbilityDependentValue<boolean> | undefined, autoDispel?: AbilityDependentValue<boolean> | undefined);
     onImpact(caster: Unit): void;
 }
+export declare class RemoveBuffsTargetAbilityBehavior extends AbilityBehavior {
+    private readonly polarity?;
+    private readonly resistanceType?;
+    private readonly includeExpirationTimers?;
+    private readonly includeAuras?;
+    private readonly autoDispel?;
+    constructor(ability: Ability, polarity?: AbilityDependentValue<BuffPolarity> | undefined, resistanceType?: AbilityDependentValue<BuffResistanceType> | undefined, includeExpirationTimers?: AbilityDependentValue<boolean> | undefined, includeAuras?: AbilityDependentValue<boolean> | undefined, autoDispel?: AbilityDependentValue<boolean> | undefined);
+    onUnitTargetImpact(_: Unit, target: Unit): void;
+}

package/engine/behaviour/ability/remove-buffs.lua

@@ -25,4 +25,25 @@ function RemoveBuffsSelfAbilityBehavior.prototype.onImpact(self, caster)
         self:resolveCurrentAbilityDependentValue(self.autoDispel)
     )
 end
+____exports.RemoveBuffsTargetAbilityBehavior = __TS__Class()
+local RemoveBuffsTargetAbilityBehavior = ____exports.RemoveBuffsTargetAbilityBehavior
+RemoveBuffsTargetAbilityBehavior.name = "RemoveBuffsTargetAbilityBehavior"
+__TS__ClassExtends(RemoveBuffsTargetAbilityBehavior, AbilityBehavior)
+function RemoveBuffsTargetAbilityBehavior.prototype.____constructor(self, ability, polarity, resistanceType, includeExpirationTimers, includeAuras, autoDispel)
+    AbilityBehavior.prototype.____constructor(self, ability)
+    self.polarity = polarity
+    self.resistanceType = resistanceType
+    self.includeExpirationTimers = includeExpirationTimers
+    self.includeAuras = includeAuras
+    self.autoDispel = autoDispel
+end
+function RemoveBuffsTargetAbilityBehavior.prototype.onUnitTargetImpact(self, _, target)
+    target:removeBuffs(
+        self:resolveCurrentAbilityDependentValue(self.polarity),
+        self:resolveCurrentAbilityDependentValue(self.resistanceType),
+        self:resolveCurrentAbilityDependentValue(self.includeExpirationTimers),
+        self:resolveCurrentAbilityDependentValue(self.includeAuras),
+        self:resolveCurrentAbilityDependentValue(self.autoDispel)
+    )
+end
 return ____exports

package/engine/behaviour/unit/stun-immunity.d.ts

@@ -4,17 +4,21 @@ import { Unit } from "../../unit";
 import { BuffTypeId } from "../../object-data/entry/buff-type";
 import { TextTagPreset } from "../../text-tag";
 import { Destructor } from "../../../destroyable";
-export type StunImmunityUnitBehaviourParameters = {
+import { BehaviorPriority } from "../../behavior";
+export type StunImmunityUnitBehaviorParameters = {
+    readonly priority?: BehaviorPriority;
     buffTypeIds?: LuaSet<BuffTypeId>;
     textTagPreset?: TextTagPreset;
     textTagText?: string;
+    additionalAction?: (this: void, unit: Unit) => void;
 };
 export declare class StunImmunityUnitBehavior extends UnitBehavior {
-    readonly parameters: Readonly<StunImmunityUnitBehaviourParameters>;
-    static defaultParameters: StunImmunityUnitBehaviourParameters;
-    constructor(unit: Unit, parameters?: Readonly<StunImmunityUnitBehaviourParameters>);
+    readonly parameters: Readonly<StunImmunityUnitBehaviorParameters>;
+    static defaultParameters: StunImmunityUnitBehaviorParameters;
+    constructor(unit: Unit, parameters?: Readonly<StunImmunityUnitBehaviorParameters>);
     protected onDestroy(): Destructor;
     onDamageReceived(): void;
     onTargetingAbilityChannelingStart(): void;
     onTargetingAbilityImpact(): void;
+    protected onEffect(): void;
 }

package/engine/behaviour/unit/stun-immunity.lua

@@ -48,8 +48,15 @@ local function process(behavior)
     for buffTypeId in pairs(behavior.parameters.buffTypeIds or DEFAULT_BUFF_TYPE_IDS) do
         hasRemovedBuffs = hasRemovedBuffs or behavior.unit:removeBuff(buffTypeId)
     end
-    if hasRemovedBuffs and behavior.parameters.textTagText ~= nil then
-        TextTag:flash(TextTag.MISS, behavior.parameters.textTagText, behavior.unit.x, behavior.unit.y)
+    if hasRemovedBuffs then
+        behavior.onEffect(behavior)
+        if behavior.parameters.textTagText ~= nil then
+            TextTag:flash(TextTag.MISS, behavior.parameters.textTagText, behavior.unit.x, behavior.unit.y)
+        end
+        local ____opt_0 = behavior.parameters.additionalAction
+        if ____opt_0 ~= nil then
+            ____opt_0(behavior.unit)
+        end
     end
 end
 ____exports.StunImmunityUnitBehavior = __TS__Class()
@@ -60,7 +67,7 @@ function StunImmunityUnitBehavior.prototype.____constructor(self, unit, paramete
     if parameters == nil then
         parameters = ____exports.StunImmunityUnitBehavior.defaultParameters
     end
-    UnitBehavior.prototype.____constructor(self, unit)
+    UnitBehavior.prototype.____constructor(self, unit, parameters.priority)
     self.parameters = parameters
     unit:decrementStunCounter()
     process(self)
@@ -79,5 +86,7 @@ end
 function StunImmunityUnitBehavior.prototype.onTargetingAbilityImpact(self)
     process(self)
 end
+function StunImmunityUnitBehavior.prototype.onEffect(self)
+end
 StunImmunityUnitBehavior.defaultParameters = {buffTypeIds = DEFAULT_BUFF_TYPE_IDS, textTagPreset = TextTag.MISS, textTagText = nil}
 return ____exports

package/engine/behaviour/unit.d.ts

@@ -1,22 +1,24 @@
 /** @noSelfInFile */
-import { Behavior } from "../behavior";
+import { Behavior, BehaviorPriority } from "../behavior";
 import { Ability } from "../internal/ability";
 import { DamageEvent, DamagingEvent, Unit } from "../internal/unit";
 import "../internal/unit+ability";
 import "../internal/unit-missile-launch";
 import { Item } from "../internal/item";
-import type { AbilityBehavior } from "./ability";
+import { AbilityBehavior } from "./ability";
 import { Event } from "../../event";
 import { Destructor } from "../../destroyable";
 import type { Widget } from "../../core/types/widget";
 import { Destructable } from "../../core/types/destructable";
 import type { Buff } from "../buff";
 import { UnitBonusType } from "../internal/unit/bonus";
+import { Player } from "../../core/types/player";
+import { UnitTypeId } from "../object-data/entry/unit-type";
 export type UnitBehaviorConstructor<Args extends any[]> = new (unit: Unit, ...args: Args) => UnitBehavior;
 export declare abstract class UnitBehavior<PeriodicActionParameters extends any[] = any[]> extends Behavior<Unit, PeriodicActionParameters> {
     readonly sourceAbilityBehavior?: AbilityBehavior;
     private _bonusIdByBonusType?;
-    constructor(unit: Unit);
+    constructor(unit: Unit, priority?: BehaviorPriority);
     protected onDestroy(): Destructor;
     get unit(): Unit;
     protected getUnitBonus(bonusType: UnitBonusType): number;
@@ -55,4 +57,6 @@ export declare abstract class UnitBehavior<PeriodicActionParameters extends any[
     onItemChargesChanged(item: Item): void;
     onKill(target: Unit): void;
     onDeath(source: Unit | undefined): void;
+    onOwnerChange(previousOwner: Player): void;
+    static bindUnitType<Args extends any[]>(this: UnitBehaviorConstructor<Args>, unitTypeId: UnitTypeId, ...args: Args): void;
 }

package/engine/behaviour/unit.lua

@@ -22,6 +22,7 @@ local addOrUpdateOrRemoveUnitBonus = ____bonus.addOrUpdateOrRemoveUnitBonus
 local getUnitBonus = ____bonus.getUnitBonus
 local removeUnitBonus = ____bonus.removeUnitBonus
 local safeCall = warpack.safeCall
+local createBehaviorFunctionsByUnitTypeId = {}
 local behaviorsByEvent = {}
 local rangeByBehaviorByEvent = {}
 local listenerByBehaviorByEvent = {}
@@ -30,8 +31,8 @@ ____exports.UnitBehavior = __TS__Class()
 local UnitBehavior = ____exports.UnitBehavior
 UnitBehavior.name = "UnitBehavior"
 __TS__ClassExtends(UnitBehavior, Behavior)
-function UnitBehavior.prototype.____constructor(self, unit)
-    Behavior.prototype.____constructor(self, unit)
+function UnitBehavior.prototype.____constructor(self, unit, priority)
+    Behavior.prototype.____constructor(self, unit, priority)
 end
 function UnitBehavior.prototype.onDestroy(self)
     local events = eventsByBehavior[self]
@@ -165,6 +166,23 @@ function UnitBehavior.prototype.onKill(self, target)
 end
 function UnitBehavior.prototype.onDeath(self, source)
 end
+function UnitBehavior.prototype.onOwnerChange(self, previousOwner)
+end
+function UnitBehavior.bindUnitType(self, unitTypeId, ...)
+    local args = {...}
+    local createBehaviorFunctions = createBehaviorFunctionsByUnitTypeId[unitTypeId]
+    if createBehaviorFunctions == nil then
+        createBehaviorFunctions = {}
+        createBehaviorFunctionsByUnitTypeId[unitTypeId] = createBehaviorFunctions
+    end
+    createBehaviorFunctions[#createBehaviorFunctions + 1] = function(unit)
+        return __TS__New(
+            self,
+            unit,
+            table.unpack(args)
+        )
+    end
+end
 __TS__SetDescriptor(
     UnitBehavior.prototype,
     "unit",
@@ -276,7 +294,18 @@ __TS__SetDescriptor(
     Unit.itemChargesChangedEvent:addListener(function(unit, item)
         ____exports.UnitBehavior:forAll(unit, "onItemChargesChanged", item)
     end)
+    Unit.onOwnerChange:addListener(function(unit, previousOwner)
+        ____exports.UnitBehavior:forAll(unit, "onOwnerChange", previousOwner)
+    end)
 end)(UnitBehavior)
+Unit.onCreate:addListener(function(unit)
+    local createBehaviorFunctions = createBehaviorFunctionsByUnitTypeId[unit.typeId]
+    if createBehaviorFunctions ~= nil then
+        for ____, createBehavior in ipairs(createBehaviorFunctions) do
+            createBehavior(unit)
+        end
+    end
+end)
 Unit.destroyEvent:addListener(function(unit)
     ____exports.UnitBehavior:forAll(unit, "destroy")
 end)

package/engine/buff.d.ts

@@ -79,6 +79,7 @@ export type BuffParameters<T extends Buff<any> = Buff> = Buff extends T ? {
     healingOnExpiration?: NumberParameterValueType;
     killsOnExpiration?: BooleanParameterValueType;
     explodesOnExpiration?: BooleanParameterValueType;
+    abilityCooldownFactor?: NumberParameterValueType;
     uniqueGroup?: BuffUniqueGroup;
 } : BuffParameters & (T extends Buff<infer AdditionalParameters> ? AdditionalParameters : object);
 declare const enum BuffPropertyKey {
@@ -124,7 +125,9 @@ declare const enum BuffPropertyKey {
     PROVIDES_INVULNERABILITY = 139,
     KILLS_ON_EXPIRATION = 140,
     EXPLODES_ON_EXPIRATION = 141,
-    MISS_PROBABILITY = 142
+    MISS_PROBABILITY = 142,
+    ABILITY_COOLDOWN_FACTOR = 143,
+    ABILITY_COOLDOWN_MODIFIER = 144
 }
 export declare const enum BuffTypeIdSelectionPolicy {
     LEAST_DURATION = 0
@@ -187,6 +190,8 @@ export declare class Buff<AdditionalParameters extends Prohibit<Record<string, a
     private [BuffPropertyKey.PROVIDES_INVULNERABILITY]?;
     private [BuffPropertyKey.KILLS_ON_EXPIRATION]?;
     private [BuffPropertyKey.EXPLODES_ON_EXPIRATION]?;
+    private [BuffPropertyKey.ABILITY_COOLDOWN_FACTOR]?;
+    private [BuffPropertyKey.ABILITY_COOLDOWN_MODIFIER]?;
     protected static readonly defaultParameters: BuffParameters;
     get source(): Unit;
     readonly typeId: ApplicableBuffTypeId;
@@ -262,6 +267,10 @@ export declare class Buff<AdditionalParameters extends Prohibit<Record<string, a
     get duration(): number;
     get remainingDuration(): number;
     set remainingDuration(remainingDuration: number);
+    get abilityCooldownFactor(): number;
+    set abilityCooldownFactor(abilityCooldownFactor: number);
+    onAbilityGained(ability: Ability): void;
+    onAbilityLost(ability: Ability): void;
     flashEffect(...parameters: [
         ...widgetOrXY: [] | [Widget] | [x: number, x: number],
         ...parametersOrDuration: [] | [EffectParameters] | [number]

package/engine/buff.lua

@@ -16,6 +16,7 @@ local internalApplyBuff = ____applicable.internalApplyBuff
 local removeBuff = ____applicable.removeBuff
 local ____ability = require("engine.internal.ability")
 local Ability = ____ability.Ability
+local UnitAbility = ____ability.UnitAbility
 local ____ability = require("engine.object-field.ability")
 local AbilityBooleanField = ____ability.AbilityBooleanField
 local AbilityNumberField = ____ability.AbilityNumberField
@@ -50,6 +51,8 @@ local ____item = require("engine.internal.item")
 local Item = ____item.Item
 local ____destructable = require("core.types.destructable")
 local Destructable = ____destructable.Destructable
+local ____ability = require("engine.standard.fields.ability")
+local COOLDOWN_ABILITY_FLOAT_LEVEL_FIELD = ____ability.COOLDOWN_ABILITY_FLOAT_LEVEL_FIELD
 local getUnitAbility = BlzGetUnitAbility
 local stringValueByBuffTypeIdByFieldId = postcompile(function()
     local stringValueByBuffTypeIdByFieldId = {}
@@ -121,7 +124,8 @@ local buffParametersKeys = {
     damageOnExpiration = true,
     healingOnExpiration = true,
     killsOnExpiration = true,
-    explodesOnExpiration = true
+    explodesOnExpiration = true,
+    abilityCooldownFactor = true
 }
 local function resolveEnumValue(ability, level, value)
     if value == nil or type(value) == "number" then
@@ -198,7 +202,8 @@ local buffNumberParameters = {
     "healingPerInterval",
     "healingOverDuration",
     "damageOnExpiration",
-    "healingOnExpiration"
+    "healingOnExpiration",
+    "abilityCooldownFactor"
 }
 local unsuccessfulApplicationMarker = {}
 local function selectBuffTypeIdWithLeastDuration(buffTypeIds, unit)
@@ -506,6 +511,22 @@ function Buff.prototype.____constructor(self, _unit, typeIdOrTypeIds, polarityOr
     self[100] = 1
     Event.invoke(buffCreatedEvent, self)
 end
+function Buff.prototype.onAbilityGained(self, ability)
+    if __TS__InstanceOf(ability, UnitAbility) then
+        local abilityCooldownModifier = self[144]
+        if abilityCooldownModifier then
+            COOLDOWN_ABILITY_FLOAT_LEVEL_FIELD:applyModifier(ability, abilityCooldownModifier)
+        end
+    end
+end
+function Buff.prototype.onAbilityLost(self, ability)
+    if __TS__InstanceOf(ability, UnitAbility) then
+        local abilityCooldownModifier = self[144]
+        if abilityCooldownModifier then
+            COOLDOWN_ABILITY_FLOAT_LEVEL_FIELD:removeModifier(ability, abilityCooldownModifier)
+        end
+    end
+end
 function Buff.prototype.flashEffect(self, widgetOrXOrParametersOrDuration, yOrParametersOrDuration, parametersOrDuration)
     if type(widgetOrXOrParametersOrDuration) == "number" and type(yOrParametersOrDuration) == "number" then
         Effect:flash(self[105], widgetOrXOrParametersOrDuration, yOrParametersOrDuration, parametersOrDuration)
@@ -580,6 +601,12 @@ function Buff.prototype.onDestroy(self)
             behavior:destroy()
         end
     end
+    local previousAbilityCooldownModifier = self[144]
+    if previousAbilityCooldownModifier then
+        for ____, ability in ipairs(self._unit.abilities) do
+            COOLDOWN_ABILITY_FLOAT_LEVEL_FIELD:removeModifier(ability, previousAbilityCooldownModifier)
+        end
+    end
     if self[139] then
         unit:decrementInvulnerabilityCounter()
     end
@@ -588,7 +615,7 @@ function Buff.prototype.onDestroy(self)
     end
     if self[136] then
         if self[137] then
-            unit:decrementStunCounter()
+            unit:decrementForceStunCounter()
         end
         unit:decrementStunCounter()
     end
@@ -987,13 +1014,13 @@ __TS__SetDescriptor(
         set = function(self, stuns)
             if not stuns and self[136] then
                 if self[137] then
-                    self.object:decrementStunCounter()
+                    self.object:decrementForceStunCounter()
                 end
                 self.object:decrementStunCounter()
                 self[136] = nil
             elseif stuns and not self[136] then
                 if self[137] then
-                    self.object:incrementStunCounter()
+                    self.object:incrementForceStunCounter()
                 end
                 self.object:incrementStunCounter()
                 self[136] = true
@@ -1016,12 +1043,12 @@ __TS__SetDescriptor(
         set = function(self, ignoresStunImmunity)
             if not ignoresStunImmunity and self[137] then
                 if self[136] then
-                    self.object:decrementStunCounter()
+                    self.object:decrementForceStunCounter()
                 end
                 self[137] = nil
             elseif ignoresStunImmunity and not self[137] then
                 if self[136] then
-                    self.object:incrementStunCounter()
+                    self.object:incrementForceStunCounter()
                 end
                 self[137] = true
             end
@@ -1269,6 +1296,32 @@ __TS__SetDescriptor(
     },
     true
 )
+__TS__SetDescriptor(
+    Buff.prototype,
+    "abilityCooldownFactor",
+    {
+        get = function(self)
+            return self[143] or 1
+        end,
+        set = function(self, abilityCooldownFactor)
+            local previousAbilityCooldownModifier = self[144]
+            if previousAbilityCooldownModifier then
+                for ____, ability in ipairs(self._unit.abilities) do
+                    COOLDOWN_ABILITY_FLOAT_LEVEL_FIELD:removeModifier(ability, previousAbilityCooldownModifier)
+                end
+            end
+            local function modifier(ability, level, cooldown)
+                return cooldown * abilityCooldownFactor
+            end
+            for ____, ability in ipairs(self._unit.abilities) do
+                COOLDOWN_ABILITY_FLOAT_LEVEL_FIELD:applyModifier(ability, modifier)
+            end
+            self[144] = modifier
+            self[143] = abilityCooldownFactor
+        end
+    },
+    true
+)
 Buff.createdEvent = buffCreatedEvent
 Buff.beingDestroyedEvent = buffBeingDestroyedEvent;
 (function(self)