warscript 0.0.1-dev.3af578d → 0.0.1-dev.3b93f8c

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
@@ -28,9 +29,6 @@ local ____math = require("math")
 local max = ____math.max
 local min = ____math.min
 local ____bonus = require("engine.internal.unit.bonus")
-local addOrUpdateOrRemoveUnitBonus = ____bonus.addOrUpdateOrRemoveUnitBonus
-local getUnitBonus = ____bonus.getUnitBonus
-local removeUnitBonus = ____bonus.removeUnitBonus
 local UnitBonusType = ____bonus.UnitBonusType
 local ____area_2Ddamage = require("engine.internal.mechanics.area-damage")
 local damageArea = ____area_2Ddamage.damageArea
@@ -44,6 +42,7 @@ local BuffType = ____buff_2Dtype.BuffType
 local ____unit = require("engine.behaviour.unit")
 local UnitBehavior = ____unit.UnitBehavior
 local ____arrays = require("utility.arrays")
+local emptyArray = ____arrays.emptyArray
 local forEach = ____arrays.forEach
 local ____event = require("event")
 local Event = ____event.Event
@@ -53,6 +52,12 @@ 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 ____ability = require("engine.behaviour.ability")
+local AbilityBehavior = ____ability.AbilityBehavior
+local ____records = require("utility.records")
+local sortedKeysUnnested = ____records.sortedKeysUnnested
 local getUnitAbility = BlzGetUnitAbility
 local stringValueByBuffTypeIdByFieldId = postcompile(function()
     local stringValueByBuffTypeIdByFieldId = {}
@@ -98,9 +103,11 @@ local buffParametersKeys = {
     damageIncrease = true,
     damageIncreaseFactor = true,
     armorIncrease = true,
-    armorIncreaseFactor = true,
+    maxHealthIncrease = true,
+    maxManaIncrease = true,
     attackSpeedIncreaseFactor = true,
     movementSpeedIncreaseFactor = true,
+    manaRegenerationRateIncreaseFactor = true,
     evasionProbability = true,
     missProbability = true,
     damageFactor = true,
@@ -109,6 +116,7 @@ local buffParametersKeys = {
     durationIncreaseOnAutoAttack = true,
     maximumDuration = true,
     maximumRemainingDuration = true,
+    turnsIntoGhost = true,
     stuns = true,
     ignoresStunImmunity = true,
     providesStunImmunity = true,
@@ -119,11 +127,15 @@ local buffParametersKeys = {
     maximumAutoAttackCount = true,
     maximumDamageDealtEventCount = true,
     maximumDamageReceivedEventCount = true,
+    absorbedDamageFactor = true,
+    maximumDamageAbsorbed = true,
+    destroysOnMaximumDamageAbsorbed = true,
     uniqueGroup = true,
     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
@@ -174,6 +186,8 @@ local function resolveAndSetNumberValue(buff, property, ability, level, value, d
     end
 end
 local buffBooleanParameters = {
+    "destroysOnMaximumDamageAbsorbed",
+    "turnsIntoGhost",
     "stuns",
     "ignoresStunImmunity",
     "disablesAutoAttack",
@@ -185,8 +199,11 @@ local buffNumberParameters = {
     "durationIncreaseOnAutoAttack",
     "attackSpeedIncreaseFactor",
     "movementSpeedIncreaseFactor",
+    "manaRegenerationRateIncreaseFactor",
     "evasionProbability",
     "armorIncrease",
+    "maxHealthIncrease",
+    "maxManaIncrease",
     "damageFactor",
     "receivedDamageFactor",
     "maximumAutoAttackCount",
@@ -199,7 +216,10 @@ local buffNumberParameters = {
     "healingPerInterval",
     "healingOverDuration",
     "damageOnExpiration",
-    "healingOnExpiration"
+    "healingOnExpiration",
+    "absorbedDamageFactor",
+    "maximumDamageAbsorbed",
+    "abilityCooldownFactor"
 }
 local unsuccessfulApplicationMarker = {}
 local function selectBuffTypeIdWithLeastDuration(buffTypeIds, unit)
@@ -302,7 +322,8 @@ buffHealingIntervalTimerCallback = function(buff)
         source:healTarget(buff[101], healingPerInterval)
     end
 end
-local buffDestroyEvent = __TS__New(Event)
+local buffCreatedEvent = __TS__New(Event)
+local buffBeingDestroyedEvent = __TS__New(Event)
 ____exports.Buff = __TS__Class()
 local Buff = ____exports.Buff
 Buff.name = "Buff"
@@ -317,33 +338,38 @@ function Buff.prototype.____constructor(self, _unit, typeIdOrTypeIds, polarityOr
     local polarity
     local resistanceType
     local ability
+    local abilityBehavior
     if type(typeIdOrTypeIds) ~= "number" then
         typeId = selectBuffTypeIdWithLeastDuration(typeIdOrTypeIds, _unit)
         polarity = resistanceTypeOrPolarity
         resistanceType = abilityOrParametersOrResistanceType
-        if __TS__InstanceOf(parametersOrAbility, Ability) then
+        if __TS__InstanceOf(parametersOrAbility, AbilityBehavior) then
+            abilityBehavior = parametersOrAbility
+            ability = abilityBehavior.ability
+        elseif __TS__InstanceOf(parametersOrAbility, Ability) then
             ability = parametersOrAbility
-        else
-            ability = nil
+        elseif parametersOrAbility ~= nil then
             parameters = parametersOrAbility
         end
     else
         typeId = typeIdOrTypeIds
         polarity = polarityOrTypeIdSelectionPolicy
         resistanceType = resistanceTypeOrPolarity
-        if __TS__InstanceOf(abilityOrParametersOrResistanceType, Ability) then
+        if __TS__InstanceOf(abilityOrParametersOrResistanceType, AbilityBehavior) then
+            abilityBehavior = abilityOrParametersOrResistanceType
+            ability = abilityBehavior.ability
+            parameters = parametersOrAbility
+        elseif __TS__InstanceOf(abilityOrParametersOrResistanceType, Ability) then
             ability = abilityOrParametersOrResistanceType
             parameters = parametersOrAbility
-        else
-            ability = nil
+        elseif abilityOrParametersOrResistanceType ~= nil then
             parameters = abilityOrParametersOrResistanceType
+        else
+            parameters = parametersOrAbility
         end
     end
+    self.sourceAbilityBehavior = abilityBehavior
     self.typeId = typeId
-    if not __TS__InstanceOf(ability, Ability) then
-        parameters = ability
-        ability = nil
-    end
     local defaultParameters = self.constructor.defaultParameters
     local level = parameters and parameters.level or defaultParameters.level
     local spellStealPriority = parameters and parameters.spellStealPriority or defaultParameters.spellStealPriority
@@ -371,14 +397,15 @@ function Buff.prototype.____constructor(self, _unit, typeIdOrTypeIds, polarityOr
     local missProbability = parameters and parameters.missProbability or defaultParameters.missProbability
     if missProbability ~= nil then
         missProbability = resolveNumberValue(ability, level, missProbability)
-        self[141] = missProbability
+        self[146] = missProbability
     end
     local buffByTypeId = buffByTypeIdByUnit[_unit]
     if buffByTypeId == nil then
         buffByTypeId = {}
         buffByTypeIdByUnit[_unit] = buffByTypeId
     end
-    local ____opt_15 = buffByTypeId[typeId]
+    self.previousBuff = buffByTypeId[typeId]
+    local ____opt_15 = self.previousBuff
     if ____opt_15 ~= nil then
         ____opt_15:destroy()
     end
@@ -454,24 +481,22 @@ function Buff.prototype.____constructor(self, _unit, typeIdOrTypeIds, polarityOr
                 abilityTypeIds = {}
                 self._abilityTypeIds = abilityTypeIds
             end
-            for abilityTypeId, abilityParameters in pairs(parametersAbilityTypeIds) do
+            for ____, abilityTypeId in ipairs(sortedKeysUnnested(parametersAbilityTypeIds)) do
+                local abilityParameters = parametersAbilityTypeIds[abilityTypeId]
                 local addedAbility = _unit:addAbility(abilityTypeId)
                 if addedAbility ~= nil then
                     _unit:makeAbilityPermanent(abilityTypeId, true)
                     _unit:setAbilityLevel(abilityTypeId, 1 + (abilityParameters.level or ability and ability.level or 0))
-                    for abilityParameterKey, abilityParameterValue in pairs(abilityParameters) do
-                        if abilityParameterKey == "isButtonVisible" then
-                            if not resolveBooleanValue(ability, level, abilityParameterValue) then
-                                _unit:hideAbility(abilityTypeId, true)
-                            end
-                        elseif abilityParameterKey ~= "level" then
-                            abilityParameterKey:setValue(
-                                addedAbility,
-                                resolveNumberValue(ability, level, abilityParameterValue)
-                            )
-                        end
+                    if abilityParameters.isButtonVisible == false then
+                        _unit:hideAbility(abilityTypeId, true)
                     end
-                    abilityTypeIds[abilityTypeId] = true
+                    for ____, field in ipairs(abilityParameters.fields or emptyArray()) do
+                        field[1]:setValue(
+                            addedAbility,
+                            resolveNumberValue(ability, level, field[2])
+                        )
+                    end
+                    abilityTypeIds[#abilityTypeIds + 1] = abilityTypeId
                 end
             end
             local behaviorConstructors = parameters and parameters.behaviorConstructors or defaultParameters.behaviorConstructors
@@ -483,18 +508,20 @@ function Buff.prototype.____constructor(self, _unit, typeIdOrTypeIds, polarityOr
                 self._behaviors = behaviors
             end
         end
-        local additionalParameters = {}
-        for key, value in pairs(parameters) do
-            if not buffParametersKeys[key] then
-                if ability then
-                    additionalParameters[key] = resolveCurrentAbilityDependentValue(ability, value)
-                else
-                    additionalParameters[key] = value
+        if parameters ~= nil then
+            local additionalParameters = {}
+            for ____, key in ipairs(sortedKeysUnnested(parameters)) do
+                if not buffParametersKeys[key] then
+                    if ability then
+                        additionalParameters[key] = resolveCurrentAbilityDependentValue(ability, parameters[key])
+                    else
+                        additionalParameters[key] = parameters[key]
+                    end
                 end
             end
-        end
-        if (next(additionalParameters)) ~= nil then
-            self.parameters = additionalParameters
+            if (next(additionalParameters)) ~= nil then
+                self.parameters = additionalParameters
+            end
         end
     end
     if duration ~= nil and duration > 0 then
@@ -504,65 +531,65 @@ function Buff.prototype.____constructor(self, _unit, typeIdOrTypeIds, polarityOr
     end
     self:onCreate()
     self[100] = 1
+    Event.invoke(buffCreatedEvent, self)
 end
-function Buff.prototype.getUnitBonus(self, bonusType)
-    local ____opt_38 = self._bonusIdByBonusType
-    local bonusId = ____opt_38 and ____opt_38[bonusType]
-    return bonusId == nil and 0 or getUnitBonus(self._unit, bonusType, bonusId)
+function Buff.prototype.onAbilityGained(self, ability)
+    if __TS__InstanceOf(ability, UnitAbility) then
+        local abilityCooldownModifier = self[148]
+        if abilityCooldownModifier then
+            COOLDOWN_ABILITY_FLOAT_LEVEL_FIELD:applyModifier(ability, abilityCooldownModifier)
+        end
+    end
 end
-function Buff.prototype.addOrUpdateOrRemoveUnitBonus(self, bonusType, value)
-    local bonusIdByBonusType = self._bonusIdByBonusType
-    if bonusIdByBonusType == nil then
-        bonusIdByBonusType = {}
-        self._bonusIdByBonusType = bonusIdByBonusType
+function Buff.prototype.onAbilityLost(self, ability)
+    if __TS__InstanceOf(ability, UnitAbility) then
+        local abilityCooldownModifier = self[148]
+        if abilityCooldownModifier then
+            COOLDOWN_ABILITY_FLOAT_LEVEL_FIELD:removeModifier(ability, abilityCooldownModifier)
+        end
     end
-    bonusIdByBonusType[bonusType] = addOrUpdateOrRemoveUnitBonus(self._unit, bonusType, bonusIdByBonusType[bonusType], value)
 end
 function Buff.prototype.flashEffect(self, widgetOrXOrParametersOrDuration, yOrParametersOrDuration, parametersOrDuration)
+    self:flash(
+        self[105],
+        stringValueByBuffTypeIdByFieldId[fourCC("feft")][self.typeId] or "origin",
+        widgetOrXOrParametersOrDuration,
+        yOrParametersOrDuration,
+        parametersOrDuration
+    )
+end
+function Buff.prototype.flashSpecialEffect(self, widgetOrXOrParametersOrDuration, yOrParametersOrDuration, parametersOrDuration)
+    self:flash(
+        self[106],
+        stringValueByBuffTypeIdByFieldId[fourCC("fspt")][self.typeId] or "origin",
+        widgetOrXOrParametersOrDuration,
+        yOrParametersOrDuration,
+        parametersOrDuration
+    )
+end
+function Buff.prototype.flash(self, modelPath, attachmentPoint, widgetOrXOrParametersOrDuration, yOrParametersOrDuration, parametersOrDuration)
     if type(widgetOrXOrParametersOrDuration) == "number" and type(yOrParametersOrDuration) == "number" then
-        Effect:flash(self[105], widgetOrXOrParametersOrDuration, yOrParametersOrDuration, parametersOrDuration)
+        Effect:flash(modelPath, widgetOrXOrParametersOrDuration, yOrParametersOrDuration, parametersOrDuration)
     else
         local isWidgetProvided = __TS__InstanceOf(widgetOrXOrParametersOrDuration, Unit) or __TS__InstanceOf(widgetOrXOrParametersOrDuration, Item) or __TS__InstanceOf(widgetOrXOrParametersOrDuration, Destructable)
-        local ____Effect_42 = Effect
-        local ____Effect_flash_43 = Effect.flash
-        local ____array_41 = __TS__SparseArrayNew(
-            self[105],
-            isWidgetProvided and widgetOrXOrParametersOrDuration or self._unit,
-            stringValueByBuffTypeIdByFieldId[fourCC("feft")][self.typeId] or "origin"
-        )
-        local ____isWidgetProvided_40
+        local ____Effect_40 = Effect
+        local ____Effect_flash_41 = Effect.flash
+        local ____array_39 = __TS__SparseArrayNew(modelPath, isWidgetProvided and widgetOrXOrParametersOrDuration or self._unit, attachmentPoint)
+        local ____isWidgetProvided_38
         if isWidgetProvided then
-            ____isWidgetProvided_40 = yOrParametersOrDuration
+            ____isWidgetProvided_38 = yOrParametersOrDuration
         else
-            ____isWidgetProvided_40 = widgetOrXOrParametersOrDuration
+            ____isWidgetProvided_38 = widgetOrXOrParametersOrDuration
         end
-        __TS__SparseArrayPush(____array_41, ____isWidgetProvided_40)
-        ____Effect_flash_43(
-            ____Effect_42,
-            __TS__SparseArraySpread(____array_41)
+        __TS__SparseArrayPush(____array_39, ____isWidgetProvided_38)
+        ____Effect_flash_41(
+            ____Effect_40,
+            __TS__SparseArraySpread(____array_39)
         )
     end
 end
-function Buff.prototype.flashSpecialEffect(self, widgetOrDuration, duration)
-    local isWidgetProvided = type(widgetOrDuration) == "table"
-    local ____Effect_46 = Effect
-    local ____Effect_flash_47 = Effect.flash
-    local ____array_45 = __TS__SparseArrayNew(
-        self[106],
-        isWidgetProvided and widgetOrDuration or self._unit,
-        stringValueByBuffTypeIdByFieldId[fourCC("fspt")][self.typeId] or "origin"
-    )
-    local ____isWidgetProvided_44
-    if isWidgetProvided then
-        ____isWidgetProvided_44 = duration
-    else
-        ____isWidgetProvided_44 = widgetOrDuration
-    end
-    __TS__SparseArrayPush(____array_45, ____isWidgetProvided_44)
-    ____Effect_flash_47(
-        ____Effect_46,
-        __TS__SparseArraySpread(____array_45)
-    )
+function Buff.prototype.expire(self)
+    expireBuff(self)
 end
 function Buff.prototype.onCreate(self)
 end
@@ -592,26 +619,33 @@ function Buff.prototype.onDestroy(self)
             behavior:destroy()
         end
     end
-    if self[137] then
+    local previousAbilityCooldownModifier = self[148]
+    if previousAbilityCooldownModifier then
+        for ____, ability in ipairs(self._unit.abilities) do
+            COOLDOWN_ABILITY_FLOAT_LEVEL_FIELD:removeModifier(ability, previousAbilityCooldownModifier)
+        end
+    end
+    if self[143] then
+        unit:decrementInvulnerabilityCounter()
+    end
+    if self[142] then
         unit:decrementDisableAutoAttackCounter()
     end
-    if self[135] then
-        if self[136] then
-            unit:decrementStunCounter()
+    if self[140] then
+        if self[141] then
+            unit:decrementForceStunCounter()
         end
         unit:decrementStunCounter()
     end
+    if self[139] then
+        unit:decrementGhostCounter()
+    end
     if self._abilityTypeIds ~= nil then
-        for abilityTypeId in pairs(self._abilityTypeIds) do
+        for ____, abilityTypeId in ipairs(self._abilityTypeIds) do
             unit:removeAbility(abilityTypeId)
         end
     end
-    if self._bonusIdByBonusType ~= nil then
-        for bonusType, bonusId in pairs(self._bonusIdByBonusType) do
-            removeUnitBonus(unit, bonusType, bonusId)
-        end
-    end
-    Event.invoke(buffDestroyEvent, self)
+    Event.invoke(buffBeingDestroyedEvent, self)
     self[100] = 3
     return UnitBehavior.prototype.onDestroy(self)
 end
@@ -640,8 +674,8 @@ function Buff.apply(self, ...)
     end
 end
 function Buff.getByTypeId(self, unit, typeId)
-    local ____opt_48 = buffByTypeIdByUnit[unit]
-    local buff = ____opt_48 and ____opt_48[typeId]
+    local ____opt_42 = buffByTypeIdByUnit[unit]
+    local buff = ____opt_42 and ____opt_42[typeId]
     if __TS__InstanceOf(buff, self) then
         return buff
     end
@@ -655,9 +689,9 @@ function Buff.prototype.onExpiration(self)
     if self[121] ~= nil then
         (self[102] or unit):healTarget(unit, self[120] or 0)
     end
-    if self[140] then
+    if self[145] then
         unit:explode()
-    elseif self[139] then
+    elseif self[144] then
         unit:kill()
     end
 end
@@ -712,6 +746,18 @@ function Buff.prototype.onDamageDealt(self, target, event)
 end
 function Buff.prototype.onDamageReceived(self, source, event)
     if event.originalAmount ~= 0 then
+        local absorbedDamage = min(event.amount * (self[135] or 1), (self[136] or 0) - (self[137] or 0))
+        if absorbedDamage > 0 then
+            event.amount = event.amount - absorbedDamage
+            self[137] = (self[137] or 0) + absorbedDamage
+            local ____self__138_44 = self[138]
+            if ____self__138_44 == nil then
+                ____self__138_44 = true
+            end
+            if ____self__138_44 and self[137] >= (self[136] or 0) then
+                self:destroy()
+            end
+        end
         local damageReceivedEventCount = (self[133] or 0) + 1
         self[133] = damageReceivedEventCount
         if damageReceivedEventCount == self[134] then
@@ -799,8 +845,8 @@ __TS__SetDescriptor(
                 return
             end
             self[112] = damageInterval
-            local ____opt_50 = self._timer
-            local elapsed = ____opt_50 and ____opt_50.elapsed or 0
+            local ____opt_45 = self._timer
+            local elapsed = ____opt_45 and ____opt_45.elapsed or 0
             local timer = self[114]
             if timer == nil then
                 timer = Timer:create()
@@ -879,8 +925,8 @@ __TS__SetDescriptor(
                 return
             end
             self[117] = healingInterval
-            local ____opt_52 = self._timer
-            local elapsed = ____opt_52 and ____opt_52.elapsed or 0
+            local ____opt_47 = self._timer
+            local elapsed = ____opt_47 and ____opt_47.elapsed or 0
             local timer = self[119]
             if timer == nil then
                 timer = Timer:create()
@@ -961,30 +1007,79 @@ __TS__SetDescriptor(
     },
     true
 )
+__TS__SetDescriptor(
+    Buff.prototype,
+    "maxHealthIncrease",
+    {
+        get = function(self)
+            return self:getUnitBonus(UnitBonusType.HEALTH)
+        end,
+        set = function(self, maxHealthIncrease)
+            self:addOrUpdateOrRemoveUnitBonus(UnitBonusType.HEALTH, maxHealthIncrease)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    Buff.prototype,
+    "maxManaIncrease",
+    {
+        get = function(self)
+            return self:getUnitBonus(UnitBonusType.MANA)
+        end,
+        set = function(self, maxManaIncrease)
+            self:addOrUpdateOrRemoveUnitBonus(UnitBonusType.MANA, maxManaIncrease)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    Buff.prototype,
+    "turnsIntoGhost",
+    {
+        get = function(self)
+            local ____self__139_49 = self[139]
+            if ____self__139_49 == nil then
+                ____self__139_49 = false
+            end
+            return ____self__139_49
+        end,
+        set = function(self, turnsIntoGhost)
+            if not turnsIntoGhost and self[139] then
+                self.object:decrementGhostCounter()
+                self[139] = nil
+            elseif turnsIntoGhost and not self[139] then
+                self.object:incrementGhostCounter()
+                self[139] = true
+            end
+        end
+    },
+    true
+)
 __TS__SetDescriptor(
     Buff.prototype,
     "stuns",
     {
         get = function(self)
-            local ____self__135_54 = self[135]
-            if ____self__135_54 == nil then
-                ____self__135_54 = false
+            local ____self__140_50 = self[140]
+            if ____self__140_50 == nil then
+                ____self__140_50 = false
             end
-            return ____self__135_54
+            return ____self__140_50
         end,
         set = function(self, stuns)
-            if not stuns and self[135] then
-                if self[136] then
-                    self.object:decrementStunCounter()
+            if not stuns and self[140] then
+                if self[141] then
+                    self.object:decrementForceStunCounter()
                 end
                 self.object:decrementStunCounter()
-                self[135] = nil
-            elseif stuns and not self[135] then
-                if self[136] then
-                    self.object:incrementStunCounter()
+                self[140] = nil
+            elseif stuns and not self[140] then
+                if self[141] then
+                    self.object:incrementForceStunCounter()
                 end
                 self.object:incrementStunCounter()
-                self[135] = true
+                self[140] = true
             end
         end
     },
@@ -995,23 +1090,23 @@ __TS__SetDescriptor(
     "ignoresStunImmunity",
     {
         get = function(self)
-            local ____self__136_55 = self[136]
-            if ____self__136_55 == nil then
-                ____self__136_55 = false
+            local ____self__141_51 = self[141]
+            if ____self__141_51 == nil then
+                ____self__141_51 = false
             end
-            return ____self__136_55
+            return ____self__141_51
         end,
         set = function(self, ignoresStunImmunity)
-            if not ignoresStunImmunity and self[136] then
-                if self[135] then
-                    self.object:decrementStunCounter()
+            if not ignoresStunImmunity and self[141] then
+                if self[140] then
+                    self.object:decrementForceStunCounter()
                 end
-                self[136] = nil
-            elseif ignoresStunImmunity and not self[136] then
-                if self[135] then
-                    self.object:incrementStunCounter()
+                self[141] = nil
+            elseif ignoresStunImmunity and not self[141] then
+                if self[140] then
+                    self.object:incrementForceStunCounter()
                 end
-                self[136] = true
+                self[141] = true
             end
         end
     },
@@ -1022,19 +1117,19 @@ __TS__SetDescriptor(
     "disablesAutoAttack",
     {
         get = function(self)
-            local ____self__137_56 = self[137]
-            if ____self__137_56 == nil then
-                ____self__137_56 = false
+            local ____self__142_52 = self[142]
+            if ____self__142_52 == nil then
+                ____self__142_52 = false
             end
-            return ____self__137_56
+            return ____self__142_52
         end,
         set = function(self, disablesAutoAttack)
-            if not disablesAutoAttack and self[137] then
+            if not disablesAutoAttack and self[142] then
                 self.object:decrementDisableAutoAttackCounter()
-                self[137] = nil
-            elseif disablesAutoAttack and not self[137] then
+                self[142] = nil
+            elseif disablesAutoAttack and not self[142] then
                 self.object:incrementDisableAutoAttackCounter()
-                self[137] = true
+                self[142] = true
             end
         end
     },
@@ -1045,19 +1140,19 @@ __TS__SetDescriptor(
     "providesInvulnerability",
     {
         get = function(self)
-            local ____self__138_57 = self[138]
-            if ____self__138_57 == nil then
-                ____self__138_57 = false
+            local ____self__143_53 = self[143]
+            if ____self__143_53 == nil then
+                ____self__143_53 = false
             end
-            return ____self__138_57
+            return ____self__143_53
         end,
         set = function(self, providesInvulnerability)
-            if not providesInvulnerability and self[138] then
+            if not providesInvulnerability and self[143] then
                 self.object:decrementInvulnerabilityCounter()
-                self[138] = nil
-            elseif providesInvulnerability and not self[138] then
+                self[143] = nil
+            elseif providesInvulnerability and not self[143] then
                 self.object:incrementInvulnerabilityCounter()
-                self[138] = true
+                self[143] = true
             end
         end
     },
@@ -1068,17 +1163,17 @@ __TS__SetDescriptor(
     "killsOnExpiration",
     {
         get = function(self)
-            local ____self__139_58 = self[139]
-            if ____self__139_58 == nil then
-                ____self__139_58 = false
+            local ____self__144_54 = self[144]
+            if ____self__144_54 == nil then
+                ____self__144_54 = false
             end
-            return ____self__139_58
+            return ____self__144_54
         end,
         set = function(self, killsOnExpiration)
-            if not killsOnExpiration and self[139] then
-                self[139] = nil
-            elseif killsOnExpiration and not self[139] then
-                self[139] = true
+            if not killsOnExpiration and self[144] then
+                self[144] = nil
+            elseif killsOnExpiration and not self[144] then
+                self[144] = true
             end
         end
     },
@@ -1089,17 +1184,17 @@ __TS__SetDescriptor(
     "explodesOnExpiration",
     {
         get = function(self)
-            local ____self__140_59 = self[140]
-            if ____self__140_59 == nil then
-                ____self__140_59 = false
+            local ____self__145_55 = self[145]
+            if ____self__145_55 == nil then
+                ____self__145_55 = false
             end
-            return ____self__140_59
+            return ____self__145_55
         end,
         set = function(self, killsOnExpiration)
-            if not killsOnExpiration and self[140] then
-                self[140] = nil
-            elseif killsOnExpiration and not self[140] then
-                self[140] = true
+            if not killsOnExpiration and self[145] then
+                self[145] = nil
+            elseif killsOnExpiration and not self[145] then
+                self[145] = true
             end
         end
     },
@@ -1208,12 +1303,39 @@ __TS__SetDescriptor(
     },
     true
 )
+__TS__SetDescriptor(
+    Buff.prototype,
+    "manaRegenerationRateIncreaseFactor",
+    {
+        get = function(self)
+            return self:getUnitBonus(UnitBonusType.MANA_REGENERATION_RATE_FACTOR)
+        end,
+        set = function(self, manaRegenerationRateIncreaseFactor)
+            self:addOrUpdateOrRemoveUnitBonus(UnitBonusType.MANA_REGENERATION_RATE_FACTOR, manaRegenerationRateIncreaseFactor)
+        end
+    },
+    true
+)
 __TS__SetDescriptor(
     Buff.prototype,
     "duration",
-    {get = function(self)
-        return self[103]
-    end},
+    {
+        get = function(self)
+            return self[103]
+        end,
+        set = function(self, duration)
+            if duration <= 0 then
+                local timer = self._timer
+                if timer ~= nil then
+                    timer:destroy()
+                    self._timer = nil
+                end
+                self[103] = 0
+            else
+                self.remainingDuration = self.remainingDuration + (duration - self[103])
+            end
+        end
+    },
     true
 )
 __TS__SetDescriptor(
@@ -1221,13 +1343,13 @@ __TS__SetDescriptor(
     "remainingDuration",
     {
         get = function(self)
-            local ____opt_60 = self._timer
-            return ____opt_60 and ____opt_60.remaining or 0
+            local ____opt_56 = self._timer
+            return ____opt_56 and ____opt_56.remaining or 0
         end,
         set = function(self, remainingDuration)
-            local ____remainingDuration_64 = remainingDuration
-            local ____opt_62 = self._timer
-            local remainingDurationDelta = ____remainingDuration_64 - (____opt_62 and ____opt_62.remaining or 0)
+            local ____remainingDuration_60 = remainingDuration
+            local ____opt_58 = self._timer
+            local remainingDurationDelta = ____remainingDuration_60 - (____opt_58 and ____opt_58.remaining or 0)
             if remainingDurationDelta ~= 0 then
                 self[103] = self[103] + remainingDurationDelta
                 if remainingDuration <= 0 then
@@ -1242,7 +1364,7 @@ __TS__SetDescriptor(
                         remainingDuration,
                         self._spellStealPriority,
                         self._learnLevelMinimum,
-                        self[141]
+                        self[146]
                     ) then
                         local timer = self._timer
                         if timer == nil then
@@ -1257,7 +1379,99 @@ __TS__SetDescriptor(
     },
     true
 )
-Buff.destroyEvent = buffDestroyEvent;
+__TS__SetDescriptor(
+    Buff.prototype,
+    "absorbedDamageFactor",
+    {
+        get = function(self)
+            return self[135] or 1
+        end,
+        set = function(self, absorbedDamageFactor)
+            if absorbedDamageFactor == 1 then
+                self[135] = nil
+            else
+                self[135] = absorbedDamageFactor
+            end
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    Buff.prototype,
+    "maximumDamageAbsorbed",
+    {
+        get = function(self)
+            return self[136] or 0
+        end,
+        set = function(self, maximumDamageAbsorbed)
+            if maximumDamageAbsorbed == 0 then
+                self[136] = nil
+            else
+                self[136] = maximumDamageAbsorbed
+            end
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    Buff.prototype,
+    "damageAbsorbed",
+    {get = function(self)
+        return self[137] or 0
+    end},
+    true
+)
+__TS__SetDescriptor(
+    Buff.prototype,
+    "destroysOnMaximumDamageAbsorbed",
+    {
+        get = function(self)
+            local ____self__138_61 = self[138]
+            if ____self__138_61 == nil then
+                ____self__138_61 = true
+            end
+            return ____self__138_61
+        end,
+        set = function(self, destroysOnMaximumDamageAbsorbed)
+            local ____destroysOnMaximumDamageAbsorbed_62
+            if destroysOnMaximumDamageAbsorbed then
+                ____destroysOnMaximumDamageAbsorbed_62 = nil
+            else
+                ____destroysOnMaximumDamageAbsorbed_62 = false
+            end
+            self[138] = ____destroysOnMaximumDamageAbsorbed_62
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    Buff.prototype,
+    "abilityCooldownFactor",
+    {
+        get = function(self)
+            return self[147] or 1
+        end,
+        set = function(self, abilityCooldownFactor)
+            local previousAbilityCooldownModifier = self[148]
+            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[148] = modifier
+            self[147] = abilityCooldownFactor
+        end
+    },
+    true
+)
+Buff.createdEvent = buffCreatedEvent
+Buff.beingDestroyedEvent = buffBeingDestroyedEvent;
 (function(self)
     local function destroyBuffIfNeeded(buff)
         if getUnitAbility(buff[101].handle, buff.typeId) ~= buff.handle and buff[100] == 1 then
@@ -1311,5 +1525,11 @@ Buff.destroyEvent = buffDestroyEvent;
             ____exports.checkBuffs(target)
         end
     )
+    buffCreatedEvent:addListener(function(buff)
+        UnitBehavior:forAll(buff.unit, "onBuffGained", buff)
+    end)
+    buffBeingDestroyedEvent:addListener(function(buff)
+        UnitBehavior:forAll(buff.unit, "onBuffLost", buff)
+    end)
 end)(Buff)
 return ____exports