1 files changed, 53 insertions, 2 deletions

diff --git a/poezio/windows/image.py b/poezio/windows/image.py
index 309fe0e6..96636ec7 100644
--- a/poezio/windows/image.py
+++ b/poezio/windows/image.py
@@ -9,8 +9,20 @@ try:
     from PIL import Image
     HAS_PIL = True
 except ImportError:
+    class Image:
+        class Image:
+            pass
     HAS_PIL = False
 
+try:
+    import gi
+    gi.require_version('Rsvg', '2.0')
+    from gi.repository import Rsvg
+    import cairo
+    HAS_RSVG = True
+except (ImportError, ValueError):
+    HAS_RSVG = False
+
 from poezio.windows.base_wins import Win
 from poezio.theming import get_theme, to_curses_attr
 from poezio.xhtml import _parse_css_color
@@ -19,13 +31,45 @@ from poezio.config import config
 from typing import Tuple, Optional, Callable
 
 
+MAX_SIZE = 16
+
+
+def render_svg(svg: bytes) -> Optional[Image.Image]:
+    if not HAS_RSVG:
+        return None
+    try:
+        handle = Rsvg.Handle.new_from_data(svg)
+        dimensions = handle.get_dimensions()
+        biggest_dimension = max(dimensions.width, dimensions.height)
+        scale = MAX_SIZE / biggest_dimension
+        translate_x = (biggest_dimension - dimensions.width) / 2
+        translate_y = (biggest_dimension - dimensions.height) / 2
+
+        surface = cairo.ImageSurface(cairo.Format.ARGB32, MAX_SIZE, MAX_SIZE)
+        context = cairo.Context(surface)
+        context.scale(scale, scale)
+        context.translate(translate_x, translate_y)
+        handle.render_cairo(context)
+        data = surface.get_data()
+        image = Image.frombytes('RGBA', (MAX_SIZE, MAX_SIZE), data.tobytes())
+        # This is required because Cairo uses a BGRA (in host endianess)
+        # format, and PIL an ABGR (in byte order) format.  Yes, this is
+        # confusing.
+        b, g, r, a = image.split()
+        return Image.merge('RGB', (r, g, b))
+    except Exception:
+        return None
+
+
 class ImageWin(Win):
     """
     A window which contains either an image or a border.
     """
 
+    __slots__ = ('_image', '_display_avatar')
+
     def __init__(self) -> None:
-        self._image = None  # type: Optional[Image]
+        self._image = None  # type: Optional[Image.Image]
         Win.__init__(self)
         if config.get('image_use_half_blocks'):
             self._display_avatar = self._display_avatar_half_blocks  # type: Callable[[int, int], None]
@@ -43,7 +87,14 @@ class ImageWin(Win):
         if data is not None and HAS_PIL:
             image_file = BytesIO(data)
             try:
-                image = Image.open(image_file)
+                try:
+                    image = Image.open(image_file)
+                except OSError:
+                    # TODO: Make the caller pass the MIME type, so we don’t
+                    # have to try all renderers like that.
+                    image = render_svg(data)
+                    if image is None:
+                        raise
             except OSError:
                 self._display_border()
             else: