git.squeep.com Git - akkoma/blob - lib/pleroma/web/media_proxy/controller.ex

   1 defmodule Pleroma.Web.MediaProxy.MediaProxyController do
   2   use Pleroma.Web, :controller
   3   alias Pleroma.{Web.MediaProxy, ReverseProxy}
   4
   5   def remote(conn, params = %{"sig" => sig64, "url" => url64}) do
   6     with config <- Pleroma.Config.get([:media_proxy]),
   7          true <- Keyword.get(config, :enabled, false),
   8          {:ok, url} <- MediaProxy.decode_url(sig64, url64),
   9          filename <- Path.basename(URI.parse(url).path),
  10          :ok <- filename_matches(Map.has_key?(params, "filename"), conn.request_path, url) do
  11       ReverseProxy.call(conn, url, Keyword.get(config, :proxy_opts, []))
  12     else
  13       false ->
  14         send_resp(conn, 404, Plug.Conn.Status.reason_phrase(404))
  15
  16       {:error, :invalid_signature} ->
  17         send_resp(conn, 403, Plug.Conn.Status.reason_phrase(403))
  18
  19       {:wrong_filename, filename} ->
  20         redirect(conn, external: MediaProxy.build_url(sig64, url64, filename))
  21     end
  22   end
  23
  24   def filename_matches(has_filename, path, url) do
  25     filename = MediaProxy.filename(url)
  26
  27     cond do
  28       has_filename && filename && Path.basename(path) != filename -> {:wrong_filename, filename}
  29       true -> :ok
  30     end
  31   end
  32 end