use crate::{ naive_now, schema::{post, post_like, post_read, post_saved}, Crud, Likeable, Readable, Saveable, }; use diesel::{dsl::*, result::Error, *}; use url::{ParseError, Url}; #[derive(Queryable, Identifiable, PartialEq, Debug)] #[table_name = "post"] pub struct Post { pub id: i32, pub name: String, pub url: Option, pub body: Option, pub creator_id: i32, pub community_id: i32, pub removed: bool, pub locked: bool, pub published: chrono::NaiveDateTime, pub updated: Option, pub deleted: bool, pub nsfw: bool, pub stickied: bool, pub embed_title: Option, pub embed_description: Option, pub embed_html: Option, pub thumbnail_url: Option, pub ap_id: String, pub local: bool, } #[derive(Insertable, AsChangeset)] #[table_name = "post"] pub struct PostForm { pub name: String, pub url: Option, pub body: Option, pub creator_id: i32, pub community_id: i32, pub removed: Option, pub locked: Option, pub published: Option, pub updated: Option, pub deleted: Option, pub nsfw: bool, pub stickied: Option, pub embed_title: Option, pub embed_description: Option, pub embed_html: Option, pub thumbnail_url: Option, pub ap_id: Option, pub local: bool, } impl PostForm { pub fn get_ap_id(&self) -> Result { Url::parse(&self.ap_id.as_ref().unwrap_or(&"not_a_url".to_string())) } } impl Post { pub fn read(conn: &PgConnection, post_id: i32) -> Result { use crate::schema::post::dsl::*; post.filter(id.eq(post_id)).first::(conn) } pub fn list_for_community( conn: &PgConnection, the_community_id: i32, ) -> Result, Error> { use crate::schema::post::dsl::*; post .filter(community_id.eq(the_community_id)) .then_order_by(published.desc()) .then_order_by(stickied.desc()) .limit(20) .load::(conn) } pub fn read_from_apub_id(conn: &PgConnection, object_id: &str) -> Result { use crate::schema::post::dsl::*; post.filter(ap_id.eq(object_id)).first::(conn) } pub fn update_ap_id(conn: &PgConnection, post_id: i32, apub_id: String) -> Result { use crate::schema::post::dsl::*; diesel::update(post.find(post_id)) .set(ap_id.eq(apub_id)) .get_result::(conn) } pub fn permadelete_for_creator( conn: &PgConnection, for_creator_id: i32, ) -> Result, Error> { use crate::schema::post::dsl::*; let perma_deleted = "*Permananently Deleted*"; let perma_deleted_url = "https://deleted.com"; diesel::update(post.filter(creator_id.eq(for_creator_id))) .set(( name.eq(perma_deleted), url.eq(perma_deleted_url), body.eq(perma_deleted), deleted.eq(true), updated.eq(naive_now()), )) .get_results::(conn) } pub fn update_deleted( conn: &PgConnection, post_id: i32, new_deleted: bool, ) -> Result { use crate::schema::post::dsl::*; diesel::update(post.find(post_id)) .set((deleted.eq(new_deleted), updated.eq(naive_now()))) .get_result::(conn) } pub fn update_removed( conn: &PgConnection, post_id: i32, new_removed: bool, ) -> Result { use crate::schema::post::dsl::*; diesel::update(post.find(post_id)) .set((removed.eq(new_removed), updated.eq(naive_now()))) .get_result::(conn) } pub fn update_removed_for_creator( conn: &PgConnection, for_creator_id: i32, for_community_id: Option, new_removed: bool, ) -> Result, Error> { use crate::schema::post::dsl::*; let mut update = diesel::update(post).into_boxed(); update = update.filter(creator_id.eq(for_creator_id)); if let Some(for_community_id) = for_community_id { update = update.filter(community_id.eq(for_community_id)); } update .set((removed.eq(new_removed), updated.eq(naive_now()))) .get_results::(conn) } pub fn update_locked(conn: &PgConnection, post_id: i32, new_locked: bool) -> Result { use crate::schema::post::dsl::*; diesel::update(post.find(post_id)) .set(locked.eq(new_locked)) .get_result::(conn) } pub fn update_stickied( conn: &PgConnection, post_id: i32, new_stickied: bool, ) -> Result { use crate::schema::post::dsl::*; diesel::update(post.find(post_id)) .set(stickied.eq(new_stickied)) .get_result::(conn) } pub fn is_post_creator(user_id: i32, post_creator_id: i32) -> bool { user_id == post_creator_id } pub fn upsert(conn: &PgConnection, post_form: &PostForm) -> Result { use crate::schema::post::dsl::*; insert_into(post) .values(post_form) .on_conflict(ap_id) .do_update() .set(post_form) .get_result::(conn) } } impl Crud for Post { fn read(conn: &PgConnection, post_id: i32) -> Result { use crate::schema::post::dsl::*; post.find(post_id).first::(conn) } fn delete(conn: &PgConnection, post_id: i32) -> Result { use crate::schema::post::dsl::*; diesel::delete(post.find(post_id)).execute(conn) } fn create(conn: &PgConnection, new_post: &PostForm) -> Result { use crate::schema::post::dsl::*; insert_into(post).values(new_post).get_result::(conn) } fn update(conn: &PgConnection, post_id: i32, new_post: &PostForm) -> Result { use crate::schema::post::dsl::*; diesel::update(post.find(post_id)) .set(new_post) .get_result::(conn) } } #[derive(Identifiable, Queryable, Associations, PartialEq, Debug)] #[belongs_to(Post)] #[table_name = "post_like"] pub struct PostLike { pub id: i32, pub post_id: i32, pub user_id: i32, pub score: i16, pub published: chrono::NaiveDateTime, } #[derive(Insertable, AsChangeset, Clone)] #[table_name = "post_like"] pub struct PostLikeForm { pub post_id: i32, pub user_id: i32, pub score: i16, } impl Likeable for PostLike { fn like(conn: &PgConnection, post_like_form: &PostLikeForm) -> Result { use crate::schema::post_like::dsl::*; insert_into(post_like) .values(post_like_form) .get_result::(conn) } fn remove(conn: &PgConnection, user_id: i32, post_id: i32) -> Result { use crate::schema::post_like::dsl; diesel::delete( dsl::post_like .filter(dsl::post_id.eq(post_id)) .filter(dsl::user_id.eq(user_id)), ) .execute(conn) } } #[derive(Identifiable, Queryable, Associations, PartialEq, Debug)] #[belongs_to(Post)] #[table_name = "post_saved"] pub struct PostSaved { pub id: i32, pub post_id: i32, pub user_id: i32, pub published: chrono::NaiveDateTime, } #[derive(Insertable, AsChangeset)] #[table_name = "post_saved"] pub struct PostSavedForm { pub post_id: i32, pub user_id: i32, } impl Saveable for PostSaved { fn save(conn: &PgConnection, post_saved_form: &PostSavedForm) -> Result { use crate::schema::post_saved::dsl::*; insert_into(post_saved) .values(post_saved_form) .get_result::(conn) } fn unsave(conn: &PgConnection, post_saved_form: &PostSavedForm) -> Result { use crate::schema::post_saved::dsl::*; diesel::delete( post_saved .filter(post_id.eq(post_saved_form.post_id)) .filter(user_id.eq(post_saved_form.user_id)), ) .execute(conn) } } #[derive(Identifiable, Queryable, Associations, PartialEq, Debug)] #[belongs_to(Post)] #[table_name = "post_read"] pub struct PostRead { pub id: i32, pub post_id: i32, pub user_id: i32, pub published: chrono::NaiveDateTime, } #[derive(Insertable, AsChangeset)] #[table_name = "post_read"] pub struct PostReadForm { pub post_id: i32, pub user_id: i32, } impl Readable for PostRead { fn mark_as_read(conn: &PgConnection, post_read_form: &PostReadForm) -> Result { use crate::schema::post_read::dsl::*; insert_into(post_read) .values(post_read_form) .get_result::(conn) } fn mark_as_unread(conn: &PgConnection, post_read_form: &PostReadForm) -> Result { use crate::schema::post_read::dsl::*; diesel::delete( post_read .filter(post_id.eq(post_read_form.post_id)) .filter(user_id.eq(post_read_form.user_id)), ) .execute(conn) } } #[cfg(test)] mod tests { use crate::{ community::*, post::*, tests::establish_unpooled_connection, user::*, ListingType, SortType, }; #[test] fn test_crud() { let conn = establish_unpooled_connection(); let new_user = UserForm { name: "jim".into(), preferred_username: None, password_encrypted: "nope".into(), email: None, matrix_user_id: None, avatar: None, banner: None, admin: false, banned: Some(false), published: None, updated: None, show_nsfw: false, theme: "browser".into(), default_sort_type: SortType::Hot as i16, default_listing_type: ListingType::Subscribed as i16, lang: "browser".into(), show_avatars: true, send_notifications_to_email: false, actor_id: None, bio: None, local: true, private_key: None, public_key: None, last_refreshed_at: None, }; let inserted_user = User_::create(&conn, &new_user).unwrap(); let new_community = CommunityForm { name: "test community_3".to_string(), title: "nada".to_owned(), description: None, category_id: 1, creator_id: inserted_user.id, removed: None, deleted: None, updated: None, nsfw: false, actor_id: None, local: true, private_key: None, public_key: None, last_refreshed_at: None, published: None, icon: None, banner: None, }; let inserted_community = Community::create(&conn, &new_community).unwrap(); let new_post = PostForm { name: "A test post".into(), url: None, body: None, creator_id: inserted_user.id, community_id: inserted_community.id, removed: None, deleted: None, locked: None, stickied: None, nsfw: false, updated: None, embed_title: None, embed_description: None, embed_html: None, thumbnail_url: None, ap_id: None, local: true, published: None, }; let inserted_post = Post::create(&conn, &new_post).unwrap(); let expected_post = Post { id: inserted_post.id, name: "A test post".into(), url: None, body: None, creator_id: inserted_user.id, community_id: inserted_community.id, published: inserted_post.published, removed: false, locked: false, stickied: false, nsfw: false, deleted: false, updated: None, embed_title: None, embed_description: None, embed_html: None, thumbnail_url: None, ap_id: inserted_post.ap_id.to_owned(), local: true, }; // Post Like let post_like_form = PostLikeForm { post_id: inserted_post.id, user_id: inserted_user.id, score: 1, }; let inserted_post_like = PostLike::like(&conn, &post_like_form).unwrap(); let expected_post_like = PostLike { id: inserted_post_like.id, post_id: inserted_post.id, user_id: inserted_user.id, published: inserted_post_like.published, score: 1, }; // Post Save let post_saved_form = PostSavedForm { post_id: inserted_post.id, user_id: inserted_user.id, }; let inserted_post_saved = PostSaved::save(&conn, &post_saved_form).unwrap(); let expected_post_saved = PostSaved { id: inserted_post_saved.id, post_id: inserted_post.id, user_id: inserted_user.id, published: inserted_post_saved.published, }; // Post Read let post_read_form = PostReadForm { post_id: inserted_post.id, user_id: inserted_user.id, }; let inserted_post_read = PostRead::mark_as_read(&conn, &post_read_form).unwrap(); let expected_post_read = PostRead { id: inserted_post_read.id, post_id: inserted_post.id, user_id: inserted_user.id, published: inserted_post_read.published, }; let read_post = Post::read(&conn, inserted_post.id).unwrap(); let updated_post = Post::update(&conn, inserted_post.id, &new_post).unwrap(); let like_removed = PostLike::remove(&conn, inserted_user.id, inserted_post.id).unwrap(); let saved_removed = PostSaved::unsave(&conn, &post_saved_form).unwrap(); let read_removed = PostRead::mark_as_unread(&conn, &post_read_form).unwrap(); let num_deleted = Post::delete(&conn, inserted_post.id).unwrap(); Community::delete(&conn, inserted_community.id).unwrap(); User_::delete(&conn, inserted_user.id).unwrap(); assert_eq!(expected_post, read_post); assert_eq!(expected_post, inserted_post); assert_eq!(expected_post, updated_post); assert_eq!(expected_post_like, inserted_post_like); assert_eq!(expected_post_saved, inserted_post_saved); assert_eq!(expected_post_read, inserted_post_read); assert_eq!(1, like_removed); assert_eq!(1, saved_removed); assert_eq!(1, read_removed); assert_eq!(1, num_deleted); } }