Add files via upload

src/core/plugins/favouriteanything/Settings.tsx

@@ -0,0 +1,13 @@
+import { Forms } from "@vendetta/ui/components";
+const { FormSection, FormText } = Forms;
+
+export default () => (
+    <FormSection title="FavouriteAnything">
+        <FormText>
+            Adds the favourite button to ALL media in the image viewer, not just GIFs!
+        </FormText>
+        <FormText style={{ marginTop: 8, color: "#b9bbbe" }}>
+            Images and videos can now be saved to your GIF favourites picker. Works by patching Discord's GIFFavButton to accept all media types.
+        </FormText>
+    </FormSection>
+)

src/core/plugins/favouriteanything/index.ts

@@ -0,0 +1,158 @@
+// FavouriteAnything - by @TheUnrealZaka
+// Ported from the Equicord desktop plugin (nin0dev & davri)
+// Patches GIFFavButton so the star shows on all media, not just GIFs.
+// Also fixes the format field when saving (images=1, videos=2)
+// and converts video thumbnails to jpeg for the mobile favourites picker.
+
+import { logger } from "@vendetta";
+import { findByProps } from "@vendetta/metro";
+import { before } from "@vendetta/patcher";
+
+let origType: Function | null = null;
+let memoWrapper: any = null;
+let retryTimeout: ReturnType<typeof setTimeout> | null = null;
+let unpatchAddFavorite: (() => void) | null = null;
+let origUseFavoriteGIFsMobile: Function | null = null;
+let favMobileModule: any = null;
+
+const VIDEO_EXT = [".mp4", ".webm", ".mov", ".avi", ".mkv", ".flv", ".wmv", ".m4v", ".gifv"];
+const processed = new WeakSet<object>();
+
+function isVideo(url: string): boolean {
+    if (!url) return false;
+    const path = url.split("?")[0].toLowerCase();
+    return VIDEO_EXT.some(e => path.endsWith(e));
+}
+
+// cdn.discordapp.com doesn't support ?format=, media.discordapp.net does
+function makeVideoThumbnail(url: string): string {
+    if (!url) return url;
+    let out = url.replace("cdn.discordapp.com", "media.discordapp.net");
+    if (out.includes("media.discordapp.net") || out.includes("images-ext")) {
+        return out + (out.includes("?") ? "&" : "?") + "format=jpeg";
+    }
+    return url;
+}
+
+function findGIFFavButton(): any {
+    const modules = (window as any).vendetta?.metro?.modules ?? (window as any).modules;
+    if (!modules) return null;
+
+    for (const id in modules) {
+        try {
+            const mod = modules[id]?.publicModule?.exports;
+            if (!mod) continue;
+            const def = mod.default;
+            if (def?.$$typeof?.toString().includes("memo") && def.type) {
+                if (def.type.displayName === "GIFFavButton" || def.type.name === "GIFFavButton") return def;
+            }
+            if (typeof mod === "function" && (mod.displayName === "GIFFavButton" || mod.name === "GIFFavButton")) return mod;
+        } catch {}
+    }
+    return null;
+}
+
+function patchSource(source: any): any {
+    if (source.isGIFV) return source;
+    return {
+        ...source,
+        isGIFV: true,
+        embedURI: source.embedURI || source.sourceURI || source.uri,
+        videoURI: source.videoURI || source.uri,
+        embedProviderName: source.embedProviderName || "",
+    };
+}
+
+function applyPatch() {
+    memoWrapper = findGIFFavButton();
+    if (!memoWrapper) return false;
+
+    origType = memoWrapper.type;
+    memoWrapper.type = function PatchedGIFFavButton(props: any) {
+        const p = { ...props };
+        if (p.source && !p.source.isGIFV) p.source = patchSource(p.source);
+        return (origType as Function)(p);
+    };
+    (memoWrapper.type as any).displayName = "GIFFavButton";
+    return true;
+}
+
+function patchAddFavorite() {
+    const favModule = findByProps("addFavoriteGIF");
+    if (!favModule) return;
+
+    unpatchAddFavorite = before("addFavoriteGIF", favModule, (args: any[]) => {
+        const data = args[0];
+        if (!data || typeof data !== "object") return;
+        const url = data.url || "";
+        const src = data.src || "";
+
+        if (isVideo(url) || isVideo(src)) {
+            data.format = 2;
+        } else if (data.format === 2) {
+            data.format = 1;
+        }
+    });
+}
+
+function patchMobileFavorites() {
+    const mod = findByProps("useFavoriteGIFsMobile");
+    if (!mod) return;
+
+    favMobileModule = mod;
+    origUseFavoriteGIFsMobile = mod.useFavoriteGIFsMobile;
+
+    let lastFavs: any = null;
+    mod.useFavoriteGIFsMobile = function (...args: any[]) {
+        const result = (origUseFavoriteGIFsMobile as Function).apply(this, args);
+        if (result?.favorites && Array.isArray(result.favorites) && result.favorites !== lastFavs) {
+            lastFavs = result.favorites;
+            for (const item of result.favorites) {
+                if (!item || processed.has(item)) continue;
+                processed.add(item);
+                if (isVideo(item.url) || isVideo(item.src)) {
+                    item.src = makeVideoThumbnail(item.src || item.url);
+                }
+            }
+        }
+        return result;
+    };
+}
+
+export default {
+    onLoad: () => {
+        if (applyPatch()) {
+            logger.log("[FavouriteAnything] Patched GIFFavButton.");
+        } else {
+            let retries = 0;
+            const tryPatch = () => {
+                if (applyPatch()) {
+                    logger.log("[FavouriteAnything] Patched GIFFavButton (after retry).");
+                    retryTimeout = null;
+                } else if (retries++ < 50) {
+                    retryTimeout = setTimeout(tryPatch, 300);
+                } else {
+                    logger.error("[FavouriteAnything] GIFFavButton not found after retries.");
+                }
+            };
+            retryTimeout = setTimeout(tryPatch, 300);
+        }
+
+        patchAddFavorite();
+        patchMobileFavorites();
+    },
+    onUnload: () => {
+        if (retryTimeout) { clearTimeout(retryTimeout); retryTimeout = null; }
+        if (memoWrapper && origType) {
+            memoWrapper.type = origType;
+            origType = null;
+            memoWrapper = null;
+        }
+        if (unpatchAddFavorite) { unpatchAddFavorite(); unpatchAddFavorite = null; }
+        if (favMobileModule && origUseFavoriteGIFsMobile) {
+            favMobileModule.useFavoriteGIFsMobile = origUseFavoriteGIFsMobile;
+            origUseFavoriteGIFsMobile = null;
+            favMobileModule = null;
+        }
+    },
+};