From e70056f2ffeea39684c8bad4ece20128d1008da4 Mon Sep 17 00:00:00 2001
From: nexy7574 <git@nexy7574.co.uk>
Date: Mon, 15 Apr 2024 22:05:53 +0100
Subject: [PATCH] Reduce the number of files that have an attempted transcode

---
 src/cogs/auto_responder.py | 39 +++++++++++++++++++++++---------------
 1 file changed, 24 insertions(+), 15 deletions(-)

diff --git a/src/cogs/auto_responder.py b/src/cogs/auto_responder.py
index d43d3f3..0026126 100644
--- a/src/cogs/auto_responder.py
+++ b/src/cogs/auto_responder.py
@@ -107,21 +107,30 @@ class AutoResponder(commands.Cog):
         links = self.extract_links(message.content, "static-assets-1.truthsocial.com")
         if links:
             for link in links:
-                self.log.info("Found link to transcode: %r", link)
-                try:
-                    file, _p = await self._transcode_hevc_to_h264(link)
-                    if file:
-                        if _p.stat().st_size <= 24.5 * 1024 * 1024:
-                            await message.reply(file=file)
-                        else:
-                            self.log.warning(
-                                "Transcoded file too large: %r (%.2f)MB",
-                                _p,
-                                _p.stat().st_size / 1024 / 1024
-                            )
-                        _p.unlink()
-                except Exception as e:
-                    self.log.error("Failed to transcode %r: %r", link, e)
+                if link.lower().endswith(
+                        (".mp4", ".mov", ".qtff", ".mkv", ".asf", ".avi", ".mxf", ".ps", ".ts", ".3gp", ".3g2")
+                ):
+                    # All containers allowed to contain HEVC
+                    # per https://en.wikipedia.org/wiki/Comparison_of_video_container_formats
+                    self.log.info("Found link to transcode: %r", link)
+                    try:
+                        async with message.channel.typing():
+                            _r = await self._transcode_hevc_to_h264(link)
+                            if not _r:
+                                continue
+                            file, _p = _r
+                            if file:
+                                if _p.stat().st_size <= 24.5 * 1024 * 1024:
+                                    await message.reply(file=file)
+                                else:
+                                    self.log.warning(
+                                        "Transcoded file too large: %r (%.2f)MB",
+                                        _p,
+                                        _p.stat().st_size / 1024 / 1024
+                                    )
+                                _p.unlink()
+                    except Exception as e:
+                        self.log.error("Failed to transcode %r: %r", link, e)
 
 
 def setup(bot):