feat: added endpoint for banning
This commit is contained in:
parent
f175c19325
commit
ad24215fef
3 changed files with 86 additions and 1 deletions
|
@ -1,3 +1,4 @@
|
|||
use axum::http::StatusCode;
|
||||
use diesel::{
|
||||
ExpressionMethods, Insertable, QueryDsl, Queryable, Selectable, SelectableHelper, delete,
|
||||
insert_into,
|
||||
|
@ -11,6 +12,7 @@ use crate::{
|
|||
error::Error,
|
||||
objects::{Me, Permissions, Role},
|
||||
schema::guild_members,
|
||||
schema::guild_bans,
|
||||
};
|
||||
|
||||
use super::{User, load_or_empty};
|
||||
|
@ -73,6 +75,13 @@ pub struct Member {
|
|||
user: User,
|
||||
}
|
||||
|
||||
#[derive(Serialize, Deserialize)]
|
||||
pub struct GuildBan {
|
||||
pub guild_uuid: Uuid,
|
||||
pub user_uuid: Uuid,
|
||||
pub reason: String,
|
||||
}
|
||||
|
||||
impl Member {
|
||||
pub async fn count(conn: &mut Conn, guild_uuid: Uuid) -> Result<i64, Error> {
|
||||
use guild_members::dsl;
|
||||
|
@ -169,6 +178,18 @@ impl Member {
|
|||
) -> Result<Self, Error> {
|
||||
let mut conn = app_state.pool.get().await?;
|
||||
|
||||
use guild_bans::dsl;
|
||||
let banned = dsl::guild_bans
|
||||
.filter(guild_bans::guild_uuid.eq(guild_uuid))
|
||||
.filter(guild_bans::user_uuid.eq(user_uuid))
|
||||
.execute(&mut conn)
|
||||
.await;
|
||||
match banned {
|
||||
Ok(_) => Err(Error::Forbidden("User banned".to_string())),
|
||||
Err(diesel::result::Error::NotFound) => Ok(()),
|
||||
Err(e) => Err(e.into()),
|
||||
}?;
|
||||
|
||||
let member_uuid = Uuid::now_v7();
|
||||
|
||||
let member = MemberBuilder {
|
||||
|
@ -195,4 +216,20 @@ impl Member {
|
|||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub async fn ban(self, conn: &mut Conn, reason: &String) -> Result<(), Error> {
|
||||
use guild_bans::dsl;
|
||||
insert_into(guild_bans::table)
|
||||
.values((
|
||||
dsl::guild_uuid.eq(self.guild_uuid),
|
||||
dsl::user_uuid.eq(self.user_uuid),
|
||||
dsl::reason.eq(reason),
|
||||
))
|
||||
.execute(conn)
|
||||
.await?;
|
||||
|
||||
self.delete(conn).await?;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue