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   @search_limit 40
  17
  18   plug(Pleroma.Web.ApiSpec.CastAndValidate)
  19
  20   # Note: Mastodon doesn't allow unauthenticated access (requires read:accounts / read:search)
  21   plug(OAuthScopesPlug, %{scopes: ["read:search"], fallback: :proceed_unauthenticated})
  22
  23   # Note: on private instances auth is required (EnsurePublicOrAuthenticatedPlug is not skipped)
  24
  25   plug(RateLimiter, [name: :search] when action in [:search, :search2, :account_search])
  26
  27   defdelegate open_api_operation(action), to: Pleroma.Web.ApiSpec.SearchOperation
  28
  29   def account_search(%{assigns: %{user: user}} = conn, %{q: query} = params) do
  30     accounts = User.search(query, search_options(params, user))
  31
  32     conn
  33     |> put_view(AccountView)
  34     |> render("index.json",
  35       users: accounts,
  36       for: user,
  37       as: :user
  38     )
  39   end
  40
  41   def search2(conn, params), do: do_search(:v2, conn, params)
  42   def search(conn, params), do: do_search(:v1, conn, params)
  43
  44   defp do_search(version, %{assigns: %{user: user}} = conn, params) do
  45     options =
  46       search_options(params, user)
  47       |> Keyword.put(:version, version)
  48
  49     search_provider = Pleroma.Config.get([:search, :provider])
  50     json(conn, search_provider.search(conn, params, options))
  51   end
  52
  53   defp search_options(params, user) do
  54     [
  55       resolve: params[:resolve],
  56       following: params[:following],
  57       limit: min(params[:limit], @search_limit),
  58       offset: params[:offset],
  59       type: params[:type],
  60       author: get_author(params),
  61       embed_relationships: ControllerHelper.embed_relationships?(params),
  62       for_user: user
  63     ]
  64     |> Enum.filter(&elem(&1, 1))
  65   end
  66
  67   defp get_author(%{account_id: account_id}) when is_binary(account_id),
  68     do: User.get_cached_by_id(account_id)
  69
  70   defp get_author(_params), do: nil
  71 end