git.squeep.com Git - akkoma/blob - lib/pleroma/http/adapter_helper/gun.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.HTTP.AdapterHelper.Gun do
   6   @behaviour Pleroma.HTTP.AdapterHelper
   7
   8   alias Pleroma.HTTP.AdapterHelper
   9   alias Pleroma.Pool.Connections
  10
  11   require Logger
  12
  13   @defaults [
  14     connect_timeout: 5_000,
  15     domain_lookup_timeout: 5_000,
  16     tls_handshake_timeout: 5_000,
  17     retry: 1,
  18     retry_timeout: 1000,
  19     await_up_timeout: 5_000
  20   ]
  21
  22   @spec options(keyword(), URI.t()) :: keyword()
  23   def options(incoming_opts \\ [], %URI{} = uri) do
  24     proxy =
  25       Pleroma.Config.get([:http, :proxy_url])
  26       |> AdapterHelper.format_proxy()
  27
  28     config_opts = Pleroma.Config.get([:http, :adapter], [])
  29
  30     @defaults
  31     |> Keyword.merge(config_opts)
  32     |> add_scheme_opts(uri)
  33     |> AdapterHelper.maybe_add_proxy(proxy)
  34     |> maybe_get_conn(uri, incoming_opts)
  35   end
  36
  37   @spec after_request(keyword()) :: :ok
  38   def after_request(opts) do
  39     if opts[:conn] && opts[:body_as] != :chunks do
  40       Connections.checkout(opts[:conn], self(), :gun_connections)
  41     end
  42
  43     :ok
  44   end
  45
  46   defp add_scheme_opts(opts, %{scheme: "http"}), do: opts
  47
  48   defp add_scheme_opts(opts, %{scheme: "https"}) do
  49     opts
  50     |> Keyword.put(:certificates_verification, true)
  51     |> Keyword.put(:tls_opts, log_level: :warning)
  52   end
  53
  54   defp maybe_get_conn(adapter_opts, uri, incoming_opts) do
  55     {receive_conn?, opts} =
  56       adapter_opts
  57       |> Keyword.merge(incoming_opts)
  58       |> Keyword.pop(:receive_conn, true)
  59
  60     if Connections.alive?(:gun_connections) and receive_conn? do
  61       checkin_conn(uri, opts)
  62     else
  63       opts
  64     end
  65   end
  66
  67   defp checkin_conn(uri, opts) do
  68     case Connections.checkin(uri, :gun_connections) do
  69       nil ->
  70         Task.start(Pleroma.Gun.Conn, :open, [uri, :gun_connections, opts])
  71         opts
  72
  73       conn when is_pid(conn) ->
  74         Keyword.merge(opts, conn: conn, close_conn: false)
  75     end
  76   end
  77 end