|
|
|
@ -1,34 +1,19 @@
|
|
|
|
|
defmodule Pleroma.Upload do
|
|
|
|
|
alias Ecto.UUID
|
|
|
|
|
alias Pleroma.Web
|
|
|
|
|
|
|
|
|
|
def store(%Plug.Upload{} = file, should_dedupe) do
|
|
|
|
|
settings = Application.get_env(:pleroma, Pleroma.Upload)
|
|
|
|
|
use_s3 = Keyword.fetch!(settings, :use_s3)
|
|
|
|
|
@storage_backend Application.get_env(:pleroma, Pleroma.Upload)
|
|
|
|
|
|> Keyword.fetch!(:uploader)
|
|
|
|
|
|
|
|
|
|
def store(%Plug.Upload{} = file, should_dedupe) do
|
|
|
|
|
content_type = get_content_type(file.path)
|
|
|
|
|
|
|
|
|
|
uuid = get_uuid(file, should_dedupe)
|
|
|
|
|
name = get_name(file, uuid, content_type, should_dedupe)
|
|
|
|
|
upload_folder = get_upload_path(uuid, should_dedupe)
|
|
|
|
|
url_path = get_url(name, uuid, should_dedupe)
|
|
|
|
|
|
|
|
|
|
strip_exif_data(content_type, file.path)
|
|
|
|
|
|
|
|
|
|
File.mkdir_p!(upload_folder)
|
|
|
|
|
result_file = Path.join(upload_folder, name)
|
|
|
|
|
|
|
|
|
|
if File.exists?(result_file) do
|
|
|
|
|
File.rm!(file.path)
|
|
|
|
|
else
|
|
|
|
|
File.cp!(file.path, result_file)
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
url_path =
|
|
|
|
|
if use_s3 do
|
|
|
|
|
put_s3_file(name, uuid, result_file, content_type)
|
|
|
|
|
else
|
|
|
|
|
url_path
|
|
|
|
|
end
|
|
|
|
|
{:ok, url_path} =
|
|
|
|
|
@storage_backend.put_file(name, uuid, file.path, content_type, should_dedupe)
|
|
|
|
|
|
|
|
|
|
%{
|
|
|
|
|
"type" => "Document",
|
|
|
|
@ -43,22 +28,16 @@ defmodule Pleroma.Upload do
|
|
|
|
|
}
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
# XXX: does this code actually work? i am skeptical. --kaniini
|
|
|
|
|
def store(%{"img" => "data:image/" <> image_data}, should_dedupe) do
|
|
|
|
|
settings = Application.get_env(:pleroma, Pleroma.Upload)
|
|
|
|
|
use_s3 = Keyword.fetch!(settings, :use_s3)
|
|
|
|
|
|
|
|
|
|
parsed = Regex.named_captures(~r/(?<filetype>jpeg|png|gif);base64,(?<data>.*)/, image_data)
|
|
|
|
|
data = Base.decode64!(parsed["data"], ignore: :whitespace)
|
|
|
|
|
uuid = UUID.generate()
|
|
|
|
|
uuidpath = Path.join(upload_path(), uuid)
|
|
|
|
|
uuid = UUID.generate()
|
|
|
|
|
|
|
|
|
|
File.mkdir_p!(upload_path())
|
|
|
|
|
tmp_path = tempfile_for_image(data)
|
|
|
|
|
|
|
|
|
|
File.write!(uuidpath, data)
|
|
|
|
|
uuid = UUID.generate()
|
|
|
|
|
|
|
|
|
|
content_type = get_content_type(uuidpath)
|
|
|
|
|
content_type = get_content_type(tmp_path)
|
|
|
|
|
strip_exif_data(content_type, tmp_path)
|
|
|
|
|
|
|
|
|
|
name =
|
|
|
|
|
create_name(
|
|
|
|
@ -67,30 +46,7 @@ defmodule Pleroma.Upload do
|
|
|
|
|
content_type
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
upload_folder = get_upload_path(uuid, should_dedupe)
|
|
|
|
|
url_path = get_url(name, uuid, should_dedupe)
|
|
|
|
|
|
|
|
|
|
File.mkdir_p!(upload_folder)
|
|
|
|
|
result_file = Path.join(upload_folder, name)
|
|
|
|
|
|
|
|
|
|
if should_dedupe do
|
|
|
|
|
if !File.exists?(result_file) do
|
|
|
|
|
File.rename(uuidpath, result_file)
|
|
|
|
|
else
|
|
|
|
|
File.rm!(uuidpath)
|
|
|
|
|
end
|
|
|
|
|
else
|
|
|
|
|
File.rename(uuidpath, result_file)
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
strip_exif_data(content_type, result_file)
|
|
|
|
|
|
|
|
|
|
url_path =
|
|
|
|
|
if use_s3 do
|
|
|
|
|
put_s3_file(name, uuid, result_file, content_type)
|
|
|
|
|
else
|
|
|
|
|
url_path
|
|
|
|
|
end
|
|
|
|
|
{:ok, url_path} = @storage_backend.put_file(name, uuid, tmp_path, content_type, should_dedupe)
|
|
|
|
|
|
|
|
|
|
%{
|
|
|
|
|
"type" => "Image",
|
|
|
|
@ -105,21 +61,28 @@ defmodule Pleroma.Upload do
|
|
|
|
|
}
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
@doc """
|
|
|
|
|
Creates a tempfile using the Plug.Upload Genserver which cleans them up
|
|
|
|
|
automatically.
|
|
|
|
|
"""
|
|
|
|
|
def tempfile_for_image(data) do
|
|
|
|
|
{:ok, tmp_path} = Plug.Upload.random_file("profile_pics")
|
|
|
|
|
{:ok, tmp_file} = File.open(tmp_path, [:write, :raw, :binary])
|
|
|
|
|
IO.binwrite(tmp_file, data)
|
|
|
|
|
|
|
|
|
|
tmp_path
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
def strip_exif_data(content_type, file) do
|
|
|
|
|
settings = Application.get_env(:pleroma, Pleroma.Upload)
|
|
|
|
|
do_strip = Keyword.fetch!(settings, :strip_exif)
|
|
|
|
|
[filetype, ext] = String.split(content_type, "/")
|
|
|
|
|
[filetype, _ext] = String.split(content_type, "/")
|
|
|
|
|
|
|
|
|
|
if filetype == "image" and do_strip == true do
|
|
|
|
|
Mogrify.open(file) |> Mogrify.custom("strip") |> Mogrify.save(in_place: true)
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
def upload_path do
|
|
|
|
|
settings = Application.get_env(:pleroma, Pleroma.Upload)
|
|
|
|
|
Keyword.fetch!(settings, :uploads)
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
defp create_name(uuid, ext, type) do
|
|
|
|
|
case type do
|
|
|
|
|
"application/octet-stream" ->
|
|
|
|
@ -163,26 +126,6 @@ defmodule Pleroma.Upload do
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
defp get_upload_path(uuid, should_dedupe) do
|
|
|
|
|
if should_dedupe do
|
|
|
|
|
upload_path()
|
|
|
|
|
else
|
|
|
|
|
Path.join(upload_path(), uuid)
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
defp get_url(name, uuid, should_dedupe) do
|
|
|
|
|
if should_dedupe do
|
|
|
|
|
url_for(:cow_uri.urlencode(name))
|
|
|
|
|
else
|
|
|
|
|
url_for(Path.join(uuid, :cow_uri.urlencode(name)))
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
defp url_for(file) do
|
|
|
|
|
"#{Web.base_url()}/media/#{file}"
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
def get_content_type(file) do
|
|
|
|
|
match =
|
|
|
|
|
File.open(file, [:read], fn f ->
|
|
|
|
@ -224,25 +167,4 @@ defmodule Pleroma.Upload do
|
|
|
|
|
_e -> "application/octet-stream"
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
defp put_s3_file(name, uuid, path, content_type) do
|
|
|
|
|
settings = Application.get_env(:pleroma, Pleroma.Upload)
|
|
|
|
|
bucket = Keyword.fetch!(settings, :bucket)
|
|
|
|
|
public_endpoint = Keyword.fetch!(settings, :public_endpoint)
|
|
|
|
|
|
|
|
|
|
{:ok, file_data} = File.read(path)
|
|
|
|
|
|
|
|
|
|
File.rm!(path)
|
|
|
|
|
|
|
|
|
|
s3_name = "#{uuid}/#{name}"
|
|
|
|
|
|
|
|
|
|
{:ok, result} =
|
|
|
|
|
ExAws.S3.put_object(bucket, s3_name, file_data, [
|
|
|
|
|
{:acl, :public_read},
|
|
|
|
|
{:content_type, content_type}
|
|
|
|
|
])
|
|
|
|
|
|> ExAws.request()
|
|
|
|
|
|
|
|
|
|
"#{public_endpoint}/#{bucket}/#{s3_name}"
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|