git.squeep.com Git - akkoma/blob - lib/pleroma/reverse_proxy/client.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.ReverseProxy.Client do
   6   @type status :: pos_integer()
   7   @type header_name :: String.t()
   8   @type header_value :: String.t()
   9   @type headers :: [{header_name(), header_value()}]
  10
  11   @callback request(atom(), String.t(), headers(), String.t(), list()) ::
  12               {:ok, status(), headers(), reference() | map()}
  13               | {:ok, status(), headers()}
  14               | {:ok, reference()}
  15               | {:error, term()}
  16
  17   @callback stream_body(map()) :: {:ok, binary(), map()} | :done | {:error, atom() | String.t()}
  18
  19   @callback close(reference() | pid() | map()) :: :ok
  20
  21   def request(method, url, headers, body \\ "", opts \\ []) do
  22     client().request(method, url, headers, body, opts)
  23   end
  24
  25   def stream_body(ref), do: client().stream_body(ref)
  26
  27   def close(ref), do: client().close(ref)
  28
  29   defp client do
  30     :tesla
  31     |> Application.get_env(:adapter)
  32     |> client()
  33   end
  34
  35   defp client(Tesla.Adapter.Hackney), do: Pleroma.ReverseProxy.Client.Hackney
  36   defp client(Tesla.Adapter.Gun), do: Pleroma.ReverseProxy.Client.Tesla
  37   defp client(_), do: Pleroma.Config.get!(Pleroma.ReverseProxy.Client)
  38 end