fix: fixes for full mongo implementation

This commit is contained in:
Paul Makles
2022-02-11 21:44:59 +00:00
parent 7a9adb45f7
commit 4801b51cef
15 changed files with 45 additions and 36 deletions

View File

@@ -1,4 +1,7 @@
use revolt_quark::{Error, Result, Ref, models::{User, Invite}, Db};
use revolt_quark::{
models::{Invite, User},
Db, Error, Ref, Result,
};
use rocket::serde::json::Value;
@@ -7,32 +10,34 @@ use crate::util::variables::MAX_SERVER_COUNT;
#[post("/<target>")]
pub async fn req(db: &Db, user: User, target: Ref) -> Result<Value> {
if user.bot.is_some() {
return Err(Error::IsBot)
return Err(Error::IsBot);
}
if !user.can_acquire_server(db).await? {
return Err(Error::TooManyServers {
max: *MAX_SERVER_COUNT
})
max: *MAX_SERVER_COUNT,
});
}
let invite = target.as_invite(db).await?;
match &invite {
Invite::Server { channel, server, .. } => {
Invite::Server {
channel, server, ..
} => {
let server = db.fetch_server(server).await?;
if db.fetch_ban(&server.id, &user.id).await.is_ok() {
return Err(Error::Banned)
return Err(Error::Banned);
}
let channel = db.fetch_channel(channel).await?;
db.insert_member(&server.id, &user.id).await?;
Ok(json!({
"type": "Server",
"channel": channel,
"server": server
}))
}
_ => unreachable!()
_ => unreachable!(),
}
}