2023-07-03 09:01:41 +00:00
|
|
|
use crate::{fetcher::resolve_actor_identifier, objects::community::ApubCommunity};
|
2023-03-21 15:03:05 +00:00
|
|
|
use activitypub_federation::config::Data;
|
2023-07-03 09:01:41 +00:00
|
|
|
use actix_web::web::{Json, Query};
|
2022-04-13 18:12:25 +00:00
|
|
|
use lemmy_api_common::{
|
2022-11-28 14:29:33 +00:00
|
|
|
context::LemmyContext,
|
2022-04-13 18:12:25 +00:00
|
|
|
site::{Search, SearchResponse},
|
2023-09-21 10:42:28 +00:00
|
|
|
utils::{check_private_instance, is_admin},
|
2022-04-13 18:12:25 +00:00
|
|
|
};
|
2022-07-30 03:55:59 +00:00
|
|
|
use lemmy_db_schema::{
|
2022-10-27 09:24:07 +00:00
|
|
|
source::{community::Community, local_site::LocalSite},
|
2023-10-16 23:37:28 +00:00
|
|
|
utils::post_to_comment_sort_type,
|
2022-07-30 03:55:59 +00:00
|
|
|
SearchType,
|
|
|
|
};
|
2023-08-29 14:47:57 +00:00
|
|
|
use lemmy_db_views::{comment_view::CommentQuery, post_view::PostQuery, structs::LocalUserView};
|
2022-08-04 19:30:17 +00:00
|
|
|
use lemmy_db_views_actor::{community_view::CommunityQuery, person_view::PersonQuery};
|
2023-06-06 16:27:22 +00:00
|
|
|
use lemmy_utils::error::LemmyError;
|
2022-04-13 18:12:25 +00:00
|
|
|
|
2023-07-03 09:01:41 +00:00
|
|
|
#[tracing::instrument(skip(context))]
|
|
|
|
pub async fn search(
|
|
|
|
data: Query<Search>,
|
|
|
|
context: Data<LemmyContext>,
|
2023-09-21 10:42:28 +00:00
|
|
|
local_user_view: Option<LocalUserView>,
|
2023-07-03 09:01:41 +00:00
|
|
|
) -> Result<Json<SearchResponse>, LemmyError> {
|
2023-07-11 13:09:59 +00:00
|
|
|
let local_site = LocalSite::read(&mut context.pool()).await?;
|
2023-07-03 09:01:41 +00:00
|
|
|
|
|
|
|
check_private_instance(&local_user_view, &local_site)?;
|
|
|
|
|
2023-08-11 09:13:14 +00:00
|
|
|
let is_admin = local_user_view
|
|
|
|
.as_ref()
|
|
|
|
.map(|luv| is_admin(luv).is_ok())
|
|
|
|
.unwrap_or_default();
|
2023-07-03 09:01:41 +00:00
|
|
|
|
|
|
|
let mut posts = Vec::new();
|
|
|
|
let mut comments = Vec::new();
|
|
|
|
let mut communities = Vec::new();
|
|
|
|
let mut users = Vec::new();
|
|
|
|
|
|
|
|
// TODO no clean / non-nsfw searching rn
|
|
|
|
|
|
|
|
let q = data.q.clone();
|
|
|
|
let page = data.page;
|
|
|
|
let limit = data.limit;
|
|
|
|
let sort = data.sort;
|
|
|
|
let listing_type = data.listing_type;
|
|
|
|
let search_type = data.type_.unwrap_or(SearchType::All);
|
|
|
|
let community_id = if let Some(name) = &data.community_name {
|
2023-07-04 11:04:38 +00:00
|
|
|
Some(
|
|
|
|
resolve_actor_identifier::<ApubCommunity, Community>(name, &context, &local_user_view, false)
|
|
|
|
.await?,
|
|
|
|
)
|
|
|
|
.map(|c| c.id)
|
2023-07-03 09:01:41 +00:00
|
|
|
} else {
|
|
|
|
data.community_id
|
|
|
|
};
|
|
|
|
let creator_id = data.creator_id;
|
2023-07-20 14:36:16 +00:00
|
|
|
let local_user = local_user_view.as_ref().map(|l| l.local_user.clone());
|
2023-07-03 09:01:41 +00:00
|
|
|
match search_type {
|
|
|
|
SearchType::Posts => {
|
2023-07-17 10:20:25 +00:00
|
|
|
posts = PostQuery {
|
|
|
|
sort: (sort),
|
|
|
|
listing_type: (listing_type),
|
|
|
|
community_id: (community_id),
|
|
|
|
creator_id: (creator_id),
|
2023-07-20 14:36:16 +00:00
|
|
|
local_user: (local_user_view.as_ref()),
|
2023-07-17 10:20:25 +00:00
|
|
|
search_term: (Some(q)),
|
|
|
|
page: (page),
|
|
|
|
limit: (limit),
|
|
|
|
..Default::default()
|
|
|
|
}
|
|
|
|
.list(&mut context.pool())
|
|
|
|
.await?;
|
2023-07-03 09:01:41 +00:00
|
|
|
}
|
|
|
|
SearchType::Comments => {
|
2023-07-17 10:20:25 +00:00
|
|
|
comments = CommentQuery {
|
|
|
|
sort: (sort.map(post_to_comment_sort_type)),
|
|
|
|
listing_type: (listing_type),
|
|
|
|
search_term: (Some(q)),
|
|
|
|
community_id: (community_id),
|
|
|
|
creator_id: (creator_id),
|
2023-07-20 14:36:16 +00:00
|
|
|
local_user: (local_user_view.as_ref()),
|
2023-07-17 10:20:25 +00:00
|
|
|
page: (page),
|
|
|
|
limit: (limit),
|
|
|
|
..Default::default()
|
|
|
|
}
|
|
|
|
.list(&mut context.pool())
|
|
|
|
.await?;
|
2023-07-03 09:01:41 +00:00
|
|
|
}
|
|
|
|
SearchType::Communities => {
|
2023-07-17 10:20:25 +00:00
|
|
|
communities = CommunityQuery {
|
|
|
|
sort: (sort),
|
|
|
|
listing_type: (listing_type),
|
|
|
|
search_term: (Some(q)),
|
|
|
|
local_user: (local_user.as_ref()),
|
|
|
|
is_mod_or_admin: (is_admin),
|
|
|
|
page: (page),
|
|
|
|
limit: (limit),
|
|
|
|
..Default::default()
|
|
|
|
}
|
|
|
|
.list(&mut context.pool())
|
|
|
|
.await?;
|
2023-07-03 09:01:41 +00:00
|
|
|
}
|
|
|
|
SearchType::Users => {
|
2023-07-17 10:20:25 +00:00
|
|
|
users = PersonQuery {
|
2023-10-16 23:37:28 +00:00
|
|
|
sort,
|
2023-07-17 10:20:25 +00:00
|
|
|
search_term: (Some(q)),
|
|
|
|
page: (page),
|
|
|
|
limit: (limit),
|
|
|
|
}
|
|
|
|
.list(&mut context.pool())
|
|
|
|
.await?;
|
2023-07-03 09:01:41 +00:00
|
|
|
}
|
|
|
|
SearchType::All => {
|
|
|
|
// If the community or creator is included, dont search communities or users
|
|
|
|
let community_or_creator_included =
|
|
|
|
data.community_id.is_some() || data.community_name.is_some() || data.creator_id.is_some();
|
|
|
|
|
2023-07-20 14:36:16 +00:00
|
|
|
let q = data.q.clone();
|
|
|
|
|
2023-07-17 10:20:25 +00:00
|
|
|
posts = PostQuery {
|
|
|
|
sort: (sort),
|
|
|
|
listing_type: (listing_type),
|
|
|
|
community_id: (community_id),
|
|
|
|
creator_id: (creator_id),
|
2023-07-20 14:36:16 +00:00
|
|
|
local_user: (local_user_view.as_ref()),
|
2023-07-17 10:20:25 +00:00
|
|
|
search_term: (Some(q)),
|
|
|
|
page: (page),
|
|
|
|
limit: (limit),
|
|
|
|
..Default::default()
|
|
|
|
}
|
|
|
|
.list(&mut context.pool())
|
|
|
|
.await?;
|
2023-07-03 09:01:41 +00:00
|
|
|
|
|
|
|
let q = data.q.clone();
|
|
|
|
|
2023-07-17 10:20:25 +00:00
|
|
|
comments = CommentQuery {
|
|
|
|
sort: (sort.map(post_to_comment_sort_type)),
|
|
|
|
listing_type: (listing_type),
|
|
|
|
search_term: (Some(q)),
|
|
|
|
community_id: (community_id),
|
|
|
|
creator_id: (creator_id),
|
2023-07-20 14:36:16 +00:00
|
|
|
local_user: (local_user_view.as_ref()),
|
2023-07-17 10:20:25 +00:00
|
|
|
page: (page),
|
|
|
|
limit: (limit),
|
|
|
|
..Default::default()
|
|
|
|
}
|
|
|
|
.list(&mut context.pool())
|
|
|
|
.await?;
|
2023-07-03 09:01:41 +00:00
|
|
|
|
|
|
|
let q = data.q.clone();
|
|
|
|
|
|
|
|
communities = if community_or_creator_included {
|
|
|
|
vec![]
|
|
|
|
} else {
|
2023-07-17 10:20:25 +00:00
|
|
|
CommunityQuery {
|
|
|
|
sort: (sort),
|
|
|
|
listing_type: (listing_type),
|
|
|
|
search_term: (Some(q)),
|
|
|
|
local_user: (local_user.as_ref()),
|
|
|
|
is_mod_or_admin: (is_admin),
|
|
|
|
page: (page),
|
|
|
|
limit: (limit),
|
|
|
|
..Default::default()
|
|
|
|
}
|
|
|
|
.list(&mut context.pool())
|
|
|
|
.await?
|
2023-07-03 09:01:41 +00:00
|
|
|
};
|
2022-04-13 18:12:25 +00:00
|
|
|
|
2023-07-03 09:01:41 +00:00
|
|
|
let q = data.q.clone();
|
2022-04-13 18:12:25 +00:00
|
|
|
|
2023-07-03 09:01:41 +00:00
|
|
|
users = if community_or_creator_included {
|
|
|
|
vec![]
|
|
|
|
} else {
|
2023-07-17 10:20:25 +00:00
|
|
|
PersonQuery {
|
2023-10-16 23:37:28 +00:00
|
|
|
sort,
|
2023-07-17 10:20:25 +00:00
|
|
|
search_term: (Some(q)),
|
|
|
|
page: (page),
|
|
|
|
limit: (limit),
|
|
|
|
}
|
|
|
|
.list(&mut context.pool())
|
|
|
|
.await?
|
2023-07-03 09:01:41 +00:00
|
|
|
};
|
|
|
|
}
|
|
|
|
SearchType::Url => {
|
2023-07-17 10:20:25 +00:00
|
|
|
posts = PostQuery {
|
|
|
|
sort: (sort),
|
|
|
|
listing_type: (listing_type),
|
|
|
|
community_id: (community_id),
|
|
|
|
creator_id: (creator_id),
|
|
|
|
url_search: (Some(q)),
|
|
|
|
page: (page),
|
|
|
|
limit: (limit),
|
|
|
|
..Default::default()
|
|
|
|
}
|
|
|
|
.list(&mut context.pool())
|
|
|
|
.await?;
|
2023-07-03 09:01:41 +00:00
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
// Return the jwt
|
|
|
|
Ok(Json(SearchResponse {
|
|
|
|
type_: search_type,
|
|
|
|
comments,
|
|
|
|
posts,
|
|
|
|
communities,
|
|
|
|
users,
|
|
|
|
}))
|
2022-04-13 18:12:25 +00:00
|
|
|
}
|