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

63 lines
1.8 KiB
Rust
Raw Normal View History

use crate::{newtypes::DbUrl, source::site::*, traits::Crud};
2020-05-16 14:04:08 +00:00
use diesel::{dsl::*, result::Error, *};
use url::Url;
impl Crud for Site {
type Form = SiteForm;
type IdType = i32;
fn read(conn: &mut PgConnection, _site_id: i32) -> Result<Self, Error> {
2021-10-16 13:33:38 +00:00
use crate::schema::site::dsl::*;
site.first::<Self>(conn)
}
fn create(conn: &mut PgConnection, new_site: &SiteForm) -> Result<Self, Error> {
2021-10-16 13:33:38 +00:00
use crate::schema::site::dsl::*;
insert_into(site).values(new_site).get_result::<Self>(conn)
}
fn update(conn: &mut PgConnection, site_id: i32, new_site: &SiteForm) -> Result<Self, Error> {
2021-10-16 13:33:38 +00:00
use crate::schema::site::dsl::*;
diesel::update(site.find(site_id))
.set(new_site)
.get_result::<Self>(conn)
}
fn delete(conn: &mut PgConnection, site_id: i32) -> Result<usize, Error> {
2021-10-16 13:33:38 +00:00
use crate::schema::site::dsl::*;
2020-12-20 05:09:20 +00:00
diesel::delete(site.find(site_id)).execute(conn)
}
}
2021-10-16 13:33:38 +00:00
impl Site {
pub fn read_local_site(conn: &mut PgConnection) -> Result<Self, Error> {
2021-10-16 13:33:38 +00:00
use crate::schema::site::dsl::*;
site.order_by(id).first::<Self>(conn)
}
pub fn upsert(conn: &mut PgConnection, site_form: &SiteForm) -> Result<Site, Error> {
use crate::schema::site::dsl::*;
insert_into(site)
.values(site_form)
.on_conflict(actor_id)
.do_update()
.set(site_form)
.get_result::<Self>(conn)
}
pub fn read_from_apub_id(conn: &mut PgConnection, object_id: Url) -> Result<Option<Self>, Error> {
use crate::schema::site::dsl::*;
let object_id: DbUrl = object_id.into();
Ok(
site
.filter(actor_id.eq(object_id))
.first::<Site>(conn)
.ok()
.map(Into::into),
)
}
pub fn read_remote_sites(conn: &mut PgConnection) -> Result<Vec<Self>, Error> {
use crate::schema::site::dsl::*;
site.order_by(id).offset(1).get_results::<Self>(conn)
}
}