Correct mimetype on bad uploads.
This commit is contained in:
parent
77e272df2d
commit
e942e1e552
@ -8,11 +8,19 @@ defmodule Pleroma.Upload do
|
|||||||
result_file = Path.join(upload_folder, file.filename)
|
result_file = Path.join(upload_folder, file.filename)
|
||||||
File.cp!(file.path, result_file)
|
File.cp!(file.path, result_file)
|
||||||
|
|
||||||
|
# fix content type on some image uploads
|
||||||
|
matches = Regex.named_captures(~r/\.(?<ext>(jpg|jpeg|png|gif))$/i, file.filename)
|
||||||
|
content_type = if file.content_type == "application/octet-stream" and matches do
|
||||||
|
if matches["ext"] == "jpg", do: "image/jpeg", else: "image/#{matches["ext"]}"
|
||||||
|
else
|
||||||
|
file.content_type
|
||||||
|
end
|
||||||
|
|
||||||
%{
|
%{
|
||||||
"type" => "Image",
|
"type" => "Image",
|
||||||
"url" => [%{
|
"url" => [%{
|
||||||
"type" => "Link",
|
"type" => "Link",
|
||||||
"mediaType" => file.content_type,
|
"mediaType" => content_type,
|
||||||
"href" => url_for(Path.join(uuid, :cow_uri.urlencode(file.filename)))
|
"href" => url_for(Path.join(uuid, :cow_uri.urlencode(file.filename)))
|
||||||
}],
|
}],
|
||||||
"name" => file.filename,
|
"name" => file.filename,
|
||||||
|
@ -9,5 +9,17 @@ defmodule Pleroma.UploadTest do
|
|||||||
assert data["name"] == "an [image.jpg"
|
assert data["name"] == "an [image.jpg"
|
||||||
assert List.first(data["url"])["href"] == "http://localhost:4001/media/#{data["uuid"]}/an%20%5Bimage.jpg"
|
assert List.first(data["url"])["href"] == "http://localhost:4001/media/#{data["uuid"]}/an%20%5Bimage.jpg"
|
||||||
end
|
end
|
||||||
|
|
||||||
|
test "fixes an incorrect content type" do
|
||||||
|
file = %Plug.Upload{content_type: "application/octet-stream", path: Path.absname("test/fixtures/image.jpg"), filename: "an [image.jpg"}
|
||||||
|
data = Upload.store(file)
|
||||||
|
assert hd(data["url"])["mediaType"] == "image/jpeg"
|
||||||
|
end
|
||||||
|
|
||||||
|
test "does not modify a valid content type" do
|
||||||
|
file = %Plug.Upload{content_type: "image/png", path: Path.absname("test/fixtures/image.jpg"), filename: "an [image.jpg"}
|
||||||
|
data = Upload.store(file)
|
||||||
|
assert hd(data["url"])["mediaType"] == "image/png"
|
||||||
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
Loading…
Reference in New Issue
Block a user