1 # Akkoma: A lightweight social networking server
2 # Copyright © 2022-2022 Akkoma Authors <https://git.ihatebeinga.live/IHBAGang/akkoma/>
3 # SPDX-License-Identifier: AGPL-3.0-only
5 defmodule Pleroma.Search.Elasticsearch do
6 @behaviour Pleroma.Search.SearchBackend
9 alias Pleroma.Object.Fetcher
10 alias Pleroma.Web.ActivityPub.Visibility
11 alias Pleroma.Search.Elasticsearch.Parsers
13 def es_query(:activity, query, offset, limit) do
14 must = Parsers.Activity.parse(query)
26 %{"_timestamp" => %{order: "desc", format: "basic_date_time"}}
37 defp maybe_fetch(:activity, search_query) do
38 with true <- Regex.match?(~r/https?:/, search_query),
39 {:ok, object} <- Fetcher.fetch_object_from_id(search_query),
40 %Activity{} = activity <- Activity.get_create_by_object_ap_id(object.data["id"]) do
47 def search(user, query, options) do
48 limit = Enum.min([Keyword.get(options, :limit), 40])
49 offset = Keyword.get(options, :offset, 0)
54 |> SearchParser.parse!()
58 maybe_fetch(:activity, String.trim(query))
63 q = es_query(:activity, parsed_query, offset, limit)
66 |> Pleroma.Search.Elasticsearch.Store.search(q)
67 |> Enum.filter(fn x ->
68 x.data["type"] == "Create" && x.object.data["type"] == "Note" &&
69 Visibility.visible_for_user?(x, user)
73 activity_results = Task.await(activity_task)
74 direct_activity = Task.await(activity_fetch_task)
77 if direct_activity == nil do
80 [direct_activity | activity_results]
87 def add_to_index(activity) do
88 Elasticsearch.put_document(Pleroma.Search.Elasticsearch.Cluster, activity, "activities")
92 def remove_from_index(object) do
93 Elasticsearch.delete_document(Pleroma.Search.Elasticsearch.Cluster, object, "activities")