continuwuity/src/admin/query/room_timeline.rs

use clap::Subcommand;
use conduwuit::{PduCount, Result, utils::stream::TryTools};
use futures::TryStreamExt;
use ruma::OwnedRoomOrAliasId;

use crate::admin_command_dispatch;

#[admin_command_dispatch]
#[derive(Debug, Subcommand)]
/// Query tables from database
pub enum RoomTimelineCommand {
	Pdus {
		room_id: OwnedRoomOrAliasId,

		from: Option<String>,

		#[arg(short, long)]
		limit: Option<usize>,
	},

	Last {
		room_id: OwnedRoomOrAliasId,
	},
}

impl crate::Context<'_> {
	pub(super) async fn last(&self, room_id: OwnedRoomOrAliasId) -> Result {
		let room_id = self.services.rooms.alias.resolve(&room_id).await?;

		let result = self
			.services
			.rooms
			.timeline
			.last_timeline_count(&room_id)
			.await?;

		self.write_str(&format!("{result:#?}")).await
	}

	pub(super) async fn pdus(
		&self,
		room_id: OwnedRoomOrAliasId,
		from: Option<String>,
		limit: Option<usize>,
	) -> Result {
		let room_id = self.services.rooms.alias.resolve(&room_id).await?;

		let from: Option<PduCount> = from.as_deref().map(str::parse).transpose()?;

		let result: Vec<_> = self
			.services
			.rooms
			.timeline
			.pdus_rev(&room_id, from)
			.try_take(limit.unwrap_or(3))
			.try_collect()
			.await?;

		self.write_str(&format!("```\n{result:#?}\n```")).await
	}
}