mirror of https://github.com/LemmyNet/lemmy.git
57 lines
1.4 KiB
Rust
57 lines
1.4 KiB
Rust
use crate::{
|
|
newtypes::{CommunityId, PersonId},
|
|
schema::community_block::dsl::{community_block, community_id, person_id},
|
|
source::community_block::{CommunityBlock, CommunityBlockForm},
|
|
traits::Blockable,
|
|
utils::{get_conn, DbPool},
|
|
};
|
|
use diesel::{
|
|
dsl::{exists, insert_into},
|
|
result::Error,
|
|
select,
|
|
QueryDsl,
|
|
};
|
|
use diesel_async::RunQueryDsl;
|
|
|
|
impl CommunityBlock {
|
|
pub async fn read(
|
|
pool: &mut DbPool<'_>,
|
|
for_person_id: PersonId,
|
|
for_community_id: CommunityId,
|
|
) -> Result<bool, Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
select(exists(
|
|
community_block.find((for_person_id, for_community_id)),
|
|
))
|
|
.get_result(conn)
|
|
.await
|
|
}
|
|
}
|
|
|
|
#[async_trait]
|
|
impl Blockable for CommunityBlock {
|
|
type Form = CommunityBlockForm;
|
|
async fn block(pool: &mut DbPool<'_>, community_block_form: &Self::Form) -> Result<Self, Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
insert_into(community_block)
|
|
.values(community_block_form)
|
|
.on_conflict((person_id, community_id))
|
|
.do_update()
|
|
.set(community_block_form)
|
|
.get_result::<Self>(conn)
|
|
.await
|
|
}
|
|
async fn unblock(
|
|
pool: &mut DbPool<'_>,
|
|
community_block_form: &Self::Form,
|
|
) -> Result<usize, Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
diesel::delete(community_block.find((
|
|
community_block_form.person_id,
|
|
community_block_form.community_id,
|
|
)))
|
|
.execute(conn)
|
|
.await
|
|
}
|
|
}
|