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