summaryrefslogtreecommitdiffstats
path: root/server/lemmy_db/src/community_view.rs
diff options
context:
space:
mode:
Diffstat (limited to 'server/lemmy_db/src/community_view.rs')
-rw-r--r--server/lemmy_db/src/community_view.rs391
1 files changed, 391 insertions, 0 deletions
diff --git a/server/lemmy_db/src/community_view.rs b/server/lemmy_db/src/community_view.rs
new file mode 100644
index 00000000..b465ddab
--- /dev/null
+++ b/server/lemmy_db/src/community_view.rs
@@ -0,0 +1,391 @@
+use super::community_view::community_fast_view::BoxedQuery;
+use crate::{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<Text>,
+ category_id -> Int4,
+ creator_id -> Int4,
+ removed -> Bool,
+ published -> Timestamp,
+ updated -> Nullable<Timestamp>,
+ 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<Text>,
+ category_name -> Varchar,
+ number_of_subscribers -> BigInt,
+ number_of_posts -> BigInt,
+ number_of_comments -> BigInt,
+ hot_rank -> Int4,
+ user_id -> Nullable<Int4>,
+ subscribed -> Nullable<Bool>,
+ }
+}
+
+table! {
+ community_fast_view (id) {
+ id -> Int4,
+ name -> Varchar,
+ title -> Varchar,
+ description -> Nullable<Text>,
+ category_id -> Int4,
+ creator_id -> Int4,
+ removed -> Bool,
+ published -> Timestamp,
+ updated -> Nullable<Timestamp>,
+ 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<Text>,
+ category_name -> Varchar,
+ number_of_subscribers -> BigInt,
+ number_of_posts -> BigInt,
+ number_of_comments -> BigInt,
+ hot_rank -> Int4,
+ user_id -> Nullable<Int4>,
+ subscribed -> Nullable<Bool>,
+ }
+}
+
+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<Text>,
+ 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<Text>,
+ 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<Text>,
+ community_actor_id -> Text,
+ community_local -> Bool,
+ community_name -> Varchar,
+ }
+}
+
+#[derive(
+ Queryable, Identifiable, PartialEq, Debug, Serialize, Deserialize, QueryableByName, Clone,
+)]
+#[table_name = "community_fast_view"]
+pub struct CommunityView {
+ pub id: i32,
+ pub name: String,
+ pub title: String,
+ pub description: Option<String>,
+ pub category_id: i32,
+ pub creator_id: i32,
+ pub removed: bool,
+ pub published: chrono::NaiveDateTime,
+ pub updated: Option<chrono::NaiveDateTime>,
+ 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<String>,
+ 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<i32>,
+ pub subscribed: Option<bool>,
+}
+
+pub struct CommunityQueryBuilder<'a> {
+ conn: &'a PgConnection,
+ query: BoxedQuery<'a, Pg>,
+ sort: &'a SortType,
+ from_user_id: Option<i32>,
+ show_nsfw: bool,
+ search_term: Option<String>,
+ page: Option<i64>,
+ limit: Option<i64>,
+}
+
+impl<'a> CommunityQueryBuilder<'a> {
+ pub fn create(conn: &'a PgConnection) -> Self {
+ use super::community_view::community_fast_view::dsl::*;
+
+ let query = community_fast_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<T: MaybeOptional<i32>>(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<T: MaybeOptional<String>>(mut self, search_term: T) -> Self {
+ self.search_term = search_term.get_optional();
+ 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<CommunityView>, Error> {
+ use super::community_view::community_fast_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::<CommunityView>(self.conn)
+ }
+}
+
+impl CommunityView {
+ pub fn read(
+ conn: &PgConnection,
+ from_community_id: i32,
+ from_user_id: Option<i32>,
+ ) -> Result<Self, Error> {
+ use super::community_view::community_fast_view::dsl::*;
+
+ let mut query = community_fast_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::<Self>(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<String>,
+ 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<Vec<Self>, Error> {
+ use super::community_view::community_moderator_view::dsl::*;
+ community_moderator_view
+ .filter(community_id.eq(from_community_id))
+ .load::<Self>(conn)
+ }
+
+ pub fn for_user(conn: &PgConnection, from_user_id: i32) -> Result<Vec<Self>, Error> {
+ use super::community_view::community_moderator_view::dsl::*;
+ community_moderator_view
+ .filter(user_id.eq(from_user_id))
+ .load::<Self>(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<String>,
+ 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<Vec<Self>, Error> {
+ use super::community_view::community_follower_view::dsl::*;
+ community_follower_view
+ .filter(community_id.eq(from_community_id))
+ .load::<Self>(conn)
+ }
+
+ pub fn for_user(conn: &PgConnection, from_user_id: i32) -> Result<Vec<Self>, Error> {
+ use super::community_view::community_follower_view::dsl::*;
+ community_follower_view
+ .filter(user_id.eq(from_user_id))
+ .load::<Self>(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<String>,
+ 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<Vec<Self>, Error> {
+ use super::community_view::community_user_ban_view::dsl::*;
+ community_user_ban_view
+ .filter(community_id.eq(from_community_id))
+ .load::<Self>(conn)
+ }
+
+ pub fn for_user(conn: &PgConnection, from_user_id: i32) -> Result<Vec<Self>, Error> {
+ use super::community_view::community_user_ban_view::dsl::*;
+ community_user_ban_view
+ .filter(user_id.eq(from_user_id))
+ .load::<Self>(conn)
+ }
+
+ pub fn get(
+ conn: &PgConnection,
+ from_user_id: i32,
+ from_community_id: i32,
+ ) -> Result<Self, Error> {
+ 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::<Self>(conn)
+ }
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-04-27 18:41:07 +0000