From ff45b072450a540ff1fcf1b655d7492819c2da54 Mon Sep 17 00:00:00 2001
From: Eduard Kuzmenko <thanwords24@gmail.com>
Date: Tue, 1 Feb 2022 23:54:00 +0400
Subject: [PATCH] Save thumb for .webm stickers

---
 src/components/wrappers.ts            | 20 ++++++++++++++++----
 src/lib/appManagers/appDocsManager.ts |  5 +++++
 2 files changed, 21 insertions(+), 4 deletions(-)

diff --git a/src/components/wrappers.ts b/src/components/wrappers.ts
index c7c96910..203a87b4 100644
--- a/src/components/wrappers.ts
+++ b/src/components/wrappers.ts
@@ -1203,7 +1203,7 @@ export function wrapSticker({doc, div, middleware, lazyLoadQueue, group, play, o
   
   let loadThumbPromise = deferredPromise<void>();
   let haveThumbCached = false;
-  if((doc.thumbs?.length || doc.stickerCachedThumbs) && !div.firstElementChild && (!downloaded || stickerType === 2 || onlyThumb) && withThumb !== false/*  && doc.thumbs[0]._ !== 'photoSizeEmpty' */) {
+  if((doc.thumbs?.length || doc.stickerCachedThumbs) && !div.firstElementChild && (!downloaded || stickerType === 2 || stickerType === 3 || onlyThumb) && withThumb !== false/*  && doc.thumbs[0]._ !== 'photoSizeEmpty' */) {
     let thumb = doc.stickerCachedThumbs && doc.stickerCachedThumbs[toneIndex] || doc.thumbs[0];
     
     //console.log('wrap sticker', thumb, div);
@@ -1565,7 +1565,7 @@ export function wrapSticker({doc, div, middleware, lazyLoadQueue, group, play, o
       }
 
       const thumbImage = div.firstElementChild !== media && div.firstElementChild;
-      needFadeIn = (needFadeIn || !downloaded || thumbImage) && rootScope.settings.animationsEnabled;
+      needFadeIn = (needFadeIn || !downloaded || (stickerType === 1 ? thumbImage : (!thumbImage || thumbImage.tagName === 'svg'))) && rootScope.settings.animationsEnabled;
 
       media.classList.add('media-sticker');
 
@@ -1584,6 +1584,18 @@ export function wrapSticker({doc, div, middleware, lazyLoadQueue, group, play, o
                 thumbImage.classList.add('fade-out');
               }
 
+              if(stickerType === 3 && !appDocsManager.isSavingLottiePreview(doc, toneIndex)) {
+                // const perf = performance.now();
+                assumeType<HTMLVideoElement>(media);
+                const canvas = document.createElement('canvas');
+                canvas.width = width * window.devicePixelRatio;
+                canvas.height = height * window.devicePixelRatio;
+                const ctx = canvas.getContext('2d');
+                ctx.drawImage(media, 0, 0, canvas.width, canvas.height);
+                appDocsManager.saveLottiePreview(doc, canvas, toneIndex);
+                // console.log('perf', performance.now() - perf);
+              }
+
               if(stickerType === 3 && group) {
                 animationIntersector.addAnimation(media as HTMLVideoElement, group);
               }
@@ -1617,11 +1629,11 @@ export function wrapSticker({doc, div, middleware, lazyLoadQueue, group, play, o
     }
   };
 
-  const loadPromise: Promise<RLottiePlayer | void> = lazyLoadQueue && (!downloaded || stickerType === 2) ? 
+  const loadPromise: Promise<RLottiePlayer | void> = lazyLoadQueue && (!downloaded || stickerType === 2 || stickerType === 3) ? 
     (lazyLoadQueue.push({div, load}), Promise.resolve()) : 
     load();
 
-  if(downloaded && (stickerType === 1 || stickerType === 3)) {
+  if(downloaded && (stickerType === 1/*  || stickerType === 3 */)) {
     loadThumbPromise = loadPromise as any;
     if(loadPromises) {
       loadPromises.push(loadThumbPromise);
diff --git a/src/lib/appManagers/appDocsManager.ts b/src/lib/appManagers/appDocsManager.ts
index 3a5b15d6..198cbfe8 100644
--- a/src/lib/appManagers/appDocsManager.ts
+++ b/src/lib/appManagers/appDocsManager.ts
@@ -391,6 +391,11 @@ export class AppDocsManager {
     return download;
   }
 
+  public isSavingLottiePreview(doc: MyDocument, toneIndex: number) {
+    const key = doc.id + '-' + toneIndex;
+    return !!this.savingLottiePreview[key];
+  }
+
   public saveLottiePreview(doc: MyDocument, canvas: HTMLCanvasElement, toneIndex: number) {
     const key = doc.id + '-' + toneIndex;
     if(this.savingLottiePreview[key]/*  || true */) return;