Starting to add post_view.
authorDessalines <tyhou13@gmx.com>
Thu, 10 Dec 2020 20:53:49 +0000 (15:53 -0500)
committerDessalines <tyhou13@gmx.com>
Thu, 10 Dec 2020 20:53:49 +0000 (15:53 -0500)
lemmy_db/src/aggregates/mod.rs
lemmy_db/src/aggregates/post_aggregates.rs [new file with mode: 0644]
lemmy_db/src/post.rs
lemmy_db/src/schema.rs
lemmy_db/src/views/community_view.rs
lemmy_db/src/views/mod.rs
lemmy_db/src/views/post_view.rs [new file with mode: 0644]
migrations/2020-12-10-152350_create_post_aggregates/down.sql [new file with mode: 0644]
migrations/2020-12-10-152350_create_post_aggregates/up.sql [new file with mode: 0644]

index 9f38f2ed0597b510c2d1836902d0aed3986b83ec..e033aede2b876e12e944769d9f0db11d3e4e0670 100644 (file)
@@ -1,3 +1,4 @@
 pub mod community_aggregates;
+pub mod post_aggregates;
 pub mod site_aggregates;
 pub mod user_aggregates;
diff --git a/lemmy_db/src/aggregates/post_aggregates.rs b/lemmy_db/src/aggregates/post_aggregates.rs
new file mode 100644 (file)
index 0000000..e009cac
--- /dev/null
@@ -0,0 +1,268 @@
+use crate::schema::post_aggregates;
+use diesel::{result::Error, *};
+use serde::Serialize;
+
+#[derive(Queryable, Associations, Identifiable, PartialEq, Debug, Serialize, Clone)]
+#[table_name = "post_aggregates"]
+pub struct PostAggregates {
+  pub id: i32,
+  pub post_id: i32,
+  pub comments: i64,
+  pub score: i64,
+  pub upvotes: i64,
+  pub downvotes: i64,
+  pub newest_comment_time: chrono::NaiveDateTime,
+}
+
+impl PostAggregates {
+  pub fn read(conn: &PgConnection, post_id: i32) -> Result<Self, Error> {
+    post_aggregates::table
+      .filter(post_aggregates::post_id.eq(post_id))
+      .first::<Self>(conn)
+  }
+}
+
+// #[cfg(test)]
+// mod tests {
+//   use crate::{
+//     aggregates::community_aggregates::CommunityAggregates,
+//     comment::{Comment, CommentForm},
+//     community::{Community, CommunityFollower, CommunityFollowerForm, CommunityForm},
+//     post::{Post, PostForm},
+//     tests::establish_unpooled_connection,
+//     user::{UserForm, User_},
+//     Crud,
+//     Followable,
+//     ListingType,
+//     SortType,
+//   };
+
+//   #[test]
+//   fn test_crud() {
+//     let conn = establish_unpooled_connection();
+
+//     let new_user = UserForm {
+//       name: "thommy_community_agg".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 another_user = UserForm {
+//       name: "jerry_community_agg".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 another_inserted_user = User_::create(&conn, &another_user).unwrap();
+
+//     let new_community = CommunityForm {
+//       name: "TIL_community_agg".into(),
+//       creator_id: inserted_user.id,
+//       title: "nada".to_owned(),
+//       description: None,
+//       category_id: 1,
+//       nsfw: false,
+//       removed: None,
+//       deleted: None,
+//       updated: None,
+//       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 another_community = CommunityForm {
+//       name: "TIL_community_agg_2".into(),
+//       creator_id: inserted_user.id,
+//       title: "nada".to_owned(),
+//       description: None,
+//       category_id: 1,
+//       nsfw: false,
+//       removed: None,
+//       deleted: None,
+//       updated: None,
+//       actor_id: None,
+//       local: true,
+//       private_key: None,
+//       public_key: None,
+//       last_refreshed_at: None,
+//       published: None,
+//       icon: None,
+//       banner: None,
+//     };
+
+//     let another_inserted_community = Community::create(&conn, &another_community).unwrap();
+
+//     let first_user_follow = CommunityFollowerForm {
+//       community_id: inserted_community.id,
+//       user_id: inserted_user.id,
+//       pending: false,
+//     };
+
+//     CommunityFollower::follow(&conn, &first_user_follow).unwrap();
+
+//     let second_user_follow = CommunityFollowerForm {
+//       community_id: inserted_community.id,
+//       user_id: another_inserted_user.id,
+//       pending: false,
+//     };
+
+//     CommunityFollower::follow(&conn, &second_user_follow).unwrap();
+
+//     let another_community_follow = CommunityFollowerForm {
+//       community_id: another_inserted_community.id,
+//       user_id: inserted_user.id,
+//       pending: false,
+//     };
+
+//     CommunityFollower::follow(&conn, &another_community_follow).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 comment_form = CommentForm {
+//       content: "A test comment".into(),
+//       creator_id: inserted_user.id,
+//       post_id: inserted_post.id,
+//       removed: None,
+//       deleted: None,
+//       read: None,
+//       parent_id: None,
+//       published: None,
+//       updated: None,
+//       ap_id: None,
+//       local: true,
+//     };
+
+//     let inserted_comment = Comment::create(&conn, &comment_form).unwrap();
+
+//     let child_comment_form = CommentForm {
+//       content: "A test comment".into(),
+//       creator_id: inserted_user.id,
+//       post_id: inserted_post.id,
+//       removed: None,
+//       deleted: None,
+//       read: None,
+//       parent_id: Some(inserted_comment.id),
+//       published: None,
+//       updated: None,
+//       ap_id: None,
+//       local: true,
+//     };
+
+//     let _inserted_child_comment = Comment::create(&conn, &child_comment_form).unwrap();
+
+//     let community_aggregates_before_delete =
+//       CommunityAggregates::read(&conn, inserted_community.id).unwrap();
+
+//     assert_eq!(2, community_aggregates_before_delete.subscribers);
+//     assert_eq!(1, community_aggregates_before_delete.posts);
+//     assert_eq!(2, community_aggregates_before_delete.comments);
+
+//     // Test the other community
+//     let another_community_aggs =
+//       CommunityAggregates::read(&conn, another_inserted_community.id).unwrap();
+//     assert_eq!(1, another_community_aggs.subscribers);
+//     assert_eq!(0, another_community_aggs.posts);
+//     assert_eq!(0, another_community_aggs.comments);
+
+//     // Unfollow test
+//     CommunityFollower::unfollow(&conn, &second_user_follow).unwrap();
+//     let after_unfollow = CommunityAggregates::read(&conn, inserted_community.id).unwrap();
+//     assert_eq!(1, after_unfollow.subscribers);
+
+//     // Follow again just for the later tests
+//     CommunityFollower::follow(&conn, &second_user_follow).unwrap();
+//     let after_follow_again = CommunityAggregates::read(&conn, inserted_community.id).unwrap();
+//     assert_eq!(2, after_follow_again.subscribers);
+
+//     // Remove a parent comment (the comment count should also be 0)
+//     Post::delete(&conn, inserted_post.id).unwrap();
+//     let after_parent_post_delete = CommunityAggregates::read(&conn, inserted_community.id).unwrap();
+//     assert_eq!(0, after_parent_post_delete.comments);
+//     assert_eq!(0, after_parent_post_delete.posts);
+
+//     // Remove the 2nd user
+//     User_::delete(&conn, another_inserted_user.id).unwrap();
+//     let after_user_delete = CommunityAggregates::read(&conn, inserted_community.id).unwrap();
+//     assert_eq!(1, after_user_delete.subscribers);
+
+//     // This should delete all the associated rows, and fire triggers
+//     let user_num_deleted = User_::delete(&conn, inserted_user.id).unwrap();
+//     assert_eq!(1, user_num_deleted);
+
+//     // Should be none found, since the creator was deleted
+//     let after_delete = CommunityAggregates::read(&conn, inserted_community.id);
+//     assert!(after_delete.is_err());
+//   }
+// }
index 530f475b4c35b68299db7162b5a7052cb11402d0..5767c72b7c3bba8c62902ecc477e3f9acf8d55a0 100644 (file)
@@ -8,9 +8,10 @@ use crate::{
   Saveable,
 };
 use diesel::{dsl::*, result::Error, *};
+use serde::Serialize;
 use url::{ParseError, Url};
 
-#[derive(Queryable, Identifiable, PartialEq, Debug)]
+#[derive(Clone, Queryable, Identifiable, PartialEq, Debug, Serialize)]
 #[table_name = "post"]
 pub struct Post {
   pub id: i32,
index e6dd6d4bd2f6b4c8dd452e51bff7c7549915573d..b0c57f5e15e23bb919f526b0a8befbb79ba58eb5 100644 (file)
@@ -329,6 +329,18 @@ table! {
     }
 }
 
+table! {
+    post_aggregates (id) {
+        id -> Int4,
+        post_id -> Int4,
+        comments -> Int8,
+        score -> Int8,
+        upvotes -> Int8,
+        downvotes -> Int8,
+        newest_comment_time -> Timestamp,
+    }
+}
+
 table! {
     post_aggregates_fast (id) {
         id -> Int4,
@@ -576,6 +588,7 @@ joinable!(mod_sticky_post -> user_ (mod_user_id));
 joinable!(password_reset_request -> user_ (user_id));
 joinable!(post -> community (community_id));
 joinable!(post -> user_ (creator_id));
+joinable!(post_aggregates -> post (post_id));
 joinable!(post_like -> post (post_id));
 joinable!(post_like -> user_ (user_id));
 joinable!(post_read -> post (post_id));
@@ -614,6 +627,7 @@ allow_tables_to_appear_in_same_query!(
   mod_sticky_post,
   password_reset_request,
   post,
+  post_aggregates,
   post_aggregates_fast,
   post_like,
   post_read,
index cbb90a4276b836fe7e6746863ef7987778b5dac4..1ddbba64ff5f22aa69b3d0320408fb3e97775de1 100644 (file)
@@ -32,7 +32,7 @@ impl CommunityView {
     // The left join below will return None in this case
     let user_id_join = my_user_id.unwrap_or(-1);
 
-    let (community, creator, category, counts, subscribed) = community::table
+    let (community, creator, category, counts, follower) = community::table
       .find(community_id)
       .inner_join(user_::table)
       .inner_join(category::table)
@@ -63,7 +63,7 @@ impl CommunityView {
       community,
       creator,
       category,
-      subscribed: subscribed.is_some(),
+      subscribed: follower.is_some(),
       counts,
     })
   }
index 0aff8eae2b44ab63061a143fb10ad8c324654841..a8112f0813cfcc5762a396b331d3f841721d39e8 100644 (file)
@@ -2,7 +2,6 @@ pub mod community_follower_view;
 pub mod community_moderator_view;
 pub mod community_user_ban_view;
 pub mod community_view;
+pub mod post_view;
 pub mod site_view;
 pub mod user_view;
-
-// TODO Every single aggregate trigger is likely broken, you need to test every one of these out
diff --git a/lemmy_db/src/views/post_view.rs b/lemmy_db/src/views/post_view.rs
new file mode 100644 (file)
index 0000000..f7b6ac7
--- /dev/null
@@ -0,0 +1,576 @@
+use crate::{
+  aggregates::post_aggregates::PostAggregates,
+  community::{Community, CommunityFollower, CommunitySafe, CommunityUserBan},
+  functions::hot_rank,
+  fuzzy_search,
+  limit_and_offset,
+  post::{Post, PostRead, PostSaved},
+  schema::{
+    community,
+    community_follower,
+    community_user_ban,
+    post,
+    post_aggregates,
+    post_like,
+    post_read,
+    post_saved,
+    user_,
+  },
+  user::{UserSafe, User_},
+  ListingType,
+  MaybeOptional,
+  SortType,
+  ToSafe,
+};
+use diesel::{result::Error, *};
+use serde::Serialize;
+
+#[derive(Debug, Serialize, Clone)]
+pub struct PostView {
+  pub post: Post,
+  pub creator: UserSafe,
+  pub community: CommunitySafe,
+  pub counts: PostAggregates,
+  pub subscribed: bool,            // Left join to CommunityFollower
+  pub banned_from_community: bool, // Left Join to CommunityUserBan
+  pub saved: bool,                 // Left join to PostSaved
+  pub read: bool,                  // Left join to PostRead
+  pub my_vote: Option<i16>,        // Left join to PostLike
+}
+
+type OutputTuple = (
+  Post,
+  UserSafe,
+  CommunitySafe,
+  PostAggregates,
+  Option<CommunityFollower>,
+  Option<CommunityUserBan>,
+  Option<PostSaved>,
+  Option<PostRead>,
+  Option<i16>,
+);
+
+impl PostView {
+  pub fn read(conn: &PgConnection, post_id: i32, my_user_id: Option<i32>) -> Result<Self, Error> {
+    // The left join below will return None in this case
+    let user_id_join = my_user_id.unwrap_or(-1);
+
+    let (post, creator, community, counts, follower, banned_from_community, saved, read, my_vote) =
+      post::table
+        .find(post_id)
+        .inner_join(user_::table)
+        .inner_join(community::table)
+        .inner_join(post_aggregates::table)
+        .left_join(
+          community_follower::table.on(
+            post::community_id
+              .eq(community_follower::community_id)
+              .and(community_follower::user_id.eq(user_id_join)),
+          ),
+        )
+        .left_join(
+          community_user_ban::table.on(
+            post::community_id
+              .eq(community_user_ban::community_id)
+              .and(community_user_ban::user_id.eq(user_id_join)),
+          ),
+        )
+        .left_join(
+          post_saved::table.on(
+            post::id
+              .eq(post_saved::post_id)
+              .and(post_saved::user_id.eq(user_id_join)),
+          ),
+        )
+        .left_join(
+          post_read::table.on(
+            post::id
+              .eq(post_read::post_id)
+              .and(post_read::user_id.eq(user_id_join)),
+          ),
+        )
+        .left_join(
+          post_like::table.on(
+            post::id
+              .eq(post_like::post_id)
+              .and(post_like::user_id.eq(user_id_join)),
+          ),
+        )
+        .select((
+          post::all_columns,
+          User_::safe_columns_tuple(),
+          Community::safe_columns_tuple(),
+          post_aggregates::all_columns,
+          community_follower::all_columns.nullable(),
+          community_user_ban::all_columns.nullable(),
+          post_saved::all_columns.nullable(),
+          post_read::all_columns.nullable(),
+          post_like::score.nullable(),
+        ))
+        .first::<OutputTuple>(conn)?;
+
+    Ok(PostView {
+      post,
+      creator,
+      community,
+      counts,
+      subscribed: follower.is_some(),
+      banned_from_community: banned_from_community.is_some(),
+      saved: saved.is_some(),
+      read: read.is_some(),
+      my_vote,
+    })
+  }
+}
+
+mod join_types {
+  use crate::schema::{
+    community,
+    community_follower,
+    community_user_ban,
+    post,
+    post_aggregates,
+    post_like,
+    post_read,
+    post_saved,
+    user_,
+  };
+  use diesel::{
+    pg::Pg,
+    query_builder::BoxedSelectStatement,
+    query_source::joins::{Inner, Join, JoinOn, LeftOuter},
+    sql_types::*,
+  };
+
+  /// TODO awful, but necessary because of the boxed join
+  pub(super) type BoxedPostJoin<'a> = BoxedSelectStatement<
+    'a,
+    (
+      (
+        Integer,
+        Text,
+        Nullable<Text>,
+        Nullable<Text>,
+        Integer,
+        Integer,
+        Bool,
+        Bool,
+        Timestamp,
+        Nullable<Timestamp>,
+        Bool,
+        Bool,
+        Bool,
+        Nullable<Text>,
+        Nullable<Text>,
+        Nullable<Text>,
+        Nullable<Text>,
+        Text,
+        Bool,
+      ),
+      (
+        Integer,
+        Text,
+        Nullable<Text>,
+        Nullable<Text>,
+        Bool,
+        Bool,
+        Timestamp,
+        Nullable<Timestamp>,
+        Nullable<Text>,
+        Text,
+        Nullable<Text>,
+        Bool,
+        Nullable<Text>,
+        Bool,
+      ),
+      (
+        Integer,
+        Text,
+        Text,
+        Nullable<Text>,
+        Integer,
+        Integer,
+        Bool,
+        Timestamp,
+        Nullable<Timestamp>,
+        Bool,
+        Bool,
+        Text,
+        Bool,
+        Nullable<Text>,
+        Nullable<Text>,
+      ),
+      (Integer, Integer, BigInt, BigInt, BigInt, BigInt, Timestamp),
+      Nullable<(Integer, Integer, Integer, Timestamp, Nullable<Bool>)>,
+      Nullable<(Integer, Integer, Integer, Timestamp)>,
+      Nullable<(Integer, Integer, Integer, Timestamp)>,
+      Nullable<(Integer, Integer, Integer, Timestamp)>,
+      Nullable<SmallInt>,
+    ),
+    JoinOn<
+      Join<
+        JoinOn<
+          Join<
+            JoinOn<
+              Join<
+                JoinOn<
+                  Join<
+                    JoinOn<
+                      Join<
+                        JoinOn<
+                          Join<
+                            JoinOn<
+                              Join<
+                                JoinOn<
+                                  Join<post::table, user_::table, Inner>,
+                                  diesel::expression::operators::Eq<
+                                    diesel::expression::nullable::Nullable<
+                                      post::columns::creator_id,
+                                    >,
+                                    diesel::expression::nullable::Nullable<user_::columns::id>,
+                                  >,
+                                >,
+                                community::table,
+                                Inner,
+                              >,
+                              diesel::expression::operators::Eq<
+                                diesel::expression::nullable::Nullable<post::columns::community_id>,
+                                diesel::expression::nullable::Nullable<community::columns::id>,
+                              >,
+                            >,
+                            post_aggregates::table,
+                            Inner,
+                          >,
+                          diesel::expression::operators::Eq<
+                            diesel::expression::nullable::Nullable<
+                              post_aggregates::columns::post_id,
+                            >,
+                            diesel::expression::nullable::Nullable<post::columns::id>,
+                          >,
+                        >,
+                        community_follower::table,
+                        LeftOuter,
+                      >,
+                      diesel::expression::operators::And<
+                        diesel::expression::operators::Eq<
+                          post::columns::community_id,
+                          community_follower::columns::community_id,
+                        >,
+                        diesel::expression::operators::Eq<
+                          community_follower::columns::user_id,
+                          diesel::expression::bound::Bound<diesel::sql_types::Integer, i32>,
+                        >,
+                      >,
+                    >,
+                    community_user_ban::table,
+                    LeftOuter,
+                  >,
+                  diesel::expression::operators::And<
+                    diesel::expression::operators::Eq<
+                      post::columns::community_id,
+                      community_user_ban::columns::community_id,
+                    >,
+                    diesel::expression::operators::Eq<
+                      community_user_ban::columns::user_id,
+                      diesel::expression::bound::Bound<diesel::sql_types::Integer, i32>,
+                    >,
+                  >,
+                >,
+                post_saved::table,
+                LeftOuter,
+              >,
+              diesel::expression::operators::And<
+                diesel::expression::operators::Eq<post::columns::id, post_saved::columns::post_id>,
+                diesel::expression::operators::Eq<
+                  post_saved::columns::user_id,
+                  diesel::expression::bound::Bound<diesel::sql_types::Integer, i32>,
+                >,
+              >,
+            >,
+            post_read::table,
+            LeftOuter,
+          >,
+          diesel::expression::operators::And<
+            diesel::expression::operators::Eq<post::columns::id, post_read::columns::post_id>,
+            diesel::expression::operators::Eq<
+              post_read::columns::user_id,
+              diesel::expression::bound::Bound<diesel::sql_types::Integer, i32>,
+            >,
+          >,
+        >,
+        post_like::table,
+        LeftOuter,
+      >,
+      diesel::expression::operators::And<
+        diesel::expression::operators::Eq<post::columns::id, post_like::columns::post_id>,
+        diesel::expression::operators::Eq<
+          post_like::columns::user_id,
+          diesel::expression::bound::Bound<diesel::sql_types::Integer, i32>,
+        >,
+      >,
+    >,
+    Pg,
+  >;
+}
+
+pub struct PostQueryBuilder<'a> {
+  conn: &'a PgConnection,
+  query: join_types::BoxedPostJoin<'a>,
+  listing_type: &'a ListingType,
+  sort: &'a SortType,
+  for_creator_id: Option<i32>,
+  for_community_id: Option<i32>,
+  for_community_name: Option<String>,
+  search_term: Option<String>,
+  url_search: Option<String>,
+  show_nsfw: bool,
+  saved_only: bool,
+  unread_only: bool,
+  page: Option<i64>,
+  limit: Option<i64>,
+}
+
+impl<'a> PostQueryBuilder<'a> {
+  pub fn create(conn: &'a PgConnection, my_user_id: Option<i32>) -> Self {
+    // The left join below will return None in this case
+    let user_id_join = my_user_id.unwrap_or(-1);
+
+    let query = post::table
+      .inner_join(user_::table)
+      .inner_join(community::table)
+      .inner_join(post_aggregates::table)
+      .left_join(
+        community_follower::table.on(
+          post::community_id
+            .eq(community_follower::community_id)
+            .and(community_follower::user_id.eq(user_id_join)),
+        ),
+      )
+      .left_join(
+        community_user_ban::table.on(
+          post::community_id
+            .eq(community_user_ban::community_id)
+            .and(community_user_ban::user_id.eq(user_id_join)),
+        ),
+      )
+      .left_join(
+        post_saved::table.on(
+          post::id
+            .eq(post_saved::post_id)
+            .and(post_saved::user_id.eq(user_id_join)),
+        ),
+      )
+      .left_join(
+        post_read::table.on(
+          post::id
+            .eq(post_read::post_id)
+            .and(post_read::user_id.eq(user_id_join)),
+        ),
+      )
+      .left_join(
+        post_like::table.on(
+          post::id
+            .eq(post_like::post_id)
+            .and(post_like::user_id.eq(user_id_join)),
+        ),
+      )
+      .select((
+        post::all_columns,
+        User_::safe_columns_tuple(),
+        Community::safe_columns_tuple(),
+        post_aggregates::all_columns,
+        community_follower::all_columns.nullable(),
+        community_user_ban::all_columns.nullable(),
+        post_saved::all_columns.nullable(),
+        post_read::all_columns.nullable(),
+        post_like::score.nullable(),
+      ))
+      .into_boxed();
+
+    PostQueryBuilder {
+      conn,
+      query,
+      listing_type: &ListingType::All,
+      sort: &SortType::Hot,
+      for_creator_id: None,
+      for_community_id: None,
+      for_community_name: None,
+      search_term: None,
+      url_search: None,
+      show_nsfw: true,
+      saved_only: false,
+      unread_only: false,
+      page: None,
+      limit: None,
+    }
+  }
+
+  pub fn listing_type(mut self, listing_type: &'a ListingType) -> Self {
+    self.listing_type = listing_type;
+    self
+  }
+
+  pub fn sort(mut self, sort: &'a SortType) -> Self {
+    self.sort = sort;
+    self
+  }
+
+  pub fn for_community_id<T: MaybeOptional<i32>>(mut self, for_community_id: T) -> Self {
+    self.for_community_id = for_community_id.get_optional();
+    self
+  }
+
+  pub fn for_community_name<T: MaybeOptional<String>>(mut self, for_community_name: T) -> Self {
+    self.for_community_name = for_community_name.get_optional();
+    self
+  }
+
+  pub fn for_creator_id<T: MaybeOptional<i32>>(mut self, for_creator_id: T) -> Self {
+    self.for_creator_id = for_creator_id.get_optional();
+    self
+  }
+
+  pub fn search_term<T: MaybeOptional<String>>(mut self, search_term: T) -> Self {
+    self.search_term = search_term.get_optional();
+    self
+  }
+
+  pub fn url_search<T: MaybeOptional<String>>(mut self, url_search: T) -> Self {
+    self.url_search = url_search.get_optional();
+    self
+  }
+
+  pub fn show_nsfw(mut self, show_nsfw: bool) -> Self {
+    self.show_nsfw = show_nsfw;
+    self
+  }
+
+  pub fn saved_only(mut self, saved_only: bool) -> Self {
+    self.saved_only = saved_only;
+    self
+  }
+
+  pub fn page<T: MaybeOptional<i64>>(mut self, page: T) -> Self {
+    self.page = page.get_optional();
+    self
+  }
+
+  pub fn limit<T: MaybeOptional<i64>>(mut self, limit: T) -> Self {
+    self.limit = limit.get_optional();
+    self
+  }
+
+  pub fn list(self) -> Result<Vec<PostView>, Error> {
+    use diesel::dsl::*;
+
+    let mut query = self.query;
+
+    query = match self.listing_type {
+      ListingType::Subscribed => query.filter(community_follower::user_id.is_not_null()), // TODO could be this: and(community_follower::user_id.eq(user_id_join)),
+      ListingType::Local => query.filter(community::local.eq(true)),
+      _ => query,
+    };
+
+    if let Some(for_community_id) = self.for_community_id {
+      query = query
+        .filter(post::community_id.eq(for_community_id))
+        .then_order_by(post::stickied.desc());
+    }
+
+    if let Some(for_community_name) = self.for_community_name {
+      query = query
+        .filter(community::name.eq(for_community_name))
+        .filter(community::local.eq(true))
+        .then_order_by(post::stickied.desc());
+    }
+
+    if let Some(url_search) = self.url_search {
+      query = query.filter(post::url.eq(url_search));
+    }
+
+    if let Some(search_term) = self.search_term {
+      let searcher = fuzzy_search(&search_term);
+      query = query.filter(
+        post::name
+          .ilike(searcher.to_owned())
+          .or(post::body.ilike(searcher)),
+      );
+    }
+
+    query = match self.sort {
+      SortType::Active => query
+        .then_order_by(
+          hot_rank(post_aggregates::score, post_aggregates::newest_comment_time).desc(),
+        )
+        .then_order_by(post::published.desc()),
+      SortType::Hot => query
+        .then_order_by(hot_rank(post_aggregates::score, post::published).desc())
+        .then_order_by(post::published.desc()),
+      SortType::New => query.then_order_by(post::published.desc()),
+      SortType::TopAll => query.then_order_by(post_aggregates::score.desc()),
+      SortType::TopYear => query
+        .filter(post::published.gt(now - 1.years()))
+        .then_order_by(post_aggregates::score.desc()),
+      SortType::TopMonth => query
+        .filter(post::published.gt(now - 1.months()))
+        .then_order_by(post_aggregates::score.desc()),
+      SortType::TopWeek => query
+        .filter(post::published.gt(now - 1.weeks()))
+        .then_order_by(post_aggregates::score.desc()),
+      SortType::TopDay => query
+        .filter(post::published.gt(now - 1.days()))
+        .then_order_by(post_aggregates::score.desc()),
+    };
+
+    // If its for a specific user, show the removed / deleted
+    if let Some(for_creator_id) = self.for_creator_id {
+      query = query.filter(post::creator_id.eq(for_creator_id));
+    }
+
+    if !self.show_nsfw {
+      query = query
+        .filter(post::nsfw.eq(false))
+        .filter(community::nsfw.eq(false));
+    };
+
+    // TODO  These two might be wrong
+    if self.saved_only {
+      query = query.filter(post_saved::id.is_not_null());
+    };
+
+    if self.unread_only {
+      query = query.filter(post_read::id.is_not_null());
+    };
+
+    let (limit, offset) = limit_and_offset(self.page, self.limit);
+
+    let res = query
+      .limit(limit)
+      .offset(offset)
+      .filter(post::removed.eq(false))
+      .filter(post::deleted.eq(false))
+      .filter(community::removed.eq(false))
+      .filter(community::deleted.eq(false))
+      .load::<OutputTuple>(self.conn)?;
+
+    Ok(to_vec(res))
+  }
+}
+
+// TODO turn this into a trait with an associated type
+fn to_vec(posts: Vec<OutputTuple>) -> Vec<PostView> {
+  posts
+    .iter()
+    .map(|a| PostView {
+      post: a.0.to_owned(),
+      creator: a.1.to_owned(),
+      community: a.2.to_owned(),
+      counts: a.3.to_owned(),
+      subscribed: a.4.is_some(),
+      banned_from_community: a.5.is_some(),
+      saved: a.6.is_some(),
+      read: a.7.is_some(),
+      my_vote: a.8,
+    })
+    .collect::<Vec<PostView>>()
+}
diff --git a/migrations/2020-12-10-152350_create_post_aggregates/down.sql b/migrations/2020-12-10-152350_create_post_aggregates/down.sql
new file mode 100644 (file)
index 0000000..113f26d
--- /dev/null
@@ -0,0 +1,9 @@
+-- post aggregates
+drop table post_aggregates;
+drop trigger post_aggregates_post on post;
+drop trigger post_aggregates_comment_count on comment;
+drop trigger post_aggregates_score on post_like;
+drop function 
+  post_aggregates_post,
+  post_aggregates_comment_count,
+  post_aggregates_score;
diff --git a/migrations/2020-12-10-152350_create_post_aggregates/up.sql b/migrations/2020-12-10-152350_create_post_aggregates/up.sql
new file mode 100644 (file)
index 0000000..f9321be
--- /dev/null
@@ -0,0 +1,107 @@
+-- Add post aggregates
+create table post_aggregates (
+  id serial primary key,
+  post_id int references post on update cascade on delete cascade not null,
+  comments bigint not null default 0,
+  score bigint not null default 0,
+  upvotes bigint not null default 0,
+  downvotes bigint not null default 0,
+  newest_comment_time timestamp not null default now(),
+  unique (post_id)
+);
+
+insert into post_aggregates (post_id, comments, score, upvotes, downvotes, newest_comment_time)
+  select 
+    p.id,
+    coalesce(ct.comments, 0::bigint) as comments,
+    coalesce(pl.score, 0::bigint) as score,
+    coalesce(pl.upvotes, 0::bigint) as upvotes,
+    coalesce(pl.downvotes, 0::bigint) as downvotes,
+    greatest(ct.recent_comment_time, p.published) as newest_activity_time
+  from post p
+  left join ( 
+    select comment.post_id,
+    count(*) as comments,
+    max(comment.published) as recent_comment_time
+    from comment
+    group by comment.post_id
+  ) ct on ct.post_id = p.id
+  left join ( 
+    select post_like.post_id,
+    sum(post_like.score) as score,
+    sum(post_like.score) filter (where post_like.score = 1) as upvotes,
+    -sum(post_like.score) filter (where post_like.score = '-1'::integer) as downvotes
+    from post_like
+    group by post_like.post_id
+  ) pl on pl.post_id = p.id;
+
+-- Add community aggregate triggers
+
+-- initial post add
+create function post_aggregates_post()
+returns trigger language plpgsql
+as $$
+begin
+  IF (TG_OP = 'INSERT') THEN
+    insert into post_aggregates (post_id) values (NEW.id);
+  ELSIF (TG_OP = 'DELETE') THEN
+    delete from post_aggregates where post_id = OLD.id;
+  END IF;
+  return null;
+end $$;
+
+create trigger post_aggregates_post
+after insert or delete on post
+for each row
+execute procedure post_aggregates_post();
+
+-- comment count
+create function post_aggregates_comment_count()
+returns trigger language plpgsql
+as $$
+begin
+  IF (TG_OP = 'INSERT') THEN
+    update post_aggregates pa
+    set comments = comments + 1,
+    newest_comment_time = NEW.published
+    where pa.post_id = NEW.post_id;
+  ELSIF (TG_OP = 'DELETE') THEN
+    update post_aggregates pa
+    set comments = comments - 1
+    where pa.post_id = OLD.post_id;
+  END IF;
+  return null;
+end $$;
+
+create trigger post_aggregates_comment_count
+after insert or delete on comment
+for each row
+execute procedure post_aggregates_comment_count();
+
+-- post score
+create function post_aggregates_score()
+returns trigger language plpgsql
+as $$
+begin
+  IF (TG_OP = 'INSERT') THEN
+    update post_aggregates pa
+    set score = score + NEW.score,
+    upvotes = case when NEW.score = 1 then upvotes + 1 else upvotes end,
+    downvotes = case when NEW.score = -1 then downvotes + 1 else downvotes end
+    where pa.post_id = NEW.post_id;
+
+  ELSIF (TG_OP = 'DELETE') THEN
+    update post_aggregates pa
+    set score = score - OLD.score,
+    upvotes = case when OLD.score = 1 then upvotes - 1 else upvotes end,
+    downvotes = case when OLD.score = -1 then downvotes - 1 else downvotes end
+    where pa.post_id = OLD.post_id;
+
+  END IF;
+  return null;
+end $$;
+
+create trigger post_aggregates_score
+after insert or delete on post_like
+for each row
+execute procedure post_aggregates_score();