lemmy/crates/apub/src/api/search.rs

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},
  utils::{check_conflicting_like_filters, check_private_instance, is_admin},
};
use lemmy_db_schema::{source::community::Community, utils::post_to_comment_sort_type, SearchType};
use lemmy_db_views::{
  comment_view::CommentQuery,
  post_view::PostQuery,
  structs::{LocalUserView, SiteView},
};
use lemmy_db_views_actor::{
  community_view::CommunityQuery,
  person_view::PersonQuery,
  structs::CommunitySortType,
};
use lemmy_utils::error::LemmyResult;

#[tracing::instrument(skip(context))]
pub async fn search(
  data: Query<Search>,
  context: Data<LemmyContext>,
  local_user_view: Option<LocalUserView>,
) -> LemmyResult<Json<SearchResponse>> {
  let local_site = SiteView::read_local(&mut context.pool()).await?;

  check_private_instance(&local_user_view, &local_site.local_site)?;

  let is_admin = local_user_view
    .as_ref()
    .map(|luv| is_admin(luv).is_ok())
    .unwrap_or_default();

  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 Query(Search {
    q,
    community_id,
    community_name,
    creator_id,
    type_,
    sort,
    listing_type,
    page,
    limit,
    title_only,
    post_url_only,
    saved_only,
    liked_only,
    disliked_only,
  }) = data;

  let q = q.clone();
  let search_type = type_.unwrap_or(SearchType::All);
  let community_id = if let Some(name) = &community_name {
    Some(
      resolve_actor_identifier::<ApubCommunity, Community>(name, &context, &local_user_view, false)
        .await?,
    )
    .map(|c| c.id)
  } else {
    community_id
  };
  let local_user = local_user_view.as_ref().map(|l| &l.local_user);

  check_conflicting_like_filters(liked_only, disliked_only)?;

  let posts_query = PostQuery {
    sort,
    listing_type,
    community_id,
    creator_id,
    local_user,
    search_term: Some(q.clone()),
    page,
    limit,
    title_only,
    url_only: post_url_only,
    liked_only,
    disliked_only,
    saved_only,
    ..Default::default()
  };

  let comment_query = CommentQuery {
    sort: sort.map(post_to_comment_sort_type),
    listing_type,
    search_term: Some(q.clone()),
    community_id,
    creator_id,
    local_user,
    page,
    limit,
    liked_only,
    disliked_only,
    saved_only,
    ..Default::default()
  };

  let community_query = CommunityQuery {
    sort: sort.map(CommunitySortType::from),
    listing_type,
    search_term: Some(q.clone()),
    title_only,
    local_user,
    is_mod_or_admin: is_admin,
    page,
    limit,
    ..Default::default()
  };

  let person_query = PersonQuery {
    sort,
    search_term: Some(q.clone()),
    listing_type,
    page,
    limit,
  };

  match search_type {
    SearchType::Posts => {
      posts = posts_query
        .list(&local_site.site, &mut context.pool())
        .await?;
    }
    SearchType::Comments => {
      comments = comment_query
        .list(&local_site.site, &mut context.pool())
        .await?;
    }
    SearchType::Communities => {
      communities = community_query
        .list(&local_site.site, &mut context.pool())
        .await?;
    }
    SearchType::Users => {
      users = person_query.list(&mut context.pool()).await?;
    }
    SearchType::All => {
      // If the community or creator is included, dont search communities or users
      let community_or_creator_included =
        community_id.is_some() || community_name.is_some() || creator_id.is_some();

      posts = posts_query
        .list(&local_site.site, &mut context.pool())
        .await?;

      comments = comment_query
        .list(&local_site.site, &mut context.pool())
        .await?;

      communities = if community_or_creator_included {
        vec![]
      } else {
        community_query
          .list(&local_site.site, &mut context.pool())
          .await?
      };

      users = if community_or_creator_included {
        vec![]
      } else {
        person_query.list(&mut context.pool()).await?
      };
    }
  };

  // Return the jwt
  Ok(Json(SearchResponse {
    type_: search_type,
    comments,
    posts,
    communities,
    users,
  }))
}