Mercurial > prosody-modules
view mod_clean_roster/mod_clean_roster.lua @ 5296:0f5657db1cfc
mod_isolate_host: handle server-generated stanzas
The hook for setting the no_host_isolation is only called for c2s
sessions. This does not work for stanzas generated by the server,
such as PEP notifications or presence probe answers.
To handle that, we do per-stanza checks for the case that the origin
is local.
author | Jonas Schäfer <jonas@wielicki.name> |
---|---|
date | Sat, 01 Apr 2023 12:03:08 +0200 |
parents | e384b91d0aa7 |
children |
line wrap: on
line source
local s_find = string.find; local pctl = require "util.prosodyctl"; local rostermanager = require "core.rostermanager"; local storagemanager = require "core.storagemanager"; local usermanager = require "core.usermanager"; -- copypaste from util.stanza local function valid_xml_cdata(str, attr) return not s_find(str, attr and "[^\1\9\10\13\20-~\128-\247]" or "[^\9\10\13\20-~\128-\247]"); end function module.command(_arg) if select(2, pctl.isrunning()) then pctl.show_warning("Stop Prosody before running this command"); return 1; end for hostname, host in pairs(prosody.hosts) do if hostname ~= "*" then if host.users.name == "null" then storagemanager.initialize_host(hostname); usermanager.initialize_host(hostname); end local fixes = 0; for username in host.users.users() do local roster = rostermanager.load_roster(username, hostname); local changed = false; for contact, item in pairs(roster) do if contact ~= false then if item.name and not valid_xml_cdata(item.name, false) then item.name = item.name:gsub("[^\9\10\13\20-~\128-\247]", "�"); fixes = fixes + 1; changed = true; end local clean_groups = {}; for group in pairs(item.groups) do if valid_xml_cdata(group, false) then clean_groups[group] = true; else clean_groups[group:gsub("[^\9\10\13\20-~\128-\247]", "�")] = true; fixes = fixes + 1; changed = true; end end item.groups = clean_groups; else -- pending entries etc end end if changed then assert(rostermanager.save_roster(username, hostname, roster)); end end pctl.show_message("Fixed %d items on host %s", fixes, hostname); end end return 0; end