warscript 0.0.1-dev.404878c

package/objutil/buff.lua

@@ -0,0 +1,769 @@
+local ____lualib = require("lualib_bundle")
+local __TS__Class = ____lualib.__TS__Class
+local __TS__ClassExtends = ____lualib.__TS__ClassExtends
+local __TS__New = ____lualib.__TS__New
+local __TS__SetDescriptor = ____lualib.__TS__SetDescriptor
+local __TS__StringSplit = ____lualib.__TS__StringSplit
+local __TS__ArrayIsArray = ____lualib.__TS__ArrayIsArray
+local __TS__ArrayMap = ____lualib.__TS__ArrayMap
+local __TS__ArrayFilter = ____lualib.__TS__ArrayFilter
+local Set = ____lualib.Set
+local __TS__ArrayFrom = ____lualib.__TS__ArrayFrom
+local __TS__InstanceOf = ____lualib.__TS__InstanceOf
+local ____exports = {}
+local ____object = require("objutil.object")
+local ObjectDefinition = ____object.ObjectDefinition
+local ____ability = require("objutil.ability")
+local AbilityDefinitionCripple = ____ability.AbilityDefinitionCripple
+local AbilityDefinitionHealingSalve = ____ability.AbilityDefinitionHealingSalve
+local AbilityDefinitionInnerFire = ____ability.AbilityDefinitionInnerFire
+local AbilityDefinitionSearingArrows = ____ability.AbilityDefinitionSearingArrows
+local AbilityDefinitionSlowPoison = ____ability.AbilityDefinitionSlowPoison
+local ____unit = require("core.types.unit")
+local AttackType = ____unit.AttackType
+local DamageType = ____unit.DamageType
+local Unit = ____unit.Unit
+local ____dummy = require("core.dummy")
+local InstantDummyCaster = ____dummy.InstantDummyCaster
+local idgen = require("objutil.idgen")
+local ____timer = require("core.types.timer")
+local Timer = ____timer.Timer
+local ____exception = require("exception")
+local IllegalStateException = ____exception.IllegalStateException
+local ____assert = _G.assert
+local ____pairs = _G.pairs
+____exports.BuffDefinition = __TS__Class()
+local BuffDefinition = ____exports.BuffDefinition
+BuffDefinition.name = "BuffDefinition"
+__TS__ClassExtends(BuffDefinition, ObjectDefinition)
+function BuffDefinition.prototype.____constructor(self, object)
+    ObjectDefinition.prototype.____constructor(self, object)
+    self.id = self.object.id
+end
+function BuffDefinition.create(self, baseId, id)
+    return __TS__New(
+        self,
+        currentMap.objects.buff:newObject(
+            id and util.id2s(id) or idgen.buff(),
+            util.id2s(baseId)
+        )
+    )
+end
+function BuffDefinition.of(self, id)
+    return __TS__New(
+        self,
+        currentMap.objects.buff:getObject(util.id2s(id))
+    )
+end
+function BuffDefinition.createDerived(derived, baseId, id)
+    return ____exports.BuffDefinition.create(derived, baseId, id)
+end
+__TS__SetDescriptor(
+    BuffDefinition.prototype,
+    "name",
+    {
+        get = function(self)
+            return self:getStringField("fnam")
+        end,
+        set = function(self, v)
+            self:setStringField("fnam", v)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    BuffDefinition.prototype,
+    "tooltip",
+    {
+        get = function(self)
+            return self:getStringField("ftip")
+        end,
+        set = function(self, v)
+            self:setStringField("ftip", v)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    BuffDefinition.prototype,
+    "tooltipExtended",
+    {
+        get = function(self)
+            return self:getStringField("fube")
+        end,
+        set = function(self, v)
+            self:setStringField("fube", v)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    BuffDefinition.prototype,
+    "icon",
+    {
+        get = function(self)
+            return self:getStringField("fart")
+        end,
+        set = function(self, v)
+            self:setStringField("fart", v)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    BuffDefinition.prototype,
+    "effectArt",
+    {
+        get = function(self)
+            return self:getStringField("feat")
+        end,
+        set = function(self, v)
+            self:setStringField("feat", v)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    BuffDefinition.prototype,
+    "specialArt",
+    {
+        get = function(self)
+            return {
+                model = self:getStringField("fsat"),
+                attachmentPoint = self:getStringField("fspt")
+            }
+        end,
+        set = function(self, v)
+            if type(v) == "string" then
+                self:setStringField("fsat", v)
+                self:setStringField("fspt", "")
+            else
+                self:setStringField("fsat", v.model)
+                self:setStringField("fspt", v.attachmentPoint)
+            end
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    BuffDefinition.prototype,
+    "targetArt",
+    {
+        get = function(self)
+            local models = __TS__StringSplit(
+                self:getStringField("ftat"),
+                ","
+            )
+            local arts = {}
+            for i = 1, #models do
+                local attachmentPoint = self:getStringField("fta" .. tostring(i - 1)) or ""
+                arts[i] = {model = models[i], attachmentPoint = attachmentPoint == "" and "origin" or attachmentPoint}
+            end
+            return arts
+        end,
+        set = function(self, v)
+            if __TS__ArrayIsArray(v) then
+                self:setStringField(
+                    "ftat",
+                    table.concat(
+                        __TS__ArrayMap(
+                            v,
+                            function(____, art) return type(art) == "string" and art or art.model end
+                        ),
+                        ","
+                    )
+                )
+                local attachmentPoints = __TS__ArrayMap(
+                    v,
+                    function(____, art) return type(art) == "string" and "origin" or art.attachmentPoint end
+                )
+                if #__TS__ArrayFilter(
+                    attachmentPoints,
+                    function(____, attachmentPoint) return attachmentPoint == "origin" or attachmentPoint == "" end
+                ) == #attachmentPoints then
+                    attachmentPoints = {}
+                end
+                for i = 0, 5 do
+                    self:setStringField(
+                        "fta" .. tostring(i),
+                        attachmentPoints[i + 1] or ""
+                    )
+                end
+                self:setNumberField("ftac", #attachmentPoints == 1 and 0 or #attachmentPoints)
+            else
+                self:setStringField(
+                    "ftat",
+                    type(v) == "string" and v or v.model
+                )
+                self:setStringField(
+                    "fta0",
+                    type(v) == "string" and "" or v.attachmentPoint
+                )
+                for i = 1, 5 do
+                    self:setStringField(
+                        "fta" .. tostring(i),
+                        ""
+                    )
+                end
+                self:setNumberField("ftac", 0)
+            end
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    BuffDefinition.prototype,
+    "missileArt",
+    {
+        get = function(self)
+            return self:getStringField("fmat")
+        end,
+        set = function(self, v)
+            self:setStringField("fmat", v)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    BuffDefinition.prototype,
+    "missileArc",
+    {
+        get = function(self)
+            return self:getNumberField("fmac")
+        end,
+        set = function(self, v)
+            self:setNumberField("fmac", v)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    BuffDefinition.prototype,
+    "missileSpeed",
+    {
+        get = function(self)
+            return self:getNumberField("fmsp")
+        end,
+        set = function(self, v)
+            self:setNumberField("fmsp", v)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    BuffDefinition.prototype,
+    "missileHoming",
+    {
+        get = function(self)
+            return self:getBooleanField("fmho")
+        end,
+        set = function(self, v)
+            self:setBooleanField("fmho", v)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    BuffDefinition.prototype,
+    "sound",
+    {
+        get = function(self)
+            return self:getStringField("fefs")
+        end,
+        set = function(self, v)
+            self:setStringField("fefs", v)
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    BuffDefinition.prototype,
+    "soundLooping",
+    {
+        get = function(self)
+            return self:getStringField("fefl")
+        end,
+        set = function(self, v)
+            self:setStringField("fefl", v)
+        end
+    },
+    true
+)
+local EVASION_ABILITY_IDS = postcompile(function()
+    local evasionParentAbilityIds = __TS__New(Set, {
+        "AEev",
+        "ACes",
+        "ACev",
+        "AIes",
+        "ANdb",
+        "Acdb",
+        "AOcr",
+        "ACct",
+        "AIcs"
+    })
+    local evasionAbilityIds = __TS__New(Set)
+    if currentMap then
+        for ____, ability in ____pairs(currentMap.objects.ability.all) do
+            if evasionParentAbilityIds:has(ability.parentId or ability.id) then
+                evasionAbilityIds:add(ability.id)
+            end
+        end
+    end
+    return __TS__ArrayMap(
+        __TS__ArrayFrom(evasionAbilityIds),
+        function(____, id) return fourCC(id) end
+    )
+end)
+local SEARING_ARROWS_ABILITY_ID = compiletime(function()
+    local ability = __TS__New(AbilityDefinitionSearingArrows)
+    ability.name = "[Buff System] Searing Arrows"
+    ability.levels = 1
+    ability.damageBonus = 0
+    ability.manaCost = 0
+    ability.missileArt = ""
+    ability.casterArt = ""
+    ability.targetArt = ""
+    ability.castRange = 99999
+    ability.requiredLevel = 6
+    ability.targetsAllowed = {
+        "ground",
+        "air",
+        "structure",
+        "ward",
+        "vulnerable",
+        "invulnerable"
+    }
+    return fourCC(ability.id)
+end)
+____exports.BuffPreset = __TS__Class()
+local BuffPreset = ____exports.BuffPreset
+BuffPreset.name = "BuffPreset"
+__TS__ClassExtends(BuffPreset, ____exports.BuffDefinition)
+function BuffPreset.prototype.____constructor(self, object)
+    BuffPreset.____super.prototype.____constructor(self, object)
+    self._positive = true
+    self._magic = true
+    self._visible = false
+    self.levels = 1
+end
+function BuffPreset.prototype.createInnerFireAbility(self)
+    local ability = __TS__New(AbilityDefinitionInnerFire)
+    ability.levels = self.levels
+    ability.damageIncrease = 0
+    ability.armorBonus = 0
+    ability.healthRegenerationBonus = 0
+    ability.autoCastRange = 0
+    ability.buffs = {self.id}
+    return ability
+end
+function BuffPreset.prototype.createCrippleAbility(self)
+    local ability = __TS__New(AbilityDefinitionCripple)
+    ability.levels = self.levels
+    ability.movementSpeedReduction = 0
+    ability.attackSpeedReduction = 0
+    ability.damageReduction = 0
+    ability.buffs = {self.id}
+    return ability
+end
+function BuffPreset.prototype.createHealingSaveAbility(self)
+    local ability = __TS__New(AbilityDefinitionHealingSalve)
+    ability.levels = self.levels
+    ability.lifeRegenerated = 1.175494e-38
+    ability.manaRegenerated = 0
+    ability.allowWhenFull = "always"
+    ability.noTargetRequired = false
+    ability.dispelOnAttack = false
+    ability.buffs = {self.id, self.id, self.id}
+    return ability
+end
+function BuffPreset.prototype.createSlowPoisonAbility(self)
+    local ability = __TS__New(AbilityDefinitionSlowPoison)
+    ability.levels = self.levels
+    ability.damagePerSecond = 0
+    ability.movementSpeedFactor = 0
+    ability.attackSpeedFactor = 0
+    ability.stackingType = 0
+    ability.buffs = {self.id, self.id}
+    return ability
+end
+function BuffPreset.prototype.compiletime(self)
+    local ability = self.positive and (self.magic and self:createInnerFireAbility() or self:createHealingSaveAbility()) or (self.magic and self:createCrippleAbility() or self:createSlowPoisonAbility())
+    ability.manaCost = 0
+    ability.cooldown = 0
+    ability.durationNormal = 0
+    ability.durationHero = 0
+    ability.missileArt = ""
+    ability.casterArt = ""
+    ability.targetArt = ""
+    ability.castRange = 99999
+    ability.areaOfEffect = 0
+    ability.checkDependencies = false
+    ability.requirements = {}
+    ability.requiredLevel = 6
+    ability.targetsAllowed = {
+        "ground",
+        "air",
+        "structure",
+        "ward",
+        "vulnerable",
+        "invulnerable"
+    }
+    return {
+        buffId = util.s2id(self.id),
+        abilityId = util.s2id(ability.id),
+        positive = self.positive,
+        magic = self._magic
+    }
+end
+function BuffPreset.create(self, baseId, id)
+    return ____exports.BuffDefinition.createDerived(self, baseId or "BHbz", id)
+end
+__TS__SetDescriptor(
+    BuffPreset.prototype,
+    "positive",
+    {
+        get = function(self)
+            return self._positive
+        end,
+        set = function(self, v)
+            self._positive = v
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    BuffPreset.prototype,
+    "magic",
+    {
+        get = function(self)
+            return self._magic
+        end,
+        set = function(self, v)
+            self._magic = v
+        end
+    },
+    true
+)
+__TS__SetDescriptor(
+    BuffPreset.prototype,
+    "visible",
+    {
+        get = function(self)
+            return self._visible
+        end,
+        set = function(self, v)
+            self._visible = v
+        end
+    },
+    true
+)
+local buffs = setmetatable(
+    {},
+    {
+        __mode = "k",
+        __index = function(self, unit)
+            local ____table = {}
+            self[unit] = ____table
+            return ____table
+        end
+    }
+)
+local IllegalBuffTargetException = {}
+Unit.abilityCastingStartEvent[SEARING_ARROWS_ABILITY_ID]:addListener(function(unit)
+    unit:removeAbility(SEARING_ARROWS_ABILITY_ID)
+end)
+____exports.Buff = __TS__Class()
+local Buff = ____exports.Buff
+Buff.name = "Buff"
+function Buff.prototype.____constructor(self, unit, source, duration, level, preset)
+    if source == nil then
+        source = unit
+    end
+    if duration == nil then
+        duration = 0
+    end
+    if level == nil then
+        level = 0
+    end
+    self.unit = unit
+    self.source = source
+    self.duration = duration
+    self.level = level
+    self.preset = preset
+    if not unit.isAlive then
+        error(IllegalBuffTargetException, 0)
+    end
+    local constructor = self.constructor
+    local instances = buffs[unit]
+    local instance = instances[constructor]
+    if instance then
+        instance:destroy()
+        if not instance.destroyed then
+            error(
+                __TS__New(
+                    IllegalStateException,
+                    (("Cannot destroy previous " .. self.constructor.name) .. " buff instance on ") .. tostring(unit)
+                ),
+                0
+            )
+        end
+    end
+    instances[constructor] = self
+    if preset then
+        local unitHandle = unit.handle
+        local slowPoisonMethod = not preset.positive and not preset.magic
+        if slowPoisonMethod then
+            for ____, abilityId in ipairs(EVASION_ABILITY_IDS) do
+                BlzUnitDisableAbility(unitHandle, abilityId, true, false)
+            end
+        end
+        local success = InstantDummyCaster:getInstance():castTarget(
+            unit.owner,
+            {preset.abilityId, SEARING_ARROWS_ABILITY_ID},
+            function(ability)
+                ability.level = level
+                local actualDuration = duration > 0 and duration or 2147483647
+                ability:setField(ABILITY_RLF_DURATION_NORMAL, level, actualDuration)
+                ability:setField(ABILITY_RLF_DURATION_HERO, level, actualDuration)
+            end,
+            preset.positive and (preset.magic and orderId("innerfire") or 852609) or (preset.magic and orderId("cripple") or orderId("flamingarrowstarg")),
+            unit
+        )
+        if slowPoisonMethod then
+            for ____, abilityId in ipairs(EVASION_ABILITY_IDS) do
+                BlzUnitDisableAbility(unitHandle, abilityId, false, false)
+            end
+        end
+        local handle = BlzGetUnitAbility(unitHandle, preset.buffId)
+        if not success or handle == nil then
+            error(
+                __TS__New(
+                    IllegalStateException,
+                    (("Cannot apply " .. self.constructor.name) .. " buff status effect on ") .. tostring(unit)
+                ),
+                0
+            )
+        end
+        self.handle = handle
+    end
+    if duration > 0 then
+        self.timer = Timer:simple(
+            duration,
+            function()
+                self.timer = nil
+                if not self.destroyed then
+                    Timer:run(function()
+                        if not self.destroyed then
+                            self:destroy()
+                        end
+                    end)
+                    self:onExpire()
+                end
+            end
+        )
+    end
+end
+function Buff.apply(self, ...)
+    local args = {...}
+    do
+        local function ____catch(e)
+            if e ~= IllegalBuffTargetException then
+                error(e, 0)
+            end
+            return true, nil
+        end
+        local ____try, ____hasReturned, ____returnValue = pcall(function()
+            return true, __TS__New(
+                self,
+                table.unpack(args)
+            )
+        end)
+        if not ____try then
+            ____hasReturned, ____returnValue = ____catch(____hasReturned)
+        end
+        if ____hasReturned then
+            return ____returnValue
+        end
+    end
+end
+function Buff.getInstance(self, unit)
+    return buffs[unit][self]
+end
+function Buff.isApplied(self, unit)
+    return buffs[unit][self] ~= nil
+end
+function Buff.ifApplied(self, unit, action)
+    local instance = buffs[unit][self]
+    if instance then
+        ____assert(__TS__InstanceOf(instance, self))
+        action(instance)
+    end
+end
+function Buff.prototype.destroy(self)
+    if self.destroyed then
+        error(
+            __TS__New(
+                IllegalStateException,
+                ((self.constructor.name .. " buff on ") .. tostring(self.unit)) .. " is already destroyed"
+            ),
+            0
+        )
+    end
+    if self.timer then
+        self.timer:destroy()
+        self.timer = nil
+    end
+    if self.preset then
+        self.unit:removeAbility(self.preset.buffId)
+    end
+    buffs[self.unit][self.constructor] = nil
+    self.destroyed = true
+end
+function Buff.prototype.onDispel(self, source)
+end
+function Buff.prototype.onExpire(self)
+end
+Unit.deathEvent:addListener(function(unit)
+    for ____, buff in pairs(buffs[unit]) do
+        buff:destroy()
+    end
+end)
+Unit.destroyEvent:addListener(function(unit)
+    for ____, buff in pairs(buffs[unit]) do
+        buff:destroy()
+    end
+end)
+local unitRemoveBuffs = UnitRemoveBuffs
+_G.UnitRemoveBuffs = function(unit, removePositive, removeNegative)
+    unitRemoveBuffs(unit, removePositive, removeNegative)
+    for ____, buff in pairs(buffs[Unit:of(unit)]) do
+        local preset = buff.preset
+        if preset then
+            local positive = preset.positive
+            if positive and removePositive or not positive and removeNegative then
+                buff:destroy()
+            end
+        end
+    end
+end
+local selfDispelAbilityIds = compiletime(__TS__ArrayMap(
+    {
+        "AHav",
+        "ANav",
+        "AOww",
+        "AHds",
+        "ACds",
+        "AIdv",
+        "AOmi",
+        "AIvu",
+        "AIvg",
+        "AIvl",
+        "ANef",
+        "Acef",
+        "AIxs",
+        "Amdf",
+        "AEtq"
+    },
+    function(____, id) return fourCC(id) end
+))
+local targetDispelAbilityIds = compiletime(__TS__ArrayMap(
+    {
+        "Acyc",
+        "SCc1",
+        "ACcy",
+        "Acny",
+        "Aams",
+        "ACam",
+        "AIcy",
+        "Aprg",
+        "Apg2",
+        "Aadm",
+        "ACpu",
+        "ACdm",
+        "ACd2",
+        "Andm",
+        "AIlp",
+        "AIpg",
+        "AIps"
+    },
+    function(____, id) return fourCC(id) end
+))
+local pointDispelAbilityIds = compiletime(__TS__ArrayMap(
+    {
+        "Adis",
+        "Adcn",
+        "Adch",
+        "Advm",
+        "Adtn",
+        "ACde",
+        "Adsm",
+        "AIdi",
+        "AIds",
+        "APdi"
+    },
+    function(____, id) return fourCC(id) end
+))
+local function checkBuffs(unit, dispel, dispelSource)
+    for ____, buff in pairs(buffs[unit]) do
+        local preset = buff.preset
+        local handle = buff.handle
+        if handle ~= nil and preset ~= nil and BlzGetUnitAbility(unit.handle, preset.buffId) ~= handle then
+            if dispel == true and preset.magic then
+                buff.onDispel(buff, dispelSource)
+            end
+            buff:destroy()
+        end
+    end
+end
+Unit.abilityChannelingStartEvent[fourCC("AOvd")]:addListener(function(caster, ability)
+    local targets = Unit:getInCollisionRange(
+        caster.x,
+        caster.y,
+        ability:getField(ABILITY_RLF_AREA_OF_EFFECT)
+    )
+    Timer:run(function()
+        for ____, target in ipairs(targets) do
+            checkBuffs(target, true, caster)
+        end
+    end)
+end)
+for ____, selfDispelAbilityId in ipairs(selfDispelAbilityIds) do
+    Unit.abilityChannelingStartEvent[selfDispelAbilityId]:addListener(function(caster)
+        Timer:run(function()
+            checkBuffs(caster, true, caster)
+        end)
+    end)
+end
+for ____, targetDispelAbilityId in ipairs(targetDispelAbilityIds) do
+    Unit.onUnitTargetCast[targetDispelAbilityId]:addListener(function(caster, ability, target)
+        Timer:run(function()
+            checkBuffs(target, true, caster)
+        end)
+    end)
+end
+for ____, pointDispelAbilityId in ipairs(pointDispelAbilityIds) do
+    Unit.onPointCast[pointDispelAbilityId]:addListener(function(caster, ability, x, y)
+        local targets = Unit:getInCollisionRange(
+            x,
+            y,
+            ability:getField(ABILITY_RLF_AREA_OF_EFFECT)
+        )
+        Timer:run(function()
+            for ____, target in ipairs(targets) do
+                checkBuffs(target, true, caster)
+            end
+        end)
+    end)
+end
+Unit.onDamaging:addListener(function(source, target, event)
+    if event.amount == 0 and event.attackType == AttackType.SPELL and event.damageType == DamageType.NORMAL and not event.isAttack then
+        Timer:run(function()
+            checkBuffs(target, true, source)
+        end)
+    end
+end)
+Unit.onItemPickup:addListener(function(unit, item)
+    if item.powerup and item:hasAbility(fourCC("APdi")) then
+    end
+end)
+return ____exports