lemmy/crates/db_schema/src/impls/site.rs

79 lines
2 KiB
Rust
Raw Normal View History

use crate::{
newtypes::{DbUrl, SiteId},
schema::site::dsl::{actor_id, id, site},
source::{
actor_language::SiteLanguage,
site::{Site, SiteInsertForm, SiteUpdateForm},
},
traits::Crud,
2022-11-09 10:05:00 +00:00
utils::{get_conn, DbPool},
};
use diesel::{dsl::insert_into, result::Error, ExpressionMethods, QueryDsl};
2022-11-09 10:05:00 +00:00
use diesel_async::RunQueryDsl;
use url::Url;
2022-11-09 10:05:00 +00:00
#[async_trait]
impl Crud for Site {
type InsertForm = SiteInsertForm;
type UpdateForm = SiteUpdateForm;
type IdType = SiteId;
2022-11-09 10:05:00 +00:00
async fn read(pool: &DbPool, _site_id: SiteId) -> Result<Self, Error> {
let conn = &mut get_conn(pool).await?;
site.first::<Self>(conn).await
}
2022-11-09 10:05:00 +00:00
async fn create(pool: &DbPool, form: &Self::InsertForm) -> Result<Self, Error> {
let conn = &mut get_conn(pool).await?;
let site_ = insert_into(site)
.values(form)
.on_conflict(actor_id)
.do_update()
.set(form)
2022-11-09 10:05:00 +00:00
.get_result::<Self>(conn)
.await?;
// initialize with all languages
2022-11-09 10:05:00 +00:00
SiteLanguage::update(pool, vec![], &site_).await?;
Ok(site_)
}
2022-11-09 10:05:00 +00:00
async fn update(
pool: &DbPool,
site_id: SiteId,
new_site: &Self::UpdateForm,
) -> Result<Self, Error> {
2022-11-09 10:05:00 +00:00
let conn = &mut get_conn(pool).await?;
diesel::update(site.find(site_id))
.set(new_site)
.get_result::<Self>(conn)
2022-11-09 10:05:00 +00:00
.await
}
2022-11-09 10:05:00 +00:00
async fn delete(pool: &DbPool, site_id: SiteId) -> Result<usize, Error> {
let conn = &mut get_conn(pool).await?;
diesel::delete(site.find(site_id)).execute(conn).await
2020-12-20 05:09:20 +00:00
}
}
2021-10-16 13:33:38 +00:00
impl Site {
2022-11-09 10:05:00 +00:00
pub async fn read_from_apub_id(pool: &DbPool, object_id: Url) -> Result<Option<Self>, Error> {
let conn = &mut get_conn(pool).await?;
let object_id: DbUrl = object_id.into();
Ok(
site
.filter(actor_id.eq(object_id))
.first::<Site>(conn)
2022-11-09 10:05:00 +00:00
.await
.ok()
.map(Into::into),
)
}
// TODO this needs fixed
2022-11-09 10:05:00 +00:00
pub async fn read_remote_sites(pool: &DbPool) -> Result<Vec<Self>, Error> {
let conn = &mut get_conn(pool).await?;
site.order_by(id).offset(1).get_results::<Self>(conn).await
}
}