From 1dd490370095bb63e870063af5ba6c36d7f01853 Mon Sep 17 00:00:00 2001
From: Snazzah <me@snazzah.com>
Date: Tue, 8 Oct 2024 15:23:29 -0500
Subject: [PATCH] support alwaysProxy in threads

---
 api/src/processing/match.js            |  1 +
 api/src/processing/services/threads.js | 59 ++++++++++++++++----------
 2 files changed, 38 insertions(+), 22 deletions(-)

diff --git a/api/src/processing/match.js b/api/src/processing/match.js
index e5cc5523..63c2df98 100644
--- a/api/src/processing/match.js
+++ b/api/src/processing/match.js
@@ -235,6 +235,7 @@ export default async function({ host, patternMatch, params }) {
                 r = await threads({
                     ...patternMatch,
                     quality: params.videoQuality,
+                    alwaysProxy: params.alwaysProxy,
                     dispatcher
                 });
                 break;
diff --git a/api/src/processing/services/threads.js b/api/src/processing/services/threads.js
index 279de16c..a7682e62 100644
--- a/api/src/processing/services/threads.js
+++ b/api/src/processing/services/threads.js
@@ -22,8 +22,8 @@ const commonHeaders = {
 
 const DATA_REGEX = /<script type="application\/json" {2}data-content-len="\d+" data-sjs>({"require":\[\["ScheduledServerJS","handle",null,\[{"__bbox":{"require":\[\["RelayPrefetchedStreamCache(?:(?:@|\\u0040)[0-9a-f]{32})?","next",\[],\["adp_BarcelonaPostPageQueryRelayPreloader_[0-9a-f]{23}",[^\n]+})<\/script>\n/;
 
-export default async function({ user, id, quality, dispatcher }) {
-    const cookie = getCookie('threads');
+export default async function({ user, id, quality, alwaysProxy, dispatcher }) {
+    const cookie = getCookie("threads");
     const response = await fetch(`https://www.threads.net/${user}/post/${id}`, {
         headers: {
             ...commonHeaders,
@@ -34,44 +34,46 @@ export default async function({ user, id, quality, dispatcher }) {
     if (cookie) updateCookie(cookie, response.headers);
 
     if (response.status !== 200) {
-        return { error: 'fetch.fail' };
+        return { error: "fetch.fail" };
     }
     const html = await response.text();
     const dataString = html.match(DATA_REGEX)?.[1];
     if (!dataString) {
-        return { error: 'fetch.fail' };
+        return { error: "fetch.fail" };
     }
 
     const data = JSON.parse(dataString);
     const post = data?.require?.[0]?.[3]?.[0]?.__bbox?.require?.[0]?.[3]?.[1]?.__bbox?.result?.data?.data?.edges[0]?.node?.thread_items[0]?.post;
     if (!post) {
-        return { error: 'fetch.fail' };
+        return { error: "fetch.fail" };
     }
 
+    const filenameBase = `threads_${post.user.username}_${post.code}`;
+
     // Video
     if (post.media_type === 2) {
         if (!post.video_versions) {
-            return { error: 'fetch.empty' };
+            return { error: "fetch.empty" };
         }
 
         // types: 640p = 101, 480p = 102, 480p-low = 103
-        const selectedQualityType = quality === 'max' ? 101 : quality && parseInt(quality) <= 480 ? 102 : 101;
+        const selectedQualityType = quality === "max" ? 101 : quality && parseInt(quality) <= 480 ? 102 : 101;
         const video = post.video_versions.find((v) => v.type === selectedQualityType) || post.video_versions.sort((a, b) => a.type - b.type)[0];
         if (!video) {
-            return { error: 'fetch.empty' };
+            return { error: "fetch.empty" };
         }
 
         return {
             urls: video.url,
-            filename: `threads_${user}_${id}.mp4`,
-            audioFilename: `threads_${user}_${id}_audio`
+            filename: `${filenameBase}.mp4`,
+            audioFilename: `${filenameBase}_audio`
         }
     }
 
     // Photo
     if (post.media_type === 1) {
         if (!post.image_versions2?.candidates) {
-            return { error: 'fetch.empty' };
+            return { error: "fetch.empty" };
         }
 
         return {
@@ -83,23 +85,36 @@ export default async function({ user, id, quality, dispatcher }) {
     // Mixed
     if (post.media_type === 8) {
         if (!post.carousel_media) {
-            return { error: 'fetch.empty' };
+            return { error: "fetch.empty" };
         }
 
         return {
-            picker: post.carousel_media.map((media) => ({
-                type: media.video_versions ? 'video' : 'photo',
-                url: media.video_versions ? media.video_versions[0].url : media.image_versions2.candidates[0].url,
-                /* thumbnails have `Cross-Origin-Resource-Policy`
-                ** set to `same-origin`, so we need to proxy them */
-                thumb: createStream({
+            picker: post.carousel_media.map((media, i) => {
+                const type = media.video_versions ? "video" : "photo";
+                let url = media.video_versions ? media.video_versions[0].url : media.image_versions2.candidates[0].url;
+                const thumbProxy = createStream({
                     service: "threads",
                     type: "proxy",
-                    u: media.image_versions2.candidates[0].url
-                })
-            }))
+                    u: media.image_versions2.candidates[0].url,
+                    filename: `${filenameBase}_${i}.jpg`,
+                });
+                if (alwaysProxy) {
+                    url = type === 'photo' ? thumbProxy : createStream({
+                        service: "threads",
+                        type: "proxy",
+                        u: media.video_versions[0].url,
+                        filename: `${filenameBase}_${i}.mp4`,
+                    });
+                }
+
+                return {
+                    type,
+                    url,
+                    thumb: thumbProxy
+                }
+            })
         }
     }
 
-    return { error: 'fetch.fail' };
+    return { error: "fetch.fail" };
 }