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

package/engine/local-client.lua

@@ -1,10 +1,10 @@
 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
 local __TS__ObjectDefineProperty = ____lualib.__TS__ObjectDefineProperty
-local __TS__New = ____lualib.__TS__New
 local ____exports = {}
+local actualizeTargetingModeState
 local ____unit = require("core.types.unit")
 local Unit = ____unit.Unit
 local ____event = require("event")
@@ -18,34 +18,97 @@ local ____timer = require("core.types.timer")
 local Timer = ____timer.Timer
 local ____color = require("core.types.color")
 local Color = ____color.Color
-local loadTOCFile = BlzLoadTOCFile
-local getLocalClientWidth = BlzGetLocalClientWidth
-local getLocalClientHeight = BlzGetLocalClientHeight
-local isLocalClientActive = BlzIsLocalClientActive
-local isHeroUnitId = IsHeroUnitId
+local ____arrays = require("utility.arrays")
+local array = ____arrays.array
+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)
@@ -62,12 +125,17 @@ 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)
+local targetingModeStateChangeEvent = __TS__New(Event)
 ____exports.LocalClient = __TS__Class()
 local LocalClient = ____exports.LocalClient
 LocalClient.name = "LocalClient"
 function LocalClient.prototype.____constructor(self)
 end
-function LocalClient.pingMinimap(self, x, y, duration, redOrColor, greenOrExtraEffects, blue, extraEffects)
+function LocalClient.pingMinimap(self, x, y, duration, redOrColor, greenOrFlashy, blue, flashy)
     if redOrColor == nil then
         pingMinimap(x, y, duration)
     elseif __TS__InstanceOf(redOrColor, Color) then
@@ -78,7 +146,7 @@ function LocalClient.pingMinimap(self, x, y, duration, redOrColor, greenOrExtraE
             redOrColor.r,
             redOrColor.g,
             redOrColor.b,
-            greenOrExtraEffects or false
+            greenOrFlashy or false
         )
     else
         pingMinimapEx(
@@ -86,12 +154,33 @@ function LocalClient.pingMinimap(self, x, y, duration, redOrColor, greenOrExtraE
             y,
             duration,
             redOrColor,
-            greenOrExtraEffects,
+            greenOrFlashy,
             blue,
-            extraEffects or false
+            flashy or false
         )
     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,
@@ -107,6 +196,13 @@ __TS__ObjectDefineProperty(
         return getLocalClientHeight()
     end}
 )
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "isSD",
+    {get = function(self)
+        return true
+    end}
+)
 __TS__ObjectDefineProperty(
     LocalClient,
     "isHD",
@@ -114,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(
@@ -128,6 +231,41 @@ __TS__ObjectDefineProperty(
         return isLocalClientActive()
     end}
 )
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "isMinimized",
+    {get = function(self)
+        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",
@@ -139,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(
@@ -174,13 +286,20 @@ __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}
 )
+__TS__ObjectDefineProperty(
+    LocalClient,
+    "isInTargetingMode",
+    {get = function(self)
+        return actualizeTargetingModeState()
+    end}
+)
+LocalClient.targetingModeEnterEvent = targetingModeEnterEvent
+LocalClient.targetingModeLeaveEvent = targetingModeLeaveEvent
+LocalClient.targetingModeStateChangeEvent = targetingModeStateChangeEvent
 LocalClient.onDisconnect = __TS__New(
     TriggerEvent,
     function(trigger)
@@ -188,11 +307,132 @@ 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 ____, container in ipairs(commandButtonContainers) do
+        if container.hideCounter > 0 then
+            setFrameVisible(container.handle, true)
+        end
+    end
+    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
+        if not lastTargetingModeState then
+            lastTargetingModeState = true
+            Event.invoke(targetingModeEnterEvent)
+            Event.invoke(targetingModeStateChangeEvent)
+        end
+        return true
+    end
+    if lastTargetingModeState then
+        lastTargetingModeState = false
+        Event.invoke(targetingModeLeaveEvent)
+        Event.invoke(targetingModeStateChangeEvent)
+    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/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/health-regeneration-type.d.ts

@@ -0,0 +1,8 @@
+/** @noSelfInFile */
+export declare const enum HealthRegenerationType {
+    NONE = 0,
+    ALWAYS = 1,
+    BLIGHT = 2,
+    DAY = 3,
+    NIGHT = 4
+}

package/engine/object-data/auxiliary/health-regeneration-type.lua

@@ -0,0 +1,2 @@
+local ____exports = {}
+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