git.squeep.com Git - akkoma/blob - lib/pleroma/reverse_proxy/client/tesla.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.ReverseProxy.Client.Tesla do
   6   @behaviour Pleroma.ReverseProxy.Client
   7
   8   @type headers() :: [{String.t(), String.t()}]
   9   @type status() :: pos_integer()
  10
  11   @spec request(atom(), String.t(), headers(), String.t(), keyword()) ::
  12           {:ok, status(), headers}
  13           | {:ok, status(), headers, map()}
  14           | {:error, atom() | String.t()}
  15           | no_return()
  16
  17   @impl true
  18   def request(method, url, headers, body, opts \\ []) do
  19     check_adapter()
  20
  21     opts = Keyword.put(opts, :body_as, :chunks)
  22
  23     with {:ok, response} <-
  24            Pleroma.HTTP.request(
  25              method,
  26              url,
  27              body,
  28              headers,
  29              opts
  30            ) do
  31       if is_map(response.body) and method != :head do
  32         {:ok, response.status, response.headers, response.body}
  33       else
  34         conn_pid = response.opts[:adapter][:conn]
  35         ConnectionPool.release_conn(conn_pid)
  36         {:ok, response.status, response.headers}
  37       end
  38     else
  39       {:error, error} -> {:error, error}
  40     end
  41   end
  42
  43   @impl true
  44   @spec stream_body(map()) ::
  45           {:ok, binary(), map()} | {:error, atom() | String.t()} | :done | no_return()
  46   def stream_body(%{pid: pid, fin: true}) do
  47     ConnectionPool.release_conn(pid)
  48     :done
  49   end
  50
  51   def stream_body(client) do
  52     case read_chunk!(client) do
  53       {:fin, body} ->
  54         {:ok, body, Map.put(client, :fin, true)}
  55
  56       {:nofin, part} ->
  57         {:ok, part, client}
  58
  59       {:error, error} ->
  60         {:error, error}
  61     end
  62   end
  63
  64   defp read_chunk!(%{pid: pid, stream: stream, opts: opts}) do
  65     adapter = check_adapter()
  66     adapter.read_chunk(pid, stream, opts)
  67   end
  68
  69   @impl true
  70   @spec close(map) :: :ok | no_return()
  71   def close(%{pid: pid}) do
  72     ConnectionPool.release_conn(pid)
  73   end
  74
  75   defp check_adapter do
  76     adapter = Application.get_env(:tesla, :adapter)
  77
  78     adapter
  79   end
  80 end