warscript 0.0.1-dev.5abc199 → 0.0.1-dev.5bc80ec

package/engine/internal/misc/unit-ability-level-natives.lua

@@ -0,0 +1,12 @@
+local ____exports = {}
+---
+-- @internal The original natives, captured before `unit+ability` wraps them to report
+-- ability level changes. For internal level juggling that must not be reported.
+____exports.rawSetUnitAbilityLevel = SetUnitAbilityLevel
+---
+-- @internal
+____exports.rawIncUnitAbilityLevel = IncUnitAbilityLevel
+---
+-- @internal
+____exports.rawDecUnitAbilityLevel = DecUnitAbilityLevel
+return ____exports

package/engine/internal/object-data/armor-bonus.d.ts

@@ -0,0 +1,2 @@
+/** @noSelfInFile */
+export {};

package/engine/internal/object-data/attribute-bonus.lua

@@ -1,6 +1,6 @@
 local ____exports = {}
-local ____attribute_2Dincrease = require("engine.object-data.entry.ability-type.attribute-increase")
-local AttributeBonusAbilityType = ____attribute_2Dincrease.AttributeBonusAbilityType
+local ____attribute_2Dbonus = require("engine.object-data.entry.ability-type.attribute-bonus")
+local AttributeBonusAbilityType = ____attribute_2Dbonus.AttributeBonusAbilityType
 ---
 -- @internal For use by internal systems.
 ____exports.ATTRIBUTE_BONUS_DUMMY_ABILITY_TYPE_ID = compiletime(function()

package/engine/internal/object-data/health-bonus.d.ts

@@ -0,0 +1,2 @@
+/** @noSelfInFile */
+export {};

package/engine/internal/object-data/health-bonus.lua

@@ -0,0 +1,16 @@
+local ____exports = {}
+local ____health_2Dbonus = require("engine.object-data.entry.ability-type.health-bonus")
+local HealthBonusAbilityType = ____health_2Dbonus.HealthBonusAbilityType
+---
+-- @internal For use by internal systems only.
+____exports.HEALTH_BONUS_DUMMY_ABILITY_TYPE_ID = compiletime(function()
+    local abilityType = HealthBonusAbilityType:create()
+    abilityType.isInternal = true
+    abilityType.isButtonVisible = false
+    abilityType.healthBonus = 0
+    return abilityType.id
+end)
+---
+-- @internal For use by internal systems only.
+____exports.HEALTH_BONUS_DUMMY_ABILITY_FIELD = ABILITY_ILF_MAX_LIFE_GAINED
+return ____exports

package/engine/internal/object-data/health-regeneration-rate-increase.d.ts

@@ -0,0 +1,2 @@
+/** @noSelfInFile */
+export {};

package/engine/internal/object-data/health-regeneration-rate-increase.lua

@@ -0,0 +1,48 @@
+local ____exports = {}
+local ____healing_2Dward_2Daura = require("engine.object-data.entry.ability-type.healing-ward-aura")
+local HealingWardAuraAbilityType = ____healing_2Dward_2Daura.HealingWardAuraAbilityType
+local ____item_2Dlife_2Dregeneration = require("engine.object-data.entry.ability-type.item-life-regeneration")
+local ItemLifeRegenerationAbilityType = ____item_2Dlife_2Dregeneration.ItemLifeRegenerationAbilityType
+local ____combat_2Dclassification = require("engine.object-data.auxiliary.combat-classification")
+local combatClassificationsOf = ____combat_2Dclassification.combatClassificationsOf
+local function createHealthRegenerationDummyAbilityType(healthRegeneration)
+    local abilityType = ItemLifeRegenerationAbilityType:create()
+    abilityType.isInternal = true
+    abilityType.isButtonVisible = false
+    abilityType.healthRegeneration = healthRegeneration
+    return abilityType.id
+end
+---
+-- @internal For use by internal systems. The ability at index `i` regenerates `2^i` hit
+-- points per second; together they compose any value from 0 to 1023.
+____exports.HEALTH_REGENERATION_RATE_INCREASE_DUMMY_ABILITY_TYPE_IDS = compiletime(function()
+    local abilityTypeIds = {}
+    for i = 0, 9 do
+        abilityTypeIds[i + 1] = createHealthRegenerationDummyAbilityType(1 << i)
+    end
+    return abilityTypeIds
+end)
+---
+-- @internal For use by internal systems.
+____exports.HEALTH_REGENERATION_RATE_DECREASE_DUMMY_VALUE = -1024
+---
+-- @internal For use by internal systems. Shifts negative values into the positive range.
+____exports.HEALTH_REGENERATION_RATE_DECREASE_DUMMY_ABILITY_TYPE_ID = compiletime(function() return createHealthRegenerationDummyAbilityType(____exports.HEALTH_REGENERATION_RATE_DECREASE_DUMMY_VALUE) end)
+---
+-- @internal For use by internal systems. Holds the fractional remainder; being an aura,
+-- it accepts field writes but applies with the usual aura delay.
+____exports.FRACTIONAL_HEALTH_REGENERATION_RATE_INCREASE_DUMMY_ABILITY_TYPE_ID = compiletime(function()
+    local abilityType = HealingWardAuraAbilityType:create()
+    abilityType.isInternal = true
+    abilityType.isButtonVisible = false
+    abilityType.healthRegeneration = 0
+    abilityType.isHealthRegenerationPercentage = false
+    abilityType.allowedTargetCombatClassifications = combatClassificationsOf(4096)
+    abilityType.buffTypeIds = {}
+    abilityType.targetEffectPresets = {}
+    return abilityType.id
+end)
+---
+-- @internal For use by internal systems.
+____exports.FRACTIONAL_HEALTH_REGENERATION_RATE_INCREASE_ABILITY_FIELD = ABILITY_RLF_AMOUNT_OF_HIT_POINTS_REGENERATED
+return ____exports

package/engine/internal/object-data/mana-bonus.d.ts

@@ -0,0 +1,2 @@
+/** @noSelfInFile */
+export {};

package/engine/internal/object-data/mana-bonus.lua

@@ -0,0 +1,16 @@
+local ____exports = {}
+local ____mana_2Dbonus = require("engine.object-data.entry.ability-type.mana-bonus")
+local ManaBonusAbilityType = ____mana_2Dbonus.ManaBonusAbilityType
+---
+-- @internal For use by internal systems only.
+____exports.MANA_BONUS_DUMMY_ABILITY_TYPE_ID = compiletime(function()
+    local abilityType = ManaBonusAbilityType:create()
+    abilityType.isInternal = true
+    abilityType.isButtonVisible = false
+    abilityType.manaBonus = 0
+    return abilityType.id
+end)
+---
+-- @internal For use by internal systems only.
+____exports.MANA_BONUS_DUMMY_ABILITY_FIELD = ABILITY_ILF_MAX_MANA_GAINED
+return ____exports

package/engine/internal/object-data/returned-melee-damage-factor.d.ts

@@ -0,0 +1,2 @@
+/** @noSelfInFile */
+export {};

package/engine/internal/object-data/returned-melee-damage-factor.lua

@@ -0,0 +1,21 @@
+local ____exports = {}
+local ____spiked_2Dcarapace = require("engine.object-data.entry.ability-type.spiked-carapace")
+local SpikedCarapaceAbilityType = ____spiked_2Dcarapace.SpikedCarapaceAbilityType
+---
+-- @internal For use by internal systems.
+-- 
+-- Spiked Carapace returns the given factor of melee attack damage to the attacker; the engine
+-- decides what counts as a melee attack.
+____exports.RETURNED_MELEE_DAMAGE_FACTOR_DUMMY_ABILITY_TYPE_ID = compiletime(function()
+    local abilityType = SpikedCarapaceAbilityType:create()
+    abilityType.isInternal = true
+    abilityType.isButtonVisible = false
+    abilityType.returnedDamageFactor = 0
+    abilityType.receivedDamageFactor = 1
+    abilityType.armorIncrease = 0
+    return abilityType.id
+end)
+---
+-- @internal For use by internal systems.
+____exports.RETURNED_MELEE_DAMAGE_FACTOR_ABILITY_FIELD = ABILITY_RLF_RETURNED_DAMAGE_FACTOR
+return ____exports

package/engine/internal/unit/ability.lua

@@ -397,6 +397,22 @@ local function invokeImpactEvent(unit, ability, ...)
     ability[impactCallbackIdAttribute] = nil
     eventInvoke(internalAbilityImpactEvent, unit, ability, ...)
 end
+--- Invokes the impact events of a cast that never reaches its native impact, with the
+-- cast's target: the common event and the one matching the target kind, as a real impact.
+-- 
+-- @internal For use by internal systems only.
+____exports.emulateAbilityImpactEvent = function(caster, ability, targetUnit, targetItem, targetDestructible, x, y)
+    eventInvoke(
+        internalAbilityImpactEvent,
+        caster,
+        ability,
+        targetUnit and targetUnit.handle,
+        targetItem and targetItem.handle,
+        targetDestructible and targetDestructible.handle,
+        x,
+        y
+    )
+end
 internalAbilityChannelingStartEvent:addListener(
     999999,
     function(unit, ability, ...)

package/engine/internal/unit/add-item-to-slot-init.lua

@@ -2,22 +2,11 @@ local ____exports = {}
 local ____unit = require("engine.internal.unit")
 local Unit = ____unit.Unit
 local ____add_2Ditem_2Dto_2Dslot = require("engine.internal.unit.add-item-to-slot")
-local fillerItems = ____add_2Ditem_2Dto_2Dslot.fillerItems
-local unitsWithFillerItems = ____add_2Ditem_2Dto_2Dslot.unitsWithFillerItems
-local setItemVisible = SetItemVisible
-local unitRemoveItem = UnitRemoveItem
+local unitRemoveFillerItems = ____add_2Ditem_2Dto_2Dslot.unitRemoveFillerItems
 Unit.itemPickedUpEvent:addListener(
-    4,
+    999999,
     function(unit)
-        local handle = unit.handle
-        if unitsWithFillerItems[handle] ~= nil then
-            for previousSlot = 1, 6 do
-                local fillerItem = fillerItems[previousSlot]
-                unitRemoveItem(handle, fillerItem)
-                setItemVisible(fillerItem, false)
-            end
-            unitsWithFillerItems[handle] = nil
-        end
+        unitRemoveFillerItems(unit.handle)
     end
 )
 return ____exports

package/engine/internal/unit/add-item-to-slot.lua

@@ -4,9 +4,14 @@ local BlankItemType = ____blank.BlankItemType
 local ____arrays = require("utility.arrays")
 local array = ____arrays.array
 local ____ignore_2Devents_2Ditems = require("engine.internal.unit.ignore-events-items")
-local ignoreEventsItems = ____ignore_2Devents_2Ditems.ignoreEventsItems
+local ignoreEventsItemTypeIds = ____ignore_2Devents_2Ditems.ignoreEventsItemTypeIds
+local ____item_2Dcontainer = require("engine.internal.unit.item-container")
+local UNIT_EXTENDED_INVENTORY_SLOT_COUNT = ____item_2Dcontainer.UNIT_EXTENDED_INVENTORY_SLOT_COUNT
+local UNIT_INVENTORY_SLOT_COUNT = ____item_2Dcontainer.UNIT_INVENTORY_SLOT_COUNT
 local setItemVisible = SetItemVisible
 local unitAddItem = UnitAddItem
+local unitInventorySize = UnitInventorySize
+local unitItemInBagSlot = UnitItemInBagSlot
 local unitItemInSlot = UnitItemInSlot
 local unitRemoveItem = UnitRemoveItem
 ---
@@ -16,14 +21,15 @@ ____exports.SLOT_FILLER_ITEM_TYPE_ID = compiletime(function()
     itemType.name = "[Warscript/Dummy] Slot Filler"
     return itemType.id
 end)
----
+ignoreEventsItemTypeIds[____exports.SLOT_FILLER_ITEM_TYPE_ID] = true
+--- Enough fillers to occupy every regular and extended inventory slot that precedes the last extended one.
+-- 
 -- @internal For use by internal systems only.
 ____exports.fillerItems = array(
-    6,
+    UNIT_INVENTORY_SLOT_COUNT + UNIT_EXTENDED_INVENTORY_SLOT_COUNT,
     function()
         local item = CreateItem(____exports.SLOT_FILLER_ITEM_TYPE_ID, 0, 0)
         setItemVisible(item, false)
-        ignoreEventsItems[item] = true
         return item
     end
 )
@@ -32,21 +38,56 @@ ____exports.fillerItems = array(
 ____exports.unitsWithFillerItems = {}
 ---
 -- @internal For use by internal systems only.
-____exports.unitAddItemToSlot = function(unit, item, slot)
-    for previousSlot = 0, slot - 1 do
-        if unitItemInSlot(unit, previousSlot) == nil then
-            unitAddItem(unit, ____exports.fillerItems[previousSlot + 1])
-            ____exports.unitsWithFillerItems[unit] = true
+____exports.unitRemoveFillerItems = function(unit)
+    if ____exports.unitsWithFillerItems[unit] ~= nil then
+        for ____, fillerItem in ipairs(____exports.fillerItems) do
+            unitRemoveItem(unit, fillerItem)
+            setItemVisible(fillerItem, false)
+        end
+        ____exports.unitsWithFillerItems[unit] = nil
+    end
+end
+--- Adds an item to a unit so that it lands in a specific slot of its regular or extended inventory.
+-- 
+-- The engine fills slots in order (the regular inventory first, then the extended one), so every empty slot
+-- preceding the target is temporarily occupied by a filler item. Unlike `moveslot` orders, this works
+-- regardless of whether the unit can currently act.
+-- 
+-- @internal For use by internal systems only.
+____exports.unitAddItemToSlot = function(unit, item, slot, containerType)
+    if containerType == nil then
+        containerType = 0
+    end
+    local fillerCount = 0
+    local function fillEmptySlots(slotCount, itemInSlot)
+        for precedingSlot = 0, slotCount - 1 do
+            if itemInSlot(unit, precedingSlot) == nil then
+                unitAddItem(unit, ____exports.fillerItems[fillerCount + 1])
+                fillerCount = fillerCount + 1
+            end
         end
     end
-    unitAddItem(unit, item)
+    if containerType == 0 then
+        fillEmptySlots(slot, unitItemInSlot)
+    else
+        fillEmptySlots(
+            unitInventorySize(unit),
+            unitItemInSlot
+        )
+        fillEmptySlots(slot, unitItemInBagSlot)
+    end
+    if fillerCount > 0 then
+        ____exports.unitsWithFillerItems[unit] = true
+    end
+    local result = unitAddItem(unit, item)
     if ____exports.unitsWithFillerItems[unit] ~= nil then
-        for previousSlot = 0, slot - 1 do
-            local fillerItem = ____exports.fillerItems[previousSlot + 1]
+        for fillerIndex = 0, fillerCount - 1 do
+            local fillerItem = ____exports.fillerItems[fillerIndex + 1]
             unitRemoveItem(unit, fillerItem)
             setItemVisible(fillerItem, false)
         end
         ____exports.unitsWithFillerItems[unit] = nil
     end
+    return result
 end
 return ____exports

package/engine/internal/unit/appearance.d.ts

@@ -0,0 +1,15 @@
+/** @noSelfInFile */
+declare module "../unit" {
+    interface Unit {
+        getAnimationDuration(this: Unit, animation: string | number): number;
+        setAurasEnabled(this: Unit, enabled: boolean, affectsUI?: boolean): void;
+        isHeroGlowAllowed: boolean;
+        maximumFlyHeight: number;
+        showsAirToGroundIndicator: boolean;
+        alwaysDisplaysHealth: boolean;
+    }
+    interface UnitWeapon {
+        resetAttack(this: UnitWeapon): void;
+    }
+}
+export {};

package/engine/internal/unit/appearance.lua

@@ -0,0 +1,85 @@
+local ____lualib = require("lualib_bundle")
+local __TS__ObjectDefineProperty = ____lualib.__TS__ObjectDefineProperty
+local ____exports = {}
+local ____unit = require("engine.internal.unit")
+local Unit = ____unit.Unit
+local UnitWeapon = ____unit.UnitWeapon
+local allowHeroGlowOnUnit = AllowHeroGlowOnUnit
+local disallowHeroGlowOnUnit = DisallowHeroGlowOnUnit
+local getUnitAnimationDuration = BlzGetUnitAnimationDuration
+local getUnitAnimationDurationByIndex = BlzGetUnitAnimationDurationByIndex
+local getUnitBooleanField = BlzGetUnitBooleanField
+local getUnitRealField = BlzGetUnitRealField
+local heroGlowIsAllowedOnUnit = HeroGlowIsAllowedOnUnit
+local resetUnitAttack = BlzResetUnitAttack
+local setUnitBooleanField = BlzSetUnitBooleanField
+local setUnitRealField = BlzSetUnitRealField
+local unitEnableAuras = BlzUnitEnableAuras
+Unit.prototype.getAnimationDuration = function(self, animation)
+    if type(animation) == "number" then
+        return getUnitAnimationDurationByIndex(self.handle, animation)
+    end
+    return getUnitAnimationDuration(self.handle, animation)
+end
+Unit.prototype.setAurasEnabled = function(self, enabled, affectsUI)
+    if affectsUI == nil then
+        affectsUI = true
+    end
+    unitEnableAuras(self.handle, enabled, affectsUI)
+end
+__TS__ObjectDefineProperty(
+    Unit.prototype,
+    "isHeroGlowAllowed",
+    {
+        get = function(self)
+            return heroGlowIsAllowedOnUnit(self.handle)
+        end,
+        set = function(self, allowed)
+            if allowed then
+                allowHeroGlowOnUnit(self.handle)
+            else
+                disallowHeroGlowOnUnit(self.handle)
+            end
+        end
+    }
+)
+__TS__ObjectDefineProperty(
+    Unit.prototype,
+    "maximumFlyHeight",
+    {
+        get = function(self)
+            return getUnitRealField(self.handle, UNIT_RF_FLY_MAX_HEIGHT)
+        end,
+        set = function(self, height)
+            setUnitRealField(self.handle, UNIT_RF_FLY_MAX_HEIGHT, height)
+        end
+    }
+)
+__TS__ObjectDefineProperty(
+    Unit.prototype,
+    "showsAirToGroundIndicator",
+    {
+        get = function(self)
+            return getUnitBooleanField(self.handle, UNIT_BF_SHOW_AIR_TO_GROUND)
+        end,
+        set = function(self, show)
+            setUnitBooleanField(self.handle, UNIT_BF_SHOW_AIR_TO_GROUND, show)
+        end
+    }
+)
+__TS__ObjectDefineProperty(
+    Unit.prototype,
+    "alwaysDisplaysHealth",
+    {
+        get = function(self)
+            return getUnitBooleanField(self.handle, UNIT_BF_FORCE_DISPLAY_HP)
+        end,
+        set = function(self, show)
+            setUnitBooleanField(self.handle, UNIT_BF_FORCE_DISPLAY_HP, show)
+        end
+    }
+)
+UnitWeapon.prototype.resetAttack = function(self)
+    resetUnitAttack(self.unit.handle, self.index)
+end
+return ____exports

package/engine/internal/unit/bonus.d.ts

@@ -5,14 +5,18 @@ export type UnitBonusId<Brand extends string = any> = number & {
     readonly __unitBonusId: unique symbol;
     readonly __unitBonusIdBrand: Brand;
 };
+export type UnitMaxHealthBonusId = UnitBonusId<"maxHealth">;
+export type UnitMaxManaBonusId = UnitBonusId<"maxMana">;
 export type UnitArmorBonusId = UnitBonusId<"armor">;
 export type UnitAttackSpeedFactorBonusId = UnitBonusId<"attackSpeedFactor">;
 export type UnitMovementSpeedFactorBonusId = UnitBonusId<"movementSpeedFactor">;
 export type UnitAutoAttackDamageBonusId = UnitBonusId<"autoAttackDamage">;
 export type UnitDamageFactorBonusId = UnitBonusId<"damageFactor">;
 export type UnitReceivedDamageFactorBonusId = UnitBonusId<"receivedDamageFactor">;
+export type UnitReturnedMeleeDamageFactorBonusId = UnitBonusId<"returnedMeleeDamageFactor">;
 export type UnitEvasionProbabilityBonusId = UnitBonusId<"evasionProbability">;
 export type UnitManaRegenerationRateFactorBonusId = UnitBonusId<"manaRegenerationRateFactor">;
+export type UnitHealthRegenerationRateBonusId = UnitBonusId<"healthRegenerationRate">;
 export type UnitBonusType<Id extends UnitBonusId = UnitBonusId> = ({
     abilityTypeId: AbilityTypeId;
     field: jabilityintegerlevelfield;
@@ -24,20 +28,31 @@ export type UnitBonusType<Id extends UnitBonusId = UnitBonusId> = ({
 } | {
     abilityTypeId?: undefined;
     valueByUnit: LuaMap<Unit, number>;
+    apply?: undefined;
+} | {
+    abilityTypeId?: undefined;
+    valueByUnit?: undefined;
+    apply: (this: void, unitHandle: junit, value: number) => void;
 }) & {
     reduce: (this: void, array: number[]) => number;
     initialValue: number;
     readonly __unitBonusTypeId?: Id;
 };
 export declare namespace UnitBonusType {
+    const HEALTH: UnitBonusType<UnitMaxHealthBonusId>;
+    const MANA: UnitBonusType<UnitMaxManaBonusId>;
     const ARMOR: UnitBonusType<UnitArmorBonusId>;
     const ATTACK_SPEED_FACTOR: UnitBonusType<UnitAttackSpeedFactorBonusId>;
     const MOVEMENT_SPEED_FACTOR: UnitBonusType<UnitAttackSpeedFactorBonusId>;
     const AUTO_ATTACK_DAMAGE: UnitBonusType<UnitAutoAttackDamageBonusId>;
     const DAMAGE_FACTOR: UnitBonusType<UnitReceivedDamageFactorBonusId>;
     const RECEIVED_DAMAGE_FACTOR: UnitBonusType<UnitReceivedDamageFactorBonusId>;
+    /** The factor of melee attack damage returned to the attacker, as Spiked Carapace. */
+    const RETURNED_MELEE_DAMAGE_FACTOR: UnitBonusType<UnitReturnedMeleeDamageFactorBonusId>;
     const EVASION_PROBABILITY: UnitBonusType<UnitEvasionProbabilityBonusId>;
     const MANA_REGENERATION_RATE_FACTOR: UnitBonusType<UnitManaRegenerationRateFactorBonusId>;
+    /** Supported range: -1000 to 1000; the total is clamped to it. */
+    const HEALTH_REGENERATION_RATE: UnitBonusType<UnitHealthRegenerationRateBonusId>;
 }
 export declare const addUnitBonus: <Id extends UnitBonusId>(unit: Unit, bonusType: UnitBonusType<Id>, value: number) => Id;
 export declare const removeUnitBonus: <Id extends UnitBonusId>(unit: Unit, bonusType: UnitBonusType<Id>, id: Id) => boolean;

package/engine/internal/unit/bonus.lua

@@ -1,11 +1,14 @@
 local ____exports = {}
+local applyAbilityBonus, applyHealthRegenerationRateBonus
 local ____unit = require("engine.internal.unit")
 local Unit = ____unit.Unit
-local ____armor_2Dincrease = require("engine.internal.object-data.armor-increase")
-local ARMOR_BONUS_DUMMY_ABILITY_FIELD = ____armor_2Dincrease.ARMOR_BONUS_DUMMY_ABILITY_FIELD
-local ARMOR_BONUS_DUMMY_ABILITY_TYPE_ID = ____armor_2Dincrease.ARMOR_BONUS_DUMMY_ABILITY_TYPE_ID
+local ____armor_2Dbonus = require("engine.internal.object-data.armor-bonus")
+local ARMOR_BONUS_DUMMY_ABILITY_FIELD = ____armor_2Dbonus.ARMOR_BONUS_DUMMY_ABILITY_FIELD
+local ARMOR_BONUS_DUMMY_ABILITY_TYPE_ID = ____armor_2Dbonus.ARMOR_BONUS_DUMMY_ABILITY_TYPE_ID
 local ____utility = require("engine.internal.utility")
 local addInternalAbility = ____utility.addInternalAbility
+local ____unit_2Dability_2Dlevel_2Dnatives = require("engine.internal.misc.unit-ability-level-natives")
+local rawSetUnitAbilityLevel = ____unit_2Dability_2Dlevel_2Dnatives.rawSetUnitAbilityLevel
 local ____arrays = require("utility.arrays")
 local product = ____arrays.product
 local sum = ____arrays.sum
@@ -24,9 +27,24 @@ local MOVEMENT_SPEED_INCREASE_FACTOR_DUMMY_ABILITY_TYPE_ID = ____movement_2Dspee
 local ____evasion_2Dprobability = require("engine.internal.object-data.evasion-probability")
 local EVASION_PROBABILITY_ABILITY_FIELD = ____evasion_2Dprobability.EVASION_PROBABILITY_ABILITY_FIELD
 local EVASION_PROBABILITY_DUMMY_ABILITY_TYPE_ID = ____evasion_2Dprobability.EVASION_PROBABILITY_DUMMY_ABILITY_TYPE_ID
+local ____health_2Dbonus = require("engine.internal.object-data.health-bonus")
+local HEALTH_BONUS_DUMMY_ABILITY_FIELD = ____health_2Dbonus.HEALTH_BONUS_DUMMY_ABILITY_FIELD
+local HEALTH_BONUS_DUMMY_ABILITY_TYPE_ID = ____health_2Dbonus.HEALTH_BONUS_DUMMY_ABILITY_TYPE_ID
+local ____mana_2Dbonus = require("engine.internal.object-data.mana-bonus")
+local MANA_BONUS_DUMMY_ABILITY_FIELD = ____mana_2Dbonus.MANA_BONUS_DUMMY_ABILITY_FIELD
+local MANA_BONUS_DUMMY_ABILITY_TYPE_ID = ____mana_2Dbonus.MANA_BONUS_DUMMY_ABILITY_TYPE_ID
 local ____mana_2Dregeneration_2Drate_2Dincrease_2Dfactor = require("engine.internal.object-data.mana-regeneration-rate-increase-factor")
 local MANA_REGENERATION_RATE_INCREASE_FACTOR_ABILITY_FIELD = ____mana_2Dregeneration_2Drate_2Dincrease_2Dfactor.MANA_REGENERATION_RATE_INCREASE_FACTOR_ABILITY_FIELD
 local MANA_REGENERATION_RATE_INCREASE_FACTOR_DUMMY_ABILITY_TYPE_ID = ____mana_2Dregeneration_2Drate_2Dincrease_2Dfactor.MANA_REGENERATION_RATE_INCREASE_FACTOR_DUMMY_ABILITY_TYPE_ID
+local ____health_2Dregeneration_2Drate_2Dincrease = require("engine.internal.object-data.health-regeneration-rate-increase")
+local FRACTIONAL_HEALTH_REGENERATION_RATE_INCREASE_ABILITY_FIELD = ____health_2Dregeneration_2Drate_2Dincrease.FRACTIONAL_HEALTH_REGENERATION_RATE_INCREASE_ABILITY_FIELD
+local FRACTIONAL_HEALTH_REGENERATION_RATE_INCREASE_DUMMY_ABILITY_TYPE_ID = ____health_2Dregeneration_2Drate_2Dincrease.FRACTIONAL_HEALTH_REGENERATION_RATE_INCREASE_DUMMY_ABILITY_TYPE_ID
+local HEALTH_REGENERATION_RATE_DECREASE_DUMMY_ABILITY_TYPE_ID = ____health_2Dregeneration_2Drate_2Dincrease.HEALTH_REGENERATION_RATE_DECREASE_DUMMY_ABILITY_TYPE_ID
+local HEALTH_REGENERATION_RATE_DECREASE_DUMMY_VALUE = ____health_2Dregeneration_2Drate_2Dincrease.HEALTH_REGENERATION_RATE_DECREASE_DUMMY_VALUE
+local HEALTH_REGENERATION_RATE_INCREASE_DUMMY_ABILITY_TYPE_IDS = ____health_2Dregeneration_2Drate_2Dincrease.HEALTH_REGENERATION_RATE_INCREASE_DUMMY_ABILITY_TYPE_IDS
+local ____returned_2Dmelee_2Ddamage_2Dfactor = require("engine.internal.object-data.returned-melee-damage-factor")
+local RETURNED_MELEE_DAMAGE_FACTOR_ABILITY_FIELD = ____returned_2Dmelee_2Ddamage_2Dfactor.RETURNED_MELEE_DAMAGE_FACTOR_ABILITY_FIELD
+local RETURNED_MELEE_DAMAGE_FACTOR_DUMMY_ABILITY_TYPE_ID = ____returned_2Dmelee_2Ddamage_2Dfactor.RETURNED_MELEE_DAMAGE_FACTOR_DUMMY_ABILITY_TYPE_ID
 local damageFactorByUnit = {}
 local receivedDamageFactorByUnit = {}
 local function atLeastOnceProbability(array)
@@ -39,6 +57,20 @@ end
 ____exports.UnitBonusType = {}
 local UnitBonusType = ____exports.UnitBonusType
 do
+    UnitBonusType.HEALTH = {
+        abilityTypeId = HEALTH_BONUS_DUMMY_ABILITY_TYPE_ID,
+        field = HEALTH_BONUS_DUMMY_ABILITY_FIELD,
+        integer = true,
+        reduce = sum,
+        initialValue = 0
+    }
+    UnitBonusType.MANA = {
+        abilityTypeId = MANA_BONUS_DUMMY_ABILITY_TYPE_ID,
+        field = MANA_BONUS_DUMMY_ABILITY_FIELD,
+        integer = true,
+        reduce = sum,
+        initialValue = 0
+    }
     UnitBonusType.ARMOR = {
         abilityTypeId = ARMOR_BONUS_DUMMY_ABILITY_TYPE_ID,
         field = ARMOR_BONUS_DUMMY_ABILITY_FIELD,
@@ -69,6 +101,14 @@ do
     }
     UnitBonusType.DAMAGE_FACTOR = {reduce = product, valueByUnit = damageFactorByUnit, initialValue = 1}
     UnitBonusType.RECEIVED_DAMAGE_FACTOR = {reduce = product, valueByUnit = receivedDamageFactorByUnit, initialValue = 1}
+    --- The factor of melee attack damage returned to the attacker, as Spiked Carapace.
+    UnitBonusType.RETURNED_MELEE_DAMAGE_FACTOR = {
+        abilityTypeId = RETURNED_MELEE_DAMAGE_FACTOR_DUMMY_ABILITY_TYPE_ID,
+        field = RETURNED_MELEE_DAMAGE_FACTOR_ABILITY_FIELD,
+        integer = false,
+        reduce = sum,
+        initialValue = 0
+    }
     UnitBonusType.EVASION_PROBABILITY = {
         abilityTypeId = EVASION_PROBABILITY_DUMMY_ABILITY_TYPE_ID,
         field = EVASION_PROBABILITY_ABILITY_FIELD,
@@ -83,33 +123,88 @@ do
         reduce = sum,
         initialValue = 0
     }
+    --- Supported range: -1000 to 1000; the total is clamped to it.
+    UnitBonusType.HEALTH_REGENERATION_RATE = {
+        apply = function(unitHandle, value) return applyHealthRegenerationRateBonus(unitHandle, value) end,
+        reduce = sum,
+        initialValue = 0
+    }
 end
 local bonusesByUnitByBonusType = {}
 local nextId = 1
+local ceil = math.ceil
+local floor = math.floor
+local max = math.max
+local min = math.min
 local function processUnitBonus(unit, bonusType, bonusesArray)
     local unitHandle = unit.handle
     local totalValue = bonusType.reduce(bonusesArray)
     local abilityTypeId = bonusType.abilityTypeId
     if abilityTypeId == nil then
         local valueByUnit = bonusType.valueByUnit
-        if totalValue == bonusType.initialValue then
+        if valueByUnit == nil then
+            bonusType.apply(unitHandle, totalValue)
+        elseif totalValue == bonusType.initialValue then
             valueByUnit[unit] = nil
         else
             valueByUnit[unit] = totalValue
         end
         return
     end
-    if totalValue == bonusType.initialValue then
+    applyAbilityBonus(
+        unitHandle,
+        abilityTypeId,
+        bonusType.field,
+        bonusType.integer,
+        totalValue,
+        bonusType.initialValue
+    )
+end
+applyAbilityBonus = function(unitHandle, abilityTypeId, field, integer, value, initialValue)
+    if value == initialValue then
         UnitRemoveAbility(unitHandle, abilityTypeId)
         return
     end
     local abilityHandle = checkNotNull(addInternalAbility(unitHandle, abilityTypeId))
-    check((bonusType.integer and BlzSetAbilityIntegerLevelField or BlzSetAbilityRealLevelField)(abilityHandle, bonusType.field, 0, totalValue))
+    check((integer and BlzSetAbilityIntegerLevelField or BlzSetAbilityRealLevelField)(abilityHandle, field, 0, value))
     BlzSetAbilityIntegerField(abilityHandle, ABILITY_IF_LEVELS, 2)
-    SetUnitAbilityLevel(unitHandle, abilityTypeId, 2)
-    SetUnitAbilityLevel(unitHandle, abilityTypeId, 1)
+    rawSetUnitAbilityLevel(unitHandle, abilityTypeId, 2)
+    rawSetUnitAbilityLevel(unitHandle, abilityTypeId, 1)
     BlzSetAbilityIntegerField(abilityHandle, ABILITY_IF_LEVELS, 1)
 end
+local function setInternalAbilityPresence(unitHandle, abilityTypeId, present)
+    if present then
+        addInternalAbility(unitHandle, abilityTypeId)
+    else
+        UnitRemoveAbility(unitHandle, abilityTypeId)
+    end
+end
+local MINIMUM_HEALTH_REGENERATION_RATE_BONUS = -1000
+local MAXIMUM_HEALTH_REGENERATION_RATE_BONUS = 1000
+applyHealthRegenerationRateBonus = function(unitHandle, value)
+    value = max(
+        MINIMUM_HEALTH_REGENERATION_RATE_BONUS,
+        min(value, MAXIMUM_HEALTH_REGENERATION_RATE_BONUS)
+    )
+    local integerValue = value < 0 and ceil(value) or floor(value)
+    local remainder = integerValue
+    setInternalAbilityPresence(unitHandle, HEALTH_REGENERATION_RATE_DECREASE_DUMMY_ABILITY_TYPE_ID, remainder < 0)
+    if remainder < 0 then
+        remainder = remainder - HEALTH_REGENERATION_RATE_DECREASE_DUMMY_VALUE
+    end
+    for ____, abilityTypeId in ipairs(HEALTH_REGENERATION_RATE_INCREASE_DUMMY_ABILITY_TYPE_IDS) do
+        setInternalAbilityPresence(unitHandle, abilityTypeId, remainder % 2 == 1)
+        remainder = floor(remainder / 2)
+    end
+    applyAbilityBonus(
+        unitHandle,
+        FRACTIONAL_HEALTH_REGENERATION_RATE_INCREASE_DUMMY_ABILITY_TYPE_ID,
+        FRACTIONAL_HEALTH_REGENERATION_RATE_INCREASE_ABILITY_FIELD,
+        false,
+        value - integerValue,
+        0
+    )
+end
 ____exports.addUnitBonus = function(unit, bonusType, value)
     local bonusesByUnit = bonusesByUnitByBonusType[bonusType]
     if bonusesByUnit == nil then

package/engine/internal/unit/buff.d.ts

@@ -32,7 +32,7 @@ declare module "../unit" {
          * @param includeAuras Whether to remove aura buffs. Default `true`.
          * @param autoDispel Unknown. Default `false`.
          */
-        removeBuffs(polarity?: BuffPolarity, resistanceType?: BuffResistanceType, includeExpirationTimers?: boolean, includeAuras?: boolean, autoDispel?: boolean): void;
-        removeBuffs(buffTypeIds: number[]): void;
+        removeBuffs(polarity?: BuffPolarity, resistanceType?: BuffResistanceType, includeExpirationTimers?: boolean, includeAuras?: boolean, autoDispel?: boolean): number;
+        removeBuffs(buffTypeIds: number[]): number;
     }
 }