|
|
|
|
@ -1,9 +1,6 @@
|
|
|
|
|
use crate::{
|
|
|
|
|
api::PerformApub,
|
|
|
|
|
fetcher::resolve_actor_identifier,
|
|
|
|
|
objects::community::ApubCommunity,
|
|
|
|
|
};
|
|
|
|
|
use crate::{fetcher::resolve_actor_identifier, objects::community::ApubCommunity};
|
|
|
|
|
use activitypub_federation::config::Data;
|
|
|
|
|
use actix_web::web::{Json, Query};
|
|
|
|
|
use lemmy_api_common::{
|
|
|
|
|
context::LemmyContext,
|
|
|
|
|
site::{Search, SearchResponse},
|
|
|
|
|
@ -18,78 +15,142 @@ use lemmy_db_views::{comment_view::CommentQuery, post_view::PostQuery};
|
|
|
|
|
use lemmy_db_views_actor::{community_view::CommunityQuery, person_view::PersonQuery};
|
|
|
|
|
use lemmy_utils::error::LemmyError;
|
|
|
|
|
|
|
|
|
|
#[async_trait::async_trait]
|
|
|
|
|
impl PerformApub for Search {
|
|
|
|
|
type Response = SearchResponse;
|
|
|
|
|
|
|
|
|
|
#[tracing::instrument(skip(context))]
|
|
|
|
|
async fn perform(&self, context: &Data<LemmyContext>) -> Result<SearchResponse, LemmyError> {
|
|
|
|
|
let data: &Search = self;
|
|
|
|
|
|
|
|
|
|
let local_user_view = local_user_view_from_jwt_opt(data.auth.as_ref(), context).await;
|
|
|
|
|
let local_site = LocalSite::read(context.conn().await?).await?;
|
|
|
|
|
|
|
|
|
|
check_private_instance(&local_user_view, &local_site)?;
|
|
|
|
|
|
|
|
|
|
let is_admin = local_user_view.as_ref().map(|luv| is_admin(luv).is_ok());
|
|
|
|
|
|
|
|
|
|
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 {
|
|
|
|
|
resolve_actor_identifier::<ApubCommunity, Community>(name, context, &local_user_view, false)
|
|
|
|
|
.await
|
|
|
|
|
.ok()
|
|
|
|
|
.map(|c| c.id)
|
|
|
|
|
} else {
|
|
|
|
|
data.community_id
|
|
|
|
|
};
|
|
|
|
|
let creator_id = data.creator_id;
|
|
|
|
|
let local_user = local_user_view.map(|l| l.local_user);
|
|
|
|
|
match search_type {
|
|
|
|
|
SearchType::Posts => {
|
|
|
|
|
posts = PostQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.sort(sort)
|
|
|
|
|
.listing_type(listing_type)
|
|
|
|
|
.community_id(community_id)
|
|
|
|
|
.creator_id(creator_id)
|
|
|
|
|
.local_user(local_user.as_ref())
|
|
|
|
|
.search_term(Some(q))
|
|
|
|
|
.is_mod_or_admin(is_admin)
|
|
|
|
|
.page(page)
|
|
|
|
|
.limit(limit)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?;
|
|
|
|
|
}
|
|
|
|
|
SearchType::Comments => {
|
|
|
|
|
comments = CommentQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.sort(sort.map(post_to_comment_sort_type))
|
|
|
|
|
.listing_type(listing_type)
|
|
|
|
|
.search_term(Some(q))
|
|
|
|
|
.community_id(community_id)
|
|
|
|
|
.creator_id(creator_id)
|
|
|
|
|
.local_user(local_user.as_ref())
|
|
|
|
|
.page(page)
|
|
|
|
|
.limit(limit)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?;
|
|
|
|
|
}
|
|
|
|
|
SearchType::Communities => {
|
|
|
|
|
communities = CommunityQuery::builder()
|
|
|
|
|
#[tracing::instrument(skip(context))]
|
|
|
|
|
pub async fn search(
|
|
|
|
|
data: Query<Search>,
|
|
|
|
|
context: Data<LemmyContext>,
|
|
|
|
|
) -> Result<Json<SearchResponse>, LemmyError> {
|
|
|
|
|
let local_user_view = local_user_view_from_jwt_opt(data.auth.as_ref(), &context).await;
|
|
|
|
|
let local_site = LocalSite::read(context.conn().await?).await?;
|
|
|
|
|
|
|
|
|
|
check_private_instance(&local_user_view, &local_site)?;
|
|
|
|
|
|
|
|
|
|
let is_admin = local_user_view.as_ref().map(|luv| is_admin(luv).is_ok());
|
|
|
|
|
|
|
|
|
|
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 {
|
|
|
|
|
resolve_actor_identifier::<ApubCommunity, Community>(name, &context, &local_user_view, false)
|
|
|
|
|
.await
|
|
|
|
|
.ok()
|
|
|
|
|
.map(|c| c.id)
|
|
|
|
|
} else {
|
|
|
|
|
data.community_id
|
|
|
|
|
};
|
|
|
|
|
let creator_id = data.creator_id;
|
|
|
|
|
let local_user = local_user_view.map(|l| l.local_user);
|
|
|
|
|
match search_type {
|
|
|
|
|
SearchType::Posts => {
|
|
|
|
|
posts = PostQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.sort(sort)
|
|
|
|
|
.listing_type(listing_type)
|
|
|
|
|
.community_id(community_id)
|
|
|
|
|
.creator_id(creator_id)
|
|
|
|
|
.local_user(local_user.as_ref())
|
|
|
|
|
.search_term(Some(q))
|
|
|
|
|
.is_mod_or_admin(is_admin)
|
|
|
|
|
.page(page)
|
|
|
|
|
.limit(limit)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?;
|
|
|
|
|
}
|
|
|
|
|
SearchType::Comments => {
|
|
|
|
|
comments = CommentQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.sort(sort.map(post_to_comment_sort_type))
|
|
|
|
|
.listing_type(listing_type)
|
|
|
|
|
.search_term(Some(q))
|
|
|
|
|
.community_id(community_id)
|
|
|
|
|
.creator_id(creator_id)
|
|
|
|
|
.local_user(local_user.as_ref())
|
|
|
|
|
.page(page)
|
|
|
|
|
.limit(limit)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?;
|
|
|
|
|
}
|
|
|
|
|
SearchType::Communities => {
|
|
|
|
|
communities = CommunityQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.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)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?;
|
|
|
|
|
}
|
|
|
|
|
SearchType::Users => {
|
|
|
|
|
users = PersonQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.sort(sort)
|
|
|
|
|
.search_term(Some(q))
|
|
|
|
|
.page(page)
|
|
|
|
|
.limit(limit)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?;
|
|
|
|
|
}
|
|
|
|
|
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();
|
|
|
|
|
|
|
|
|
|
let local_user_ = local_user.clone();
|
|
|
|
|
posts = PostQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.sort(sort)
|
|
|
|
|
.listing_type(listing_type)
|
|
|
|
|
.community_id(community_id)
|
|
|
|
|
.creator_id(creator_id)
|
|
|
|
|
.local_user(local_user_.as_ref())
|
|
|
|
|
.search_term(Some(q))
|
|
|
|
|
.is_mod_or_admin(is_admin)
|
|
|
|
|
.page(page)
|
|
|
|
|
.limit(limit)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?;
|
|
|
|
|
|
|
|
|
|
let q = data.q.clone();
|
|
|
|
|
|
|
|
|
|
let local_user_ = local_user.clone();
|
|
|
|
|
comments = CommentQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.sort(sort.map(post_to_comment_sort_type))
|
|
|
|
|
.listing_type(listing_type)
|
|
|
|
|
.search_term(Some(q))
|
|
|
|
|
.community_id(community_id)
|
|
|
|
|
.creator_id(creator_id)
|
|
|
|
|
.local_user(local_user_.as_ref())
|
|
|
|
|
.page(page)
|
|
|
|
|
.limit(limit)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?;
|
|
|
|
|
|
|
|
|
|
let q = data.q.clone();
|
|
|
|
|
|
|
|
|
|
communities = if community_or_creator_included {
|
|
|
|
|
vec![]
|
|
|
|
|
} else {
|
|
|
|
|
CommunityQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.sort(sort)
|
|
|
|
|
.listing_type(listing_type)
|
|
|
|
|
@ -100,116 +161,48 @@ impl PerformApub for Search {
|
|
|
|
|
.limit(limit)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?;
|
|
|
|
|
}
|
|
|
|
|
SearchType::Users => {
|
|
|
|
|
users = PersonQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.sort(sort)
|
|
|
|
|
.search_term(Some(q))
|
|
|
|
|
.page(page)
|
|
|
|
|
.limit(limit)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?;
|
|
|
|
|
}
|
|
|
|
|
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();
|
|
|
|
|
|
|
|
|
|
let local_user_ = local_user.clone();
|
|
|
|
|
posts = PostQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.sort(sort)
|
|
|
|
|
.listing_type(listing_type)
|
|
|
|
|
.community_id(community_id)
|
|
|
|
|
.creator_id(creator_id)
|
|
|
|
|
.local_user(local_user_.as_ref())
|
|
|
|
|
.search_term(Some(q))
|
|
|
|
|
.is_mod_or_admin(is_admin)
|
|
|
|
|
.page(page)
|
|
|
|
|
.limit(limit)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?;
|
|
|
|
|
.await?
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
let q = data.q.clone();
|
|
|
|
|
let q = data.q.clone();
|
|
|
|
|
|
|
|
|
|
let local_user_ = local_user.clone();
|
|
|
|
|
comments = CommentQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.sort(sort.map(post_to_comment_sort_type))
|
|
|
|
|
.listing_type(listing_type)
|
|
|
|
|
.search_term(Some(q))
|
|
|
|
|
.community_id(community_id)
|
|
|
|
|
.creator_id(creator_id)
|
|
|
|
|
.local_user(local_user_.as_ref())
|
|
|
|
|
.page(page)
|
|
|
|
|
.limit(limit)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?;
|
|
|
|
|
|
|
|
|
|
let q = data.q.clone();
|
|
|
|
|
|
|
|
|
|
communities = if community_or_creator_included {
|
|
|
|
|
vec![]
|
|
|
|
|
} else {
|
|
|
|
|
CommunityQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.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)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
let q = data.q.clone();
|
|
|
|
|
|
|
|
|
|
users = if community_or_creator_included {
|
|
|
|
|
vec![]
|
|
|
|
|
} else {
|
|
|
|
|
PersonQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.sort(sort)
|
|
|
|
|
.search_term(Some(q))
|
|
|
|
|
.page(page)
|
|
|
|
|
.limit(limit)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
SearchType::Url => {
|
|
|
|
|
posts = PostQuery::builder()
|
|
|
|
|
users = if community_or_creator_included {
|
|
|
|
|
vec![]
|
|
|
|
|
} else {
|
|
|
|
|
PersonQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.sort(sort)
|
|
|
|
|
.listing_type(listing_type)
|
|
|
|
|
.community_id(community_id)
|
|
|
|
|
.creator_id(creator_id)
|
|
|
|
|
.url_search(Some(q))
|
|
|
|
|
.is_mod_or_admin(is_admin)
|
|
|
|
|
.search_term(Some(q))
|
|
|
|
|
.page(page)
|
|
|
|
|
.limit(limit)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?;
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
// Return the jwt
|
|
|
|
|
Ok(SearchResponse {
|
|
|
|
|
type_: search_type,
|
|
|
|
|
comments,
|
|
|
|
|
posts,
|
|
|
|
|
communities,
|
|
|
|
|
users,
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
.await?
|
|
|
|
|
};
|
|
|
|
|
}
|
|
|
|
|
SearchType::Url => {
|
|
|
|
|
posts = PostQuery::builder()
|
|
|
|
|
.conn(context.conn().await?)
|
|
|
|
|
.sort(sort)
|
|
|
|
|
.listing_type(listing_type)
|
|
|
|
|
.community_id(community_id)
|
|
|
|
|
.creator_id(creator_id)
|
|
|
|
|
.url_search(Some(q))
|
|
|
|
|
.is_mod_or_admin(is_admin)
|
|
|
|
|
.page(page)
|
|
|
|
|
.limit(limit)
|
|
|
|
|
.build()
|
|
|
|
|
.list()
|
|
|
|
|
.await?;
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
// Return the jwt
|
|
|
|
|
Ok(Json(SearchResponse {
|
|
|
|
|
type_: search_type,
|
|
|
|
|
comments,
|
|
|
|
|
posts,
|
|
|
|
|
communities,
|
|
|
|
|
users,
|
|
|
|
|
}))
|
|
|
|
|
}
|
|
|
|
|
|
dullbananas
1 year ago
committed by
GitHub