git.squeep.com Git - akkoma/blob - lib/pleroma/helpers/media_helper.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2020 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Helpers.MediaHelper do
   6   @moduledoc """
   7   Handles common media-related operations.
   8   """
   9
  10   def image_resize(url, %{max_width: max_width, max_height: max_height} = options) do
  11     quality = options[:quality] || 85
  12
  13     cmd = ~s"""
  14     convert - -resize '#{max_width}x#{max_height}>' -quality #{quality} -
  15     """
  16
  17     pid = Port.open({:spawn, cmd}, [:use_stdio, :in, :stream, :exit_status, :binary])
  18     {:ok, env} = url |> Pleroma.Web.MediaProxy.url() |> Pleroma.HTTP.get()
  19     image = env.body
  20     Port.command(pid, image)
  21     loop_recv(pid)
  22   end
  23
  24   defp loop_recv(pid) do
  25     loop_recv(pid, <<>>)
  26   end
  27
  28   defp loop_recv(pid, acc) do
  29     receive do
  30       {^pid, {:data, data}} ->
  31         loop_recv(pid, acc <> data)
  32
  33       {^pid, {:exit_status, 0}} ->
  34         {:ok, acc}
  35
  36       {^pid, {:exit_status, status}} ->
  37         {:error, status}
  38     end
  39   end
  40 end