mirrors/prosody
1
0
Fork 0
mirror of https://github.com/bjc/prosody.git synced 2025-04-04 05:37:39 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

mod_csi_simple: Add command to test importance algorithm on stream of stanzas

Browse source 
This won't include behavior provided by extra modules tho.
This commit is contained in:
Kim Alvefur 2021-02-23 19:52:57 +01:00
parent 4ed5700a19
commit 40252c2cc7
1 changed files with 32 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

32
plugins/mod_csi_simple.lua
View file

@ -218,3 +218,35 @@ function module.unload()
end
end
end
function module.command(arg)
if arg[1] ~= "test" then
print("Usage: "..module.name.." test < test-stream.xml")
print("");
print("Provide a series of stanzas to test against importance algoritm");
return 1;
end
-- luacheck: ignore 212/self
local xmppstream = require "util.xmppstream";
local input_session = { notopen = true }
local stream_callbacks = { stream_ns = "jabber:client", default_ns = "jabber:client" };
function stream_callbacks:handlestanza(stanza)
local important, because = is_important(stanza);
print("--");
print(stanza:indent(nil, " "));
-- :pretty_print() maybe?
if important then
print((because or "unspecified reason").. " -> important");
else
print((because or "unspecified reason").. " -> unimportant");
end
end
local input_stream = xmppstream.new(input_session, stream_callbacks);
input_stream:reset();
input_stream:feed(st.stanza("stream", { xmlns = "jabber:client" }):top_tag());
input_session.notopen = nil;
for line in io.lines() do
input_stream:feed(line);
end
end