crates/db_queries/src/source/site.rs

   1 use crate::Crud;
   2 use diesel::{dsl::*, result::Error, *};
   3 use lemmy_db_schema::{naive_now, source::site::*, PersonId};
   4
   5 impl Crud<SiteForm, i32> for Site {
   6   fn read(conn: &PgConnection, _site_id: i32) -> Result<Self, Error> {
   7     use lemmy_db_schema::schema::site::dsl::*;
   8     site.first::<Self>(conn)
   9   }
  10
  11   fn create(conn: &PgConnection, new_site: &SiteForm) -> Result<Self, Error> {
  12     use lemmy_db_schema::schema::site::dsl::*;
  13     insert_into(site).values(new_site).get_result::<Self>(conn)
  14   }
  15
  16   fn update(conn: &PgConnection, site_id: i32, new_site: &SiteForm) -> Result<Self, Error> {
  17     use lemmy_db_schema::schema::site::dsl::*;
  18     diesel::update(site.find(site_id))
  19       .set(new_site)
  20       .get_result::<Self>(conn)
  21   }
  22   fn delete(conn: &PgConnection, site_id: i32) -> Result<usize, Error> {
  23     use lemmy_db_schema::schema::site::dsl::*;
  24     diesel::delete(site.find(site_id)).execute(conn)
  25   }
  26 }
  27
  28 pub trait Site_ {
  29   fn transfer(conn: &PgConnection, new_creator_id: PersonId) -> Result<Site, Error>;
  30   fn read_simple(conn: &PgConnection) -> Result<Site, Error>;
  31 }
  32
  33 impl Site_ for Site {
  34   fn transfer(conn: &PgConnection, new_creator_id: PersonId) -> Result<Site, Error> {
  35     use lemmy_db_schema::schema::site::dsl::*;
  36     diesel::update(site.find(1))
  37       .set((creator_id.eq(new_creator_id), updated.eq(naive_now())))
  38       .get_result::<Self>(conn)
  39   }
  40
  41   fn read_simple(conn: &PgConnection) -> Result<Self, Error> {
  42     use lemmy_db_schema::schema::site::dsl::*;
  43     site.first::<Self>(conn)
  44   }
  45 }