akkoma/lib/pleroma/helpers/media_helper.ex

135 lines
3.1 KiB
Elixir
Raw Normal View History

# Pleroma: A lightweight social networking server
# Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
# SPDX-License-Identifier: AGPL-3.0-only
defmodule Pleroma.Helpers.MediaHelper do
@moduledoc """
Handles common media-related operations.
"""
2020-08-26 14:12:34 +00:00
@tmp_base "/tmp/pleroma-media_preview-pipe"
def image_resize(url, options) do
with executable when is_binary(executable) <- System.find_executable("convert"),
{:ok, args} <- prepare_image_resize_args(options),
2020-08-26 14:12:34 +00:00
url = Pleroma.Web.MediaProxy.url(url),
{:ok, env} <- Pleroma.HTTP.get(url),
{:ok, fifo_path} <- mkfifo() do
args = List.flatten([fifo_path, args])
2020-08-26 14:12:34 +00:00
run_fifo(fifo_path, env, executable, args)
else
nil -> {:error, {:convert, :command_not_found}}
{:error, _} = error -> error
end
end
defp prepare_image_resize_args(%{max_width: max_width, max_height: max_height} = options) do
quality = options[:quality] || 85
2020-08-26 14:12:34 +00:00
resize = Enum.join([max_width, "x", max_height, ">"])
2020-08-26 14:12:34 +00:00
args = [
"-interlace",
"Plane",
"-resize",
resize,
"-quality",
to_string(quality),
"jpg:-"
2020-08-26 14:12:34 +00:00
]
2020-08-26 14:12:34 +00:00
{:ok, args}
end
defp prepare_image_resize_args(_), do: {:error, :missing_options}
def video_framegrab(url) do
with executable when is_binary(executable) <- System.find_executable("ffmpeg"),
url = Pleroma.Web.MediaProxy.url(url),
{:ok, env} <- Pleroma.HTTP.get(url),
{:ok, fifo_path} <- mkfifo(),
args = [
"-y",
"-i",
fifo_path,
"-vframes",
"1",
"-f",
"mjpeg",
"-loglevel",
"error",
"-"
] do
run_fifo(fifo_path, env, executable, args)
else
nil -> {:error, {:ffmpeg, :command_not_found}}
{:error, _} = error -> error
end
end
2020-08-26 14:12:34 +00:00
defp run_fifo(fifo_path, env, executable, args) do
pid =
Port.open({:spawn_executable, executable}, [
:use_stdio,
:stream,
:exit_status,
:binary,
args: args
])
2020-08-26 14:12:34 +00:00
fifo = Port.open(to_charlist(fifo_path), [:eof, :binary, :stream, :out])
fix = Pleroma.Helpers.QtFastStart.fix(env.body)
true = Port.command(fifo, fix)
2020-08-26 14:12:34 +00:00
:erlang.port_close(fifo)
2020-08-21 17:02:57 +00:00
loop_recv(pid)
2020-08-26 14:12:34 +00:00
after
File.rm(fifo_path)
end
defp mkfifo() do
path = "#{@tmp_base}#{to_charlist(:erlang.phash2(self()))}"
2020-08-26 14:12:34 +00:00
case System.cmd("mkfifo", [path]) do
{_, 0} ->
spawn(fifo_guard(path))
{:ok, path}
2020-08-26 14:12:34 +00:00
{_, err} ->
{:error, {:fifo_failed, err}}
end
end
defp fifo_guard(path) do
pid = self()
fn ->
2020-08-26 14:12:34 +00:00
ref = Process.monitor(pid)
2020-08-26 14:12:34 +00:00
receive do
{:DOWN, ^ref, :process, ^pid, _} ->
File.rm(path)
end
end
2020-08-21 17:02:57 +00:00
end
defp loop_recv(pid) do
loop_recv(pid, <<>>)
end
2020-08-21 17:02:57 +00:00
defp loop_recv(pid, acc) do
receive do
{^pid, {:data, data}} ->
2020-08-21 17:02:57 +00:00
loop_recv(pid, acc <> data)
2020-08-21 17:02:57 +00:00
{^pid, {:exit_status, 0}} ->
{:ok, acc}
2020-08-21 17:02:57 +00:00
{^pid, {:exit_status, status}} ->
{:error, status}
after
5000 ->
:erlang.port_close(pid)
{:error, :timeout}
end
end
end