continuwuity/src/admin/room/alias.rs

use std::fmt::Write;

use clap::Subcommand;
use conduwuit::Result;
use futures::StreamExt;
use ruma::{
	OwnedRoomAliasId, OwnedRoomId, RoomId, events::room::message::RoomMessageEventContent,
};

use crate::{Command, escape_html};

#[derive(Debug, Subcommand)]
pub(crate) enum RoomAliasCommand {
	/// - Make an alias point to a room.
	Set {
		#[arg(short, long)]
		/// Set the alias even if a room is already using it
		force: bool,

		/// The room id to set the alias on
		room_id: Box<RoomId>,

		/// The alias localpart to use (`alias`, not `#alias:servername.tld`)
		room_alias_localpart: String,
	},

	/// - Remove a local alias
	Remove {
		/// The alias localpart to remove (`alias`, not `#alias:servername.tld`)
		room_alias_localpart: String,
	},

	/// - Show which room is using an alias
	Which {
		/// The alias localpart to look up (`alias`, not
		/// `#alias:servername.tld`)
		room_alias_localpart: String,
	},

	/// - List aliases currently being used
	List {
		/// If set, only list the aliases for this room
		room_id: Option<Box<RoomId>>,
	},
}

pub(super) async fn process(command: RoomAliasCommand, context: &Command<'_>) -> Result {
	let c = reprocess(command, context).await?;
	context.write_str(c.body()).await?;

	Ok(())
}

pub(super) async fn reprocess(
	command: RoomAliasCommand,
	context: &Command<'_>,
) -> Result<RoomMessageEventContent> {
	let services = context.services;
	let server_user = &services.globals.server_user;

	match command {
		| RoomAliasCommand::Set { ref room_alias_localpart, .. }
		| RoomAliasCommand::Remove { ref room_alias_localpart }
		| RoomAliasCommand::Which { ref room_alias_localpart } => {
			let room_alias_str =
				format!("#{}:{}", room_alias_localpart, services.globals.server_name());
			let room_alias = match OwnedRoomAliasId::parse(room_alias_str) {
				| Ok(alias) => alias,
				| Err(err) => {
					return Ok(RoomMessageEventContent::text_plain(format!(
						"Failed to parse alias: {err}"
					)));
				},
			};
			match command {
				| RoomAliasCommand::Set { force, room_id, .. } => {
					match (force, services.rooms.alias.resolve_local_alias(&room_alias).await) {
						| (true, Ok(id)) => {
							match services.rooms.alias.set_alias(
								&room_alias,
								&room_id,
								server_user,
							) {
								| Ok(()) => Ok(RoomMessageEventContent::text_plain(format!(
									"Successfully overwrote alias (formerly {id})"
								))),
								| Err(err) => Ok(RoomMessageEventContent::text_plain(format!(
									"Failed to remove alias: {err}"
								))),
							}
						},
						| (false, Ok(id)) => Ok(RoomMessageEventContent::text_plain(format!(
							"Refusing to overwrite in use alias for {id}, use -f or --force to \
							 overwrite"
						))),
						| (_, Err(_)) => {
							match services.rooms.alias.set_alias(
								&room_alias,
								&room_id,
								server_user,
							) {
								| Ok(()) => Ok(RoomMessageEventContent::text_plain(
									"Successfully set alias",
								)),
								| Err(err) => Ok(RoomMessageEventContent::text_plain(format!(
									"Failed to remove alias: {err}"
								))),
							}
						},
					}
				},
				| RoomAliasCommand::Remove { .. } => {
					match services.rooms.alias.resolve_local_alias(&room_alias).await {
						| Ok(id) => match services
							.rooms
							.alias
							.remove_alias(&room_alias, server_user)
							.await
						{
							| Ok(()) => Ok(RoomMessageEventContent::text_plain(format!(
								"Removed alias from {id}"
							))),
							| Err(err) => Ok(RoomMessageEventContent::text_plain(format!(
								"Failed to remove alias: {err}"
							))),
						},
						| Err(_) =>
							Ok(RoomMessageEventContent::text_plain("Alias isn't in use.")),
					}
				},
				| RoomAliasCommand::Which { .. } => {
					match services.rooms.alias.resolve_local_alias(&room_alias).await {
						| Ok(id) => Ok(RoomMessageEventContent::text_plain(format!(
							"Alias resolves to {id}"
						))),
						| Err(_) =>
							Ok(RoomMessageEventContent::text_plain("Alias isn't in use.")),
					}
				},
				| RoomAliasCommand::List { .. } => unreachable!(),
			}
		},
		| RoomAliasCommand::List { room_id } =>
			if let Some(room_id) = room_id {
				let aliases: Vec<OwnedRoomAliasId> = services
					.rooms
					.alias
					.local_aliases_for_room(&room_id)
					.map(Into::into)
					.collect()
					.await;

				let plain_list = aliases.iter().fold(String::new(), |mut output, alias| {
					writeln!(output, "- {alias}")
						.expect("should be able to write to string buffer");
					output
				});

				let html_list = aliases.iter().fold(String::new(), |mut output, alias| {
					writeln!(output, "<li>{}</li>", escape_html(alias.as_ref()))
						.expect("should be able to write to string buffer");
					output
				});

				let plain = format!("Aliases for {room_id}:\n{plain_list}");
				let html = format!("Aliases for {room_id}:\n<ul>{html_list}</ul>");
				Ok(RoomMessageEventContent::text_html(plain, html))
			} else {
				let aliases = services
					.rooms
					.alias
					.all_local_aliases()
					.map(|(room_id, localpart)| (room_id.into(), localpart.into()))
					.collect::<Vec<(OwnedRoomId, String)>>()
					.await;

				let server_name = services.globals.server_name();
				let plain_list = aliases
					.iter()
					.fold(String::new(), |mut output, (alias, id)| {
						writeln!(output, "- `{alias}` -> #{id}:{server_name}")
							.expect("should be able to write to string buffer");
						output
					});

				let html_list = aliases
					.iter()
					.fold(String::new(), |mut output, (alias, id)| {
						writeln!(
							output,
							"<li><code>{}</code> -> #{}:{}</li>",
							escape_html(alias.as_ref()),
							escape_html(id),
							server_name
						)
						.expect("should be able to write to string buffer");
						output
					});

				let plain = format!("Aliases:\n{plain_list}");
				let html = format!("Aliases:\n<ul>{html_list}</ul>");
				Ok(RoomMessageEventContent::text_html(plain, html))
			},
	}
}