Add support for poToken and visitor data

Fixes api requests iv-org/invidious#4789
ajayyy · Aug 12, 2024 · 090e185 · 090e185
1 parent d2df5ce
commit 090e185
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 4 deletions.
diff --git a/src/config.ts b/src/config.ts
@@ -195,7 +195,12 @@ addDefaults(config, {
     useCacheForSegmentGroups: false,
     maxConnections: 100,
     maxResponseTime: 1000,
-    maxResponseTimeWhileLoadingCache: 2000
+    maxResponseTimeWhileLoadingCache: 2000,
+    etagExpiry: 5000,
+    youTubeKeys: {
+        visitorData: null,
+        poToken: null
+    }
 });
 loadFromEnv(config);
 migrate(config);

diff --git a/src/types/config.model.ts b/src/types/config.model.ts
@@ -109,6 +109,11 @@ export interface SBSConfig {
     maxConnections: number;
     maxResponseTime: number;
     maxResponseTimeWhileLoadingCache: number;
+    etagExpiry: number;
+    youTubeKeys: {
+        visitorData: string | null;
+        poToken: string | null;
+    }
 }
 
 export interface WebhookConfig {

diff --git a/src/utils/innerTubeAPI.ts b/src/utils/innerTubeAPI.ts
@@ -2,6 +2,7 @@ import axios from "axios";
 import { Logger } from "./logger";
 import { innerTubeVideoDetails } from "../types/innerTubeApi.model";
 import DiskCache from "./diskCache";
+import { config } from "../config";
 
 const privateResponse = (videoId: string, reason: string): innerTubeVideoDetails => ({
     videoId,
@@ -34,13 +35,20 @@ export async function getFromITube (videoID: string): Promise<innerTubeVideoDeta
         context: {
             client: {
                 clientName: "WEB",
-                clientVersion: "2.20221215.04.01"
+                clientVersion: "2.20221215.04.01",
+                visitorData: config.youTubeKeys.visitorData
             }
         },
-        videoId: videoID
+        videoId: videoID,
+        serviceIntegrityDimensions: {
+            poToken: config.youTubeKeys.poToken
+        }
     };
     const result = await axios.post(url, data, {
-        timeout: 3500
+        timeout: 3500,
+        headers: {
+            "X-Goog-Visitor-Id": config.youTubeKeys.visitorData
+        }
     });
     /* istanbul ignore else */
     if (result.status === 200) {