aace1bd71e
Add post_read back in, do some cleanup Add some delete functions back in Merge branch 'main' into remove-dead-code Replace body of unused db functions with unimplemented!() Remove dead code Remove remaining usages of unwrap() from activitypub code Remove most usage of Option::unwrap() from activitypub code Co-authored-by: Felix Ableitner <me@nutomic.com> Reviewed-on: https://yerbamate.dev/LemmyNet/lemmy/pulls/81
62 lines
1.9 KiB
Rust
62 lines
1.9 KiB
Rust
use crate::{naive_now, schema::site, Crud};
|
|
use diesel::{dsl::*, result::Error, *};
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
#[derive(Queryable, Identifiable, PartialEq, Debug, Serialize, Deserialize)]
|
|
#[table_name = "site"]
|
|
pub struct Site {
|
|
pub id: i32,
|
|
pub name: String,
|
|
pub description: Option<String>,
|
|
pub creator_id: i32,
|
|
pub published: chrono::NaiveDateTime,
|
|
pub updated: Option<chrono::NaiveDateTime>,
|
|
pub enable_downvotes: bool,
|
|
pub open_registration: bool,
|
|
pub enable_nsfw: bool,
|
|
pub icon: Option<String>,
|
|
pub banner: Option<String>,
|
|
}
|
|
|
|
#[derive(Insertable, AsChangeset, Clone, Serialize, Deserialize)]
|
|
#[table_name = "site"]
|
|
pub struct SiteForm {
|
|
pub name: String,
|
|
pub description: Option<String>,
|
|
pub creator_id: i32,
|
|
pub updated: Option<chrono::NaiveDateTime>,
|
|
pub enable_downvotes: bool,
|
|
pub open_registration: bool,
|
|
pub enable_nsfw: bool,
|
|
// 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.
|
|
pub icon: Option<Option<String>>,
|
|
pub banner: Option<Option<String>>,
|
|
}
|
|
|
|
impl Crud<SiteForm> for Site {
|
|
fn read(conn: &PgConnection, _site_id: i32) -> Result<Self, Error> {
|
|
use crate::schema::site::dsl::*;
|
|
site.first::<Self>(conn)
|
|
}
|
|
|
|
fn create(conn: &PgConnection, new_site: &SiteForm) -> Result<Self, Error> {
|
|
use crate::schema::site::dsl::*;
|
|
insert_into(site).values(new_site).get_result::<Self>(conn)
|
|
}
|
|
|
|
fn update(conn: &PgConnection, site_id: i32, new_site: &SiteForm) -> Result<Self, Error> {
|
|
use crate::schema::site::dsl::*;
|
|
diesel::update(site.find(site_id))
|
|
.set(new_site)
|
|
.get_result::<Self>(conn)
|
|
}
|
|
}
|
|
|
|
impl Site {
|
|
pub fn transfer(conn: &PgConnection, new_creator_id: i32) -> Result<Self, Error> {
|
|
use crate::schema::site::dsl::*;
|
|
diesel::update(site.find(1))
|
|
.set((creator_id.eq(new_creator_id), updated.eq(naive_now())))
|
|
.get_result::<Self>(conn)
|
|
}
|
|
}
|