1 use crate::{fetcher::resolve_actor_identifier, objects::community::ApubCommunity};
2 use activitypub_federation::config::Data;
3 use actix_web::web::{Json, Query};
4 use lemmy_api_common::{
6 site::{Search, SearchResponse},
7 utils::{check_private_instance, is_admin, local_user_view_from_jwt_opt},
10 source::{community::Community, local_site::LocalSite},
11 utils::post_to_comment_sort_type,
14 use lemmy_db_views::{comment_view::CommentQuery, post_view::PostQuery};
15 use lemmy_db_views_actor::{community_view::CommunityQuery, person_view::PersonQuery};
16 use lemmy_utils::error::LemmyError;
18 #[tracing::instrument(skip(context))]
21 context: Data<LemmyContext>,
22 ) -> Result<Json<SearchResponse>, LemmyError> {
23 let local_user_view = local_user_view_from_jwt_opt(data.auth.as_ref(), &context).await;
24 let local_site = LocalSite::read(context.pool()).await?;
26 check_private_instance(&local_user_view, &local_site)?;
28 let is_admin = local_user_view.as_ref().map(|luv| is_admin(luv).is_ok());
30 let mut posts = Vec::new();
31 let mut comments = Vec::new();
32 let mut communities = Vec::new();
33 let mut users = Vec::new();
35 // TODO no clean / non-nsfw searching rn
37 let q = data.q.clone();
39 let limit = data.limit;
41 let listing_type = data.listing_type;
42 let search_type = data.type_.unwrap_or(SearchType::All);
43 let community_id = if let Some(name) = &data.community_name {
45 resolve_actor_identifier::<ApubCommunity, Community>(name, &context, &local_user_view, false)
52 let creator_id = data.creator_id;
53 let local_user = local_user_view.map(|l| l.local_user);
55 SearchType::Posts => {
56 posts = PostQuery::builder()
59 .listing_type(listing_type)
60 .community_id(community_id)
61 .creator_id(creator_id)
62 .local_user(local_user.as_ref())
64 .is_mod_or_admin(is_admin)
71 SearchType::Comments => {
72 comments = CommentQuery::builder()
74 .sort(sort.map(post_to_comment_sort_type))
75 .listing_type(listing_type)
77 .community_id(community_id)
78 .creator_id(creator_id)
79 .local_user(local_user.as_ref())
86 SearchType::Communities => {
87 communities = CommunityQuery::builder()
90 .listing_type(listing_type)
92 .local_user(local_user.as_ref())
93 .is_mod_or_admin(is_admin)
100 SearchType::Users => {
101 users = PersonQuery::builder()
102 .pool(context.pool())
104 .search_term(Some(q))
112 // If the community or creator is included, dont search communities or users
113 let community_or_creator_included =
114 data.community_id.is_some() || data.community_name.is_some() || data.creator_id.is_some();
116 let local_user_ = local_user.clone();
117 posts = PostQuery::builder()
118 .pool(context.pool())
120 .listing_type(listing_type)
121 .community_id(community_id)
122 .creator_id(creator_id)
123 .local_user(local_user_.as_ref())
124 .search_term(Some(q))
125 .is_mod_or_admin(is_admin)
132 let q = data.q.clone();
134 let local_user_ = local_user.clone();
135 comments = CommentQuery::builder()
136 .pool(context.pool())
137 .sort(sort.map(post_to_comment_sort_type))
138 .listing_type(listing_type)
139 .search_term(Some(q))
140 .community_id(community_id)
141 .creator_id(creator_id)
142 .local_user(local_user_.as_ref())
149 let q = data.q.clone();
151 communities = if community_or_creator_included {
154 CommunityQuery::builder()
155 .pool(context.pool())
157 .listing_type(listing_type)
158 .search_term(Some(q))
159 .local_user(local_user.as_ref())
160 .is_mod_or_admin(is_admin)
168 let q = data.q.clone();
170 users = if community_or_creator_included {
173 PersonQuery::builder()
174 .pool(context.pool())
176 .search_term(Some(q))
185 posts = PostQuery::builder()
186 .pool(context.pool())
188 .listing_type(listing_type)
189 .community_id(community_id)
190 .creator_id(creator_id)
192 .is_mod_or_admin(is_admin)
202 Ok(Json(SearchResponse {