From aa0a5ffb4849880b5adbcc9188de01ef778381e3 Mon Sep 17 00:00:00 2001
From: Ivan Tashkinov <ivantashkinov@gmail.com>
Date: Fri, 21 Aug 2020 08:59:08 +0300
Subject: [PATCH] [#2497] Media preview proxy: added `quality` config setting,
 adjusted width/height defaults.

---
 config/config.exs                                     | 5 +++--
 config/description.exs                                | 5 +++++
 lib/pleroma/helpers/media_helper.ex                   | 6 ++++--
 lib/pleroma/web/media_proxy/media_proxy_controller.ex | 4 +++-
 4 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/config/config.exs b/config/config.exs
index 6e6231cf8..b399ce6d7 100644
--- a/config/config.exs
+++ b/config/config.exs
@@ -441,8 +441,9 @@
 # Note: media preview proxy depends on media proxy to be enabled
 config :pleroma, :media_preview_proxy,
   enabled: false,
-  thumbnail_max_width: 400,
-  thumbnail_max_height: 200,
+  thumbnail_max_width: 600,
+  thumbnail_max_height: 600,
+  quality: 2,
   proxy_opts: [
     head_request_max_read_duration: 5_000
   ]
diff --git a/config/description.exs b/config/description.exs
index 90d8eca65..22da60900 100644
--- a/config/description.exs
+++ b/config/description.exs
@@ -1925,6 +1925,11 @@
         type: :integer,
         description: "Max height of preview thumbnail."
       },
+      %{
+        key: :quality,
+        type: :integer,
+        description: "Quality of the output. Ranges from 1 (max quality) to 31 (lowest quality)."
+      },
       %{
         key: :proxy_opts,
         type: :keyword,
diff --git a/lib/pleroma/helpers/media_helper.ex b/lib/pleroma/helpers/media_helper.ex
index ca46698cc..e11038052 100644
--- a/lib/pleroma/helpers/media_helper.ex
+++ b/lib/pleroma/helpers/media_helper.ex
@@ -7,13 +7,15 @@ defmodule Pleroma.Helpers.MediaHelper do
   Handles common media-related operations.
   """
 
-  def ffmpeg_resize(uri_or_path, %{max_width: max_width, max_height: max_height}) do
+  def ffmpeg_resize(uri_or_path, %{max_width: max_width, max_height: max_height} = options) do
+    quality = options[:quality] || 1
+
     cmd = ~s"""
     ffmpeg -i #{uri_or_path} -f lavfi -i color=c=white \
       -filter_complex "[0:v] scale='min(#{max_width},iw)':'min(#{max_height},ih)': \
         force_original_aspect_ratio=decrease [scaled]; \
         [1][scaled] scale2ref [bg][img]; [bg] setsar=1 [bg]; [bg][img] overlay=shortest=1" \
-      -loglevel quiet -f image2 -vcodec mjpeg -frames:v 1 pipe:1
+      -loglevel quiet -f image2 -vcodec mjpeg -frames:v 1 -q:v #{quality} pipe:1
     """
 
     pid = Port.open({:spawn, cmd}, [:use_stdio, :in, :stream, :exit_status, :binary])
diff --git a/lib/pleroma/web/media_proxy/media_proxy_controller.ex b/lib/pleroma/web/media_proxy/media_proxy_controller.ex
index 5513432f0..1c51aa5e3 100644
--- a/lib/pleroma/web/media_proxy/media_proxy_controller.ex
+++ b/lib/pleroma/web/media_proxy/media_proxy_controller.ex
@@ -78,12 +78,14 @@ defp handle_preview(content_type, conn, _url) do
   end
 
   defp handle_image_or_video_preview(%{params: params} = conn, url) do
+    quality = Config.get!([:media_preview_proxy, :quality])
+
     with {thumbnail_max_width, thumbnail_max_height} <- thumbnail_max_dimensions(params),
          media_proxy_url <- MediaProxy.url(url),
          {:ok, thumbnail_binary} <-
            MediaHelper.ffmpeg_resize(
              media_proxy_url,
-             %{max_width: thumbnail_max_width, max_height: thumbnail_max_height}
+             %{max_width: thumbnail_max_width, max_height: thumbnail_max_height, quality: quality}
            ) do
       conn
       |> put_resp_header("content-type", "image/jpeg")