From 7c1b2625cf8f315bced5e560574c0c64eedd368f Mon Sep 17 00:00:00 2001
From: Torsten Flammiger <git@netfg.net>
Date: Fri, 24 Dec 2021 23:06:54 +0100
Subject: [PATCH] Prepare to add an option to list local user accounts from
 your homeserver

---
 src/database/admin.rs | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/database/admin.rs b/src/database/admin.rs
index 1e5c47c9..5ea872e8 100644
--- a/src/database/admin.rs
+++ b/src/database/admin.rs
@@ -13,6 +13,7 @@ use tracing::warn;
 pub enum AdminCommand {
     RegisterAppservice(serde_yaml::Value),
     ListAppservices,
+    CountUsers,
     SendMessage(RoomMessageEventContent),
 }
 
@@ -93,6 +94,16 @@ impl Admin {
                         let state_lock = mutex_state.lock().await;
 
                         match event {
+                            AdminCommand::CountUsers => {
+                                // count() does not return an error on failure...
+                                if let Ok(usercount) = guard.users.count() {
+                                    let message = format!("Found {} total user accounts", usercount);
+                                    send_message(RoomMessageEventContent::text_plain(message), guard, &state_lock);
+                                } else {
+                                    // ... so we simply spit out a generic non-explaining-info in case count() did not return Ok()
+                                    send_message(RoomMessageEventContent::text_plain("Unable to count users"), guard, &state_lock);
+                                }
+                            }
                             AdminCommand::RegisterAppservice(yaml) => {
                                 guard.appservice.register_appservice(yaml).unwrap(); // TODO handle error
                             }