mirror of
https://github.com/LemmyNet/lemmy.git
synced 2024-11-16 01:14:02 +00:00
6bfbb9332d
* Adding listMedia endpoint, to view all your local image uploads. - Fixes #4445 * Fix ts import. * Forgot to order by published desc * Adding an endpoint to list all images, for admins only. * Forgot to add file. * Add additional test. * Use better logic for no-limit version. * Better call sites. * Adding another test. * Fix tests. * Moving list_media to /account action. * Addressing PR comments. * Removing pointless comment. --------- Co-authored-by: SleeplessOne1917 <28871516+SleeplessOne1917@users.noreply.github.com>
113 lines
2.9 KiB
Rust
113 lines
2.9 KiB
Rust
use crate::{
|
|
newtypes::{DbUrl, LocalUserId},
|
|
schema::{local_image, remote_image},
|
|
source::images::{LocalImage, LocalImageForm, RemoteImage, RemoteImageForm},
|
|
utils::{get_conn, limit_and_offset, DbPool},
|
|
};
|
|
use diesel::{
|
|
dsl::exists,
|
|
insert_into,
|
|
result::Error,
|
|
select,
|
|
ExpressionMethods,
|
|
NotFound,
|
|
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 get_all_paged_by_local_user_id(
|
|
pool: &mut DbPool<'_>,
|
|
user_id: LocalUserId,
|
|
page: Option<i64>,
|
|
limit: Option<i64>,
|
|
) -> Result<Vec<Self>, Error> {
|
|
Self::get_all_helper(pool, Some(user_id), page, limit, false).await
|
|
}
|
|
|
|
pub async fn get_all_by_local_user_id(
|
|
pool: &mut DbPool<'_>,
|
|
user_id: LocalUserId,
|
|
) -> Result<Vec<Self>, Error> {
|
|
Self::get_all_helper(pool, Some(user_id), None, None, true).await
|
|
}
|
|
|
|
pub async fn get_all(
|
|
pool: &mut DbPool<'_>,
|
|
page: Option<i64>,
|
|
limit: Option<i64>,
|
|
) -> Result<Vec<Self>, Error> {
|
|
Self::get_all_helper(pool, None, page, limit, false).await
|
|
}
|
|
|
|
async fn get_all_helper(
|
|
pool: &mut DbPool<'_>,
|
|
user_id: Option<LocalUserId>,
|
|
page: Option<i64>,
|
|
limit: Option<i64>,
|
|
ignore_page_limits: bool,
|
|
) -> Result<Vec<Self>, Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
let mut query = local_image::table
|
|
.select(local_image::all_columns)
|
|
.order_by(local_image::published.desc())
|
|
.into_boxed();
|
|
|
|
if let Some(user_id) = user_id {
|
|
query = query.filter(local_image::local_user_id.eq(user_id))
|
|
}
|
|
|
|
if !ignore_page_limits {
|
|
let (limit, offset) = limit_and_offset(page, limit)?;
|
|
query = query.limit(limit).offset(offset);
|
|
}
|
|
|
|
query.load::<LocalImage>(conn).await
|
|
}
|
|
|
|
pub async fn delete_by_alias(pool: &mut DbPool<'_>, alias: &str) -> Result<usize, Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
diesel::delete(local_image::table.filter(local_image::pictrs_alias.eq(alias)))
|
|
.execute(conn)
|
|
.await
|
|
}
|
|
}
|
|
|
|
impl RemoteImage {
|
|
pub async fn create(pool: &mut DbPool<'_>, links: Vec<Url>) -> Result<usize, Error> {
|
|
let conn = &mut get_conn(pool).await?;
|
|
let forms = links
|
|
.into_iter()
|
|
.map(|url| RemoteImageForm { link: url.into() })
|
|
.collect::<Vec<_>>();
|
|
insert_into(remote_image::table)
|
|
.values(forms)
|
|
.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 exists = select(exists(
|
|
remote_image::table.filter(remote_image::link.eq(link_)),
|
|
))
|
|
.get_result::<bool>(conn)
|
|
.await?;
|
|
if exists {
|
|
Ok(())
|
|
} else {
|
|
Err(NotFound)
|
|
}
|
|
}
|
|
}
|