Pleroma/lib/pleroma/http/http.ex

122 lines
3.7 KiB
Elixir
Raw Normal View History

# Pleroma: A lightweight social networking server
2020-03-03 23:44:49 +01:00
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
# SPDX-License-Identifier: AGPL-3.0-only
defmodule Pleroma.HTTP do
2018-12-04 15:51:49 +01:00
@moduledoc """
2020-02-11 08:12:57 +01:00
Wrapper for `Tesla.request/2`.
2018-12-04 15:51:49 +01:00
"""
alias Pleroma.HTTP.Connection
2020-02-11 08:12:57 +01:00
alias Pleroma.HTTP.Request
alias Pleroma.HTTP.RequestBuilder, as: Builder
2020-02-11 08:12:57 +01:00
alias Tesla.Client
alias Tesla.Env
require Logger
@type t :: __MODULE__
@type method() :: :get | :post | :put | :delete | :head
2018-12-04 15:48:55 +01:00
@doc """
2020-02-11 08:12:57 +01:00
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)
@spec head(Request.url(), Request.headers(), keyword()) :: {:ok, Env.t()} | {:error, any()}
def head(url, headers \\ [], options \\ []), do: request(:head, url, "", headers, options)
2020-02-11 08:12:57 +01:00
@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.
2018-12-04 15:51:49 +01:00
# Arguments:
`method` - :get, :post, :put, :delete, :head
2020-02-11 08:12:57 +01:00
`url` - full url
`body` - request body
2018-12-04 15:51:49 +01:00
`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}`
2018-12-04 15:48:55 +01:00
"""
@spec request(method(), Request.url(), String.t(), Request.headers(), keyword()) ::
2020-02-11 08:12:57 +01:00
{:ok, Env.t()} | {:error, any()}
def request(method, url, body, headers, options) when is_binary(url) do
2020-03-03 13:11:48 +01:00
uri = URI.parse(url)
2020-03-12 16:28:54 +01:00
adapter_opts = Connection.options(uri, options[:adapter] || [])
2020-03-03 13:11:48 +01:00
options = put_in(options[:adapter], adapter_opts)
2020-03-12 16:28:54 +01:00
params = options[:params] || []
2020-03-03 13:11:48 +01:00
request = build_request(method, headers, options, url, body, params)
adapter = Application.get_env(:tesla, :adapter)
client = Tesla.client([Tesla.Middleware.FollowRedirects], adapter)
pid = Process.whereis(adapter_opts[:pool])
pool_alive? =
if adapter == Tesla.Adapter.Gun && pid do
Process.alive?(pid)
else
false
end
request_opts =
adapter_opts
|> Enum.into(%{})
|> Map.put(:env, Pleroma.Config.get([:env]))
|> Map.put(:pool_alive?, pool_alive?)
response = request(client, request, request_opts)
Connection.after_request(adapter_opts)
response
2020-02-11 08:12:57 +01:00
end
@spec request(Client.t(), keyword(), map()) :: {:ok, Env.t()} | {:error, any()}
def request(%Client{} = client, request, %{env: :test}), do: request(client, request)
2020-02-11 08:12:57 +01:00
def request(%Client{} = client, request, %{body_as: :chunks}), do: request(client, request)
2020-02-11 08:12:57 +01:00
def request(%Client{} = client, request, %{pool_alive?: false}), do: request(client, request)
2020-02-11 08:12:57 +01:00
def request(%Client{} = client, request, %{pool: pool, timeout: timeout}) do
2020-03-03 15:27:56 +01:00
:poolboy.transaction(
pool,
&Pleroma.Pool.Request.execute(&1, client, request, timeout),
timeout
)
end
@spec request(Client.t(), keyword()) :: {:ok, Env.t()} | {:error, any()}
def request(client, request), do: Tesla.request(client, request)
2020-02-11 08:12:57 +01:00
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()
end
end