# HG changeset patch
# User Kim Alvefur <zash@zash.se>
# Date 1680760915 -7200
# Node ID b7d0c1d75a37f2d4dc71c02d95924279c12b28f0
# Parent  7558fd152459bef3e362fc18510ecdb6dcf9c6aa
mod_csi: Add metrics, covering changes and totals

Motivation: Investigating clients that seem to forget to set CSI.
Also, of course, MORE GRAPHS!

diff -r 7558fd152459 -r b7d0c1d75a37 plugins/mod_csi.lua
--- a/plugins/mod_csi.lua	Wed Apr 05 15:33:56 2023 +0100
+++ b/plugins/mod_csi.lua	Thu Apr 06 08:01:55 2023 +0200
@@ -2,6 +2,17 @@
 local xmlns_csi = "urn:xmpp:csi:0";
 local csi_feature = st.stanza("csi", { xmlns = xmlns_csi });
 
+local sum = module:metric("gauge", "sessions_per_state", "sessions", "CSI state per session", { "csi_state" })
+local change = module:metric("counter", "changes", "events", "CSI state changes", {"csi_state"});
+
+module:hook_global("stats-update", function ()
+	for _, session in pairs(prosody.full_sessions) do
+		if session.host == module.host then
+			sum:with_labels(session.state or "none"):add(1);
+		end
+	end
+end);
+
 local csi_handler_available = nil;
 module:hook("stream-features", function (event)
 	if event.origin.username and csi_handler_available then
@@ -13,6 +24,7 @@
 	return function (event)
 		if event.origin.username then
 			event.origin.state = event.stanza.name;
+			change:with_labels(event.stanza.name):add(1);
 			module:fire_event(name, event);
 			return true;
 		end