1 # Pleroma: A lightweight social networking server
2 # Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
3 # SPDX-License-Identifier: AGPL-3.0-only
5 defmodule Pleroma.Web.MediaProxy.MediaProxyController do
6 use Pleroma.Web, :controller
9 alias Pleroma.Helpers.MediaHelper
10 alias Pleroma.ReverseProxy
11 alias Pleroma.Web.MediaProxy
14 def remote(conn, %{"sig" => sig64, "url" => url64}) do
15 with {_, true} <- {:enabled, MediaProxy.enabled?()},
16 {:ok, url} <- MediaProxy.decode_url(sig64, url64),
17 {_, false} <- {:in_banned_urls, MediaProxy.in_banned_urls(url)},
18 :ok <- MediaProxy.verify_request_path_and_url(conn, url) do
19 ReverseProxy.call(conn, url, media_proxy_opts())
22 send_resp(conn, 404, Conn.Status.reason_phrase(404))
24 {:in_banned_urls, true} ->
25 send_resp(conn, 404, Conn.Status.reason_phrase(404))
27 {:error, :invalid_signature} ->
28 send_resp(conn, 403, Conn.Status.reason_phrase(403))
30 {:wrong_filename, filename} ->
31 redirect(conn, external: MediaProxy.build_url(sig64, url64, filename))
35 def preview(%Conn{} = conn, %{"sig" => sig64, "url" => url64}) do
36 with {_, true} <- {:enabled, MediaProxy.preview_enabled?()},
37 {:ok, url} <- MediaProxy.decode_url(sig64, url64) do
38 handle_preview(conn, url)
41 send_resp(conn, 404, Conn.Status.reason_phrase(404))
43 {:error, :invalid_signature} ->
44 send_resp(conn, 403, Conn.Status.reason_phrase(403))
46 {:wrong_filename, filename} ->
47 redirect(conn, external: MediaProxy.build_preview_url(sig64, url64, filename))
51 defp handle_preview(conn, url) do
52 media_proxy_url = MediaProxy.url(url)
54 with {:ok, %{status: status} = head_response} when status in 200..299 <-
55 Pleroma.HTTP.request("head", media_proxy_url, [], [], pool: :media) do
56 content_type = Tesla.get_header(head_response, "content-type")
57 handle_preview(content_type, conn, media_proxy_url)
59 {_, %{status: status}} ->
60 send_resp(conn, :failed_dependency, "Can't fetch HTTP headers (HTTP #{status}).")
62 {:error, :recv_response_timeout} ->
63 send_resp(conn, :failed_dependency, "HEAD request timeout.")
66 send_resp(conn, :failed_dependency, "Can't fetch HTTP headers.")
71 "image/" <> _ = _content_type,
72 %{params: %{"output_format" => "jpeg"}} = conn,
75 handle_jpeg_preview(conn, media_proxy_url)
78 defp handle_preview("image/gif" = _content_type, conn, media_proxy_url) do
79 redirect(conn, external: media_proxy_url)
82 defp handle_preview("image/png" <> _ = _content_type, conn, media_proxy_url) do
83 handle_png_preview(conn, media_proxy_url)
86 defp handle_preview("image/" <> _ = _content_type, conn, media_proxy_url) do
87 handle_jpeg_preview(conn, media_proxy_url)
90 defp handle_preview("video/" <> _ = _content_type, conn, media_proxy_url) do
91 handle_video_preview(conn, media_proxy_url)
94 defp handle_preview(content_type, conn, _media_proxy_url) do
95 send_resp(conn, :unprocessable_entity, "Unsupported content type: #{content_type}.")
98 defp handle_png_preview(conn, media_proxy_url) do
99 quality = Config.get!([:media_preview_proxy, :image_quality])
101 with {thumbnail_max_width, thumbnail_max_height} <- thumbnail_max_dimensions(),
102 {:ok, thumbnail_binary} <-
103 MediaHelper.image_resize(
106 max_width: thumbnail_max_width,
107 max_height: thumbnail_max_height,
113 |> put_preview_response_headers(["image/png", "preview.png"])
114 |> send_resp(200, thumbnail_binary)
117 send_resp(conn, :failed_dependency, "Can't handle preview.")
121 defp handle_jpeg_preview(conn, media_proxy_url) do
122 quality = Config.get!([:media_preview_proxy, :image_quality])
124 with {thumbnail_max_width, thumbnail_max_height} <- thumbnail_max_dimensions(),
125 {:ok, thumbnail_binary} <-
126 MediaHelper.image_resize(
128 %{max_width: thumbnail_max_width, max_height: thumbnail_max_height, quality: quality}
131 |> put_preview_response_headers()
132 |> send_resp(200, thumbnail_binary)
135 send_resp(conn, :failed_dependency, "Can't handle preview.")
139 defp handle_video_preview(conn, media_proxy_url) do
140 with {:ok, thumbnail_binary} <-
141 MediaHelper.video_framegrab(media_proxy_url) do
143 |> put_preview_response_headers()
144 |> send_resp(200, thumbnail_binary)
147 send_resp(conn, :failed_dependency, "Can't handle preview.")
151 defp put_preview_response_headers(
153 [content_type, filename] = _content_info \\ ["image/jpeg", "preview.jpg"]
156 |> put_resp_header("content-type", content_type)
157 |> put_resp_header("content-disposition", "inline; filename=\"#{filename}\"")
158 |> put_resp_header("cache-control", ReverseProxy.default_cache_control_header())
161 defp thumbnail_max_dimensions() do
162 config = Config.get([:media_preview_proxy], [])
164 thumbnail_max_width = Keyword.fetch!(config, :thumbnail_max_width)
165 thumbnail_max_height = Keyword.fetch!(config, :thumbnail_max_height)
167 {thumbnail_max_width, thumbnail_max_height}
170 defp media_proxy_opts do
171 Config.get([:media_proxy, :proxy_opts], [])