warscript 0.0.1-dev.459629d → 0.0.1-dev.45a73a4

package/engine/local-client.lua

@@ -1,5 +1,4 @@
 local ____lualib = require("lualib_bundle")
-local __TS__ArrayMap = ____lualib.__TS__ArrayMap
 local __TS__New = ____lualib.__TS__New
 local __TS__Class = ____lualib.__TS__Class
 local __TS__InstanceOf = ____lualib.__TS__InstanceOf
@@ -21,34 +20,95 @@ local ____color = require("core.types.color")
 local Color = ____color.Color
 local ____arrays = require("utility.arrays")
 local array = ____arrays.array
-local loadTOCFile = BlzLoadTOCFile
-local getLocalClientWidth = BlzGetLocalClientWidth
-local getLocalClientHeight = BlzGetLocalClientHeight
-local isLocalClientActive = BlzIsLocalClientActive
-local isHeroUnitId = IsHeroUnitId
+local frameToPixelX = BlzFrameToPixelX
+local frameToPixelY = BlzFrameToPixelY
 local getHandleId = GetHandleId
+local getLocalClientHeight = BlzGetLocalClientHeight
+local getLocalClientWidth = BlzGetLocalClientWidth
+local getLocale = BlzGetLocale
 local getMouseFocusUnit = BlzGetMouseFocusUnit
+local getMouseScreenPosX = BlzGetMouseScreenPosX
+local getMouseScreenPosY = BlzGetMouseScreenPosY
+local frameGetText = BlzFrameGetText
 local getUnitRealField = BlzGetUnitRealField
+local getUnitState = GetUnitState
 local getUnitTypeId = GetUnitTypeId
-local getLocale = BlzGetLocale
+local isHeroUnitId = IsHeroUnitId
+local isKeyPressed = BlzIsKeyPressed
+local isLocalClientActive = BlzIsLocalClientActive
+local isMetaKeyPressed = BlzIsMetaKeyPressed
+local isMouseButtonPressed = BlzIsMouseButtonPressed
+local loadTOCFile = BlzLoadTOCFile
 local pingMinimap = PingMinimap
 local pingMinimapEx = PingMinimapEx
+local pixelToFrameX = BlzPixelToFrameX
+local pixelToFrameY = BlzPixelToFrameY
+local setFrameVisible = BlzFrameSetVisible
 local tableSort = table.sort
-local tocPath = "_warscript\\IsHD.toc"
+local hdTocPath = "_warscript\\IsHD.toc"
+local deTocPath = "_warscript\\IsDE.toc"
 compiletime(function()
     if currentMap then
-        local fdfPath = "_warscript\\IsHD.fdf"
-        currentMap:addFileString("_HD.w3mod\\" .. fdfPath, "\r\n")
-        currentMap:addFileString("_HD.w3mod\\" .. tocPath, fdfPath .. "\r\n")
+        local hdFdfPath = "_warscript\\IsHD.fdf"
+        currentMap:addFileString("_HD.w3mod\\" .. hdFdfPath, "\r\n")
+        currentMap:addFileString("_HD.w3mod\\" .. hdTocPath, hdFdfPath .. "\r\n")
+        local deFdfPath = "_warscript\\IsDE.fdf"
+        currentMap:addFileString("_DE.w3mod\\" .. deFdfPath, "\r\n")
+        currentMap:addFileString("_DE.w3mod\\" .. deTocPath, deFdfPath .. "\r\n")
     end
 end)
-local selectionButtons
-Timer:run(function()
-    selectionButtons = __TS__ArrayMap(
-        Frame:byName("SimpleInfoPanelUnitDetail").parent:getChild(5):getChild(0).children,
-        function(____, frame) return frame:getChild(1) end
+--- The portrait's health and mana texts show the focused unit's current and maximum values
+-- as "current / maximum". Once they exist, these origin frames live for the whole game, so
+-- the polling may read them every tick; the group panel's buttons, by contrast, are rebuilt
+-- at any selection change and a retained button handle crashes the game on the first native
+-- called with it. The frames are not there at map load, though, and appear at different
+-- times on different clients: `Frame.byOrigin` returns undefined until then and balances
+-- the handle allocation across clients, so it is asked again once a second.
+local portraitHpText
+local portraitManaText
+local function actualizePortraitFrames()
+    portraitHpText = Frame:byOrigin(ORIGIN_FRAME_PORTRAIT_HP_TEXT)
+    portraitManaText = Frame:byOrigin(ORIGIN_FRAME_PORTRAIT_MANA_TEXT)
+end
+--- Finds "current / maximum" in the text, which may also carry color codes and other
+-- decoration; returns -1 for both when there is no such pair.
+local function parsePortraitValues(frame)
+    if frame == nil then
+        return -1, -1
+    end
+    local text = frameGetText(frame.handle)
+    local current, maximum = string.match(text, "(%d+)%s*/%s*(%d+)")
+    if current == nil or maximum == nil then
+        return -1, -1
+    end
+    return tonumber(current), tonumber(maximum)
+end
+--- Whether a unit value can be what the portrait shows as `shown`; the rounding rule is unknown.
+local function matchesPortraitValue(value, shown)
+    return math.abs(value - shown) < 1
+end
+--- Whether the unit's health matches what the portrait shows; the health text is required.
+local function matchesPortraitHealth(unit, hp, maxHp)
+    local handle = unit.handle
+    return maxHp >= 0 and matchesPortraitValue(
+        getUnitState(handle, UNIT_STATE_LIFE),
+        hp
+    ) and matchesPortraitValue(
+        getUnitState(handle, UNIT_STATE_MAX_LIFE),
+        maxHp
     )
-end)
+end
+--- Whether the unit has mana and it matches what the portrait shows. The mana text is not
+-- redrawn for a unit without mana (it keeps the previously focused unit's values), so a
+-- mana match is evidence for a unit, but a mismatch is no evidence against one without mana.
+local function matchesPortraitMana(unit, mana, maxMana)
+    local handle = unit.handle
+    local unitMaxMana = getUnitState(handle, UNIT_STATE_MAX_MANA)
+    return maxMana >= 0 and unitMaxMana > 0 and matchesPortraitValue(
+        getUnitState(handle, UNIT_STATE_MANA),
+        mana
+    ) and matchesPortraitValue(unitMaxMana, maxMana)
+end
 local localSelectedUnits = {}
 local indexByLocalSelectedUnit = {}
 local function compareUnitsSelectionPriority(a, b)
@@ -65,6 +125,7 @@ local function compareUnitsSelectionPriority(a, b)
 end
 local mainSelectedUnitChangeEvent
 local previousMainSelectedUnit
+local currentMainSelectedUnit
 local lastTargetingModeState = false
 local targetingModeEnterEvent = __TS__New(Event)
 local targetingModeLeaveEvent = __TS__New(Event)
@@ -99,6 +160,27 @@ function LocalClient.pingMinimap(self, x, y, duration, redOrColor, greenOrFlashy
         )
     end
 end
+function LocalClient.isKeyPressed(self, key)
+    return isKeyPressed(key)
+end
+function LocalClient.isMetaKeyPressed(self, key)
+    return isMetaKeyPressed(key)
+end
+function LocalClient.isMouseButtonPressed(self, button)
+    return isMouseButtonPressed(button)
+end
+function LocalClient.screenToFrameX(self, x)
+    return pixelToFrameX(x)
+end
+function LocalClient.screenToFrameY(self, y)
+    return pixelToFrameY(y)
+end
+function LocalClient.frameToScreenX(self, x)
+    return frameToPixelX(x)
+end
+function LocalClient.frameToScreenY(self, y)
+    return frameToPixelY(y)
+end
 LocalClient.locale = getLocale()
 __TS__ObjectDefineProperty(
     LocalClient,
@@ -114,6 +196,13 @@ __TS__ObjectDefineProperty(
         return getLocalClientHeight()
     end}
 )
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "isSD",
+    {get = function(self)
+        return true
+    end}
+)
 __TS__ObjectDefineProperty(
     LocalClient,
     "isHD",
@@ -121,11 +210,18 @@ __TS__ObjectDefineProperty(
         return false
     end}
 )
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "isDE",
+    {get = function(self)
+        return false
+    end}
+)
 __TS__ObjectDefineProperty(
     LocalClient,
     "graphicsMode",
     {get = function(self)
-        return self.isHD and 1 or 0
+        return 0
     end}
 )
 __TS__ObjectDefineProperty(
@@ -142,6 +238,34 @@ __TS__ObjectDefineProperty(
         return getLocalClientHeight() == 0
     end}
 )
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "mouseScreenX",
+    {get = function(self)
+        return getMouseScreenPosX()
+    end}
+)
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "mouseScreenY",
+    {get = function(self)
+        return getMouseScreenPosY()
+    end}
+)
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "mouseFrameX",
+    {get = function(self)
+        return pixelToFrameX(getMouseScreenPosX())
+    end}
+)
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "mouseFrameY",
+    {get = function(self)
+        return pixelToFrameY(getMouseScreenPosY())
+    end}
+)
 __TS__ObjectDefineProperty(
     LocalClient,
     "mouseFocusUnit",
@@ -153,33 +277,7 @@ __TS__ObjectDefineProperty(
     LocalClient,
     "mainSelectedUnit",
     {get = function(self)
-        Unit:getSelectionOf(Player["local"], localSelectedUnits)
-        for i = 1, #localSelectedUnits do
-            indexByLocalSelectedUnit[localSelectedUnits[i]] = i
-        end
-        tableSort(localSelectedUnits, compareUnitsSelectionPriority)
-        local mainSelectedUnitIndex
-        if selectionButtons and #localSelectedUnits > 1 then
-            local maxButtonWidth = 0
-            for i = 0, #selectionButtons - 1 do
-                local width = selectionButtons[i + 1].width
-                if width > maxButtonWidth then
-                    maxButtonWidth = width
-                    mainSelectedUnitIndex = i
-                end
-            end
-        end
-        local mainSelectedUnit = localSelectedUnits[(mainSelectedUnitIndex or 0) + 1]
-        for i = 1, #localSelectedUnits do
-            indexByLocalSelectedUnit[localSelectedUnits[i]] = nil
-            localSelectedUnits[i] = nil
-        end
-        if mainSelectedUnitChangeEvent ~= nil and mainSelectedUnit ~= previousMainSelectedUnit then
-            local previousPreviousMainSelectedUnit = previousMainSelectedUnit
-            previousMainSelectedUnit = mainSelectedUnit
-            Event.invoke(mainSelectedUnitChangeEvent, previousPreviousMainSelectedUnit, previousMainSelectedUnit)
-        end
-        return mainSelectedUnit
+        return currentMainSelectedUnit
     end}
 )
 __TS__ObjectDefineProperty(
@@ -188,9 +286,6 @@ __TS__ObjectDefineProperty(
     {get = function(self)
         if mainSelectedUnitChangeEvent == nil then
             mainSelectedUnitChangeEvent = __TS__New(Event)
-            Timer.onPeriod[1 / 64]:addListener(function()
-                local _ = ____exports.LocalClient.mainSelectedUnit
-            end)
         end
         return mainSelectedUnitChangeEvent
     end}
@@ -212,17 +307,104 @@ LocalClient.onDisconnect = __TS__New(
     end,
     function() return  end
 )
+local function actualizeMainSelectedUnit()
+    Unit:getSelectionOf(Player["local"], localSelectedUnits)
+    local selectionCount = #localSelectedUnits
+    for i = 1, selectionCount do
+        indexByLocalSelectedUnit[localSelectedUnits[i]] = i
+    end
+    tableSort(localSelectedUnits, compareUnitsSelectionPriority)
+    local mainSelectedUnit
+    if selectionCount <= 1 then
+        mainSelectedUnit = localSelectedUnits[1]
+    else
+        local hp, maxHp = parsePortraitValues(portraitHpText)
+        local mana, maxMana = parsePortraitValues(portraitManaText)
+        if maxHp >= 0 then
+            local healthMatch
+            for i = 1, selectionCount do
+                local unit = localSelectedUnits[i]
+                if matchesPortraitHealth(unit, hp, maxHp) then
+                    if matchesPortraitMana(unit, mana, maxMana) then
+                        mainSelectedUnit = unit
+                        break
+                    end
+                    if healthMatch == nil then
+                        healthMatch = unit
+                    end
+                end
+            end
+            if mainSelectedUnit == nil then
+                mainSelectedUnit = healthMatch
+            end
+        end
+        if mainSelectedUnit == nil and previousMainSelectedUnit ~= nil then
+            for i = 1, selectionCount do
+                if localSelectedUnits[i] == previousMainSelectedUnit then
+                    mainSelectedUnit = previousMainSelectedUnit
+                    break
+                end
+            end
+        end
+        if mainSelectedUnit == nil then
+            mainSelectedUnit = localSelectedUnits[1]
+        end
+    end
+    for i = 1, selectionCount do
+        indexByLocalSelectedUnit[localSelectedUnits[i]] = nil
+        localSelectedUnits[i] = nil
+    end
+    currentMainSelectedUnit = mainSelectedUnit
+    if mainSelectedUnitChangeEvent ~= nil and mainSelectedUnit ~= previousMainSelectedUnit then
+        local previousPreviousMainSelectedUnit = previousMainSelectedUnit
+        previousMainSelectedUnit = mainSelectedUnit
+        Event.invoke(mainSelectedUnitChangeEvent, previousPreviousMainSelectedUnit, previousMainSelectedUnit)
+    end
+end
 local commandButtons = array(
     12,
     function(i) return Frame:byOrigin(ORIGIN_FRAME_COMMAND_BUTTON, i) end
 )
+local commandButtonContainers = array(
+    12,
+    function(i) return Frame:byName("CommandButton_" .. tostring(i)) end
+)
 local function getTargetingModeState()
-    for i = 0, 10 do
-        if commandButtons[i + 1].visible then
-            return false
+    for ____, container in ipairs(commandButtonContainers) do
+        if container.hideCounter > 0 then
+            setFrameVisible(container.handle, true)
         end
     end
-    return commandButtons[12].visible
+    local targeting = false
+    local failed = false
+    local failure
+    do
+        local function ____catch(____error)
+            failed = true
+            failure = ____error
+        end
+        local ____try, ____hasReturned = pcall(function()
+            targeting = commandButtons[12].visible
+            for i = 0, 10 do
+                if commandButtons[i + 1].visible then
+                    targeting = false
+                    break
+                end
+            end
+        end)
+        if not ____try then
+            ____catch(____hasReturned)
+        end
+    end
+    for ____, container in ipairs(commandButtonContainers) do
+        if container.hideCounter > 0 then
+            setFrameVisible(container.handle, false)
+        end
+    end
+    if failed then
+        error(failure, 0)
+    end
+    return targeting
 end
 actualizeTargetingModeState = function()
     if getTargetingModeState() then
@@ -240,14 +422,17 @@ actualizeTargetingModeState = function()
     end
     return false
 end
+Timer.onPeriod[1]:addListener(actualizePortraitFrames)
 Timer.onPeriod[1 / 64]:addListener(function()
+    actualizeMainSelectedUnit()
     actualizeTargetingModeState()
 end)
 warpack.afterMapInit(function()
-    rawset(
-        ____exports.LocalClient,
-        "isHD",
-        loadTOCFile(tocPath)
-    )
+    local isDE = loadTOCFile(deTocPath)
+    local isHD = not isDE and loadTOCFile(hdTocPath)
+    rawset(____exports.LocalClient, "isDE", isDE)
+    rawset(____exports.LocalClient, "isHD", isHD)
+    rawset(____exports.LocalClient, "isSD", not isDE and not isHD)
+    rawset(____exports.LocalClient, "graphicsMode", isDE and 2 or (isHD and 1 or 0))
 end)
 return ____exports

package/engine/object-data/auxiliary/attachment-preset.d.ts

@@ -8,6 +8,13 @@ export type AttachmentPreset = {
     nodeName: ModelNodeName;
     nodeQualifiers: ModelNodeQualifier[];
 };
+/** Where an effect attaches: a model node, e.g. the chest or the left hand. */
+export type AttachmentPoint = {
+    nodeName: ModelNodeName;
+    nodeQualifiers: ModelNodeQualifier[];
+};
+export type AttachmentPointInput = Optional<AttachmentPoint, "nodeQualifiers">;
+export declare const attachmentPointToNodeFQN: (attachmentPoint: AttachmentPointInput) => string;
 export type EffectPresetWithParameters = AttachmentPreset & {
     parameters?: EffectParameters;
 };

package/engine/object-data/auxiliary/attachment-preset.lua

@@ -3,6 +3,13 @@ local __TS__ArrayJoin = ____lualib.__TS__ArrayJoin
 local __TS__StringSplit = ____lualib.__TS__StringSplit
 local __TS__ArraySlice = ____lualib.__TS__ArraySlice
 local ____exports = {}
+____exports.attachmentPointToNodeFQN = function(attachmentPoint) return table.concat(
+    {
+        attachmentPoint.nodeName,
+        table.unpack(attachmentPoint.nodeQualifiers or ({}))
+    },
+    ","
+) end
 ____exports.toEffectPreset = function(effectPresetInput)
     return type(effectPresetInput) == "string" and ({modelPath = effectPresetInput, nodeName = "origin", nodeQualifiers = {}}) or ({modelPath = effectPresetInput.modelPath, nodeName = effectPresetInput.nodeName or "origin", nodeQualifiers = effectPresetInput.nodeQualifiers or ({}), parameters = effectPresetInput.parameters})
 end

package/engine/object-data/auxiliary/equipment-slot.d.ts

@@ -0,0 +1,12 @@
+/** @noSelfInFile */
+export declare const enum EquipmentSlot {
+    HEAD = 0,
+    CHEST = 1,
+    GLOVES = 2,
+    BOOTS = 3,
+    RING = 4,
+    ALTERNATE_RING = 5,
+    PRIMARY = 6,
+    OFFHAND = 7,
+    TRINKET = 8
+}

package/engine/object-data/auxiliary/equipment-slot.lua

@@ -0,0 +1,2 @@
+local ____exports = {}
+return ____exports

package/engine/object-data/auxiliary/equipment-type.d.ts

@@ -0,0 +1,12 @@
+/** @noSelfInFile */
+export declare const enum EquipmentType {
+    NONE = 0,
+    HEAD = 1,
+    CHEST = 2,
+    GLOVES = 3,
+    BOOTS = 4,
+    RING = 5,
+    PRIMARY = 6,
+    OFFHAND = 7,
+    TRINKET = 8
+}

package/engine/object-data/auxiliary/equipment-type.lua

@@ -0,0 +1,26 @@
+local ____exports = {}
+local ____records = require("utility.records")
+local invertRecord = ____records.invertRecord
+local stringByEquipmentType = {
+    [0] = "None",
+    [1] = "Head",
+    [2] = "Chest",
+    [3] = "Gloves",
+    [4] = "Boots",
+    [5] = "Ring",
+    [6] = "Primary",
+    [7] = "Offhand",
+    [8] = "Trinket"
+}
+local equipmentTypeByString = invertRecord(stringByEquipmentType)
+---
+-- @internal For use by internal systems only.
+____exports.equipmentTypeToString = function(equipmentType)
+    return stringByEquipmentType[equipmentType]
+end
+---
+-- @internal For use by internal systems only.
+____exports.stringToEquipmentType = function(____string)
+    return equipmentTypeByString[____string] or 0
+end
+return ____exports

package/engine/object-data/auxiliary/item-class.d.ts

@@ -0,0 +1,11 @@
+/** @noSelfInFile */
+export declare const enum ItemClass {
+    PERMANENT = "Permanent",
+    CHARGED = "Charged",
+    POWER_UP = "PowerUp",
+    ARTIFACT = "Artifact",
+    PURCHASABLE = "Purchasable",
+    CAMPAIGN = "Campaign",
+    MISCELLANEOUS = "Miscellaneous",
+    EQUIPMENT = "Equipment"
+}

package/engine/object-data/auxiliary/item-class.lua

@@ -0,0 +1,2 @@
+local ____exports = {}
+return ____exports

package/engine/object-data/auxiliary/item-tag.d.ts

@@ -0,0 +1,12 @@
+/** @noSelfInFile */
+export declare const enum ItemTag {
+    UNDEFINED = 0,
+    DROPPABLE = 1,
+    QUEST_REWARD = 2,
+    BOSS_DROP = 3,
+    SECRET = 4,
+    PUZZLE = 5,
+    WORLD = 6,
+    SHOP = 7,
+    ANY = 8
+}

package/engine/object-data/auxiliary/item-tag.lua

@@ -0,0 +1,2 @@
+local ____exports = {}
+return ____exports