forget-dead-body.lua

-- Removes emotions associated with seeing a dead body
--@ module = true

local help = [====[

forget-dead-body
================
Removes emotions associated with seeing a dead body.

]====]

local utils = require 'utils'

local validArgs = utils.invert({
    'help',
    'all',
})

function forgetDeadBody(unit)
    for _, t in ipairs(unit.status.current_soul.personality.emotions) do
        if t.thought == df.unit_thought_type.SawDeadBody then
            t.type = df.emotion_type.ANYTHING
        end
    end
end

function main(...)
    local args = utils.processArgs({...}, validArgs)

    if args.help then
        print(help)
        return
    end

    if args.all then
        for _, unit in ipairs(df.global.world.units.active) do
            if dfhack.units.isCitizen(unit) then
                forgetDeadBody(unit)
            end
        end
    else
        local unit = dfhack.gui.getSelectedUnit()
        if unit then
            forgetDeadBody(unit)
        else
            qerror('Invalid usage: No unit selected and -all argument not given.')
        end
    end
end

if not dfhack_flags.module then
    main(...)
end