warscript 0.0.1-dev.d6e2c10 → 0.0.1-dev.d77b529

package/engine/internal/unit.lua

@@ -47,12 +47,10 @@ local checkNotNull = ____preconditions.checkNotNull
 local ____constants = require("engine.constants")
 local HEALTH_REGENERATION_RATE_BONUS_PER_STRENGTH_POINT = ____constants.HEALTH_REGENERATION_RATE_BONUS_PER_STRENGTH_POINT
 local MANA_REGENERATION_RATE_BONUS_PER_INTELLIGENCE_POINT = ____constants.MANA_REGENERATION_RATE_BONUS_PER_INTELLIGENCE_POINT
-local ____arrays = require("utility.arrays")
-local forEach = ____arrays.forEach
-local ____math = require("math")
-local min = ____math.min
 local ____ignore_2Devents_2Ditems = require("engine.internal.unit.ignore-events-items")
-local ignoreEventsItems = ____ignore_2Devents_2Ditems.ignoreEventsItems
+local isItemIgnoredInEvents = ____ignore_2Devents_2Ditems.isItemIgnoredInEvents
+local ____owner_2Dcache = require("engine.internal.item.owner-cache")
+local ownerByItem = ____owner_2Dcache.ownerByItem
 local ____attack_2Dtype = require("engine.object-data.auxiliary.attack-type")
 local attackTypeToNative = ____attack_2Dtype.attackTypeToNative
 local nativeToAttackType = ____attack_2Dtype.nativeToAttackType
@@ -64,6 +62,8 @@ local ____ability = require("engine.internal.item.ability")
 local doUnitAbilityAction = ____ability.doUnitAbilityAction
 local ____synchronization = require("engine.synchronization")
 local synchronizer = ____synchronization.synchronizer
+local ____linked_2Dmap = require("utility.linked-map")
+local LinkedMap = ____linked_2Dmap.LinkedMap
 local match = string.match
 local ____tostring = _G.tostring
 local setUnitAnimation = SetUnitAnimation
@@ -73,16 +73,12 @@ local getUnitRealField = BlzGetUnitRealField
 local getHeroStr = GetHeroStr
 local getHeroAgi = GetHeroAgi
 local getHeroInt = GetHeroInt
-local setHeroStr = SetHeroStr
-local setHeroAgi = SetHeroAgi
-local setHeroInt = SetHeroInt
 local getUnitBooleanField = BlzGetUnitBooleanField
 local getUnitStringField = BlzGetUnitStringField
 local setUnitIntegerField = BlzSetUnitIntegerField
 local setUnitRealField = BlzSetUnitRealField
 local setUnitBooleanField = BlzSetUnitBooleanField
 local setUnitStringField = BlzSetUnitStringField
-local setUnitScale = SetUnitScale
 local setUnitPosition = SetUnitPosition
 local setUnitTimeScale = SetUnitTimeScale
 local getHandleId = GetHandleId
@@ -110,7 +106,6 @@ local getUnitWeaponStringField = BlzGetUnitWeaponStringField
 local setUnitWeaponStringField = BlzSetUnitWeaponStringField
 local getUnitAbilityLevel = GetUnitAbilityLevel
 local unitDisableAbility = BlzUnitDisableAbility
-local unitInterruptAttack = BlzUnitInterruptAttack
 local isUnitInvisible = IsUnitInvisible
 local isUnitVisible = IsUnitVisible
 local getUnitX = GetUnitX
@@ -386,15 +381,27 @@ local modifiers = {
     end,
     armor = function(unit, value)
         if UnitAddAbility(unit, armorBonusAbilityId) then
-            assert(UnitMakeAbilityPermanent(unit, true, armorBonusAbilityId))
+            assert(
+                UnitMakeAbilityPermanent(unit, true, armorBonusAbilityId),
+                "armor bonus ability must be made permanent",
+                unit
+            )
         end
-        local ability = assert(BlzGetUnitAbility(unit, armorBonusAbilityId))
-        assert(BlzSetAbilityRealLevelField(
-            ability,
-            armorBonusField,
-            0,
-            BlzGetAbilityRealLevelField(ability, armorBonusField, 0) + value
-        ))
+        local ability = assert(
+            BlzGetUnitAbility(unit, armorBonusAbilityId),
+            "armor bonus ability must be existing",
+            unit
+        )
+        assert(
+            BlzSetAbilityRealLevelField(
+                ability,
+                armorBonusField,
+                0,
+                BlzGetAbilityRealLevelField(ability, armorBonusField, 0) + value
+            ),
+            "armor bonus ability field must be set",
+            unit
+        )
     end
 }
 local getters = {
@@ -583,8 +590,6 @@ __TS__SetDescriptor(
     },
     true
 )
-local unitInventorySize = UnitInventorySize
-local unitItemInSlot = UnitItemInSlot
 local getItemAbility = BlzGetItemAbility
 local getUnitAbility = BlzGetUnitAbility
 local getUnitAbilityByIndex = BlzGetUnitAbilityByIndex
@@ -602,8 +607,8 @@ local function retrieveAbility(unit, ability, abilityId)
             ____exports.Unit:of(unit)
         )
     end
-    for i = 0, unitInventorySize(unit) - 1 do
-        local item = unitItemInSlot(unit, i)
+    for i = 0, UnitInventorySize(unit) - 1 do
+        local item = UnitItemInSlot(unit, i)
         if getItemAbility(item, abilityId) == ability then
             return ItemAbility:of(
                 ability,
@@ -648,13 +653,17 @@ local fieldSetters = {
 }
 local dummies = {}
 for ____, player in ipairs(Player.all) do
-    local dummy = assert(createUnit(
-        player.handle,
-        dummyUnitId,
-        0,
-        0,
-        270
-    ))
+    local dummy = assert(
+        createUnit(
+            player.handle,
+            dummyUnitId,
+            0,
+            0,
+            270
+        ),
+        "dummy must be created",
+        player
+    )
     ShowUnit(dummy, false)
     dummies[player] = dummy
 end
@@ -697,41 +706,57 @@ function Unit.prototype.____constructor(self, handle)
     nextSyncId = ____nextSyncId_1 + 1
     self.syncId = ____nextSyncId_1
     self._owner = Player:of(getOwningPlayer(handle))
-    assert(unitAddAbility(handle, leaveDetectAbilityId) and UnitMakeAbilityPermanent(handle, true, leaveDetectAbilityId))
-    assert(unitAddAbility(handle, morphDetectAbilityId))
+    assert(
+        unitAddAbility(handle, leaveDetectAbilityId) and UnitMakeAbilityPermanent(handle, true, leaveDetectAbilityId),
+        "leave detection ability must be added",
+        self
+    )
+    assert(
+        unitAddAbility(handle, morphDetectAbilityId),
+        "morph detection ability must be added",
+        self
+    )
     if unitAddAbility(
         handle,
         fourCC("Amrf")
     ) then
-        assert(unitRemoveAbility(
-            handle,
-            fourCC("Amrf")
-        ))
+        assert(
+            unitRemoveAbility(
+                handle,
+                fourCC("Amrf")
+            ),
+            "fly ability must be removed after addition",
+            self
+        )
     end
     unitBySyncId[self.syncId] = self
     local ____ = self.abilities
 end
-function Unit.prototype.getEvent(self, event, collector)
-    self.events = self.events or ({})
-    local eventId = GetHandleId(event)
-    if not self.events[eventId] then
-        self.events[eventId] = __TS__New(
+function Unit.prototype.getEvent(self, jevent, collector)
+    self.events = self.events or __TS__New(LinkedMap)
+    local eventId = GetHandleId(jevent)
+    local event = self.events:get(eventId)
+    if event == nil then
+        event = __TS__New(
             TriggerEvent,
             function(trigger)
-                TriggerRegisterUnitEvent(trigger, self.handle, event)
+                TriggerRegisterUnitEvent(trigger, self.handle, jevent)
             end,
             collector or (function() return {} end)
         )
+        self.events:put(eventId, event)
     end
-    return self.events[eventId]
+    return event
+end
+function Unit.prototype.saveData(self)
+    local handle = self.handle
+    self[108] = self[108] or getUnitX(handle)
+    self[109] = self[109] or getUnitY(handle)
+    self._owner = self._owner or Player:of(getOwningPlayer(handle))
 end
 function Unit.prototype.onDestroy(self)
     local handle = self.handle
-    self[108] = getUnitX(handle)
-    self[109] = getUnitY(handle)
-    if not self._owner then
-        self._owner = Player:of(getOwningPlayer(handle))
-    end
+    self:saveData()
     local abilities = self.abilities
     for ____, ability in ipairs(abilities) do
         ability:destroy()
@@ -749,7 +774,9 @@ function Unit.prototype.onDestroy(self)
     end
     local eventsToDestroy = self._eventsToDestroy
     if eventsToDestroy ~= nil then
-        forEach(eventsToDestroy, "destroy")
+        for ____, event in ipairs(eventsToDestroy) do
+            event:destroy()
+        end
     end
     if getUnitAbilityLevel(handle, leaveDetectAbilityId) > 0 then
         unitRemoveAbility(handle, leaveDetectAbilityId)
@@ -767,7 +794,11 @@ function Unit.prototype.addAttackHandler(self, condition, action)
     self._attackHandlers = handlers
     if #handlers == 1 then
         local handle = self.handle
-        assert(unitAddAbility(handle, attackHandlerAbilityId) and UnitMakeAbilityPermanent(handle, true, attackHandlerAbilityId))
+        assert(
+            unitAddAbility(handle, attackHandlerAbilityId) and UnitMakeAbilityPermanent(handle, true, attackHandlerAbilityId),
+            "attack handler ability must be added",
+            self
+        )
     end
     return handler
 end
@@ -926,7 +957,7 @@ function Unit.prototype.dropItemSlot(self, item, slot)
     return UnitDropItemSlot(self.handle, item.handle, slot)
 end
 function Unit.prototype.itemInSlot(self, slot)
-    return Item:of(unitItemInSlot(self.handle, slot))
+    return Item:of(UnitItemInSlot(self.handle, slot))
 end
 function Unit.prototype.addAbility(self, abilityId)
     local ability = UnitAbility:of(
@@ -980,24 +1011,6 @@ end
 function Unit.prototype.endAbilityCooldown(self, abilityId)
     BlzEndUnitAbilityCooldown(self.handle, abilityId)
 end
-function Unit.prototype.interruptMovement(self)
-    local handle = self.handle
-    unitDisableAbility(
-        handle,
-        fourCC("Amov"),
-        true,
-        false
-    )
-    unitDisableAbility(
-        handle,
-        fourCC("Amov"),
-        false,
-        false
-    )
-end
-function Unit.prototype.interruptAttack(self)
-    unitInterruptAttack(self.handle)
-end
 function Unit.prototype.interruptCast(self, abilityId)
     local handle = self.handle
     unitDisableAbility(handle, abilityId, true, false)
@@ -1302,7 +1315,7 @@ __TS__SetDescriptor(
     Unit.prototype,
     "isStunned",
     {get = function(self)
-        return getUnitCurrentOrder(self.handle) == orderId("stunned")
+        return getUnitCurrentOrder(self.handle) == 851973
     end},
     true
 )
@@ -1392,19 +1405,6 @@ __TS__SetDescriptor(
     },
     true
 )
-__TS__SetDescriptor(
-    Unit.prototype,
-    "strengthBase",
-    {
-        get = function(self)
-            return getHeroStr(self.handle, false)
-        end,
-        set = function(self, strengthBase)
-            setHeroStr(self.handle, strengthBase, true)
-        end
-    },
-    true
-)
 __TS__SetDescriptor(
     Unit.prototype,
     "strengthBonus",
@@ -1422,19 +1422,6 @@ __TS__SetDescriptor(
     end},
     true
 )
-__TS__SetDescriptor(
-    Unit.prototype,
-    "agilityBase",
-    {
-        get = function(self)
-            return getHeroAgi(self.handle, false)
-        end,
-        set = function(self, agilityBase)
-            setHeroAgi(self.handle, agilityBase, true)
-        end
-    },
-    true
-)
 __TS__SetDescriptor(
     Unit.prototype,
     "agilityBonus",
@@ -1452,19 +1439,6 @@ __TS__SetDescriptor(
     end},
     true
 )
-__TS__SetDescriptor(
-    Unit.prototype,
-    "intelligenceBase",
-    {
-        get = function(self)
-            return getHeroInt(self.handle, false)
-        end,
-        set = function(self, intelligenceBase)
-            setHeroInt(self.handle, intelligenceBase, true)
-        end
-    },
-    true
-)
 __TS__SetDescriptor(
     Unit.prototype,
     "intelligenceBonus",
@@ -1903,7 +1877,7 @@ __TS__SetDescriptor(
     Unit.prototype,
     "pathingCollisionRange",
     {get = function(self)
-        return 16 * (min(
+        return 16 * (math.min(
             getUnitCollisionSize(self.handle) // 16,
             3
         ) + 1)
@@ -2013,30 +1987,6 @@ __TS__SetDescriptor(
     end},
     true
 )
-__TS__SetDescriptor(
-    Unit.prototype,
-    "onUnitInRange",
-    {get = function(self)
-        local handle = self.handle
-        local onUnitInRange = setmetatable(
-            {},
-            {__index = function(self, value)
-                local event = __TS__New(
-                    TriggerEvent,
-                    function(trigger)
-                        TriggerRegisterUnitInRangeSimple(trigger, value, handle)
-                    end,
-                    function() return ____exports.Unit:of(handle) end
-                )
-                rawset(self, value, event)
-                return event
-            end}
-        )
-        rawset(self, "onUnitInRange", onUnitInRange)
-        return onUnitInRange
-    end},
-    true
-)
 __TS__SetDescriptor(
     Unit.prototype,
     "onManaEqual",
@@ -2172,6 +2122,11 @@ __TS__SetDescriptor(
     end},
     true
 )
+Unit.levelChangedEvent = __TS__New(
+    ____exports.UnitTriggerEvent,
+    EVENT_PLAYER_HERO_LEVEL,
+    function() return ____exports.Unit:of(getTriggerUnit()) end
+)
 Unit.deathEvent = __TS__New(
     ____exports.UnitTriggerEvent,
     EVENT_PLAYER_UNIT_DEATH,
@@ -2189,7 +2144,7 @@ Unit.onResurrect = __TS__New(
         ____exports.Unit.deathEvent:addListener(function(unit)
             dead[unit] = true
         end)
-        ____exports.Unit.onImmediateOrder[orderId("undefend")]:addListener(function(unit)
+        ____exports.Unit.onImmediateOrder[852056]:addListener(function(unit)
             if dead[unit] and not IsUnitType(unit.handle, UNIT_TYPE_DEAD) then
                 dead[unit] = false
                 invoke(event, unit)
@@ -2206,7 +2161,7 @@ Unit.morphEvent = __TS__New(
                 invoke(event, unit)
             end
         end
-        ____exports.Unit.onImmediateOrder[orderId("undefend")]:addListener(function(unit)
+        ____exports.Unit.onImmediateOrder[852056]:addListener(function(unit)
             if getUnitAbilityLevel(unit.handle, morphDetectAbilityId) == 0 then
                 Timer:run(ifNotLeft, unit)
             end
@@ -2673,7 +2628,7 @@ Unit.itemDroppedEvent = __TS__New(
     function()
         local unit = getTriggerUnit()
         local item = getManipulatedItem()
-        if getUnitTypeId(unit) ~= dummyUnitId and not (ignoreEventsItems[item] ~= nil) then
+        if getUnitTypeId(unit) ~= dummyUnitId and not isItemIgnoredInEvents(item) then
             return ____exports.Unit:of(unit), Item:of(item)
         end
         return IgnoreEvent
@@ -2685,10 +2640,10 @@ Unit.itemPickedUpEvent = __TS__New(
     function()
         local unitHandle = getTriggerUnit()
         local itemHandle = getManipulatedItem()
-        if getUnitTypeId(unitHandle) ~= dummyUnitId and not (ignoreEventsItems[itemHandle] ~= nil) then
+        if getUnitTypeId(unitHandle) ~= dummyUnitId and not isItemIgnoredInEvents(itemHandle) then
             local unit = ____exports.Unit:of(unitHandle)
             local item = Item:of(itemHandle)
-            if item.owner ~= unit then
+            if ownerByItem[item] ~= unit then
                 return unit, item
             end
         end
@@ -2701,17 +2656,12 @@ Unit.itemUsedEvent = __TS__New(
     function()
         local unit = getTriggerUnit()
         local item = getManipulatedItem()
-        if getUnitTypeId(unit) ~= dummyUnitId and not (ignoreEventsItems[item] ~= nil) then
+        if getUnitTypeId(unit) ~= dummyUnitId and not isItemIgnoredInEvents(item) then
             return ____exports.Unit:of(unit), Item:of(item)
         end
         return IgnoreEvent
     end
 )
-Unit.itemStackedEvent = __TS__New(
-    ____exports.UnitTriggerEvent,
-    EVENT_PLAYER_UNIT_STACK_ITEM,
-    function() return ____exports.Unit:of(getTriggerUnit()), Item:of(BlzGetStackingItemTarget()), Item:of(BlzGetStackingItemSource()) end
-)
 __TS__ObjectDefineProperty(
     Unit,
     "itemChargesChangedEvent",
@@ -2727,51 +2677,6 @@ __TS__ObjectDefineProperty(
         return event
     end}
 )
-__TS__ObjectDefineProperty(
-    Unit,
-    "itemUseOrderEvent",
-    {get = function(self)
-        local event = __TS__New(Event)
-        for order = orderId("useslot0"), orderId("useslot5") do
-            local slot = order - orderId("useslot0")
-            local function listener(unit)
-                local item = unit.items[slot + 1]
-                if item ~= nil then
-                    invoke(event, unit, item)
-                end
-            end
-            self.onImmediateOrder[order]:addListener(listener)
-            self.onTargetOrder[order]:addListener(listener)
-            self.onPointOrder[order]:addListener(listener)
-        end
-        rawset(self, "itemUseOrderEvent", event)
-        return event
-    end}
-)
-__TS__ObjectDefineProperty(
-    Unit,
-    "itemMoveOrderEvent",
-    {get = function(self)
-        local event = __TS__New(Event)
-        for order = orderId("moveslot0"), orderId("moveslot5") do
-            local slotTo = order - orderId("moveslot0")
-            self.onTargetOrder[order]:addListener(function(unit, item)
-                local slotFrom = unit.items:findSlot(item)
-                if slotFrom ~= nil then
-                    invoke(
-                        event,
-                        unit,
-                        item,
-                        slotFrom,
-                        slotTo
-                    )
-                end
-            end)
-        end
-        rawset(self, "itemMoveOrderEvent", event)
-        return event
-    end}
-)
 __TS__ObjectDefineProperty(
     Unit,
     "onCreate",
@@ -2818,11 +2723,7 @@ Unit.synchronize = synchronizer(
             function(____, id) return fourCC(id) end
         )
     end)
-    for ____, leaveOrderId in ipairs({
-        orderId("undefend"),
-        orderId("magicundefense"),
-        orderId("unimmolation")
-    }) do
+    for ____, leaveOrderId in ipairs({852056, 852479, 852178}) do
         ____exports.Unit.onImmediateOrder[leaveOrderId]:addListener(function(unit)
             local handle = unit.handle
             for i = 1, #leaveAbilityIds do
@@ -2830,7 +2731,8 @@ Unit.synchronize = synchronizer(
                     return
                 end
             end
-            unit:destroy()
+            unit:saveData()
+            Timer:run(unit, "destroy")
         end)
     end
 end)(Unit)

package/engine/local-client.d.ts

@@ -9,15 +9,41 @@ export declare class LocalClient {
     static readonly locale: string;
     static get width(): number;
     static get height(): number;
+    static get isSD(): boolean;
     static get isHD(): boolean;
+    static get isDE(): boolean;
     static get graphicsMode(): GraphicsMode;
     static get isActive(): boolean;
-    static pingMinimap(x: number, y: number, duration: number, ...parameters: [] | [red: number, green: number, blue: number, extraEffects?: boolean] | [color: Color, extraEffects?: boolean]): void;
+    static get isMinimized(): boolean;
+    static pingMinimap(x: number, y: number, duration: number, ...parameters: [] | [red: number, green: number, blue: number, flashy?: boolean] | [color: Color, flashy?: boolean]): void;
+    /** Local input state; do not use to branch synchronized game logic. */
+    static isKeyPressed(key: joskeytype): boolean;
+    /** Local input state. Uses the same modifier mask as Player key events. */
+    static isMetaKeyPressed(key: oskeymeta): boolean;
+    /** Local input state. */
+    static isMouseButtonPressed(button: jmousebuttontype): boolean;
+    static get mouseScreenX(): number;
+    static get mouseScreenY(): number;
+    static screenToFrameX(x: number): number;
+    static screenToFrameY(y: number): number;
+    static frameToScreenX(x: number): number;
+    static frameToScreenY(y: number): number;
+    static get mouseFrameX(): number;
+    static get mouseFrameY(): number;
     static get mouseFocusUnit(): Async<Unit> | undefined;
+    /**
+     * Local selection sampled every 1/64 second; undefined before the first tick. Among
+     * several selected units, the one the portrait shows, identified by its health and
+     * mana; units with identical values cannot be told apart.
+     */
     static get mainSelectedUnit(): Async<Unit> | undefined;
     static get mainSelectedUnitChangeEvent(): Event<[
         previousMainSelectedUnit: Unit | undefined,
         newMainSelectedUnit: Unit | undefined
     ]>;
+    static get isInTargetingMode(): boolean;
+    static readonly targetingModeEnterEvent: Event<[]>;
+    static readonly targetingModeLeaveEvent: Event<[]>;
+    static readonly targetingModeStateChangeEvent: Event<[]>;
     static readonly onDisconnect: TriggerEvent<[]>;
 }