continuwuity/src/service/admin/grant.rs

use std::collections::BTreeMap;

use conduwuit::{error, implement, Result};
use ruma::{
	events::{
		room::{
			member::{MembershipState, RoomMemberEventContent},
			message::RoomMessageEventContent,
			power_levels::RoomPowerLevelsEventContent,
		},
		tag::{TagEvent, TagEventContent, TagInfo},
		RoomAccountDataEventType,
	},
	RoomId, UserId,
};

use crate::pdu::PduBuilder;

/// Invite the user to the conduwuit admin room.
///
/// This is equivalent to granting server admin privileges.
#[implement(super::Service)]
pub async fn make_user_admin(&self, user_id: &UserId) -> Result<()> {
	let Ok(room_id) = self.get_admin_room().await else {
		return Ok(());
	};

	let state_lock = self.services.state.mutex.lock(&room_id).await;

	// Use the server user to grant the new admin's power level
	let server_user = &self.services.globals.server_user;

	// Invite and join the real user
	self.services
		.timeline
		.build_and_append_pdu(
			PduBuilder::state(
				user_id.to_string(),
				&RoomMemberEventContent::new(MembershipState::Invite),
			),
			server_user,
			&room_id,
			&state_lock,
		)
		.await?;
	self.services
		.timeline
		.build_and_append_pdu(
			PduBuilder::state(
				user_id.to_string(),
				&RoomMemberEventContent::new(MembershipState::Join),
			),
			user_id,
			&room_id,
			&state_lock,
		)
		.await?;

	// Set power level
	let users = BTreeMap::from_iter([
		(server_user.clone(), 100.into()),
		(user_id.to_owned(), 100.into()),
	]);

	self.services
		.timeline
		.build_and_append_pdu(
			PduBuilder::state(String::new(), &RoomPowerLevelsEventContent {
				users,
				..Default::default()
			}),
			server_user,
			&room_id,
			&state_lock,
		)
		.await?;

	// Set room tag
	let room_tag = &self.services.server.config.admin_room_tag;
	if !room_tag.is_empty() {
		if let Err(e) = self.set_room_tag(&room_id, user_id, room_tag).await {
			error!(?room_id, ?user_id, ?room_tag, ?e, "Failed to set tag for admin grant");
		}
	}

	if self.services.server.config.admin_room_notices {
		let welcome_message = String::from("## Thank you for trying out conduwuit!\n\nconduwuit is technically a hard fork of Conduit, which is in Beta. The Beta status initially was inherited from Conduit, however overtime this Beta status is rapidly becoming less and less relevant as our codebase significantly diverges more and more. conduwuit is quite stable and very usable as a daily driver and for a low-medium sized homeserver. There is still a lot of more work to be done, but it is in a far better place than the project was in early 2024.\n\nHelpful links:\n> GitHub Repo: https://github.com/girlbossceo/conduwuit\n> Documentation: https://conduwuit.puppyirl.gay/\n> Report issues: https://github.com/girlbossceo/conduwuit/issues\n\nFor a list of available commands, send the following message in this room: `!admin --help`\n\nHere are some rooms you can join (by typing the command into your client) -\n\nconduwuit space: `/join #conduwuit-space:puppygock.gay`\nconduwuit main room (Ask questions and get notified on updates): `/join #conduwuit:puppygock.gay`\nconduwuit offtopic room: `/join #conduwuit-offtopic:puppygock.gay`");

		// Send welcome message
		self.services
			.timeline
			.build_and_append_pdu(
				PduBuilder::timeline(&RoomMessageEventContent::text_markdown(welcome_message)),
				server_user,
				&room_id,
				&state_lock,
			)
			.await?;
	}

	Ok(())
}

#[implement(super::Service)]
async fn set_room_tag(&self, room_id: &RoomId, user_id: &UserId, tag: &str) -> Result<()> {
	let mut event = self
		.services
		.account_data
		.get_room(room_id, user_id, RoomAccountDataEventType::Tag)
		.await
		.unwrap_or_else(|_| TagEvent {
			content: TagEventContent { tags: BTreeMap::new() },
		});

	event
		.content
		.tags
		.insert(tag.to_owned().into(), TagInfo::new());

	self.services
		.account_data
		.update(
			Some(room_id),
			user_id,
			RoomAccountDataEventType::Tag,
			&serde_json::to_value(event)?,
		)
		.await?;

	Ok(())
}