http.ex 3.19 KB
Newer Older
1
# Pleroma: A lightweight social networking server
feld's avatar
feld committed
2
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
3
4
# SPDX-License-Identifier: AGPL-3.0-only

5
defmodule Pleroma.HTTP do
Maksim's avatar
Maksim committed
6
  @moduledoc """
Alexander Strizhakov's avatar
Alexander Strizhakov committed
7
    Wrapper for `Tesla.request/2`.
Maksim's avatar
Maksim committed
8
9
  """

10
  alias Pleroma.HTTP.AdapterHelper
Alexander Strizhakov's avatar
Alexander Strizhakov committed
11
  alias Pleroma.HTTP.Request
12
  alias Pleroma.HTTP.RequestBuilder, as: Builder
Alexander Strizhakov's avatar
Alexander Strizhakov committed
13
14
15
16
  alias Tesla.Client
  alias Tesla.Env

  require Logger
kaniini's avatar
kaniini committed
17

18
  @type t :: __MODULE__
19
  @type method() :: :get | :post | :put | :delete | :head
20

Maksim's avatar
Maksim committed
21
  @doc """
Alexander Strizhakov's avatar
Alexander Strizhakov committed
22
23
24
25
26
27
28
29
30
31
  Performs GET request.

  See `Pleroma.HTTP.request/5`
  """
  @spec get(Request.url() | nil, Request.headers(), keyword()) ::
          nil | {:ok, Env.t()} | {:error, any()}
  def get(url, headers \\ [], options \\ [])
  def get(nil, _, _), do: nil
  def get(url, headers, options), do: request(:get, url, "", headers, options)

32
33
34
  @spec head(Request.url(), Request.headers(), keyword()) :: {:ok, Env.t()} | {:error, any()}
  def head(url, headers \\ [], options \\ []), do: request(:head, url, "", headers, options)

Alexander Strizhakov's avatar
Alexander Strizhakov committed
35
36
37
38
39
40
41
42
43
44
45
46
  @doc """
  Performs POST request.

  See `Pleroma.HTTP.request/5`
  """
  @spec post(Request.url(), String.t(), Request.headers(), keyword()) ::
          {:ok, Env.t()} | {:error, any()}
  def post(url, body, headers \\ [], options \\ []),
    do: request(:post, url, body, headers, options)

  @doc """
  Builds and performs http request.
Maksim's avatar
Maksim committed
47
48

  # Arguments:
49
  `method` - :get, :post, :put, :delete, :head
Alexander Strizhakov's avatar
Alexander Strizhakov committed
50
51
  `url` - full url
  `body` - request body
Maksim's avatar
Maksim committed
52
53
54
55
56
57
  `headers` - a keyworld list of headers, e.g. `[{"content-type", "text/plain"}]`
  `options` - custom, per-request middleware or adapter options

  # Returns:
  `{:ok, %Tesla.Env{}}` or `{:error, error}`

Maksim's avatar
Maksim committed
58
  """
59
  @spec request(method(), Request.url(), String.t(), Request.headers(), keyword()) ::
Alexander Strizhakov's avatar
Alexander Strizhakov committed
60
61
          {:ok, Env.t()} | {:error, any()}
  def request(method, url, body, headers, options) when is_binary(url) do
Alexander Strizhakov's avatar
Alexander Strizhakov committed
62
    uri = URI.parse(url)
63
    adapter_opts = AdapterHelper.options(uri, options || [])
Alexander Strizhakov's avatar
Alexander Strizhakov committed
64

65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
    options = put_in(options[:adapter], adapter_opts)
    params = options[:params] || []
    request = build_request(method, headers, options, url, body, params)

    adapter = Application.get_env(:tesla, :adapter)

    client = Tesla.client(adapter_middlewares(adapter), adapter)

    maybe_limit(
      fn ->
        request(client, request)
      end,
      adapter,
      adapter_opts
    )
kaniini's avatar
kaniini committed
80
81
  end

82
83
  @spec request(Client.t(), keyword()) :: {:ok, Env.t()} | {:error, any()}
  def request(client, request), do: Tesla.request(client, request)
84

Alexander Strizhakov's avatar
Alexander Strizhakov committed
85
86
87
88
89
90
91
92
93
  defp build_request(method, headers, options, url, body, params) do
    Builder.new()
    |> Builder.method(method)
    |> Builder.headers(headers)
    |> Builder.opts(options)
    |> Builder.url(url)
    |> Builder.add_param(:body, :body, body)
    |> Builder.add_param(:query, :query, params)
    |> Builder.convert_to_keyword()
94
  end
rinpatch's avatar
rinpatch committed
95
96
97
98
99
100
101
102
103

  @prefix Pleroma.Gun.ConnectionPool
  defp maybe_limit(fun, Tesla.Adapter.Gun, opts) do
    ConcurrentLimiter.limit(:"#{@prefix}.#{opts[:pool] || :default}", fun)
  end

  defp maybe_limit(fun, _, _) do
    fun.()
  end
104
105

  defp adapter_middlewares(Tesla.Adapter.Gun) do
106
    [Tesla.Middleware.FollowRedirects, Pleroma.Tesla.Middleware.ConnectionPool]
107
108
109
  end

  defp adapter_middlewares(_), do: []
110
end