]> Untitled Git - lemmy.git/blob - crates/api_crud/src/post/list.rs
Add default post listing type (fixes #2195) (#2209)
[lemmy.git] / crates / api_crud / src / post / list.rs
1 use crate::PerformCrud;
2 use actix_web::web::Data;
3 use lemmy_api_common::{
4   blocking,
5   check_private_instance,
6   get_local_user_view_from_jwt_opt,
7   post::{GetPosts, GetPostsResponse},
8 };
9 use lemmy_apub::{fetcher::resolve_actor_identifier, objects::community::ApubCommunity};
10 use lemmy_db_schema::{
11   from_opt_str_to_opt_enum,
12   source::{community::Community, site::Site},
13   traits::DeleteableOrRemoveable,
14   ListingType,
15   SortType,
16 };
17 use lemmy_db_views::post_view::PostQueryBuilder;
18 use lemmy_utils::{ConnectionId, LemmyError};
19 use lemmy_websocket::LemmyContext;
20 use std::str::FromStr;
21
22 #[async_trait::async_trait(?Send)]
23 impl PerformCrud for GetPosts {
24   type Response = GetPostsResponse;
25
26   #[tracing::instrument(skip(context, _websocket_id))]
27   async fn perform(
28     &self,
29     context: &Data<LemmyContext>,
30     _websocket_id: Option<ConnectionId>,
31   ) -> Result<GetPostsResponse, LemmyError> {
32     let data: &GetPosts = self;
33     let local_user_view =
34       get_local_user_view_from_jwt_opt(data.auth.as_ref(), context.pool(), context.secret())
35         .await?;
36
37     check_private_instance(&local_user_view, context.pool()).await?;
38
39     let person_id = local_user_view.to_owned().map(|l| l.person.id);
40
41     let show_nsfw = local_user_view.as_ref().map(|t| t.local_user.show_nsfw);
42     let show_bot_accounts = local_user_view
43       .as_ref()
44       .map(|t| t.local_user.show_bot_accounts);
45     let show_read_posts = local_user_view
46       .as_ref()
47       .map(|t| t.local_user.show_read_posts);
48
49     let sort: Option<SortType> = from_opt_str_to_opt_enum(&data.sort);
50     let listing_type: ListingType = match from_opt_str_to_opt_enum(&data.type_) {
51       Some(l) => l,
52       None => {
53         let site = blocking(context.pool(), Site::read_local_site).await??;
54         ListingType::from_str(&site.default_post_listing_type)?
55       }
56     };
57
58     let page = data.page;
59     let limit = data.limit;
60     let community_id = data.community_id;
61     let community_actor_id = if let Some(name) = &data.community_name {
62       resolve_actor_identifier::<ApubCommunity, Community>(name, context)
63         .await
64         .ok()
65         .map(|c| c.actor_id)
66     } else {
67       None
68     };
69     let saved_only = data.saved_only;
70
71     let mut posts = blocking(context.pool(), move |conn| {
72       PostQueryBuilder::create(conn)
73         .listing_type(listing_type)
74         .sort(sort)
75         .show_nsfw(show_nsfw)
76         .show_bot_accounts(show_bot_accounts)
77         .show_read_posts(show_read_posts)
78         .community_id(community_id)
79         .community_actor_id(community_actor_id)
80         .saved_only(saved_only)
81         .my_person_id(person_id)
82         .page(page)
83         .limit(limit)
84         .list()
85     })
86     .await?
87     .map_err(|e| LemmyError::from_error_message(e, "couldnt_get_posts"))?;
88
89     // Blank out deleted or removed info for non-logged in users
90     if person_id.is_none() {
91       for pv in posts
92         .iter_mut()
93         .filter(|p| p.post.deleted || p.post.removed)
94       {
95         pv.post = pv.to_owned().post.blank_out_deleted_or_removed_info();
96       }
97
98       for pv in posts
99         .iter_mut()
100         .filter(|p| p.community.deleted || p.community.removed)
101       {
102         pv.community = pv.to_owned().community.blank_out_deleted_or_removed_info();
103       }
104     }
105
106     Ok(GetPostsResponse { posts })
107   }
108 }