warscript 0.0.1-dev.c37d12c → 0.0.1-dev.c47d565

package/engine/internal/unit/item-order-events.lua

@@ -0,0 +1,67 @@
+local ____lualib = require("lualib_bundle")
+local __TS__New = ____lualib.__TS__New
+local __TS__ObjectDefineProperty = ____lualib.__TS__ObjectDefineProperty
+local ____exports = {}
+local ____unit = require("engine.internal.unit")
+local Unit = ____unit.Unit
+local ____event = require("event")
+local Event = ____event.Event
+local ____item_2Dslot = require("engine.internal.unit.item-slot")
+local UnitItemSlot = ____item_2Dslot.UnitItemSlot
+local invoke = Event.invoke
+__TS__ObjectDefineProperty(
+    Unit,
+    "itemUseOrderEvent",
+    {get = function(self)
+        local event = __TS__New(Event)
+        local function addListeners(firstOrderType, lastOrderType, containerType)
+            for orderType = firstOrderType, lastOrderType do
+                local slot = UnitItemSlot:get(containerType, orderType - firstOrderType)
+                local function listener(unit)
+                    local item = slot:getItem(unit)
+                    if item ~= nil then
+                        invoke(event, unit, item)
+                    end
+                end
+                Unit.onImmediateOrder[orderType]:addListener(listener)
+                Unit.onTargetOrder[orderType]:addListener(listener)
+                Unit.onPointOrder[orderType]:addListener(listener)
+            end
+        end
+        addListeners(852008, 852013, 0)
+        addListeners(852771, 852800, 1)
+        addListeners(852801, 852809, 2)
+        rawset(Unit, "itemUseOrderEvent", event)
+        return event
+    end}
+)
+__TS__ObjectDefineProperty(
+    Unit,
+    "itemMoveOrderEvent",
+    {get = function(self)
+        local event = __TS__New(Event)
+        local function addListeners(firstOrderType, lastOrderType, containerType)
+            for orderType = firstOrderType, lastOrderType do
+                local slotTo = UnitItemSlot:get(containerType, orderType - firstOrderType)
+                Unit.onTargetOrder[orderType]:addListener(function(unit, item)
+                    local slotFrom = UnitItemSlot:find(unit, item)
+                    if slotFrom ~= nil then
+                        invoke(
+                            event,
+                            unit,
+                            item,
+                            slotFrom,
+                            slotTo
+                        )
+                    end
+                end)
+            end
+        end
+        addListeners(852002, 852007, 0)
+        addListeners(852714, 852743, 1)
+        addListeners(852744, 852752, 2)
+        rawset(Unit, "itemMoveOrderEvent", event)
+        return event
+    end}
+)
+return ____exports

package/engine/internal/unit/item-slot.d.ts

@@ -0,0 +1,27 @@
+/** @noSelfInFile */
+import { Item } from "../item";
+import { Unit } from "../unit";
+import { EquipmentSlot } from "../../object-data/auxiliary/equipment-slot";
+import { UNIT_EQUIPMENT_INVENTORY_SLOT_COUNT, UNIT_EXTENDED_INVENTORY_SLOT_COUNT, UNIT_INVENTORY_SLOT_COUNT, UnitItemContainerType } from "./item-container";
+export { UNIT_EQUIPMENT_INVENTORY_SLOT_COUNT, UNIT_EXTENDED_INVENTORY_SLOT_COUNT, UNIT_INVENTORY_SLOT_COUNT, UnitItemContainerType, };
+/**
+ * A slot of one of a unit's item containers (the inventory, the extended inventory or the equipment inventory).
+ *
+ * There is exactly one instance per (container type, index) pair, so slots can be compared by identity.
+ */
+export declare class UnitItemSlot {
+    readonly containerType: UnitItemContainerType;
+    readonly index: number;
+    private static readonly inventorySlots;
+    private static readonly extendedInventorySlots;
+    private static readonly equipmentInventorySlots;
+    private constructor();
+    static inventory(index: 0 | 1 | 2 | 3 | 4 | 5): UnitItemSlot;
+    static extendedInventory(index: number): UnitItemSlot;
+    static equipmentInventory(slot: EquipmentSlot): UnitItemSlot;
+    static get(containerType: UnitItemContainerType, index: number): UnitItemSlot;
+    /** Returns the slot the item currently occupies in the unit's inventory, extended inventory or equipment inventory, if any. */
+    static find(unit: Readonly<Unit>, item: Item): UnitItemSlot | undefined;
+    getItem(unit: Readonly<Unit>): Item | undefined;
+    private static readonly slotsByContainerType;
+}

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

@@ -0,0 +1,92 @@
+local ____lualib = require("lualib_bundle")
+local __TS__Class = ____lualib.__TS__Class
+local __TS__New = ____lualib.__TS__New
+local ____exports = {}
+local itemInSlotByContainerType
+local ____item = require("engine.internal.item")
+local Item = ____item.Item
+local ____item_2Dcontainer = require("engine.internal.unit.item-container")
+local UNIT_EQUIPMENT_INVENTORY_SLOT_COUNT = ____item_2Dcontainer.UNIT_EQUIPMENT_INVENTORY_SLOT_COUNT
+local UNIT_EXTENDED_INVENTORY_SLOT_COUNT = ____item_2Dcontainer.UNIT_EXTENDED_INVENTORY_SLOT_COUNT
+local UNIT_INVENTORY_SLOT_COUNT = ____item_2Dcontainer.UNIT_INVENTORY_SLOT_COUNT
+____exports.UNIT_EQUIPMENT_INVENTORY_SLOT_COUNT = UNIT_EQUIPMENT_INVENTORY_SLOT_COUNT
+____exports.UNIT_EXTENDED_INVENTORY_SLOT_COUNT = UNIT_EXTENDED_INVENTORY_SLOT_COUNT
+____exports.UNIT_INVENTORY_SLOT_COUNT = UNIT_INVENTORY_SLOT_COUNT
+local convertLoadoutSlot = ConvertLoadoutSlot
+local unitExtendedInventorySize = UnitExtendedInventorySize
+local unitInventorySize = UnitInventorySize
+local unitItemInBagSlot = UnitItemInBagSlot
+local unitItemInEquipmentSlot = UnitItemInEquipmentSlot
+local unitItemInSlot = UnitItemInSlot
+--- A slot of one of a unit's item containers (the inventory, the extended inventory or the equipment inventory).
+-- 
+-- There is exactly one instance per (container type, index) pair, so slots can be compared by identity.
+____exports.UnitItemSlot = __TS__Class()
+local UnitItemSlot = ____exports.UnitItemSlot
+UnitItemSlot.name = "UnitItemSlot"
+function UnitItemSlot.prototype.____constructor(self, containerType, index)
+    self.containerType = containerType
+    self.index = index
+end
+function UnitItemSlot.inventory(self, index)
+    return ____exports.UnitItemSlot.inventorySlots[index + 1]
+end
+function UnitItemSlot.extendedInventory(self, index)
+    return ____exports.UnitItemSlot.extendedInventorySlots[index + 1]
+end
+function UnitItemSlot.equipmentInventory(self, slot)
+    return ____exports.UnitItemSlot.equipmentInventorySlots[slot + 1]
+end
+function UnitItemSlot.get(self, containerType, index)
+    return ____exports.UnitItemSlot.slotsByContainerType[containerType][index + 1]
+end
+function UnitItemSlot.find(self, unit, item)
+    local unitHandle = unit.handle
+    local itemHandle = item.handle
+    for index = 0, unitInventorySize(unitHandle) - 1 do
+        if unitItemInSlot(unitHandle, index) == itemHandle then
+            return ____exports.UnitItemSlot.inventorySlots[index + 1]
+        end
+    end
+    for index = 0, unitExtendedInventorySize(unitHandle) - 1 do
+        if unitItemInBagSlot(unitHandle, index) == itemHandle then
+            return ____exports.UnitItemSlot.extendedInventorySlots[index + 1]
+        end
+    end
+    for index = 0, UNIT_EQUIPMENT_INVENTORY_SLOT_COUNT - 1 do
+        if unitItemInEquipmentSlot(
+            unitHandle,
+            convertLoadoutSlot(index)
+        ) == itemHandle then
+            return ____exports.UnitItemSlot.equipmentInventorySlots[index + 1]
+        end
+    end
+    return nil
+end
+function UnitItemSlot.prototype.getItem(self, unit)
+    return Item:of(itemInSlotByContainerType[self.containerType](unit.handle, self.index))
+end
+UnitItemSlot.inventorySlots = {}
+UnitItemSlot.extendedInventorySlots = {}
+UnitItemSlot.equipmentInventorySlots = {};
+(function(self)
+    for index = 0, UNIT_INVENTORY_SLOT_COUNT - 1 do
+        ____exports.UnitItemSlot.inventorySlots[index + 1] = __TS__New(____exports.UnitItemSlot, 0, index)
+    end
+    for index = 0, UNIT_EXTENDED_INVENTORY_SLOT_COUNT - 1 do
+        ____exports.UnitItemSlot.extendedInventorySlots[index + 1] = __TS__New(____exports.UnitItemSlot, 1, index)
+    end
+    for index = 0, UNIT_EQUIPMENT_INVENTORY_SLOT_COUNT - 1 do
+        ____exports.UnitItemSlot.equipmentInventorySlots[index + 1] = __TS__New(____exports.UnitItemSlot, 2, index)
+    end
+end)(UnitItemSlot)
+UnitItemSlot.slotsByContainerType = {[0] = ____exports.UnitItemSlot.inventorySlots, [1] = ____exports.UnitItemSlot.extendedInventorySlots, [2] = ____exports.UnitItemSlot.equipmentInventorySlots}
+itemInSlotByContainerType = {
+    [0] = unitItemInSlot,
+    [1] = unitItemInBagSlot,
+    [2] = function(unitHandle, index) return unitItemInEquipmentSlot(
+        unitHandle,
+        convertLoadoutSlot(index)
+    ) end
+}
+return ____exports

package/engine/internal/unit/main-selected.d.ts

@@ -0,0 +1,13 @@
+/** @noSelfInFile */
+import { Player } from "../../../core/types/player";
+import { Event } from "../../../event";
+declare module "../unit" {
+    namespace Unit {
+        const mainSelectedUnitChangeEvent: Event<[Player]>;
+    }
+}
+declare module "../unit" {
+    namespace Unit {
+        const getMainSelectedOf: (player: Player) => Unit | undefined;
+    }
+}

package/engine/internal/unit/main-selected.lua

@@ -0,0 +1,36 @@
+local ____lualib = require("lualib_bundle")
+local __TS__New = ____lualib.__TS__New
+local ____exports = {}
+local ____local_2Dclient = require("engine.local-client")
+local LocalClient = ____local_2Dclient.LocalClient
+local ____unit = require("engine.internal.unit")
+local Unit = ____unit.Unit
+local ____event = require("event")
+local Event = ____event.Event
+local ____synchronization = require("engine.synchronization")
+local ObjectBus = ____synchronization.ObjectBus
+local mainSelectedUnitChangeEvent = __TS__New(Event)
+rawset(Unit, "mainSelectedUnitChangeEvent", mainSelectedUnitChangeEvent)
+local mainSelectedUnitByPlayer = {}
+local unitBus = __TS__New(
+    ObjectBus,
+    function(unit) return unit.syncId end,
+    function(syncId) return Unit:getBySyncId(syncId) end
+)
+LocalClient.mainSelectedUnitChangeEvent:addListener(function()
+    unitBus:send(LocalClient.mainSelectedUnit)
+end)
+unitBus.event:addListener(function(player, unit)
+    if unit ~= mainSelectedUnitByPlayer[player] then
+        mainSelectedUnitByPlayer[player] = unit
+        Event.invoke(mainSelectedUnitChangeEvent, player)
+    end
+end)
+rawset(
+    Unit,
+    "getMainSelectedOf",
+    function(player)
+        return mainSelectedUnitByPlayer[player]
+    end
+)
+return ____exports

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

@@ -0,0 +1,20 @@
+/** @noSelfInFile */
+import { Widget } from "../../../core/types/widget";
+export declare const enum OrderKind {
+    IMMEDIATE = 0,
+    POINT = 1,
+    TARGET = 2
+}
+declare module "../unit" {
+    interface Unit {
+        readonly currentOrderKind: OrderKind;
+        readonly currentOrderId: number;
+        readonly currentOrderStartTime: number;
+        readonly currentOrderStartX: number;
+        readonly currentOrderStartY: number;
+        readonly currentOrderTargetX: number;
+        readonly currentOrderTargetY: number;
+        readonly currentOrderTarget?: Widget;
+        issueOrder(...order: [orderType: OrderKind.IMMEDIATE, orderId: number] | [orderType: OrderKind.POINT, orderId: number, x: number, y: number] | [orderType: OrderKind.TARGET, orderId: number, target: Widget]): boolean;
+    }
+}

package/engine/internal/unit/order.lua

@@ -0,0 +1,136 @@
+local ____lualib = require("lualib_bundle")
+local __TS__ObjectDefineProperty = ____lualib.__TS__ObjectDefineProperty
+local ____exports = {}
+local ____attributes = require("attributes")
+local attribute = ____attributes.attribute
+local ____unit = require("engine.internal.unit")
+local Unit = ____unit.Unit
+local ____game = require("core.game")
+local elapsedTime = ____game.elapsedTime
+local getUnitCurrentOrder = GetUnitCurrentOrder
+local issueImmediateOrderById = IssueImmediateOrderById
+local issuePointOrderById = IssuePointOrderById
+local issueTargetOrderById = IssueTargetOrderById
+local unitLastOrderKindAttribute = attribute()
+local unitLastOrderIdAttribute = attribute()
+local unitLastOrderStartTimeAttribute = attribute()
+local unitLastOrderStartXAttribute = attribute()
+local unitLastOrderStartYAttribute = attribute()
+local unitLastOrderTargetXAttribute = attribute()
+local unitLastOrderTargetYAttribute = attribute()
+local unitLastOrderTargetAttribute = attribute()
+Unit.onImmediateOrder:addListener(
+    4,
+    function(unit, orderId)
+        unit[unitLastOrderKindAttribute] = 0
+        unit[unitLastOrderIdAttribute] = orderId
+        unit[unitLastOrderStartTimeAttribute] = elapsedTime()
+        unit[unitLastOrderStartXAttribute] = unit.x
+        unit[unitLastOrderStartYAttribute] = unit.y
+        unit[unitLastOrderTargetXAttribute] = nil
+        unit[unitLastOrderTargetYAttribute] = nil
+        unit[unitLastOrderTargetAttribute] = nil
+    end
+)
+Unit.onPointOrder:addListener(
+    4,
+    function(unit, orderId, x, y)
+        unit[unitLastOrderKindAttribute] = 1
+        unit[unitLastOrderIdAttribute] = orderId
+        unit[unitLastOrderStartTimeAttribute] = elapsedTime()
+        unit[unitLastOrderStartXAttribute] = unit.x
+        unit[unitLastOrderStartYAttribute] = unit.y
+        unit[unitLastOrderTargetXAttribute] = x
+        unit[unitLastOrderTargetYAttribute] = y
+        unit[unitLastOrderTargetAttribute] = nil
+    end
+)
+Unit.onTargetOrder:addListener(
+    4,
+    function(unit, orderId, target)
+        unit[unitLastOrderKindAttribute] = 2
+        unit[unitLastOrderIdAttribute] = orderId
+        unit[unitLastOrderStartTimeAttribute] = elapsedTime()
+        unit[unitLastOrderStartXAttribute] = unit.x
+        unit[unitLastOrderStartYAttribute] = unit.y
+        unit[unitLastOrderTargetXAttribute] = target.x
+        unit[unitLastOrderTargetYAttribute] = target.y
+        unit[unitLastOrderTargetAttribute] = target
+    end
+)
+local function toUndefinedIfCurrentOrderDoesNotMatchLast(unit, value)
+    local currentOrderId = getUnitCurrentOrder(unit.handle)
+    local lastOrderId = unit[unitLastOrderIdAttribute]
+    local ____temp_0
+    if currentOrderId == lastOrderId or currentOrderId == 851991 and lastOrderId == 851990 then
+        ____temp_0 = value
+    else
+        ____temp_0 = nil
+    end
+    return ____temp_0
+end
+__TS__ObjectDefineProperty(
+    Unit.prototype,
+    "currentOrderKind",
+    {get = function(self)
+        return toUndefinedIfCurrentOrderDoesNotMatchLast(self, self[unitLastOrderKindAttribute]) or 0
+    end}
+)
+__TS__ObjectDefineProperty(
+    Unit.prototype,
+    "currentOrderId",
+    {get = function(self)
+        return toUndefinedIfCurrentOrderDoesNotMatchLast(self, self[unitLastOrderIdAttribute]) or 0
+    end}
+)
+__TS__ObjectDefineProperty(
+    Unit.prototype,
+    "currentOrderStartTime",
+    {get = function(self)
+        return toUndefinedIfCurrentOrderDoesNotMatchLast(self, self[unitLastOrderStartTimeAttribute]) or 0
+    end}
+)
+__TS__ObjectDefineProperty(
+    Unit.prototype,
+    "currentOrderStartX",
+    {get = function(self)
+        return toUndefinedIfCurrentOrderDoesNotMatchLast(self, self[unitLastOrderStartXAttribute]) or 0
+    end}
+)
+__TS__ObjectDefineProperty(
+    Unit.prototype,
+    "currentOrderStartY",
+    {get = function(self)
+        return toUndefinedIfCurrentOrderDoesNotMatchLast(self, self[unitLastOrderStartYAttribute]) or 0
+    end}
+)
+__TS__ObjectDefineProperty(
+    Unit.prototype,
+    "currentOrderTargetX",
+    {get = function(self)
+        return toUndefinedIfCurrentOrderDoesNotMatchLast(self, self[unitLastOrderTargetXAttribute]) or 0
+    end}
+)
+__TS__ObjectDefineProperty(
+    Unit.prototype,
+    "currentOrderTargetY",
+    {get = function(self)
+        return toUndefinedIfCurrentOrderDoesNotMatchLast(self, self[unitLastOrderTargetYAttribute]) or 0
+    end}
+)
+__TS__ObjectDefineProperty(
+    Unit.prototype,
+    "currentOrderTarget",
+    {get = function(self)
+        return toUndefinedIfCurrentOrderDoesNotMatchLast(self, self[unitLastOrderTargetAttribute])
+    end}
+)
+local issueOrderByKind = {
+    [0] = issueImmediateOrderById,
+    [1] = issuePointOrderById,
+    [2] = function(unitHandle, orderId, widget) return issueTargetOrderById(unitHandle, orderId, widget.handle) end
+}
+Unit.prototype.issueOrder = function(self, orderKind, orderId, xOrTarget, y)
+    return issueOrderByKind[orderKind](self.handle, orderId, xOrTarget, y)
+end
+return ____exports

package/engine/internal/unit/range-event.d.ts

@@ -0,0 +1,12 @@
+/** @noSelfInFile */
+import { Event } from "../../../event";
+declare module "../unit" {
+    interface Unit {
+        readonly unitInRangeEvent: Record<number, Event<[unit: Unit, range: number, unitInRange: Unit]>>;
+    }
+}
+declare module "../unit" {
+    interface Unit {
+        readonly unitOutOfRangeEvent: Record<number, Event<[unit: Unit, range: number, unitOutOfRange: Unit]>>;
+    }
+}

package/engine/internal/unit/range-event.lua

@@ -0,0 +1,90 @@
+local ____lualib = require("lualib_bundle")
+local __TS__New = ____lualib.__TS__New
+local __TS__ObjectDefineProperty = ____lualib.__TS__ObjectDefineProperty
+local ____exports = {}
+local ____event = require("event")
+local Event = ____event.Event
+local InitializingEvent = ____event.InitializingEvent
+local TriggerEvent = ____event.TriggerEvent
+local ____unit = require("engine.internal.unit")
+local Unit = ____unit.Unit
+local ____timer = require("core.types.timer")
+local Timer = ____timer.Timer
+local ____attributes = require("attributes")
+local attribute = ____attributes.attribute
+local ____linked_2Dmap = require("utility.linked-map")
+local mutableLinkedMap = ____linked_2Dmap.mutableLinkedMap
+local ____linked_2Dset = require("utility.linked-set")
+local LinkedSet = ____linked_2Dset.LinkedSet
+local mutableLinkedSet = ____linked_2Dset.mutableLinkedSet
+local getTriggerUnit = GetTriggerUnit
+local triggerRegisterUnitInRange = TriggerRegisterUnitInRange
+local eventInvoke = Event.invoke
+__TS__ObjectDefineProperty(
+    Unit.prototype,
+    "unitInRangeEvent",
+    {get = function(self)
+        local unit = self
+        local handle = self.handle
+        local unitInRangeEvent = setmetatable(
+            {},
+            {__index = function(self, value)
+                local event = __TS__New(
+                    TriggerEvent,
+                    function(trigger)
+                        triggerRegisterUnitInRange(trigger, handle, value, nil)
+                    end,
+                    function() return unit, value, Unit:of(getTriggerUnit()) end
+                )
+                rawset(self, value, event)
+                return event
+            end}
+        )
+        rawset(self, "unitInRangeEvent", unitInRangeEvent)
+        return unitInRangeEvent
+    end}
+)
+local units = __TS__New(LinkedSet)
+local unitsInRangeByRangeAttribute = attribute()
+local function registerUnitOfRangeEvent(unit, range, unitInRange)
+    units:add(unit)
+    unit:getOrPut(unitsInRangeByRangeAttribute, mutableLinkedMap):getOrPut(range, mutableLinkedSet):add(unitInRange)
+end
+__TS__ObjectDefineProperty(
+    Unit.prototype,
+    "unitOutOfRangeEvent",
+    {get = function(self)
+        local unit = self
+        local unitOutOfRangeEvent = setmetatable(
+            {},
+            {__index = function(self, value)
+                local event = __TS__New(
+                    InitializingEvent,
+                    function()
+                        unit.unitInRangeEvent[value]:addListener(999999, registerUnitOfRangeEvent)
+                    end
+                )
+                rawset(self, value, event)
+                return event
+            end}
+        )
+        rawset(self, "unitOutOfRangeEvent", unitOutOfRangeEvent)
+        return unitOutOfRangeEvent
+    end}
+)
+Timer.onPeriod[1]:addListener(function()
+    for unit in pairs(units) do
+        local unitsInRangeByRange = unit[unitsInRangeByRangeAttribute]
+        if unitsInRangeByRange ~= nil then
+            for range, unitsInRange in pairs(unitsInRangeByRange) do
+                for unitInRange in pairs(unitsInRange) do
+                    if unit:getDistanceTo(unitInRange) > range then
+                        unitsInRange:remove(unitInRange)
+                        eventInvoke(unit.unitOutOfRangeEvent[range], unit, range, unitInRange)
+                    end
+                end
+            end
+        end
+    end
+end)
+return ____exports

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

@@ -0,0 +1,7 @@
+/** @noSelfInFile */
+declare module "../unit" {
+    interface Unit {
+        scale: number;
+    }
+}
+export {};

package/engine/internal/unit/scale.lua

@@ -0,0 +1,20 @@
+local ____lualib = require("lualib_bundle")
+local __TS__ObjectDefineProperty = ____lualib.__TS__ObjectDefineProperty
+local ____exports = {}
+local ____unit = require("engine.internal.unit")
+local Unit = ____unit.Unit
+local ____unit = require("engine.standard.fields.unit")
+local SCALING_VALUE_UNIT_FIELD = ____unit.SCALING_VALUE_UNIT_FIELD
+__TS__ObjectDefineProperty(
+    Unit.prototype,
+    "scale",
+    {
+        get = function(self)
+            return SCALING_VALUE_UNIT_FIELD:getValue(self)
+        end,
+        set = function(self, value)
+            SCALING_VALUE_UNIT_FIELD:setValue(self, value)
+        end
+    }
+)
+return ____exports

package/engine/internal/unit+ability.d.ts

@@ -1,9 +1,10 @@
 /** @noSelfInFile */
-import { Ability } from "./ability";
+import { Ability, UnitAbility } from "./ability";
 import { DispatchingEvent } from "../../event";
 declare module "./unit" {
     namespace Unit {
         const abilityGainedEvent: DispatchingEvent<[Unit, Ability]>;
         const abilityLostEvent: DispatchingEvent<[Unit, Ability]>;
+        const abilityLevelChangedEvent: DispatchingEvent<[Unit, UnitAbility]>;
     }
 }

package/engine/internal/unit+ability.lua

@@ -7,8 +7,12 @@ local UnitAbility = ____ability.UnitAbility
 local ____unit = require("engine.internal.unit")
 local Unit = ____unit.Unit
 local ____event = require("event")
-local Event = ____event.Event
 local createDispatchingEvent = ____event.createDispatchingEvent
+local Event = ____event.Event
+local ____unit_2Dability_2Dlevel_2Dnatives = require("engine.internal.misc.unit-ability-level-natives")
+local rawDecUnitAbilityLevel = ____unit_2Dability_2Dlevel_2Dnatives.rawDecUnitAbilityLevel
+local rawIncUnitAbilityLevel = ____unit_2Dability_2Dlevel_2Dnatives.rawIncUnitAbilityLevel
+local rawSetUnitAbilityLevel = ____unit_2Dability_2Dlevel_2Dnatives.rawSetUnitAbilityLevel
 local abilityGainedEvent = createDispatchingEvent(
     __TS__New(Event),
     function(unit, ability) return ability.typeId end
@@ -19,6 +23,63 @@ local abilityLostEvent = createDispatchingEvent(
     function(unit, ability) return ability.typeId end
 )
 rawset(Unit, "abilityLostEvent", abilityLostEvent)
+local abilityLevelChangedEvent = createDispatchingEvent(
+    __TS__New(Event),
+    function(unit, ability) return ability.typeId end
+)
+rawset(Unit, "abilityLevelChangedEvent", abilityLevelChangedEvent)
+local getUnitAbilityLevel = GetUnitAbilityLevel
+local function invokeAbilityLevelChangedEvent(unitHandle, abilityTypeId)
+    local unit = Unit:of(unitHandle)
+    if unit ~= nil then
+        local ability = unit:getAbility(abilityTypeId)
+        if ability ~= nil then
+            Event.invoke(abilityLevelChangedEvent, unit, ability)
+        end
+    end
+end
+local function invokeAbilityLevelChangedEventIfChanged(unitHandle, abilityTypeId, previousLevel)
+    if previousLevel ~= 0 then
+        local level = getUnitAbilityLevel(unitHandle, abilityTypeId)
+        if level ~= 0 and level ~= previousLevel then
+            invokeAbilityLevelChangedEvent(unitHandle, abilityTypeId)
+        end
+    end
+end
+_G.SetUnitAbilityLevel = function(whichUnit, abilcode, level)
+    local previousLevel = getUnitAbilityLevel(whichUnit, abilcode)
+    local result = rawSetUnitAbilityLevel(whichUnit, abilcode, level)
+    invokeAbilityLevelChangedEventIfChanged(whichUnit, abilcode, previousLevel)
+    return result
+end
+_G.IncUnitAbilityLevel = function(whichUnit, abilcode)
+    local previousLevel = getUnitAbilityLevel(whichUnit, abilcode)
+    local result = rawIncUnitAbilityLevel(whichUnit, abilcode)
+    invokeAbilityLevelChangedEventIfChanged(whichUnit, abilcode, previousLevel)
+    return result
+end
+_G.DecUnitAbilityLevel = function(whichUnit, abilcode)
+    local previousLevel = getUnitAbilityLevel(whichUnit, abilcode)
+    local result = rawDecUnitAbilityLevel(whichUnit, abilcode)
+    invokeAbilityLevelChangedEventIfChanged(whichUnit, abilcode, previousLevel)
+    return result
+end
+local heroSkillTrigger = CreateTrigger()
+TriggerRegisterAnyUnitEventBJ(heroSkillTrigger, EVENT_PLAYER_HERO_SKILL)
+TriggerAddCondition(
+    heroSkillTrigger,
+    Condition(function()
+        if GetLearnedSkillLevel() > 1 then
+            local unitHandle = GetTriggerUnit()
+            if unitHandle ~= nil then
+                invokeAbilityLevelChangedEvent(
+                    unitHandle,
+                    GetLearnedSkill()
+                )
+            end
+        end
+    end)
+)
 UnitAbility.onCreate:addListener(
     0,
     function(ability)
@@ -40,16 +101,25 @@ ItemAbility.onCreate:addListener(
         end
     end
 )
-Unit.onItemPickup:addListener(
-    0,
+ItemAbility.destroyEvent:addListener(
+    4,
+    function(ability)
+        local unit = ability.owner.owner
+        if unit ~= nil then
+            Event.invoke(abilityLostEvent, unit, ability)
+        end
+    end
+)
+Unit.itemPickedUpEvent:addListener(
+    -999999,
     function(unit, item)
         for ____, ability in ipairs(item.abilities) do
             Event.invoke(abilityGainedEvent, unit, ability)
         end
     end
 )
-Unit.onItemDrop:addListener(
-    4,
+Unit.itemDroppedEvent:addListener(
+    999999,
     function(unit, item)
         for ____, ability in ipairs(item.abilities) do
             Event.invoke(abilityLostEvent, unit, ability)

package/engine/internal/unit+bonus.lua

@@ -3,9 +3,9 @@ local __TS__ObjectDefineProperty = ____lualib.__TS__ObjectDefineProperty
 local ____exports = {}
 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 getUnitAbility = BlzGetUnitAbility
 local setAbilityRealLevelField = BlzSetAbilityRealLevelField
 local unitAddAbility = UnitAddAbility