continuwuity/src/admin/admin.rs

use clap::Parser;
use conduwuit::Result;

use crate::{
	appservice, appservice::AppserviceCommand, check, check::CheckCommand, context::Context,
	debug, debug::DebugCommand, federation, federation::FederationCommand, media,
	media::MediaCommand, query, query::QueryCommand, room, room::RoomCommand, server,
	server::ServerCommand, user, user::UserCommand,
};

#[derive(Debug, Parser)]
#[command(name = conduwuit_core::name(), version = conduwuit_core::version())]
pub enum AdminCommand {
	#[command(subcommand)]
	/// - Commands for managing appservices
	Appservices(AppserviceCommand),

	#[command(subcommand)]
	/// - Commands for managing local users
	Users(UserCommand),

	#[command(subcommand)]
	/// - Commands for managing rooms
	Rooms(RoomCommand),

	#[command(subcommand)]
	/// - Commands for managing federation
	Federation(FederationCommand),

	#[command(subcommand)]
	/// - Commands for managing the server
	Server(ServerCommand),

	#[command(subcommand)]
	/// - Commands for managing media
	Media(MediaCommand),

	#[command(subcommand)]
	/// - Commands for checking integrity
	Check(CheckCommand),

	#[command(subcommand)]
	/// - Commands for debugging things
	Debug(DebugCommand),

	#[command(subcommand)]
	/// - Low-level queries for database getters and iterators
	Query(QueryCommand),
}

#[tracing::instrument(skip_all, name = "command")]
pub(super) async fn process(command: AdminCommand, context: &Context<'_>) -> Result {
	use AdminCommand::*;

	match command {
		| Appservices(command) => {
			// appservice commands are all restricted
			context.bail_restricted()?;
			appservice::process(command, context).await
		},
		| Media(command) => media::process(command, context).await,
		| Users(command) => {
			// user commands are all restricted
			context.bail_restricted()?;
			user::process(command, context).await
		},
		| Rooms(command) => room::process(command, context).await,
		| Federation(command) => federation::process(command, context).await,
		| Server(command) => server::process(command, context).await,
		| Debug(command) => debug::process(command, context).await,
		| Query(command) => {
			// query commands are all restricted
			context.bail_restricted()?;
			query::process(command, context).await
		},
		| Check(command) => check::process(command, context).await,
	}
}