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.Plugs.UploadedMedia do
10 import Pleroma.Web.Gettext
17 @default_cache_control_header "public, max-age=1209600"
22 |> Keyword.put(:from, "__unconfigured_media_plug")
23 |> Keyword.put(:at, "/__unconfigured_media_plug")
26 %{static_plug_opts: static_plug_opts}
29 def call(%{request_path: <<"/", @path, "/", file::binary>>} = conn, opts) do
31 case fetch_query_params(conn) do
32 %{query_params: %{"name" => name}} = conn ->
33 name = String.replace(name, "\"", "\\\"")
36 |> put_resp_header("content-disposition", "filename=\"#{name}\"")
42 config = Pleroma.Config.get(Pleroma.Upload)
44 with uploader <- Keyword.fetch!(config, :uploader),
45 proxy_remote = Keyword.get(config, :proxy_remote, false),
46 {:ok, get_method} <- uploader.get_file(file) do
47 get_media(conn, get_method, proxy_remote, opts)
51 |> send_resp(:internal_server_error, dgettext("errors", "Failed"))
56 def call(conn, _opts), do: conn
58 defp get_media(conn, {:static_dir, directory}, _, opts) do
60 Map.get(opts, :static_plug_opts)
61 |> Map.put(:at, [@path])
62 |> Map.put(:from, directory)
63 |> Map.put(:cache_control_for_etags, @default_cache_control_header)
64 |> Map.put(:headers, %{
65 "cache-control" => @default_cache_control_header
68 conn = Plug.Static.call(conn, static_opts)
74 |> send_resp(:not_found, dgettext("errors", "Not found"))
79 defp get_media(conn, {:url, url}, true, _) do
81 |> Pleroma.ReverseProxy.call(url, Pleroma.Config.get([Pleroma.Upload, :proxy_opts], []))
84 defp get_media(conn, {:url, url}, _, _) do
86 |> Phoenix.Controller.redirect(external: url)
90 defp get_media(conn, unknown, _, _) do
91 Logger.error("#{__MODULE__}: Unknown get startegy: #{inspect(unknown)}")
94 |> send_resp(:internal_server_error, dgettext("errors", "Internal Error"))