]> Untitled Git - lemmy.git/blob - crates/api_crud/src/post/read.rs
Making more fields optional in the API.
[lemmy.git] / crates / api_crud / src / post / read.rs
1 use crate::PerformCrud;
2 use actix_web::web::Data;
3 use lemmy_api_common::{blocking, get_local_user_view_from_jwt_opt, post::*, user_show_bot_accounts, user_show_nsfw};
4 use lemmy_db_queries::{from_opt_str_to_opt_enum, ListingType, SortType};
5 use lemmy_db_views::{
6   comment_view::CommentQueryBuilder,
7   post_view::{PostQueryBuilder, PostView},
8 };
9 use lemmy_db_views_actor::{
10   community_moderator_view::CommunityModeratorView,
11   community_view::CommunityView,
12 };
13 use lemmy_utils::{ApiError, ConnectionId, LemmyError};
14 use lemmy_websocket::{messages::GetPostUsersOnline, LemmyContext};
15
16 #[async_trait::async_trait(?Send)]
17 impl PerformCrud for GetPost {
18   type Response = GetPostResponse;
19
20   async fn perform(
21     &self,
22     context: &Data<LemmyContext>,
23     _websocket_id: Option<ConnectionId>,
24   ) -> Result<GetPostResponse, LemmyError> {
25     let data: &GetPost = &self;
26     let local_user_view = get_local_user_view_from_jwt_opt(&data.auth, context.pool()).await?;
27
28     let show_bot_accounts = user_show_bot_accounts(&local_user_view);
29
30     let person_id = local_user_view.map(|u| u.person.id);
31
32     let id = data.id;
33     let post_view = blocking(context.pool(), move |conn| {
34       PostView::read(conn, id, person_id)
35     })
36     .await?
37     .map_err(|_| ApiError::err("couldnt_find_post"))?;
38
39     let id = data.id;
40     let comments = blocking(context.pool(), move |conn| {
41       CommentQueryBuilder::create(conn)
42         .my_person_id(person_id)
43         .show_bot_accounts(show_bot_accounts)
44         .post_id(id)
45         .limit(9999)
46         .list()
47     })
48     .await??;
49
50     let community_id = post_view.community.id;
51     let moderators = blocking(context.pool(), move |conn| {
52       CommunityModeratorView::for_community(conn, community_id)
53     })
54     .await??;
55
56     // Necessary for the sidebar
57     let community_view = blocking(context.pool(), move |conn| {
58       CommunityView::read(conn, community_id, person_id)
59     })
60     .await?
61     .map_err(|_| ApiError::err("couldnt_find_community"))?;
62
63     let online = context
64       .chat_server()
65       .send(GetPostUsersOnline { post_id: data.id })
66       .await
67       .unwrap_or(1);
68
69     // Return the jwt
70     Ok(GetPostResponse {
71       post_view,
72       community_view,
73       comments,
74       moderators,
75       online,
76     })
77   }
78 }
79
80 #[async_trait::async_trait(?Send)]
81 impl PerformCrud for GetPosts {
82   type Response = GetPostsResponse;
83
84   async fn perform(
85     &self,
86     context: &Data<LemmyContext>,
87     _websocket_id: Option<ConnectionId>,
88   ) -> Result<GetPostsResponse, LemmyError> {
89     let data: &GetPosts = &self;
90     let local_user_view = get_local_user_view_from_jwt_opt(&data.auth, context.pool()).await?;
91
92     let person_id = local_user_view.to_owned().map(|l| l.person.id);
93
94     let show_nsfw = user_show_nsfw(&local_user_view);
95     let show_bot_accounts = user_show_bot_accounts(&local_user_view);
96
97     let sort: Option<SortType> = from_opt_str_to_opt_enum(&data.sort);
98     let listing_type: Option<ListingType> = from_opt_str_to_opt_enum(&data.type_);
99
100     let page = data.page;
101     let limit = data.limit;
102     let community_id = data.community_id;
103     let community_name = data.community_name.to_owned();
104     let saved_only = data.saved_only;
105
106     let posts = blocking(context.pool(), move |conn| {
107       PostQueryBuilder::create(conn)
108         .listing_type(listing_type)
109         .sort(sort)
110         .show_nsfw(show_nsfw)
111         .show_bot_accounts(show_bot_accounts)
112         .community_id(community_id)
113         .community_name(community_name)
114         .saved_only(saved_only)
115         .my_person_id(person_id)
116         .page(page)
117         .limit(limit)
118         .list()
119     })
120     .await?
121     .map_err(|_| ApiError::err("couldnt_get_posts"))?;
122
123     Ok(GetPostsResponse { posts })
124   }
125 }