From e10fad3d4e447fee3fe2ad924885016de80a4d76 Mon Sep 17 00:00:00 2001
From: wukko <me@wukko.me>
Date: Fri, 14 Mar 2025 08:53:26 +0600
Subject: [PATCH] api/config: add `CUSTOM_INNERTUBE_CLIENT` env

---
 api/src/config.js | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/api/src/config.js b/api/src/config.js
index 191e8441..bff1eda3 100644
--- a/api/src/config.js
+++ b/api/src/config.js
@@ -1,3 +1,4 @@
+import { Constants } from "youtubei.js";
 import { getVersion } from "@imput/version-info";
 import { services } from "./processing/service-config.js";
 import { supportsReusePort } from "./misc/cluster.js";
@@ -52,6 +53,7 @@ const env = {
     keyReloadInterval: 900,
 
     enabledServices,
+    customInnertubeClient: process.env.CUSTOM_INNERTUBE_CLIENT,
 }
 
 const genericUserAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36";
@@ -74,6 +76,12 @@ if (env.instanceCount > 1 && !env.redisURL) {
     throw new Error('SO_REUSEPORT is not supported');
 }
 
+if (env.customInnertubeClient && !Constants.SUPPORTED_CLIENTS.includes(env.customInnertubeClient)) {
+    console.error("CUSTOM_INNERTUBE_CLIENT is invalid. Provided client is not supported.");
+    console.error(`Supported clients are: ${Constants.SUPPORTED_CLIENTS.join(', ')}\n`);
+    throw new Error("Invalid CUSTOM_INNERTUBE_CLIENT");
+}
+
 export {
     env,
     genericUserAgent,