morrowind-no-more-friendly-fire/MWSE/mods/celediel/NoMoreFriendlyFire/main.lua

local common = require("celediel.NoMoreFriendlyFire.common")
local config = require("celediel.NoMoreFriendlyFire.config").getConfig()

local mag
pcall(function() mag = require("celediel.MoreAttentiveGuards.interop") end)

-- todo: make this not hardcoded somehow
local followMatches = {"follow", "together", "travel", "wait", "stay"}
local postDialogueTimer

local function log(level, ...)
    if config.debugLevel >= level then mwse.log("[%s] %s", common.modName, string.format(...)) end
end

-- keep track of followers
local followers = {}

-- {{{ internal functions
local function notActuallyFriend(friend)
    if not friend then return false end
    local obj = friend.baseObject and friend.baseObject or
                    (friend.object.baseObject and friend.object.baseObject or friend.object)
    local magGuard = mag and mag.getGuardFollower() or nil
    local ignoredId = config.ignored[obj.id:lower()]
    local ignoredMod = config.ignored[obj.sourceMod:lower()]

    log(common.logLevels.small, "Friend %s: Is MAG Guard:%s, Ignored id:%s, Ignored mod:%s", obj.name,
        friend == magGuard, ignoredId or "false", ignoredMod or "false")

    return friend == magGuard or ignoredId or ignoredMod
end

local function followerCheck(attacker, target)
    return followers[attacker.object.id] ~= nil and followers[target.object.id] ~= nil and attacker ~= target
end

local function buildFollowerList()
    local friends = {}

    local msg = ""

    for friend in tes3.iterate(tes3.mobilePlayer.friendlyActors) do
        -- nice double negative
        if not notActuallyFriend(friend) then
            friends[friend.object.id] = true
            msg = msg .. friend.object.name .. " "
        end
    end
    if msg ~= "" then log(common.logLevels.small, "Friends: %s", msg) end
    return friends
end

local function isHarmfulSpell(source)
    -- look for harmful effects in the source spell
    for i = 1, 8 do
        local effect = source.effects[i]
        if effect.object then
            -- todo: get name of spell effect instead of id
            log(common.logLevels.big, "effect #%s id:%s: harmful:%s", i, effect.object.id, effect.object.isHarmful)
            if effect.object.isHarmful then return true end -- found one!
        end
    end
    return false -- found nothing
end
-- }}}

-- {{{ Event functions
local eventFunctions = {}

-- {{{ events to actually do the stuff
eventFunctions.onDamage = function(e)
    if not e.attackerReference then return end

    if followerCheck(e.attackerReference, e.reference) then
        if config.stopDamage then
            log(common.logLevels.small, "%s hit %s for %s friendly damage, nullifying", e.attackerReference.object.name,
                e.reference.object.name, e.damage)
            e.damage = 0
            return false -- I don't know if this makes a difference or not
        else
            log(common.logLevels.small, "%s hit %s for %s friendly damage", e.attackerReference.object.name,
                e.reference.object.name, e.damage)
        end
    else
        log(common.logLevels.big, "%s hit %s for %s damage", e.attackerReference.object.name, e.reference.object.name,
            e.damage)
    end
end

eventFunctions.onSpellResist = function(e)
    -- ignore non-targeted magicka or without caster
    if not e.caster or not e.target then return end
    -- shortcuts
    local caster = e.caster
    local target = e.target
    local source = e.source

    if followerCheck(caster, target) then
        log(common.logLevels.small, "%s hit %s with friendly magicka %s", caster, target, source)
        if config.stopDamage and isHarmfulSpell(e.source) then
            e.resistedPercent = 100
            log(common.logLevels.small, "%s spell resist vs harmful %s maxified", target, source)
        end
    end
end

-- prevent combat from even happening between friendly actors
eventFunctions.onCombatStart = function(e)
    if config.stopCombat and followerCheck(e.actor, e.target) then
        log(common.logLevels.small, "Friendly combat attempted, putting a stop to it!")
        return false
    end
end
-- }}}

-- {{{ events to rebuild follower list
eventFunctions.onCellChanged = function(e) followers = buildFollowerList() end

-- hopefully, when telling a follower to follow or wait, rebuild followers list
eventFunctions.onInfoResponse = function(e)
    -- the dialogue option clicked on
    local dialogue = tostring(e.dialogue):lower()
    -- what that dialogue option triggers; this will catch AIFollow commands
    local command = e.command:lower()

    for _, item in pairs(followMatches) do
        if command:match(item) or dialogue:match(item) then
            -- wait until game time restarts, and don't set multiple timers
            if not postDialogueTimer or postDialogueTimer.state ~= timer.active then
                log(common.logLevels.big, "Found %s in dialogue, rebuilding followers", item)
                postDialogueTimer = timer.start({
                    type = timer.simulate,
                    duration = 0.25,
                    iteration = 1,
                    callback = function() followers = buildFollowerList() end
                })
            end
        end
    end
end

-- rebuild followers list when player casts conjuration, in case its a summon spell
-- false positives are okay because we're not doing anything destructive
-- doesn't work if Skill Based Magicka Progression is activated because e.expGainSchool becomes nil
eventFunctions.onSpellCasted = function(e)
    if e.caster == tes3.player and e.expGainSchool == tes3.magicSchool.conjuration then
        log(common.logLevels.big, "Player cast conjuration spell %s, rebuilding followers list...", e.source.id)
        -- wait for summon to be loaded
        timer.start({
            type = timer.simulate,
            duration = 0.25,
            iterations = 1,
            callback = function() followers = buildFollowerList() end
        })
    end
end
-- }}}
-- }}}

-- Register events
local function onInitialized()
    for name, func in pairs(eventFunctions) do
        event.register(name:gsub("on(%u)", string.lower), func)
        log(common.logLevels.small, "%s event registered", name)
    end

    log(common.logLevels.no, "Successfully initialized%s", mag and " with More Attentive Guards interop" or "")
end

event.register("modConfigReady", function() mwse.mcm.register(require("celediel.NoMoreFriendlyFire.mcm")) end)
event.register("initialized", onInitialized)