From 971f619e181e32112dbbaa0c59780b1a49122589 Mon Sep 17 00:00:00 2001
From: "Mr. Stallion" <mrstallion@nobody.nowhere.fauxdomain.ext>
Date: Sun, 27 Oct 2019 14:31:45 -0500
Subject: [PATCH] Fixed Imgur gallery count

---
 chat/ImagePreview.vue         | 15 ++++++++-------
 chat/image-preview-mutator.ts | 26 +++++++++++++++++++++-----
 2 files changed, 29 insertions(+), 12 deletions(-)

diff --git a/chat/ImagePreview.vue b/chat/ImagePreview.vue
index 85efc21..3311412 100644
--- a/chat/ImagePreview.vue
+++ b/chat/ImagePreview.vue
@@ -318,17 +318,18 @@
                 (event: Event) => {
                     if (this.debug)
                         console.log('ImagePreview did-finish-load', event);
-
-                    webview.getWebContents().session.on(
-                        'will-download',
-                        (e: Event) => {
-                            e.preventDefault();
-                        }
-                    );
                 }
             );
 
 
+            /* webview.getWebContents().session.on(
+                'will-download',
+                (e: Event) => {
+                    e.preventDefault();
+                }
+            );*/
+
+
             _.each(
                 ['did-start-loading', 'load-commit', 'dom-ready', 'will-navigate', 'did-navigate', 'did-navigate-in-page', 'update-target-url'],
                 (en: string) => {
diff --git a/chat/image-preview-mutator.ts b/chat/image-preview-mutator.ts
index f4df7d9..7034dae 100644
--- a/chat/image-preview-mutator.ts
+++ b/chat/image-preview-mutator.ts
@@ -123,7 +123,6 @@ export class ImagePreviewMutator {
         this.add('pornhub.com', this.getBaseJsMutatorScript(['.mainPlayerDiv video', '.photoImageSection img']));
         this.add('sex.com', this.getBaseJsMutatorScript(['.image_frame video', '.image_frame img']));
         this.add('redirect.media.tumblr.com', this.getBaseJsMutatorScript(['picture video', 'picture img']));
-        this.add('i.imgur.com', this.getBaseJsMutatorScript(['video', 'img']));
         this.add('postimg.cc', this.getBaseJsMutatorScript(['video', '#main-image']));
         this.add('gifsauce.com', this.getBaseJsMutatorScript(['video']));
         this.add('motherless.com', this.getBaseJsMutatorScript(['.content video', '.content img']));
@@ -132,6 +131,23 @@ export class ImagePreviewMutator {
         this.add(/^media[0-9]\.tenor\.com$/, this.getBaseJsMutatorScript(['#view .file video', '#view .file img']));
         this.add('tenor.com', this.getBaseJsMutatorScript(['#view video', '#view img']));
 
+        // tslint:disable max-line-length
+        this.add(
+            'i.imgur.com',
+            `
+                const imageCount = (new URL(window.location.href)).searchParams.get('flist_gallery_image_count');
+
+                ${this.getBaseJsMutatorScript(['video', 'img'])}
+
+                if(imageCount > 1) {
+                    ${this.injectHtmlJs('<div id="imageCount" style="z-index: 1000000; position: absolute; bottom: 0; right: 0; background: green; border: 2px solid lightgreen; width: 5rem; height: 5rem; font-size: 2rem; font-weight: bold; color: white; border-radius: 5rem; margin: 0.75rem;"><div id="imageCountInner" style="position: absolute; top: 50%; left: 50%; transform: translateY(-50%) translateX(-50%);"></div></div>')}
+
+                    const imageCountEl = document.getElementById('imageCountInner');
+                    imageCountEl.innerHTML = '+' + (imageCount - 1);
+                }
+            `
+        );
+
         // tslint:disable max-line-length
         this.add(
             'imgur.com',
@@ -189,11 +205,11 @@ export class ImagePreviewMutator {
                 selected = selected.concat(selectedElements);
             }
 
-            ${true /*this.debug*/ ? `console.log('Selector', '${elSelector}'); console.log('Selected', selected);` : ''}
+            ${this.debug ? `console.log('Selector', '${elSelector}'); console.log('Selected', selected);` : ''}
 
             const img = selected.shift();
 
-            ${true /*this.debug*/ ? `console.log('Img', img);` : ''}
+            ${this.debug ? `console.log('Img', img);` : ''}
 
             if (!img) { return; }
 
@@ -246,7 +262,7 @@ export class ImagePreviewMutator {
             ${this.debug ? "console.log('Wrapper', el);" : ''}
 
             document.addEventListener('DOMContentLoaded', (event) => {
-                ${true /*this.debug*/ ? "console.log('on DOMContentLoaded');" : ''}
+                ${this.debug ? "console.log('on DOMContentLoaded');" : ''}
 
                 if (
                     (img.play)
@@ -256,7 +272,7 @@ export class ImagePreviewMutator {
             });
 
             document.addEventListener('load', (event) => {
-                ${true /*this.debug*/ ? "console.log('on load');" : ''}
+                ${this.debug ? "console.log('on load');" : ''}
 
                 if (
                     (img.play)