git.squeep.com Git - akkoma/blob - lib/pleroma/web/mastodon_api/controllers/search_controller.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Web.MastodonAPI.SearchController do
   6   use Pleroma.Web, :controller
   7
   8   alias Pleroma.User
   9   alias Pleroma.Web.ControllerHelper
  10   alias Pleroma.Web.MastodonAPI.AccountView
  11   alias Pleroma.Web.Plugs.OAuthScopesPlug
  12   alias Pleroma.Web.Plugs.RateLimiter
  13
  14   require Logger
  15
  16   plug(Pleroma.Web.ApiSpec.CastAndValidate)
  17
  18   # Note: Mastodon doesn't allow unauthenticated access (requires read:accounts / read:search)
  19   plug(OAuthScopesPlug, %{scopes: ["read:search"], fallback: :proceed_unauthenticated})
  20
  21   # Note: on private instances auth is required (EnsurePublicOrAuthenticatedPlug is not skipped)
  22
  23   plug(RateLimiter, [name: :search] when action in [:search, :search2, :account_search])
  24
  25   defdelegate open_api_operation(action), to: Pleroma.Web.ApiSpec.SearchOperation
  26
  27   def account_search(%{assigns: %{user: user}} = conn, %{q: query} = params) do
  28     accounts = User.search(query, search_options(params, user))
  29
  30     conn
  31     |> put_view(AccountView)
  32     |> render("index.json",
  33       users: accounts,
  34       for: user,
  35       as: :user
  36     )
  37   end
  38
  39   def search2(conn, params), do: do_search(:v2, conn, params)
  40   def search(conn, params), do: do_search(:v1, conn, params)
  41
  42   defp do_search(version, %{assigns: %{user: user}} = conn, params) do
  43     options =
  44       search_options(params, user)
  45       |> Keyword.put(:version, version)
  46
  47     search_provider = Pleroma.Config.get([:search, :provider])
  48     json(conn, search_provider.search(conn, params, options))
  49   end
  50
  51   defp search_options(params, user) do
  52     [
  53       resolve: params[:resolve],
  54       following: params[:following],
  55       limit: params[:limit],
  56       offset: params[:offset],
  57       type: params[:type],
  58       author: get_author(params),
  59       embed_relationships: ControllerHelper.embed_relationships?(params),
  60       for_user: user
  61     ]
  62     |> Enum.filter(&elem(&1, 1))
  63   end
  64
  65   defp get_author(%{account_id: account_id}) when is_binary(account_id),
  66     do: User.get_cached_by_id(account_id)
  67
  68   defp get_author(_params), do: nil
  69 end