forked from AkkomaGang/akkoma
add the behaviour, work on actually making it work.
This commit is contained in:
parent
8d2d7a8859
commit
dad39b24a1
3 changed files with 45 additions and 37 deletions
|
@ -1,11 +1,13 @@
|
||||||
defmodule Pleroma.Upload do
|
defmodule Pleroma.Upload do
|
||||||
alias Ecto.UUID
|
alias Ecto.UUID
|
||||||
|
import Logger
|
||||||
|
|
||||||
@storage_backend Application.get_env(:pleroma, Pleroma.Upload)
|
@storage_backend Application.get_env(:pleroma, Pleroma.Upload)
|
||||||
|> Keyword.fetch!(:uploader)
|
|> Keyword.fetch!(:uploader)
|
||||||
|
|
||||||
def store(%Plug.Upload{} = file, should_dedupe) do
|
def store(%Plug.Upload{} = file, should_dedupe) do
|
||||||
content_type = get_content_type(file.path)
|
content_type = get_content_type(file.path)
|
||||||
|
|
||||||
uuid = get_uuid(file, should_dedupe)
|
uuid = get_uuid(file, should_dedupe)
|
||||||
name = get_name(file, uuid, content_type, should_dedupe)
|
name = get_name(file, uuid, content_type, should_dedupe)
|
||||||
|
|
||||||
|
@ -26,23 +28,21 @@ def store(%Plug.Upload{} = file, should_dedupe) do
|
||||||
}
|
}
|
||||||
end
|
end
|
||||||
|
|
||||||
"""
|
|
||||||
# XXX: does this code actually work? i am skeptical. --kaniini
|
# XXX: does this code actually work? i am skeptical. --kaniini
|
||||||
def store(%{"img" => "data:image/" <> image_data}, should_dedupe) do
|
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)
|
parsed = Regex.named_captures(~r/(?<filetype>jpeg|png|gif);base64,(?<data>.*)/, image_data)
|
||||||
data = Base.decode64!(parsed["data"], ignore: :whitespace)
|
data = Base.decode64!(parsed["data"], ignore: :whitespace)
|
||||||
uuid = UUID.generate()
|
|
||||||
uuidpath = Path.join(upload_path(), uuid)
|
tmp_path = mkupload_for_image(data)
|
||||||
|
|
||||||
uuid = UUID.generate()
|
uuid = UUID.generate()
|
||||||
|
|
||||||
File.mkdir_p!(upload_path())
|
# create temp local storage, like plug upload provides for us.
|
||||||
|
|
||||||
File.write!(uuidpath, data)
|
Logger.info(tmp_path)
|
||||||
|
|
||||||
content_type = get_content_type(uuidpath)
|
content_type = get_content_type(tmp_path)
|
||||||
|
strip_exif_data(content_type, tmp_path)
|
||||||
|
|
||||||
name =
|
name =
|
||||||
create_name(
|
create_name(
|
||||||
|
@ -51,30 +51,7 @@ def store(%{"img" => "data:image/" <> image_data}, should_dedupe) do
|
||||||
content_type
|
content_type
|
||||||
)
|
)
|
||||||
|
|
||||||
upload_folder = get_upload_path(uuid, should_dedupe)
|
url_path = @storage_backend.put_file(name, uuid, tmp_path, content_type, 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
|
|
||||||
|
|
||||||
%{
|
%{
|
||||||
"type" => "Image",
|
"type" => "Image",
|
||||||
|
@ -88,7 +65,12 @@ def store(%{"img" => "data:image/" <> image_data}, should_dedupe) do
|
||||||
"name" => name
|
"name" => name
|
||||||
}
|
}
|
||||||
end
|
end
|
||||||
"""
|
|
||||||
|
def mkupload_for_image(data) do
|
||||||
|
{:ok, tmp_path} = Plug.Upload.random_file("profile_pics")
|
||||||
|
:file.write_file(tmp_path, data, [:write, :raw, :exclusive, :binary])
|
||||||
|
tmp_path
|
||||||
|
end
|
||||||
|
|
||||||
def strip_exif_data(content_type, file) do
|
def strip_exif_data(content_type, file) do
|
||||||
settings = Application.get_env(:pleroma, Pleroma.Upload)
|
settings = Application.get_env(:pleroma, Pleroma.Upload)
|
||||||
|
|
|
@ -3,7 +3,7 @@ defmodule Pleroma.Uploaders.Local do
|
||||||
|
|
||||||
alias Pleroma.Web
|
alias Pleroma.Web
|
||||||
|
|
||||||
def put_file(name, uuid, file, _content_type, should_dedupe) do
|
def put_file(name, uuid, tmpfile, _content_type, should_dedupe) do
|
||||||
upload_folder = get_upload_path(uuid, should_dedupe)
|
upload_folder = get_upload_path(uuid, should_dedupe)
|
||||||
url_path = get_url(name, uuid, should_dedupe)
|
url_path = get_url(name, uuid, should_dedupe)
|
||||||
|
|
||||||
|
@ -12,9 +12,9 @@ def put_file(name, uuid, file, _content_type, should_dedupe) do
|
||||||
result_file = Path.join(upload_folder, name)
|
result_file = Path.join(upload_folder, name)
|
||||||
|
|
||||||
if File.exists?(result_file) do
|
if File.exists?(result_file) do
|
||||||
File.rm!(file.path)
|
File.rm!(tmpfile)
|
||||||
else
|
else
|
||||||
File.cp!(file.path, result_file)
|
File.cp!(tmpfile, result_file)
|
||||||
end
|
end
|
||||||
|
|
||||||
url_path
|
url_path
|
||||||
|
|
26
lib/pleroma/uploaders/uploader.ex
Normal file
26
lib/pleroma/uploaders/uploader.ex
Normal file
|
@ -0,0 +1,26 @@
|
||||||
|
defmodule Pleroma.Uploaders.Uploader do
|
||||||
|
@moduledoc """
|
||||||
|
Defines the contract to put an uploaded file to any backend.
|
||||||
|
"""
|
||||||
|
|
||||||
|
@doc """
|
||||||
|
Put a file to the backend.
|
||||||
|
|
||||||
|
Returns a `String.t` containing the path of the uploaded file.
|
||||||
|
"""
|
||||||
|
@callback put_file(
|
||||||
|
name :: String.t(),
|
||||||
|
uuid :: String.t(),
|
||||||
|
file :: File.t(),
|
||||||
|
content_type :: String.t(),
|
||||||
|
should_dedupe :: Boolean.t()
|
||||||
|
) :: String.t()
|
||||||
|
|
||||||
|
@callback put_file(
|
||||||
|
name :: String.t(),
|
||||||
|
uuid :: String.t(),
|
||||||
|
image_data :: String.t(),
|
||||||
|
content_type :: String.t(),
|
||||||
|
should_dedupe :: String.t()
|
||||||
|
) :: String.t()
|
||||||
|
end
|
Loading…
Reference in a new issue