forked from jmug/stoatchat
fix: fixes for full mongo implementation
This commit is contained in:
@@ -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!(),
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user