refactor(quark): port member_edit, member_experimental_query, member_fetch_all, member_remove, permissions_set_default, permissions_set

#283
This commit is contained in:
Paul Makles
2024-04-07 21:35:45 +01:00
parent 03340a2b5a
commit 9de963fb56
14 changed files with 308 additions and 216 deletions

View File

@@ -1,9 +1,12 @@
use revolt_quark::{
models::{Member, User},
perms, Db, Error, Ref, Result,
use futures::future::join_all;
use revolt_database::{
util::{permissions::DatabasePermissionQuery, reference::Reference},
Database, Member, User,
};
use rocket::serde::json::Json;
use revolt_models::v0;
use revolt_permissions::PermissionQuery;
use revolt_result::{create_error, Result};
use rocket::{serde::json::Json, State};
use serde::{Deserialize, Serialize};
/// # Query Parameters
@@ -20,9 +23,9 @@ pub struct OptionsQueryMembers {
#[derive(Serialize, JsonSchema)]
pub struct MemberQueryResponse {
/// List of members
members: Vec<Member>,
members: Vec<v0::Member>,
/// List of users
users: Vec<User>,
users: Vec<v0::User>,
}
/// # Query members by name
@@ -31,17 +34,20 @@ pub struct MemberQueryResponse {
#[openapi(tag = "Server Members")]
#[get("/<target>/members_experimental_query?<options..>")]
pub async fn member_experimental_query(
db: &Db,
db: &State<Database>,
user: User,
target: Ref,
target: Reference,
options: OptionsQueryMembers,
) -> Result<Json<MemberQueryResponse>> {
if !options.experimental_api {
return Err(Error::InternalError);
return Err(create_error!(InternalError));
}
let server = target.as_server(db).await?;
perms(&user).server(&server).calc(db).await?;
let mut query = DatabasePermissionQuery::new(db, &user).server(&server);
if !query.are_we_a_member().await {
return Err(create_error!(NotFound));
}
let mut members = db.fetch_all_members(&server.id).await?;
@@ -50,7 +56,7 @@ pub async fn member_experimental_query(
user_ids.push(member.id.user.clone());
}
let mut users = User::fetch_foreign_users(db, &user_ids).await?;
let mut users = db.fetch_users(&user_ids).await?;
// Ensure the lists match up exactly
members.sort_by(|a, b| a.id.user.cmp(&b.id.user));
@@ -76,7 +82,15 @@ pub async fn member_experimental_query(
left.len().cmp(&right.len())
});
// Take the first five and return them
let (members, users) = zipped_vec.into_iter().take(10).unzip();
Ok(Json(MemberQueryResponse { members, users }))
// Take the first ten and return them
let (members, users): (Vec<Member>, Vec<User>) = zipped_vec.into_iter().take(10).unzip();
Ok(Json(MemberQueryResponse {
members: members.into_iter().map(Into::into).collect(),
users: join_all(
users
.into_iter()
.map(|other_user| other_user.into(db, &user)),
)
.await,
}))
}