Mercurial > prosody-modules
view mod_export_skeletons/skeleton_filter.lua @ 5536:96dec7681af8
mod_firewall: Update user marks to store instantly via map store
The original approach was to keep marks in memory only, and persist them at
shutdown. That saves I/O, at the cost of potentially losing marks on an
unclean shutdown.
This change persists marks instantly, which may have some performance overhead
but should be more "correct".
It also splits the marking/unmarking into an event which may be watched or
even fired by other modules.
author | Matthew Wild <mwild1@gmail.com> |
---|---|
date | Thu, 08 Jun 2023 16:20:42 +0100 |
parents | b4cc6ee9fc8c |
children |
line wrap: on
line source
#!/usr/bin/env lua package.path = package.path:gsub("([^;]*)(?[^;]*)", "%1prosody/%2;%1%2"); package.cpath = package.cpath:gsub("([^;]*)(?[^;]*)", "%1prosody/%2;%1%2"); local t_insert = table.insert; local t_sort = table.sort; local jid = require "util.jid"; local st = require "util.stanza"; local xs = require "util.xmppstream"; local function skeleton(s) local o = st.stanza(s.name, { xmlns = s.attr.xmlns }); local children = {}; for _, child in ipairs(s.tags) do t_insert(children, skeleton(child)) end t_sort(children, function(a, b) if a.attr.xmlns == b.attr.xmlns then return a.name < b.name; end return (a.attr.xmlns or "") < (b.attr.xmlns or ""); end); for _, child in ipairs(children) do o:add_direct_child(child); end return o; end local function classify_jid(s) if not s then return "" end local u, h, r = jid.split(s); if r then return "full" elseif u then return "bare" elseif h then return "host" else return "invalid" end end local stream_session = { notopen = true }; local stream_callbacks = { stream_ns = "jabber:client"; default_ns = "jabber:client" }; function stream_callbacks:handlestanza(item) local clean = skeleton(item); -- Normalize top level attributes clean.attr.type = item.attr.type; if clean.attr.type == nil and clean.name == "message" then clean.attr.type = "normal"; end clean.attr.id = string.rep("x", math.floor(math.log(1 + #(item.attr.id or ""), 2))); clean.attr.from = classify_jid(item.attr.from); clean.attr.to = classify_jid(item.attr.to); print(clean); end local stream = xs.new(stream_session, stream_callbacks); assert(stream:feed(st.stanza("stream", { xmlns = "jabber:client" }):top_tag())); stream_session.notopen = nil; local data = io.read(4096); while data do stream:feed(data); data = io.read(4096); end assert(stream:feed("</stream>"));