api/internal: also transplant youtube HEAD requests

imputnet · Mar 6, 2025 · 068ae2f · 068ae2f
1 parent 187b1f8
commit 068ae2f
Showing 1 changed file with 18 additions and 7 deletions.
diff --git a/api/src/stream/internal.js b/api/src/stream/internal.js
@@ -53,14 +53,25 @@ async function handleYoutubeStream(streamInfo, res) {
     const cleanup = () => (res.end(), closeRequest(streamInfo.controller));
 
     try {
-        const req = await fetch(streamInfo.url, {
-            headers: getHeaders('youtube'),
-            method: 'HEAD',
-            dispatcher: streamInfo.dispatcher,
-            signal
-        });
+        let req, attempts = 3;
+        while (attempts--) {
+            req = await fetch(streamInfo.url, {
+                headers: getHeaders('youtube'),
+                method: 'HEAD',
+                dispatcher: streamInfo.dispatcher,
+                signal
+            });
+
+            streamInfo.url = req.url;
+            if (req.status === 403 && streamInfo.transplant) {
+                try {
+                    await streamInfo.transplant(streamInfo.dispatcher);
+                } catch {
+                    break;
+                }
+            } else break;
+        }
 
-        streamInfo.url = req.url;
         const size = BigInt(req.headers.get('content-length'));
 
         if (req.status !== 200 || !size) {