use clap::Subcommand;
use conduwuit::Result;
use ruma::{OwnedEventId, OwnedRoomOrAliasId};

use crate::{admin_command, admin_command_dispatch};

#[admin_command_dispatch]
#[derive(Debug, Subcommand)]
/// Query tables from database
pub(crate) enum ShortCommand {
	ShortEventId {
		event_id: OwnedEventId,
	},

	ShortRoomId {
		room_id: OwnedRoomOrAliasId,
	},
}

#[admin_command]
pub(super) async fn short_event_id(&self, event_id: OwnedEventId) -> Result {
	let shortid = self
		.services
		.rooms
		.short
		.get_shorteventid(&event_id)
		.await?;

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

#[admin_command]
pub(super) async fn short_room_id(&self, room_id: OwnedRoomOrAliasId) -> Result {
	let room_id = self.services.rooms.alias.resolve(&room_id).await?;

	let shortid = self.services.rooms.short.get_shortroomid(&room_id).await?;

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