refactor a ton of the admin room code (50% done)

Signed-off-by: strawberry <strawberry@puppygock.gay>

src/service/admin/server/mod.rs

@@ -0,0 +1,58 @@
+pub(crate) mod server_commands;
+
+use clap::Subcommand;
+use ruma::events::room::message::RoomMessageEventContent;
+
+use self::server_commands::{
+	backup_database, clear_database_caches, clear_service_caches, list_backups, list_database_files, memory_usage,
+	show_config,
+};
+use crate::Result;
+
+#[cfg_attr(test, derive(Debug))]
+#[derive(Subcommand)]
+pub(crate) enum ServerCommand {
+	/// - Show configuration values
+	ShowConfig,
+
+	/// - Print database memory usage statistics
+	MemoryUsage,
+
+	/// - Clears all of Conduit's database caches with index smaller than the
+	///   amount
+	ClearDatabaseCaches {
+		amount: u32,
+	},
+
+	/// - Clears all of Conduit's service caches with index smaller than the
+	///   amount
+	ClearServiceCaches {
+		amount: u32,
+	},
+
+	/// - Performs an online backup of the database (only available for RocksDB
+	///   at the moment)
+	BackupDatabase,
+
+	/// - List database backups
+	ListBackups,
+
+	/// - List database files
+	ListDatabaseFiles,
+}
+
+pub(crate) async fn process(command: ServerCommand, body: Vec<&str>) -> Result<RoomMessageEventContent> {
+	Ok(match command {
+		ServerCommand::ShowConfig => show_config(body).await?,
+		ServerCommand::MemoryUsage => memory_usage(body).await?,
+		ServerCommand::ClearDatabaseCaches {
+			amount,
+		} => clear_database_caches(body, amount).await?,
+		ServerCommand::ClearServiceCaches {
+			amount,
+		} => clear_service_caches(body, amount).await?,
+		ServerCommand::ListBackups => list_backups(body).await?,
+		ServerCommand::BackupDatabase => backup_database(body).await?,
+		ServerCommand::ListDatabaseFiles => list_database_files(body).await?,
+	})
+}

src/service/admin/server/server_commands.rs

@@ -0,0 +1,71 @@
+use ruma::events::room::message::RoomMessageEventContent;
+
+use crate::{services, Result};
+
+pub(super) async fn show_config(_body: Vec<&str>) -> Result<RoomMessageEventContent> {
+	// Construct and send the response
+	Ok(RoomMessageEventContent::text_plain(format!("{}", services().globals.config)))
+}
+
+pub(super) async fn memory_usage(_body: Vec<&str>) -> Result<RoomMessageEventContent> {
+	let response1 = services().memory_usage().await;
+	let response2 = services().globals.db.memory_usage();
+
+	Ok(RoomMessageEventContent::text_plain(format!(
+		"Services:\n{response1}\n\nDatabase:\n{response2}"
+	)))
+}
+
+pub(super) async fn clear_database_caches(_body: Vec<&str>, amount: u32) -> Result<RoomMessageEventContent> {
+	services().globals.db.clear_caches(amount);
+
+	Ok(RoomMessageEventContent::text_plain("Done."))
+}
+
+pub(super) async fn clear_service_caches(_body: Vec<&str>, amount: u32) -> Result<RoomMessageEventContent> {
+	services().clear_caches(amount).await;
+
+	Ok(RoomMessageEventContent::text_plain("Done."))
+}
+
+pub(super) async fn list_backups(_body: Vec<&str>) -> Result<RoomMessageEventContent> {
+	let result = services().globals.db.backup_list()?;
+
+	if result.is_empty() {
+		Ok(RoomMessageEventContent::text_plain("No backups found."))
+	} else {
+		Ok(RoomMessageEventContent::text_plain(result))
+	}
+}
+
+pub(super) async fn backup_database(_body: Vec<&str>) -> Result<RoomMessageEventContent> {
+	if !cfg!(feature = "rocksdb") {
+		return Ok(RoomMessageEventContent::text_plain(
+			"Only RocksDB supports online backups in conduwuit.",
+		));
+	}
+
+	let mut result = tokio::task::spawn_blocking(move || match services().globals.db.backup() {
+		Ok(()) => String::new(),
+		Err(e) => (*e).to_string(),
+	})
+	.await
+	.unwrap();
+
+	if result.is_empty() {
+		result = services().globals.db.backup_list()?;
+	}
+
+	Ok(RoomMessageEventContent::text_plain(&result))
+}
+
+pub(super) async fn list_database_files(_body: Vec<&str>) -> Result<RoomMessageEventContent> {
+	if !cfg!(feature = "rocksdb") {
+		return Ok(RoomMessageEventContent::text_plain(
+			"Only RocksDB supports listing files in conduwuit.",
+		));
+	}
+
+	let result = services().globals.db.file_list()?;
+	Ok(RoomMessageEventContent::notice_html(String::new(), result))
+}