mirror of
https://github.com/girlbossceo/conduwuit.git
synced 2025-03-14 18:55:37 +00:00
135 lines
3.4 KiB
Rust
135 lines
3.4 KiB
Rust
use std::fmt::Write;
|
|
|
|
use conduwuit::Result;
|
|
use futures::StreamExt;
|
|
use ruma::{
|
|
events::room::message::RoomMessageEventContent, OwnedRoomId, RoomId, ServerName, UserId,
|
|
};
|
|
|
|
use crate::{admin_command, get_room_info};
|
|
|
|
#[admin_command]
|
|
pub(super) async fn disable_room(&self, room_id: Box<RoomId>) -> Result<RoomMessageEventContent> {
|
|
self.services.rooms.metadata.disable_room(&room_id, true);
|
|
Ok(RoomMessageEventContent::text_plain("Room disabled."))
|
|
}
|
|
|
|
#[admin_command]
|
|
pub(super) async fn enable_room(&self, room_id: Box<RoomId>) -> Result<RoomMessageEventContent> {
|
|
self.services.rooms.metadata.disable_room(&room_id, false);
|
|
Ok(RoomMessageEventContent::text_plain("Room enabled."))
|
|
}
|
|
|
|
#[admin_command]
|
|
pub(super) async fn incoming_federation(&self) -> Result<RoomMessageEventContent> {
|
|
let map = self
|
|
.services
|
|
.rooms
|
|
.event_handler
|
|
.federation_handletime
|
|
.read()
|
|
.expect("locked");
|
|
let mut msg = format!("Handling {} incoming pdus:\n", map.len());
|
|
|
|
for (r, (e, i)) in map.iter() {
|
|
let elapsed = i.elapsed();
|
|
writeln!(msg, "{} {}: {}m{}s", r, e, elapsed.as_secs() / 60, elapsed.as_secs() % 60)?;
|
|
}
|
|
|
|
Ok(RoomMessageEventContent::text_plain(&msg))
|
|
}
|
|
|
|
#[admin_command]
|
|
pub(super) async fn fetch_support_well_known(
|
|
&self,
|
|
server_name: Box<ServerName>,
|
|
) -> Result<RoomMessageEventContent> {
|
|
let response = self
|
|
.services
|
|
.client
|
|
.default
|
|
.get(format!("https://{server_name}/.well-known/matrix/support"))
|
|
.send()
|
|
.await?;
|
|
|
|
let text = response.text().await?;
|
|
|
|
if text.is_empty() {
|
|
return Ok(RoomMessageEventContent::text_plain("Response text/body is empty."));
|
|
}
|
|
|
|
if text.len() > 1500 {
|
|
return Ok(RoomMessageEventContent::text_plain(
|
|
"Response text/body is over 1500 characters, assuming no support well-known.",
|
|
));
|
|
}
|
|
|
|
let json: serde_json::Value = match serde_json::from_str(&text) {
|
|
| Ok(json) => json,
|
|
| Err(_) => {
|
|
return Ok(RoomMessageEventContent::text_plain(
|
|
"Response text/body is not valid JSON.",
|
|
));
|
|
},
|
|
};
|
|
|
|
let pretty_json: String = match serde_json::to_string_pretty(&json) {
|
|
| Ok(json) => json,
|
|
| Err(_) => {
|
|
return Ok(RoomMessageEventContent::text_plain(
|
|
"Response text/body is not valid JSON.",
|
|
));
|
|
},
|
|
};
|
|
|
|
Ok(RoomMessageEventContent::notice_markdown(format!(
|
|
"Got JSON response:\n\n```json\n{pretty_json}\n```"
|
|
)))
|
|
}
|
|
|
|
#[admin_command]
|
|
pub(super) async fn remote_user_in_rooms(
|
|
&self,
|
|
user_id: Box<UserId>,
|
|
) -> Result<RoomMessageEventContent> {
|
|
if user_id.server_name() == self.services.globals.config.server_name {
|
|
return Ok(RoomMessageEventContent::text_plain(
|
|
"User belongs to our server, please use `list-joined-rooms` user admin command \
|
|
instead.",
|
|
));
|
|
}
|
|
|
|
if !self.services.users.exists(&user_id).await {
|
|
return Ok(RoomMessageEventContent::text_plain(
|
|
"Remote user does not exist in our database.",
|
|
));
|
|
}
|
|
|
|
let mut rooms: Vec<(OwnedRoomId, u64, String)> = self
|
|
.services
|
|
.rooms
|
|
.state_cache
|
|
.rooms_joined(&user_id)
|
|
.then(|room_id| get_room_info(self.services, room_id))
|
|
.collect()
|
|
.await;
|
|
|
|
if rooms.is_empty() {
|
|
return Ok(RoomMessageEventContent::text_plain("User is not in any rooms."));
|
|
}
|
|
|
|
rooms.sort_by_key(|r| r.1);
|
|
rooms.reverse();
|
|
|
|
let output = format!(
|
|
"Rooms {user_id} shares with us ({}):\n```\n{}\n```",
|
|
rooms.len(),
|
|
rooms
|
|
.iter()
|
|
.map(|(id, members, name)| format!("{id} | Members: {members} | Name: {name}"))
|
|
.collect::<Vec<_>>()
|
|
.join("\n")
|
|
);
|
|
|
|
Ok(RoomMessageEventContent::text_markdown(output))
|
|
}
|