continuwuity/src/admin/room/directory.rs

use clap::Subcommand;
use conduwuit::{Err, Result};
use futures::StreamExt;
use ruma::OwnedRoomId;

use crate::{Context, PAGE_SIZE, get_room_info};

#[derive(Debug, Subcommand)]
pub enum RoomDirectoryCommand {
	/// - Publish a room to the room directory
	Publish {
		/// The room id of the room to publish
		room_id: OwnedRoomId,
	},

	/// - Unpublish a room to the room directory
	Unpublish {
		/// The room id of the room to unpublish
		room_id: OwnedRoomId,
	},

	/// - List rooms that are published
	List {
		page: Option<usize>,
	},
}

pub(super) async fn process(command: RoomDirectoryCommand, context: &Context<'_>) -> Result {
	let services = context.services;
	match command {
		| RoomDirectoryCommand::Publish { room_id } => {
			services.rooms.directory.set_public(&room_id);
			context.write_str("Room published").await
		},
		| RoomDirectoryCommand::Unpublish { room_id } => {
			services.rooms.directory.set_not_public(&room_id);
			context.write_str("Room unpublished").await
		},
		| RoomDirectoryCommand::List { page } => {
			// TODO: i know there's a way to do this with clap, but i can't seem to find it
			let page = page.unwrap_or(1);
			let mut rooms: Vec<_> = services
				.rooms
				.directory
				.public_rooms()
				.then(|room_id| get_room_info(services, room_id))
				.collect()
				.await;

			rooms.sort_by_key(|r| r.1);
			rooms.reverse();

			let rooms: Vec<_> = rooms
				.into_iter()
				.skip(page.saturating_sub(1).saturating_mul(PAGE_SIZE))
				.take(PAGE_SIZE)
				.collect();

			if rooms.is_empty() {
				return Err!("No more rooms.");
			}

			let body = rooms
				.iter()
				.map(|(id, members, name)| format!("{id} | Members: {members} | Name: {name}"))
				.collect::<Vec<_>>()
				.join("\n");

			context
				.write_str(&format!("Rooms (page {page}):\n```\n{body}\n```",))
				.await
		},
	}
}