Comparison

mod_admin_shell: Add watch:stanzas() command

comparison

equal deleted inserted replaced

-:788048158982
+:f4c59af273a3
 	if not logger.remove_sink(sink) then
 		module:log("warn", "Unable to remove watch:log() sink");
 	end
 end
+local stanza_watchers = module:require("mod_debug_stanzas/watcher");
+function def_env.watch:stanzas(target_spec, filter_spec)
+	local function handler(event_type, stanza, session)
+		if stanza then
+			if event_type == "sent" then
+				self.session.print(("\n<!-- sent to %s -->"):format(session.id));
+			elseif event_type == "received" then
+				self.session.print(("\n<!-- received from %s -->"):format(session.id));
+			else
+				self.session.print(("\n<!-- %s (%s) -->"):format(event_type, session.id));
+			end
+			self.session.print(stanza);
+		elseif session then
+			self.session.print("\n<!-- session "..session.id.." "..event_type.." -->");
+		elseif event_type then
+			self.session.print("\n<!-- "..event_type.." -->");
+		end
+	end
+	stanza_watchers.add({
+		target_spec = {
+			jid = target_spec;
+		};
+		filter_spec = filter_spec and {
+			with_jid = filter_spec;
+		};
+	}, handler);
+	while self.session.is_connected() do
+		async.sleep(3);
+	end
+	stanza_watchers.remove(handler);
+end
 def_env.debug = {};
 function def_env.debug:logevents(host)
 	helpers.log_host_events(host);
 	end
 	return displayed_stats;
 end
+function module.unload()
+	stanza_watchers.cleanup();
+end
 -------------
 function printbanner(session)
 	local option = module:get_option_string("console_banner", "full");