88 lines
2.1 KiB
Rust
88 lines
2.1 KiB
Rust
use crate::database::*;
|
|
use crate::util::result::{Error, Result};
|
|
|
|
use futures::StreamExt;
|
|
use mongodb::options::FindOptions;
|
|
use serde::{Serialize, Deserialize};
|
|
use rocket::serde::json::Value;
|
|
use mongodb::bson::{doc, from_document};
|
|
|
|
#[derive(Serialize, Deserialize)]
|
|
struct BannedUser {
|
|
_id: String,
|
|
username: String,
|
|
avatar: Option<File>
|
|
}
|
|
|
|
#[get("/<target>/bans")]
|
|
pub async fn req(user: User, target: Ref) -> Result<Value> {
|
|
let target = target.fetch_server().await?;
|
|
|
|
let perm = permissions::PermissionCalculator::new(&user)
|
|
.with_server(&target)
|
|
.for_server()
|
|
.await?;
|
|
|
|
if !perm.get_ban_members() {
|
|
return Err(Error::MissingPermission);
|
|
}
|
|
|
|
let mut cursor = get_collection("server_bans")
|
|
.find(
|
|
doc! {
|
|
"_id.server": target.id
|
|
},
|
|
None,
|
|
)
|
|
.await
|
|
.map_err(|_| Error::DatabaseError {
|
|
operation: "find",
|
|
with: "server_bans",
|
|
})?;
|
|
|
|
let mut bans = vec![];
|
|
let mut user_ids = vec![];
|
|
while let Some(result) = cursor.next().await {
|
|
if let Ok(doc) = result {
|
|
if let Ok(ban) = from_document::<Ban>(doc) {
|
|
user_ids.push(ban.id.user.clone());
|
|
bans.push(ban);
|
|
}
|
|
}
|
|
}
|
|
|
|
let mut cursor = get_collection("users")
|
|
.find(
|
|
doc! {
|
|
"_id": {
|
|
"$in": user_ids
|
|
}
|
|
},
|
|
FindOptions::builder()
|
|
.projection(doc! {
|
|
"username": 1,
|
|
"avatar": 1
|
|
})
|
|
.build(),
|
|
)
|
|
.await
|
|
.map_err(|_| Error::DatabaseError {
|
|
operation: "find",
|
|
with: "users",
|
|
})?;
|
|
|
|
let mut users = vec![];
|
|
while let Some(result) = cursor.next().await {
|
|
if let Ok(doc) = result {
|
|
if let Ok(user) = from_document::<BannedUser>(doc) {
|
|
users.push(user);
|
|
}
|
|
}
|
|
}
|
|
|
|
Ok(json!({
|
|
"users": users,
|
|
"bans": bans
|
|
}))
|
|
}
|