mirror of https://github.com/LemmyNet/lemmy.git
85 lines
2.1 KiB
Rust
85 lines
2.1 KiB
Rust
use crate::{
|
|
newtypes::DbUrl,
|
|
schema::{image_details, local_image, remote_image},
|
|
source::images::{
|
|
ImageDetails,
|
|
ImageDetailsForm,
|
|
LocalImage,
|
|
LocalImageForm,
|
|
RemoteImage,
|
|
RemoteImageForm,
|
|
},
|
|
utils::{get_conn, DbPool},
|
|
};
|
|
use diesel::{
|
|
insert_into,
|
|
result::Error,
|
|
ExpressionMethods,
|
|
NotFound,
|
|
OptionalExtension,
|
|
QueryDsl,
|
|
};
|
|
use diesel_async::RunQueryDsl;
|
|
use url::Url;
|
|
|
|
impl LocalImage {
|
|
pub async fn create(pool: &mut DbPool<'_>, form: &LocalImageForm) -> Result<Self, Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
insert_into(local_image::table)
|
|
.values(form)
|
|
.get_result::<Self>(conn)
|
|
.await
|
|
}
|
|
|
|
pub async fn delete_by_alias(pool: &mut DbPool<'_>, alias: &str) -> Result<Self, Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
diesel::delete(local_image::table.filter(local_image::pictrs_alias.eq(alias)))
|
|
.get_result(conn)
|
|
.await
|
|
}
|
|
|
|
pub async fn delete_by_url(pool: &mut DbPool<'_>, url: &DbUrl) -> Result<Self, Error> {
|
|
let alias = url.as_str().split('/').last().ok_or(NotFound)?;
|
|
Self::delete_by_alias(pool, alias).await
|
|
}
|
|
}
|
|
|
|
impl RemoteImage {
|
|
pub async fn create(pool: &mut DbPool<'_>, link_: &Url) -> Result<usize, Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
let form = RemoteImageForm {
|
|
link: link_.clone().into(),
|
|
};
|
|
insert_into(remote_image::table)
|
|
.values(form)
|
|
.on_conflict_do_nothing()
|
|
.execute(conn)
|
|
.await
|
|
}
|
|
|
|
pub async fn validate(pool: &mut DbPool<'_>, link_: DbUrl) -> Result<(), Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
let res = remote_image::table
|
|
.find(link_)
|
|
.first::<RemoteImage>(conn)
|
|
.await
|
|
.optional()?;
|
|
if res.is_some() {
|
|
Ok(())
|
|
} else {
|
|
Err(NotFound)
|
|
}
|
|
}
|
|
}
|
|
|
|
impl ImageDetails {
|
|
pub async fn create(pool: &mut DbPool<'_>, form: &ImageDetailsForm) -> Result<Self, Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
insert_into(image_details::table)
|
|
.values(form)
|
|
.on_conflict_do_nothing()
|
|
.get_result::<Self>(conn)
|
|
.await
|
|
}
|
|
}
|