From 08cd37b814d7303e1264fc6f8a47b66dd49c6b89 Mon Sep 17 00:00:00 2001
From: Comrade DOS <dos@hidevlab.com>
Date: Fri, 13 Mar 2020 14:09:34 +0600
Subject: [PATCH] [preview] Adding detection of supported mime types at runtime

---
 url_image_preview/url_image_preview.py | 25 +++++++++++++++++--------
 1 file changed, 17 insertions(+), 8 deletions(-)

diff --git a/url_image_preview/url_image_preview.py b/url_image_preview/url_image_preview.py
index 6e66fe7..d668aa7 100644
--- a/url_image_preview/url_image_preview.py
+++ b/url_image_preview/url_image_preview.py
@@ -26,6 +26,7 @@ from gi.repository import Gtk
 from gi.repository import Gdk
 from gi.repository import GLib
 from gi.repository import Soup
+from gi.repository import GdkPixbuf
 
 from gajim.common import app
 from gajim.common import configpaths
@@ -52,6 +53,7 @@ ERROR_MSG = None
 try:
     from PIL import Image  # pylint: disable=unused-import
 except ImportError:
+    Image = None
     log.error('Pillow not available')
     ERROR_MSG = _('Please install python-pillow')
 
@@ -73,14 +75,21 @@ if ERROR_MSG is None:
     from url_image_preview.utils import filename_from_uri
 # pylint: enable=ungrouped-imports
 
-ACCEPTED_MIME_TYPES = [
-    'image/png',
-    'image/jpeg',
-    'image/gif',
-    'image/raw',
-    'image/svg+xml',
-    'image/x-ms-bmp',
-]
+def get_accepted_mime_types():
+    accepted_mime_types = set()
+    for fmt in GdkPixbuf.Pixbuf.get_formats():
+        for mime_type in fmt.get_mime_types():
+            accepted_mime_types.add(mime_type.lower())
+    if Image is not None:
+        Image.init()
+        for mime_type in Image.MIME.values():
+            accepted_mime_types.add(mime_type.lower())
+    return tuple(filter(
+        lambda mime_type: mime_type.startswith('image'),
+        accepted_mime_types
+    ))
+
+ACCEPTED_MIME_TYPES = get_accepted_mime_types()
 
 
 class UrlImagePreviewPlugin(GajimPlugin):