From 4ceca3f87e0d60773fdb9395d4077c7310ca0423 Mon Sep 17 00:00:00 2001
From: Martin/Geno <geno+dev@fireorbit.de>
Date: Thu, 17 Oct 2019 09:13:00 +0200
Subject: s2s_in/out + muc

---
 ejabberdrpc.py | 59 +++++++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 56 insertions(+), 3 deletions(-)

(limited to 'ejabberdrpc.py')

diff --git a/ejabberdrpc.py b/ejabberdrpc.py
index 4dd089b..180579e 100755
--- a/ejabberdrpc.py
+++ b/ejabberdrpc.py
@@ -13,9 +13,12 @@ class EjabberdMetrics():
 
     def _cmd(self, command, data):
         fn = getattr(self._server, command)
-        if self._login is not None:
-            return fn(self._login, data)
-        return fn(data)
+        try:
+            if self._login is not None:
+                return fn(self._login, data)
+            return fn(data)
+        except:
+            return {}
 
     def fetch_onlineuser(self):
         data = None
@@ -51,6 +54,18 @@ class EjabberdMetrics():
             data.append(vhost["vhost"])
         return data
 
+    def fetch_s2s_in(self):
+        result = self._cmd("incoming_s2s_number",{})
+        if "s2s_incoming" not in result:
+            return None
+        return result["s2s_incoming"]
+
+    def fetch_s2s_out(self):
+        result = self._cmd("outgoing_s2s_number",{})
+        if "s2s_outgoing" not in result:
+            return None
+        return result["s2s_outgoing"]
+
     def fetch_registered(self, vhost=None):
         if vhost is None:
             result = self._cmd("stats", {"name":"registeredusers"})
@@ -61,6 +76,14 @@ class EjabberdMetrics():
             if "stat" in result:
                 return result["stat"]
 
+    def fetch_muc(self, vhost=None):
+        host = "global"
+        if vhost is not None:
+            host = "conference." + vhost
+        result = self._cmd("muc_online_rooms", {"host": host})
+        if "rooms" in result:
+            return len(result["rooms"])
+
     def update(self):
         # nodes
         self._nodes = self.fetch_nodes()
@@ -70,10 +93,19 @@ class EjabberdMetrics():
         if not hasattr(self, "_registered"):
             self._registered = {}
         self._registered[None] = self.fetch_registered()
+        # muc
+        if not hasattr(self, "_muc"):
+            self._muc = {}
+        self._muc[None] = self.fetch_muc()
+        # registered + muc
         for vhost in self._vhosts:
             self._registered[vhost] = self.fetch_registered(vhost)
+            self._muc[vhost] = self.fetch_muc(vhost)
         # online user
         self._onlineuser = self.fetch_onlineuser()
+        # s2s
+        self._s2s_in = self.fetch_s2s_in()
+        self._s2s_out = self.fetch_s2s_out()
 
 
     def get_online_by(self, by="node", parse=None, vhost=None, node=None):
@@ -147,14 +179,32 @@ class EjabberdMetrics():
             self._registered[vhost] = self.fetch_registered(vhost)
         return self._registered[vhost]
 
+    def get_muc(self, vhost=None):
+        if not hasattr(self, "_muc"):
+            self._muc = {}
+        if vhost not in self._muc:
+            self._muc[vhost] = self.fetch_muc(vhost)
+        return self._muc[vhost]
+
     def get_vhosts(self):
         if not hasattr(self, "_vhosts"):
             self._vhosts = self.fetch_vhosts()
         return self._vhosts
 
+    def get_s2s_in(self):
+        if not hasattr(self, "_s2s_in"):
+            self._s2s_in = self.fetch_s2s_in()
+        return self._s2s_in
+
+    def get_s2s_out(self):
+        if not hasattr(self, "_s2s_out"):
+            self._s2s_out = self.fetch_s2s_out()
+        return self._s2s_out
+
     def get_vhost_metrics(self, vhost): 
         data = {}
         data["registered"] = self.get_registered(vhost)
+        data["muc"] = self.get_muc(vhost)
         data["online_by_status"] = self.get_online_by_status(vhost)
         data["online_by_client"] = self.get_online_by_client(vhost)
         data["online_by_ipversion"] = self.get_online_by_ipversion(vhost)
@@ -181,6 +231,7 @@ class EjabberdMetrics():
     def get_all(self):
         data = {}
         data["registered"] = self.get_registered()
+        data["muc"] = self.get_muc()
         data["online_by_status"] = self.get_online_by_status()
         data["online_by_client"] = self.get_online_by_client()
         data["online_by_ipversion"] = self.get_online_by_ipversion()
@@ -199,6 +250,8 @@ class EjabberdMetrics():
             nodes[node] = self.get_node_metrics(node)
         data["nodes"] = nodes
         
+        data["s2s_in"] = self.get_s2s_in()
+        data["s2s_out"] = self.get_s2s_out()
         return data
 
 
-- 
cgit v1.2.3-54-g00ecf