git.squeep.com Git - akkoma/blob - lib/pleroma/upload/filter/analyze_metadata.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Pleroma.Upload.Filter.AnalyzeMetadata do
   6   @moduledoc """
   7   Extracts metadata about the upload, such as width/height
   8   """
   9   require Logger
  10
  11   @behaviour Pleroma.Upload.Filter
  12
  13   @spec filter(Pleroma.Upload.t()) ::
  14           {:ok, :filtered, Pleroma.Upload.t()} | {:ok, :noop} | {:error, String.t()}
  15   def filter(%Pleroma.Upload{tempfile: file, content_type: "image" <> _} = upload) do
  16     try do
  17       image =
  18         file
  19         |> Mogrify.open()
  20         |> Mogrify.verbose()
  21
  22       upload =
  23         upload
  24         |> Map.put(:width, image.width)
  25         |> Map.put(:height, image.height)
  26         |> Map.put(:blurhash, get_blurhash(file))
  27
  28       {:ok, :filtered, upload}
  29     rescue
  30       e in ErlangError ->
  31         Logger.warn("#{__MODULE__}: #{inspect(e)}")
  32         {:ok, :noop}
  33     end
  34   end
  35
  36   def filter(_), do: {:ok, :noop}
  37
  38   defp get_blurhash(file) do
  39     with {:ok, blurhash} <- :eblurhash.magick(file) do
  40       blurhash
  41     else
  42       _ -> nil
  43     end
  44   end
  45 end