git.squeep.com Git - akkoma/blob - lib/pleroma/uploaders/s3.ex

   1 defmodule Pleroma.Uploaders.S3 do
   2   alias Pleroma.Web.MediaProxy
   3
   4   @behaviour Pleroma.Uploaders.Uploader
   5
   6   def put_file(name, uuid, path, content_type, _should_dedupe) do
   7     settings = Application.get_env(:pleroma, Pleroma.Uploaders.S3)
   8     bucket = Keyword.fetch!(settings, :bucket)
   9     public_endpoint = Keyword.fetch!(settings, :public_endpoint)
  10     force_media_proxy = Keyword.fetch!(settings, :force_media_proxy)
  11
  12     {:ok, file_data} = File.read(path)
  13
  14     File.rm!(path)
  15
  16     s3_name = "#{uuid}/#{encode(name)}"
  17
  18     {:ok, _} =
  19       ExAws.S3.put_object(bucket, s3_name, file_data, [
  20         {:acl, :public_read},
  21         {:content_type, content_type}
  22       ])
  23       |> ExAws.request()
  24
  25     url_base = "#{public_endpoint}/#{bucket}/#{s3_name}"
  26
  27     public_url =
  28       if force_media_proxy do
  29         MediaProxy.url(url_base)
  30       else
  31         url_base
  32       end
  33
  34     {:ok, public_url}
  35   end
  36
  37   defp encode(name) do
  38     String.replace(name, ~r/[^0-9a-zA-Z!.*'()_-]/, "-")
  39   end
  40 end