continuwuity/src/admin/admin.rs

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

use crate::{
	appservice::{self, AppserviceCommand},
	check::{self, CheckCommand},
	context::Context,
	debug::{self, DebugCommand},
	federation::{self, FederationCommand},
	media::{self, MediaCommand},
	query::{self, QueryCommand},
	room::{self, RoomCommand},
	server::{self, ServerCommand},
	token::{self, TokenCommand},
	user::{self, 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 registration tokens
	Token(TokenCommand),

	#[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", level = "info")]
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
		},
		| Token(command) => {
			// token commands are all restricted
			context.bail_restricted()?;
			token::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,
	}
}