warscript 0.0.1-dev.0efff64 → 0.0.1-dev.10e558e

package/engine/internal/item/owner-cache.lua

@@ -0,0 +1,4 @@
+local ____exports = {}
+--- The last unit known (from item events) to carry each item; validated against the engine on read.
+____exports.ownerByItem = setmetatable({}, {__mode = "kv"})
+return ____exports

package/engine/internal/item+owner.d.ts

@@ -1,7 +1,9 @@
 /** @noSelfInFile */
 import { Unit } from "./unit";
+import "./unit/equipment-inventory";
 declare module "./item" {
     interface Item {
+        /** The unit carrying the item in any of its inventories (regular, extended or equipment), if any. */
         readonly owner?: Unit;
     }
 }

package/engine/internal/item+owner.lua

@@ -5,15 +5,33 @@ local ____item = require("engine.internal.item")
 local Item = ____item.Item
 local ____unit = require("engine.internal.unit")
 local Unit = ____unit.Unit
-local ownerByItem = setmetatable({}, {__mode = "kv"})
+require("engine.internal.unit.equipment-inventory")
+local ____owner_2Dcache = require("engine.internal.item.owner-cache")
+local ownerByItem = ____owner_2Dcache.ownerByItem
+local ____item_2Dcontainer = require("engine.internal.unit.item-container")
+local unitOwnsItem = ____item_2Dcontainer.unitOwnsItem
 Unit.itemPickedUpEvent:addListener(
-    4,
+    999999,
     function(unit, item)
         ownerByItem[item] = unit
     end
 )
+Unit.itemEquippedEvent:addListener(
+    999999,
+    function(unit, item)
+        ownerByItem[item] = unit
+    end
+)
+Unit.itemUnequippedEvent:addListener(
+    999999,
+    function(unit, item)
+        if not unitOwnsItem(unit.handle, item.handle) then
+            ownerByItem[item] = nil
+        end
+    end
+)
 Unit.itemDroppedEvent:addListener(
-    4,
+    999999,
     function(unit, item)
         ownerByItem[item] = nil
     end
@@ -22,7 +40,12 @@ __TS__ObjectDefineProperty(
     Item.prototype,
     "owner",
     {get = function(self)
-        return ownerByItem[self]
+        local owner = ownerByItem[self]
+        if owner ~= nil and not unitOwnsItem(owner.handle, self.handle) then
+            ownerByItem[self] = nil
+            return nil
+        end
+        return owner
     end}
 )
 return ____exports

package/engine/internal/item.d.ts

@@ -2,19 +2,37 @@
 import { Handle, HandleDestructor } from "../../core/types/handle";
 import { Color } from "../../core/types/color";
 import { Event } from "../../event";
+import { PlayerColor } from "../../core/types/playerColor";
+import { EquipmentType } from "../object-data/auxiliary/equipment-type";
+import { ItemTag } from "../object-data/auxiliary/item-tag";
 import { ReadonlyRect } from "../../core/types/rect";
 import { ItemAbility } from "./ability";
 import { AbilityTypeId } from "../object-data/entry/ability-type";
 import type { ItemTypeId } from "../object-data/entry/item-type";
 type DefenseType = 0 | 1 | 2 | 3 | 4 | 5;
 declare const enum ItemPropertyKey {
-    ABILITIES = 100
+    ABILITIES = 100,
+    DEATH_TRIGGER = 101,
+    COLOR = 102
 }
 export declare class Item extends Handle<jitem> {
     private readonly [ItemPropertyKey.ABILITIES];
+    private readonly [ItemPropertyKey.DEATH_TRIGGER];
+    private [ItemPropertyKey.COLOR]?;
     constructor(handle: jitem);
     protected onDestroy(): HandleDestructor;
     static create<T extends Item>(this: typeof Item & (new (handle: jitem) => T), id: number, x: number, y: number, skinId?: number): T;
+    /** Whether the item is owned by a unit, i.e. carried in any of its inventories (regular, extended or equipment). */
+    get isOwned(): boolean;
+    /** Whether the item is carried in a unit's (regular) inventory, as opposed to its extended or equipment inventory. */
+    get isInInventory(): boolean;
+    get isInExtendedInventory(): boolean;
+    get isInEquipmentInventory(): boolean;
+    get equipmentType(): EquipmentType;
+    get tag(): ItemTag;
+    get color(): PlayerColor;
+    set color(color: PlayerColor);
+    static getRandomTypeId(level: number, equipmentType?: EquipmentType | undefined, tag?: ItemTag, itemType?: jitemtype): ItemTypeId | undefined;
     get typeId(): ItemTypeId;
     set skinId(v: number);
     get skinId(): number;
@@ -78,6 +96,7 @@ export declare class Item extends Handle<jitem> {
     static getInRect(rect: ReadonlyRect): Item[];
     static get onCreate(): Event<[Item]>;
     static get destroyEvent(): Event<[Item]>;
+    static readonly deathEvent: Event<[Item]>;
     static readonly chargesChangedEvent: Event<[Item]>;
 }
 export {};

package/engine/internal/item.lua

@@ -13,6 +13,8 @@ local ____color = require("core.types.color")
 local Color = ____color.Color
 local ____event = require("event")
 local Event = ____event.Event
+local ____playerColor = require("core.types.playerColor")
+local PlayerColor = ____playerColor.PlayerColor
 local ____rect = require("core.types.rect")
 local Rect = ____rect.Rect
 local ____ability = require("engine.internal.ability")
@@ -27,37 +29,58 @@ local ____add_2Ditem_2Dto_2Dslot = require("engine.internal.unit.add-item-to-slo
 local SLOT_FILLER_ITEM_TYPE_ID = ____add_2Ditem_2Dto_2Dslot.SLOT_FILLER_ITEM_TYPE_ID
 local ____vec2 = require("math.vec2")
 local distance = ____vec2.distance
-local itemChargesChangeEvent = __TS__New(Event)
-local itemAddAbility = BlzItemAddAbility
-local itemRemoveAbility = BlzItemRemoveAbility
-local getItemAbility = BlzGetItemAbility
-local isItemPowerup = IsItemPowerup
-local getItemAbilityByIndex = BlzGetItemAbilityByIndex
-local getAbilityId = BlzGetAbilityId
-local getWidgetLife = GetWidgetLife
-local removeItem = RemoveItem
-local getHandleId = GetHandleId
-local setRect = SetRect
+local ____timer = require("core.types.timer")
+local Timer = ____timer.Timer
+local chooseRandomItemExWithFilter = ChooseRandomItemExWithFilter
+local condition = Condition
+local convertEquipmentType = ConvertEquipmentType
+local convertItemTag = ConvertItemTag
+local createTrigger = CreateTrigger
+local destroyTrigger = DestroyTrigger
 local enumItemsInRect = EnumItemsInRect
+local getAbilityId = BlzGetAbilityId
 local getEnumItem = GetEnumItem
+local getHandleId = GetHandleId
+local getItemAbility = BlzGetItemAbility
+local getItemAbilityByIndex = BlzGetItemAbilityByIndex
+local getItemBooleanField = BlzGetItemBooleanField
+local getItemCharges = GetItemCharges
+local getItemEquipmentType = GetItemEquipmentType
+local getItemIntegerField = BlzGetItemIntegerField
+local getItemTag = GetItemTag
 local getItemTypeId = GetItemTypeId
 local getItemX = GetItemX
 local getItemY = GetItemY
-local getItemCharges = GetItemCharges
+local getManipulatedItem = GetManipulatedItem
+local getTriggerWidget = GetTriggerWidget
+local getWidgetLife = GetWidgetLife
+local isItemEquipped = IsItemEquipped
+local isItemInBag = IsItemInBag
+local isItemOwned = IsItemOwned
+local isItemPawnable = IsItemPawnable
+local isItemPowerup = IsItemPowerup
+local itemAddAbility = BlzItemAddAbility
+local itemRemoveAbility = BlzItemRemoveAbility
+local loadItemHandle = LoadItemHandle
+local removeItem = RemoveItem
+local saveWidgetHandle = SaveWidgetHandle
+local setItemBooleanField = BlzSetItemBooleanField
 local setItemCharges = SetItemCharges
+local setItemColor = SetItemColor
+local setItemDropOnDeath = SetItemDropOnDeath
+local setItemDroppable = SetItemDroppable
+local setItemPawnable = SetItemPawnable
+local setRect = SetRect
+local triggerAddCondition = TriggerAddCondition
+local triggerRegisterDeathEvent = TriggerRegisterDeathEvent
 local unitRemoveAbility = UnitRemoveAbility
 local unitRemoveItem = UnitRemoveItem
 local unitUseItem = UnitUseItem
 local unitUseItemPoint = UnitUseItemPoint
 local unitUseItemTarget = UnitUseItemTarget
-local setItemDropOnDeath = SetItemDropOnDeath
-local setItemDroppable = SetItemDroppable
-local setItemPawnable = SetItemPawnable
-local isItemPawnable = IsItemPawnable
-local getItemIntegerField = BlzGetItemIntegerField
-local setItemBooleanField = BlzSetItemBooleanField
-local getItemBooleanField = BlzGetItemBooleanField
 local tableRemove = table.remove
+local itemChargesChangeEvent = __TS__New(Event)
+local itemDeathEvent = __TS__New(Event)
 _G.SetItemCharges = function(whichItem, charges)
     setItemCharges(whichItem, charges)
     invoke(
@@ -137,6 +160,25 @@ local function collectIntoTargetRange()
         targetCollectionNextIndex = targetCollectionNextIndex + 1
     end
 end
+local hashtable = InitHashtable()
+local deathTriggerCondition = condition(function()
+    saveWidgetHandle(
+        hashtable,
+        1,
+        1,
+        getTriggerWidget()
+    )
+    local item = ____exports.Item:of(loadItemHandle(hashtable, 1, 1))
+    if item ~= nil then
+        invoke(itemDeathEvent, item)
+    end
+end)
+itemDeathEvent:addListener(
+    999999,
+    function(item)
+        Timer:run(item, "destroy")
+    end
+)
 ____exports.Item = __TS__Class()
 local Item = ____exports.Item
 Item.name = "Item"
@@ -144,6 +186,10 @@ __TS__ClassExtends(Item, Handle)
 function Item.prototype.____constructor(self, handle)
     Handle.prototype.____constructor(self, handle)
     self[100] = doAbilityAction(handle, getItemAbilities, self)
+    local deathTrigger = createTrigger()
+    triggerRegisterDeathEvent(deathTrigger, handle)
+    triggerAddCondition(deathTrigger, deathTriggerCondition)
+    self[101] = deathTrigger
 end
 function Item.prototype.onDestroy(self)
     local owner = self.owner
@@ -155,11 +201,27 @@ function Item.prototype.onDestroy(self)
         abilities[i]:destroy()
     end
     removeItem(self.handle)
+    destroyTrigger(self[101])
     return Handle.prototype.onDestroy(self)
 end
 function Item.create(self, id, x, y, skinId)
     return self:of(BlzCreateItemWithSkin(id, x, y, skinId or id))
 end
+function Item.getRandomTypeId(self, level, equipmentType, tag, itemType)
+    if tag == nil then
+        tag = 8
+    end
+    if itemType == nil then
+        itemType = ITEM_TYPE_ANY
+    end
+    local id = chooseRandomItemExWithFilter(
+        itemType,
+        level,
+        convertEquipmentType(equipmentType or 9),
+        convertItemTag(tag)
+    )
+    return id ~= 0 and id or nil
+end
 function Item.prototype.consumeCharge(self)
     return self:consumeCharges(1)
 end
@@ -244,6 +306,70 @@ function Item.getInRect(self, rect)
     enumItemsInRect(rect.handle, nil, collectIntoTarget)
     return targetCollection
 end
+__TS__SetDescriptor(
+    Item.prototype,
+    "isOwned",
+    {get = function(self)
+        local handle = self.handle
+        return isItemOwned(handle) or isItemInBag(handle) or isItemEquipped(handle)
+    end},
+    true
+)
+__TS__SetDescriptor(
+    Item.prototype,
+    "isInInventory",
+    {get = function(self)
+        local handle = self.handle
+        return isItemOwned(handle) and not isItemInBag(handle) and not isItemEquipped(handle)
+    end},
+    true
+)
+__TS__SetDescriptor(
+    Item.prototype,
+    "isInExtendedInventory",
+    {get = function(self)
+        return isItemInBag(self.handle)
+    end},
+    true
+)
+__TS__SetDescriptor(
+    Item.prototype,
+    "isInEquipmentInventory",
+    {get = function(self)
+        return isItemEquipped(self.handle)
+    end},
+    true
+)
+__TS__SetDescriptor(
+    Item.prototype,
+    "equipmentType",
+    {get = function(self)
+        return getHandleId(getItemEquipmentType(self.handle))
+    end},
+    true
+)
+__TS__SetDescriptor(
+    Item.prototype,
+    "tag",
+    {get = function(self)
+        return getHandleId(getItemTag(self.handle))
+    end},
+    true
+)
+__TS__SetDescriptor(
+    Item.prototype,
+    "color",
+    {
+        get = function(self)
+            return self[102] or PlayerColor.red
+        end,
+        set = function(self, color)
+            setItemColor(self.handle, color.handle)
+            self[102] = color
+        end
+    },
+    true
+)
 __TS__SetDescriptor(
     Item.prototype,
     "typeId",
@@ -635,8 +761,8 @@ __TS__ObjectDefineProperty(
         return self.onDestroyEvent
     end}
 )
+Item.deathEvent = itemDeathEvent
 Item.chargesChangedEvent = itemChargesChangeEvent
-local getManipulatedItem = GetManipulatedItem
 local trigger = CreateTrigger()
 TriggerRegisterAnyUnitEventBJ(trigger, EVENT_PLAYER_UNIT_PICKUP_ITEM)
 TriggerAddAction(

package/engine/internal/misc/unit-ability-level-natives.d.ts

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

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/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/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

@@ -15,6 +15,7 @@ export type UnitDamageFactorBonusId = UnitBonusId<"damageFactor">;
 export type UnitReceivedDamageFactorBonusId = UnitBonusId<"receivedDamageFactor">;
 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;
@@ -26,6 +27,11 @@ 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;
@@ -42,6 +48,8 @@ export declare namespace UnitBonusType {
     const RECEIVED_DAMAGE_FACTOR: UnitBonusType<UnitReceivedDamageFactorBonusId>;
     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;