qb_akkoma/lib/pleroma/uploaders/s3.ex

41 lines
992 B
Elixir
Raw Normal View History

2018-08-28 01:20:54 +00:00
defmodule Pleroma.Uploaders.S3 do
alias Pleroma.Web.MediaProxy
@behaviour Pleroma.Uploaders.Uploader
2018-08-28 01:20:54 +00:00
2018-08-28 01:45:53 +00:00
def put_file(name, uuid, path, content_type, _should_dedupe) do
2018-08-28 01:20:54 +00:00
settings = Application.get_env(:pleroma, Pleroma.Uploaders.S3)
bucket = Keyword.fetch!(settings, :bucket)
public_endpoint = Keyword.fetch!(settings, :public_endpoint)
force_media_proxy = Keyword.fetch!(settings, :force_media_proxy)
2018-08-28 01:20:54 +00:00
{:ok, file_data} = File.read(path)
File.rm!(path)
s3_name = "#{uuid}/#{encode(name)}"
2018-08-28 01:20:54 +00:00
2018-08-28 01:45:53 +00:00
{:ok, _} =
2018-08-28 01:20:54 +00:00
ExAws.S3.put_object(bucket, s3_name, file_data, [
{:acl, :public_read},
{:content_type, content_type}
])
|> ExAws.request()
url_base = "#{public_endpoint}/#{bucket}/#{s3_name}"
public_url =
if force_media_proxy do
MediaProxy.url(url_base)
else
url_base
end
{:ok, public_url}
2018-08-28 01:20:54 +00:00
end
defp encode(name) do
String.replace(name, ~r/[^0-9a-zA-Z!.*'()_-]/, "-")
end
2018-08-28 01:20:54 +00:00
end