use super::community_view::community_view::BoxedQuery; use crate::db::{fuzzy_search, limit_and_offset, MaybeOptional, SortType}; use diesel::{pg::Pg, result::Error, *}; use serde::{Deserialize, Serialize}; table! { community_view (id) { id -> Int4, name -> Varchar, title -> Varchar, description -> Nullable, category_id -> Int4, creator_id -> Int4, removed -> Bool, published -> Timestamp, updated -> Nullable, deleted -> Bool, nsfw -> Bool, actor_id -> Text, local -> Bool, last_refreshed_at -> Timestamp, creator_actor_id -> Text, creator_local -> Bool, creator_name -> Varchar, creator_avatar -> Nullable, category_name -> Varchar, number_of_subscribers -> BigInt, number_of_posts -> BigInt, number_of_comments -> BigInt, hot_rank -> Int4, user_id -> Nullable, subscribed -> Nullable, } } table! { community_moderator_view (id) { id -> Int4, community_id -> Int4, user_id -> Int4, published -> Timestamp, user_actor_id -> Text, user_local -> Bool, user_name -> Varchar, avatar -> Nullable, community_actor_id -> Text, community_local -> Bool, community_name -> Varchar, } } table! { community_follower_view (id) { id -> Int4, community_id -> Int4, user_id -> Int4, published -> Timestamp, user_actor_id -> Text, user_local -> Bool, user_name -> Varchar, avatar -> Nullable, community_actor_id -> Text, community_local -> Bool, community_name -> Varchar, } } table! { community_user_ban_view (id) { id -> Int4, community_id -> Int4, user_id -> Int4, published -> Timestamp, user_actor_id -> Text, user_local -> Bool, user_name -> Varchar, avatar -> Nullable, community_actor_id -> Text, community_local -> Bool, community_name -> Varchar, } } #[derive( Queryable, Identifiable, PartialEq, Debug, Serialize, Deserialize, QueryableByName, Clone, )] #[table_name = "community_view"] pub struct CommunityView { pub id: i32, pub name: String, pub title: String, pub description: Option, pub category_id: i32, pub creator_id: i32, pub removed: bool, pub published: chrono::NaiveDateTime, pub updated: Option, pub deleted: bool, pub nsfw: bool, pub actor_id: String, pub local: bool, pub last_refreshed_at: chrono::NaiveDateTime, pub creator_actor_id: String, pub creator_local: bool, pub creator_name: String, pub creator_avatar: Option, pub category_name: String, pub number_of_subscribers: i64, pub number_of_posts: i64, pub number_of_comments: i64, pub hot_rank: i32, pub user_id: Option, pub subscribed: Option, } pub struct CommunityQueryBuilder<'a> { conn: &'a PgConnection, query: BoxedQuery<'a, Pg>, sort: &'a SortType, from_user_id: Option, show_nsfw: bool, search_term: Option, page: Option, limit: Option, } impl<'a> CommunityQueryBuilder<'a> { pub fn create(conn: &'a PgConnection) -> Self { use super::community_view::community_view::dsl::*; let query = community_view.into_boxed(); CommunityQueryBuilder { conn, query, sort: &SortType::Hot, from_user_id: None, show_nsfw: true, search_term: None, page: None, limit: None, } } pub fn sort(mut self, sort: &'a SortType) -> Self { self.sort = sort; self } pub fn for_user>(mut self, from_user_id: T) -> Self { self.from_user_id = from_user_id.get_optional(); self } pub fn show_nsfw(mut self, show_nsfw: bool) -> Self { self.show_nsfw = show_nsfw; self } pub fn search_term>(mut self, search_term: T) -> Self { self.search_term = search_term.get_optional(); self } pub fn page>(mut self, page: T) -> Self { self.page = page.get_optional(); self } pub fn limit>(mut self, limit: T) -> Self { self.limit = limit.get_optional(); self } pub fn list(self) -> Result, Error> { use super::community_view::community_view::dsl::*; let mut query = self.query; if let Some(search_term) = self.search_term { let searcher = fuzzy_search(&search_term); query = query .filter(name.ilike(searcher.to_owned())) .or_filter(title.ilike(searcher.to_owned())) .or_filter(description.ilike(searcher)); }; // The view lets you pass a null user_id, if you're not logged in match self.sort { SortType::Hot => { query = query .order_by(hot_rank.desc()) .then_order_by(number_of_subscribers.desc()) .filter(user_id.is_null()) } SortType::New => query = query.order_by(published.desc()).filter(user_id.is_null()), SortType::TopAll => match self.from_user_id { Some(from_user_id) => { query = query .filter(user_id.eq(from_user_id)) .order_by((subscribed.asc(), number_of_subscribers.desc())) } None => { query = query .order_by(number_of_subscribers.desc()) .filter(user_id.is_null()) } }, _ => (), }; if !self.show_nsfw { query = query.filter(nsfw.eq(false)); }; let (limit, offset) = limit_and_offset(self.page, self.limit); query .limit(limit) .offset(offset) .filter(removed.eq(false)) .filter(deleted.eq(false)) .load::(self.conn) } } impl CommunityView { pub fn read( conn: &PgConnection, from_community_id: i32, from_user_id: Option, ) -> Result { use super::community_view::community_view::dsl::*; let mut query = community_view.into_boxed(); query = query.filter(id.eq(from_community_id)); // The view lets you pass a null user_id, if you're not logged in if let Some(from_user_id) = from_user_id { query = query.filter(user_id.eq(from_user_id)); } else { query = query.filter(user_id.is_null()); }; query.first::(conn) } } #[derive( Queryable, Identifiable, PartialEq, Debug, Serialize, Deserialize, QueryableByName, Clone, )] #[table_name = "community_moderator_view"] pub struct CommunityModeratorView { pub id: i32, pub community_id: i32, pub user_id: i32, pub published: chrono::NaiveDateTime, pub user_actor_id: String, pub user_local: bool, pub user_name: String, pub avatar: Option, pub community_actor_id: String, pub community_local: bool, pub community_name: String, } impl CommunityModeratorView { pub fn for_community(conn: &PgConnection, from_community_id: i32) -> Result, Error> { use super::community_view::community_moderator_view::dsl::*; community_moderator_view .filter(community_id.eq(from_community_id)) .load::(conn) } pub fn for_user(conn: &PgConnection, from_user_id: i32) -> Result, Error> { use super::community_view::community_moderator_view::dsl::*; community_moderator_view .filter(user_id.eq(from_user_id)) .load::(conn) } } #[derive( Queryable, Identifiable, PartialEq, Debug, Serialize, Deserialize, QueryableByName, Clone, )] #[table_name = "community_follower_view"] pub struct CommunityFollowerView { pub id: i32, pub community_id: i32, pub user_id: i32, pub published: chrono::NaiveDateTime, pub user_actor_id: String, pub user_local: bool, pub user_name: String, pub avatar: Option, pub community_actor_id: String, pub community_local: bool, pub community_name: String, } impl CommunityFollowerView { pub fn for_community(conn: &PgConnection, from_community_id: i32) -> Result, Error> { use super::community_view::community_follower_view::dsl::*; community_follower_view .filter(community_id.eq(from_community_id)) .load::(conn) } pub fn for_user(conn: &PgConnection, from_user_id: i32) -> Result, Error> { use super::community_view::community_follower_view::dsl::*; community_follower_view .filter(user_id.eq(from_user_id)) .load::(conn) } } #[derive( Queryable, Identifiable, PartialEq, Debug, Serialize, Deserialize, QueryableByName, Clone, )] #[table_name = "community_user_ban_view"] pub struct CommunityUserBanView { pub id: i32, pub community_id: i32, pub user_id: i32, pub published: chrono::NaiveDateTime, pub user_actor_id: String, pub user_local: bool, pub user_name: String, pub avatar: Option, pub community_actor_id: String, pub community_local: bool, pub community_name: String, } impl CommunityUserBanView { pub fn for_community(conn: &PgConnection, from_community_id: i32) -> Result, Error> { use super::community_view::community_user_ban_view::dsl::*; community_user_ban_view .filter(community_id.eq(from_community_id)) .load::(conn) } pub fn for_user(conn: &PgConnection, from_user_id: i32) -> Result, Error> { use super::community_view::community_user_ban_view::dsl::*; community_user_ban_view .filter(user_id.eq(from_user_id)) .load::(conn) } pub fn get( conn: &PgConnection, from_user_id: i32, from_community_id: i32, ) -> Result { use super::community_view::community_user_ban_view::dsl::*; community_user_ban_view .filter(user_id.eq(from_user_id)) .filter(community_id.eq(from_community_id)) .first::(conn) } }