52 lines
1.3 KiB
Rust
52 lines
1.3 KiB
Rust
use revolt_quark::{
|
|
models::{server_member::MemberCompositeKey, Invite, Member, User},
|
|
Db, Error, Ref, Result,
|
|
};
|
|
|
|
use rocket::serde::json::Value;
|
|
|
|
use crate::util::variables::MAX_SERVER_COUNT;
|
|
|
|
/// # Join Invite
|
|
///
|
|
/// Join an invite by its ID.
|
|
#[openapi(tag = "Invites")]
|
|
#[post("/<target>")]
|
|
pub async fn req(db: &Db, user: User, target: Ref) -> Result<Value> {
|
|
if user.bot.is_some() {
|
|
return Err(Error::IsBot);
|
|
}
|
|
|
|
if !user.can_acquire_server(db).await? {
|
|
return Err(Error::TooManyServers {
|
|
max: *MAX_SERVER_COUNT,
|
|
});
|
|
}
|
|
|
|
let invite = target.as_invite(db).await?;
|
|
match &invite {
|
|
Invite::Server {
|
|
channel, server, ..
|
|
} => {
|
|
let server = db.fetch_server(server).await?;
|
|
let channel = db.fetch_channel(channel).await?;
|
|
let member = Member {
|
|
id: MemberCompositeKey {
|
|
server: server.id.clone(),
|
|
user: user.id.clone(),
|
|
},
|
|
..Default::default()
|
|
};
|
|
|
|
member.create(db).await?;
|
|
|
|
Ok(json!({
|
|
"type": "Server",
|
|
"channel": channel,
|
|
"server": server
|
|
}))
|
|
}
|
|
_ => unreachable!(),
|
|
}
|
|
}
|