forked from jmug/stoatchat
113 lines
3.5 KiB
Rust
113 lines
3.5 KiB
Rust
use revolt_database::{
|
|
util::{permissions::DatabasePermissionQuery, reference::Reference}, voice::{delete_voice_state, get_channel_node, get_voice_channel_members, VoiceClient}, Channel, Database, PartialChannel, User, AMQP
|
|
};
|
|
use revolt_models::v0;
|
|
use revolt_permissions::{calculate_channel_permissions, ChannelPermission};
|
|
use revolt_result::{create_error, Result};
|
|
use rocket::State;
|
|
use rocket_empty::EmptyResponse;
|
|
|
|
/// # Close Channel
|
|
///
|
|
/// Deletes a server channel, leaves a group or closes a group.
|
|
#[openapi(tag = "Channel Information")]
|
|
#[delete("/<target>?<options..>")]
|
|
pub async fn delete(
|
|
db: &State<Database>,
|
|
voice_client: &State<VoiceClient>,
|
|
amqp: &State<AMQP>,
|
|
user: User,
|
|
target: Reference,
|
|
options: v0::OptionsChannelDelete,
|
|
) -> Result<EmptyResponse> {
|
|
let mut channel = target.as_channel(db).await?;
|
|
let mut query = DatabasePermissionQuery::new(db, &user).channel(&channel);
|
|
let permissions = calculate_channel_permissions(&mut query).await;
|
|
|
|
permissions.throw_if_lacking_channel_permission(ChannelPermission::ViewChannel)?;
|
|
|
|
match &channel {
|
|
Channel::SavedMessages { .. } => Err(create_error!(NoEffect))?,
|
|
Channel::DirectMessage { .. } => channel
|
|
.update(
|
|
db,
|
|
PartialChannel {
|
|
active: Some(false),
|
|
..Default::default()
|
|
},
|
|
vec![],
|
|
)
|
|
.await?,
|
|
Channel::Group { .. } => channel
|
|
.remove_user_from_group(
|
|
db,
|
|
amqp,
|
|
&user,
|
|
None,
|
|
options.leave_silently.unwrap_or_default(),
|
|
)
|
|
.await?,
|
|
Channel::TextChannel { .. } | Channel::VoiceChannel { .. } => {
|
|
permissions.throw_if_lacking_channel_permission(ChannelPermission::ManageChannel)?;
|
|
channel.delete(db).await?
|
|
}
|
|
};
|
|
|
|
if let Some(users) = get_voice_channel_members(channel.id()).await? {
|
|
let node = get_channel_node(channel.id()).await?;
|
|
|
|
for user in users {
|
|
voice_client.remove_user(&node, &user, channel.id()).await?;
|
|
delete_voice_state(channel.id(), channel.server(), &user).await?;
|
|
}
|
|
};
|
|
|
|
Ok(EmptyResponse)
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod test {
|
|
use crate::{rocket, util::test::TestHarness};
|
|
use revolt_database::{events::client::EventV1, Channel};
|
|
use revolt_models::v0::DataCreateGroup;
|
|
use rocket::http::{Header, Status};
|
|
|
|
#[rocket::async_test]
|
|
async fn success_delete_group() {
|
|
let mut harness = TestHarness::new().await;
|
|
let (_, session, user) = harness.new_user().await;
|
|
|
|
let group = Channel::create_group(
|
|
&harness.db,
|
|
DataCreateGroup {
|
|
..Default::default()
|
|
},
|
|
user.id.clone(),
|
|
)
|
|
.await
|
|
.expect("`Channel`");
|
|
|
|
let response = harness
|
|
.client
|
|
.delete(format!("/channels/{}", group.id()))
|
|
.header(Header::new("x-session-token", session.token.to_string()))
|
|
.dispatch()
|
|
.await;
|
|
|
|
assert_eq!(response.status(), Status::NoContent);
|
|
drop(response);
|
|
|
|
harness
|
|
.wait_for_event(group.id(), |event| match event {
|
|
EventV1::ChannelDelete { id, .. } => id == group.id(),
|
|
_ => false,
|
|
})
|
|
.await;
|
|
}
|
|
|
|
// TEST: member leaves group (no delete)
|
|
// TEST: no effect with saved messages
|
|
// TEST: DM set to inactive
|
|
// TEST: server channel deleted
|
|
}
|