]> Untitled Git - lemmy.git/blob - server/lemmy_db/src/site.rs
Merge remote-tracking branch 'weblate/main' into main
[lemmy.git] / server / lemmy_db / src / site.rs
1 use crate::{naive_now, schema::site, Crud};
2 use diesel::{dsl::*, result::Error, *};
3 use serde::{Deserialize, Serialize};
4
5 #[derive(Queryable, Identifiable, PartialEq, Debug, Serialize, Deserialize)]
6 #[table_name = "site"]
7 pub struct Site {
8   pub id: i32,
9   pub name: String,
10   pub description: Option<String>,
11   pub creator_id: i32,
12   pub published: chrono::NaiveDateTime,
13   pub updated: Option<chrono::NaiveDateTime>,
14   pub enable_downvotes: bool,
15   pub open_registration: bool,
16   pub enable_nsfw: bool,
17   pub icon: Option<String>,
18   pub banner: Option<String>,
19 }
20
21 #[derive(Insertable, AsChangeset, Clone, Serialize, Deserialize)]
22 #[table_name = "site"]
23 pub struct SiteForm {
24   pub name: String,
25   pub description: Option<String>,
26   pub creator_id: i32,
27   pub updated: Option<chrono::NaiveDateTime>,
28   pub enable_downvotes: bool,
29   pub open_registration: bool,
30   pub enable_nsfw: bool,
31   // when you want to null out a column, you have to send Some(None)), since sending None means you just don't want to update that column.
32   pub icon: Option<Option<String>>,
33   pub banner: Option<Option<String>>,
34 }
35
36 impl Crud<SiteForm> for Site {
37   fn read(conn: &PgConnection, _site_id: i32) -> Result<Self, Error> {
38     use crate::schema::site::dsl::*;
39     site.first::<Self>(conn)
40   }
41
42   fn delete(conn: &PgConnection, site_id: i32) -> Result<usize, Error> {
43     use crate::schema::site::dsl::*;
44     diesel::delete(site.find(site_id)).execute(conn)
45   }
46
47   fn create(conn: &PgConnection, new_site: &SiteForm) -> Result<Self, Error> {
48     use crate::schema::site::dsl::*;
49     insert_into(site).values(new_site).get_result::<Self>(conn)
50   }
51
52   fn update(conn: &PgConnection, site_id: i32, new_site: &SiteForm) -> Result<Self, Error> {
53     use crate::schema::site::dsl::*;
54     diesel::update(site.find(site_id))
55       .set(new_site)
56       .get_result::<Self>(conn)
57   }
58 }
59
60 impl Site {
61   pub fn transfer(conn: &PgConnection, new_creator_id: i32) -> Result<Self, Error> {
62     use crate::schema::site::dsl::*;
63     diesel::update(site.find(1))
64       .set((creator_id.eq(new_creator_id), updated.eq(naive_now())))
65       .get_result::<Self>(conn)
66   }
67 }