Add support for searching media status

w/ some refactors and updates

lib/pleroma/search/database_search.ex

@@ -26,6 +26,7 @@ def search(user, search_query, options \\ []) do
     limit = Enum.min([Keyword.get(options, :limit), 40])
     offset = Keyword.get(options, :offset, 0)
     author = Keyword.get(options, :author)
+    only_media = Keyword.get(options, :only_media, false)
 
     search_function =
       if :persistent_term.get({Pleroma.Repo, :postgres_version}) >= 11 do
@@ -43,6 +44,7 @@ def search(user, search_query, options \\ []) do
       |> maybe_restrict_local(user)
       |> maybe_restrict_author(author)
       |> maybe_restrict_blocked(user)
+      |> maybe_only_media(only_media)
       |> Pagination.fetch_paginated(
         %{"offset" => offset, "limit" => limit, "skip_order" => index_type == :rum},
         :offset
@@ -173,4 +175,12 @@ def maybe_fetch(activities, user, search_query) do
       _ -> activities
     end
   end
+
+  def maybe_only_media(q, true) do
+    from([a, o] in q,
+      where: fragment("not (?)->'attachment' = (?)", o.data, ^[])
+    )
+  end
+
+  def maybe_only_media(q, _), do: q
 end

lib/pleroma/search/elasticsearch.ex

@@ -43,6 +43,7 @@ defp maybe_fetch(:activity, search_query) do
   def search(user, query, options) do
     limit = Enum.min([Keyword.get(options, :limit), 40])
     offset = Keyword.get(options, :offset, 0)
+    only_media = Keyword.get(options, :only_media, false)
 
     parsed_query =
       query
@@ -62,7 +63,10 @@ def search(user, query, options) do
         |> Pleroma.Search.Elasticsearch.Store.search(q)
         |> Enum.filter(fn x ->
           x.data["type"] == "Create" && x.object.data["type"] == "Note" &&
-            Visibility.visible_for_user?(x, user)
+            Visibility.visible_for_user?(x, user) &&
+            # i'm not going to implement this very seriously because the support for
+            # elasticsearch is already kinda broken
+            if only_media, do: length(x.object.data["attachment"]) > 0, else: true
         end)
       end)
 

lib/pleroma/search/elasticsearch/store.ex

@@ -32,7 +32,7 @@ def search(:raw, index, q) do
       {:ok, results}
     else
       {:error, e} ->
-        Logger.error(e)
+        Logger.error("Unable to search with Elasiticsearch: #{inspect(e)}")
         {:error, e}
     end
   end
@@ -44,7 +44,7 @@ def search(:activities, q) do
       |> Pleroma.Activity.all_by_ids_with_object()
     else
       e ->
-        Logger.error(e)
+        Logger.error("Unable to search with Elasiticsearch: #{inspect(e)}")
         []
     end
   end

lib/pleroma/search/meilisearch.ex

@@ -79,6 +79,7 @@ def search(user, query, options \\ []) do
     limit = Enum.min([Keyword.get(options, :limit), 40])
     offset = Keyword.get(options, :offset, 0)
     author = Keyword.get(options, :author)
+    only_media = Keyword.get(options, :only_media, false)
 
     res =
       meili_post(
@@ -93,11 +94,11 @@ def search(user, query, options \\ []) do
         hits
         |> Activity.create_by_object_ap_id()
         |> Activity.with_preloaded_object()
-        |> Activity.with_preloaded_object()
         |> Activity.restrict_deactivated_users()
         |> maybe_restrict_local(user)
         |> maybe_restrict_author(author)
         |> maybe_restrict_blocked(user)
+        |> maybe_only_media(only_media)
         |> maybe_fetch(user, query)
         |> order_by([object: obj], desc: obj.data["published"])
         |> Pleroma.Repo.all()

lib/pleroma/web/api_spec/operations/search_operation.ex

@@ -75,7 +75,7 @@ def search2_operation do
         Operation.parameter(
           :type,
           :query,
-          %Schema{type: :string, enum: ["accounts", "hashtags", "statuses"]},
+          %Schema{type: :string, enum: ["accounts", "hashtags", "statuses", "media"]},
           "Search type"
         ),
         Operation.parameter(:q, :query, %Schema{type: :string}, "What to search for",

lib/pleroma/web/mastodon_api/controllers/search_controller.ex

@@ -47,7 +47,7 @@ defp do_search(version, %{assigns: %{user: user}} = conn, %{q: query} = params)
     query = String.trim(query)
     options = search_options(params, user)
     timeout = Keyword.get(Repo.config(), :timeout, 15_000)
-    default_values = %{"statuses" => [], "accounts" => [], "hashtags" => []}
+    default_values = %{"statuses" => [], "media" => [], "accounts" => [], "hashtags" => []}
 
     result =
       default_values
@@ -107,6 +107,22 @@ defp resource_search(_, "statuses", query, options) do
     )
   end
 
+  defp resource_search(_, "media", query, options) do
+    statuses =
+      with_fallback(fn ->
+        Pleroma.Search.search(
+          query,
+          Keyword.put(options, :only_media, true)
+        )
+      end)
+
+    StatusView.render("index.json",
+      activities: statuses,
+      for: options[:for_user],
+      as: :activity
+    )
+  end
+
   defp resource_search(:v2, "hashtags", query, options) do
     tags_path = Endpoint.url() <> "/tag/"