feat: allow fetching channels with server
This commit is contained in:
@@ -1,17 +1,65 @@
|
|||||||
use revolt_quark::{
|
use revolt_quark::{
|
||||||
models::{Server, User},
|
models::{Channel, Server, User},
|
||||||
perms, Db, Ref, Result,
|
perms, Db, Ref, Result,
|
||||||
};
|
};
|
||||||
use rocket::serde::json::Json;
|
use rocket::serde::json::Json;
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
|
||||||
|
/// # Query Parameters
|
||||||
|
#[derive(Deserialize, JsonSchema, FromForm)]
|
||||||
|
pub struct OptionsFetchServer {
|
||||||
|
/// Whether to include channels
|
||||||
|
include_channels: Option<bool>,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// # Fetch server route response
|
||||||
|
#[derive(Serialize, JsonSchema)]
|
||||||
|
#[serde(untagged)]
|
||||||
|
pub enum FetchServerResponse {
|
||||||
|
JustServer(Server),
|
||||||
|
ServerWithChannels {
|
||||||
|
#[serde(flatten)]
|
||||||
|
server: Server,
|
||||||
|
channels: Vec<Channel>,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
/// # Fetch Server
|
/// # Fetch Server
|
||||||
///
|
///
|
||||||
/// Fetch a server by its id.
|
/// Fetch a server by its id.
|
||||||
#[openapi(tag = "Server Information")]
|
#[openapi(tag = "Server Information")]
|
||||||
#[get("/<target>")]
|
#[get("/<target>?<options..>")]
|
||||||
pub async fn req(db: &Db, user: User, target: Ref) -> Result<Json<Server>> {
|
pub async fn req(
|
||||||
|
db: &Db,
|
||||||
|
user: User,
|
||||||
|
target: Ref,
|
||||||
|
options: OptionsFetchServer,
|
||||||
|
) -> Result<Json<FetchServerResponse>> {
|
||||||
let server = target.as_server(db).await?;
|
let server = target.as_server(db).await?;
|
||||||
perms(&user).server(&server).calc(db).await?;
|
let mut perms = perms(&user).server(&server);
|
||||||
|
perms.calc(db).await?;
|
||||||
|
|
||||||
Ok(Json(server))
|
if let Some(true) = options.include_channels {
|
||||||
|
let all_channels = db.fetch_channels(&server.channels).await?;
|
||||||
|
let mut visible_channels = vec![];
|
||||||
|
|
||||||
|
for channel in all_channels {
|
||||||
|
if perms
|
||||||
|
.clone()
|
||||||
|
.channel(&channel)
|
||||||
|
.calc(db)
|
||||||
|
.await?
|
||||||
|
.can_view_channel()
|
||||||
|
{
|
||||||
|
visible_channels.push(channel);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(Json(FetchServerResponse::ServerWithChannels {
|
||||||
|
server,
|
||||||
|
channels: visible_channels,
|
||||||
|
}))
|
||||||
|
} else {
|
||||||
|
Ok(Json(FetchServerResponse::JustServer(server)))
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user