git.squeep.com Git - akkoma/blob - lib/pleroma/web/plugs/set_format_plug.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.Plugs.SetFormatPlug do
   6   import Plug.Conn, only: [assign: 3, fetch_query_params: 1]
   7
   8   def init(_), do: nil
   9
  10   def call(conn, _) do
  11     case get_format(conn) do
  12       nil -> conn
  13       format -> assign(conn, :format, format)
  14     end
  15   end
  16
  17   defp get_format(conn) do
  18     conn.private[:phoenix_format] ||
  19       case fetch_query_params(conn) do
  20         %{query_params: %{"_format" => format}} -> format
  21         _ -> nil
  22       end
  23   end
  24 end