From 29be9e6ca1814f04f861d515b9024316300ba144 Mon Sep 17 00:00:00 2001
From: Stephen Paul Weber <singpolyma@singpolyma.net>
Date: Wed, 17 May 2023 14:53:44 -0500
Subject: [PATCH] Send thumbhash when sending media

---
 .../siacs/conversations/entities/Message.java | 16 +++++++++
 .../persistance/FileBackend.java              | 34 ++++++++++++++++++-
 2 files changed, 49 insertions(+), 1 deletion(-)

diff --git a/src/main/java/eu/siacs/conversations/entities/Message.java b/src/main/java/eu/siacs/conversations/entities/Message.java
index 9d122a247be19ffe01329ffa63348a57348e8e63..e50540c61c06c62a45612f57f7a9c542f93595af 100644
--- a/src/main/java/eu/siacs/conversations/entities/Message.java
+++ b/src/main/java/eu/siacs/conversations/entities/Message.java
@@ -1456,6 +1456,22 @@ public class Message extends AbstractEntity implements AvatarService.Avatarable
             return cids;
         }
 
+        public void addThumbnail(int width, int height, String mimeType, String uri) {
+            for (Element thumb : getThumbnails()) {
+                if (uri.equals(thumb.getAttribute("uri"))) return;
+            }
+
+            if (sims == null) toSims();
+            Element file = getFileElement();
+            file.addChild(
+                new Element("thumbnail", "urn:xmpp:thumbs:1")
+                    .setAttribute("width", Integer.toString(width))
+                    .setAttribute("height", Integer.toString(height))
+                    .setAttribute("type", mimeType)
+                    .setAttribute("uri", uri)
+            );
+        }
+
         public List<Element> getThumbnails() {
             List<Element> thumbs = new ArrayList<>();
             Element file = getFileElement();
diff --git a/src/main/java/eu/siacs/conversations/persistance/FileBackend.java b/src/main/java/eu/siacs/conversations/persistance/FileBackend.java
index f8d03e11fd4836fba80859bc4e1f3f35cc629bab..266553b8f08e57725c65712fa228a6f10ed2b700 100644
--- a/src/main/java/eu/siacs/conversations/persistance/FileBackend.java
+++ b/src/main/java/eu/siacs/conversations/persistance/FileBackend.java
@@ -1125,7 +1125,12 @@ public class FileBackend {
                             return cached;
                         }
                     } else if (parts[0].equals("image/thumbhash")) {
-                        ThumbHash.Image image = ThumbHash.thumbHashToRGBA(data);
+                        ThumbHash.Image image;
+                        try {
+                            image = ThumbHash.thumbHashToRGBA(data);
+                        } catch (final Exception e) {
+                            continue;
+                        }
                         int[] pixels = new int[image.width * image.height];
                         for (int i = 0; i < pixels.length; i++) {
                             pixels[i] = Color.argb(image.rgba[(i*4)+3] & 0xff, image.rgba[i*4] & 0xff, image.rgba[(i*4)+1] & 0xff, image.rgba[(i*4)+2] & 0xff);
@@ -1233,6 +1238,12 @@ public class FileBackend {
           return drawDrawable(drawable);
     }
 
+    public Bitmap getThumbnailBitmap(DownloadableFile file, Resources res, int size, String cacheKey) throws IOException {
+          final Drawable drawable = getThumbnail(file, res, size, false, cacheKey);
+          if (drawable == null) return null;
+          return drawDrawable(drawable);
+    }
+
     public static Rect rectForSize(int w, int h, int size) {
         int scalledW;
         int scalledH;
@@ -1871,6 +1882,27 @@ public class FileBackend {
             } else if (audio) {
                 fileParams.runtime = getMediaRuntime(file);
             }
+            try {
+                Bitmap thumb = getThumbnailBitmap(file, mXmppConnectionService.getResources(), 100, file.getAbsolutePath() + " x 100");
+                if (thumb != null) {
+                    int[] pixels = new int[thumb.getWidth() * thumb.getHeight()];
+                    byte[] rgba = new byte[pixels.length * 4];
+                    try {
+                        thumb.getPixels(pixels, 0, thumb.getWidth(), 0, 0, thumb.getWidth(), thumb.getHeight());
+                    } catch (final IllegalStateException e) {
+                        Bitmap softThumb = thumb.copy(Bitmap.Config.ARGB_8888, false);
+                        softThumb.getPixels(pixels, 0, thumb.getWidth(), 0, 0, thumb.getWidth(), thumb.getHeight());
+                        softThumb.recycle();
+                    }
+                    for (int i = 0; i < pixels.length; i++) {
+                        rgba[i*4] = (byte)((pixels[i] >> 16) & 0xff);
+                        rgba[(i*4)+1] = (byte)((pixels[i] >> 8) & 0xff);
+                        rgba[(i*4)+2] = (byte)(pixels[i] & 0xff);
+                        rgba[(i*4)+3] = (byte)((pixels[i] >> 24) & 0xff);
+                    }
+                    fileParams.addThumbnail(thumb.getWidth(), thumb.getHeight(), "image/thumbhash", "data:image/thumbhash;base64," + Base64.encodeToString(ThumbHash.rgbaToThumbHash(thumb.getWidth(), thumb.getHeight(), rgba), Base64.NO_WRAP));
+                }
+            } catch (final IOException e) { }
         }
         message.setFileParams(fileParams);
         message.setDeleted(false);