Guess mime type when reported wrongly

RikudouSage · Feb 19, 2024 · fcdbeed · fcdbeed
1 parent 3c5f387
commit fcdbeed
Show file tree

Hide file tree

Showing 3 changed files with 178 additions and 2 deletions.
diff --git a/composer.json b/composer.json
@@ -30,6 +30,7 @@
         "symfony/framework-bundle": "7.0.*",
         "symfony/http-client": "7.0.*",
         "symfony/messenger": "7.0.*",
+        "symfony/mime": "7.0.*",
         "symfony/property-access": "7.0.*",
         "symfony/property-info": "7.0.*",
         "symfony/psr-http-message-bridge": "7.0.*",

diff --git a/composer.lock b/composer.lock
diff --git a/src/Service/ImageFetcher.php b/src/Service/ImageFetcher.php
@@ -7,6 +7,7 @@
 use SapientPro\ImageComparator\ImageComparator;
 use Symfony\Component\ErrorHandler\Error\FatalError;
 use Symfony\Component\HttpFoundation\Request;
+use Symfony\Component\Mime\MimeTypeGuesserInterface;
 use Symfony\Contracts\Cache\ItemInterface;
 use Symfony\Contracts\HttpClient\HttpClientInterface;
 use Throwable;
@@ -17,6 +18,7 @@ public function __construct(
         private CacheItemPoolInterface $cache,
         private HttpClientInterface $httpClient,
         private ImageComparator $imageComparator,
+        private MimeTypeGuesserInterface $typeGuesser,
     ) {
     }
 
@@ -33,7 +35,13 @@ public function getImageHash(string $url): ?string
             $cacheItem->set(null);
         } else {
             $contentType = $response->getHeaders(false)['content-type'][0] ?? '';
-            if (str_starts_with($contentType, 'image/')) {
+            $isImage = str_starts_with($contentType, 'image/');
+            if (!$isImage) {
+                $filepath = tempnam(sys_get_temp_dir(), 'lemmy_automod');
+                file_put_contents($filepath, $response->getContent());
+                $isImage = str_starts_with($this->typeGuesser->guessMimeType($filepath), 'image/');
+            }
+            if ($isImage) {
                 try {
                     if ($this->isAnimatedWebP($response->getContent())) {
                         $image = false;