From f0b029d7933bc85a7ca794c3edea1afccc8d4f28 Mon Sep 17 00:00:00 2001
From: Munif Tanjim <hello@muniftanjim.dev>
Date: Sun, 29 Dec 2024 01:49:36 +0600
Subject: [PATCH] feat(debrid/stremthru): use default url only for specific
 services

---
 comet/utils/config.py | 5 ++++-
 comet/utils/models.py | 7 ++++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/comet/utils/config.py b/comet/utils/config.py
index f67e635..402efa9 100644
--- a/comet/utils/config.py
+++ b/comet/utils/config.py
@@ -36,5 +36,8 @@ def prepare_debrid_config(config: dict[str, Any]):
         config["debridService"] = settings.PROXY_DEBRID_STREAM_DEBRID_DEFAULT_SERVICE
         config["debridApiKey"] = settings.PROXY_DEBRID_STREAM_DEBRID_DEFAULT_APIKEY
 
-    if not config["stremthruUrl"]:
+    if (
+        not config["stremthruUrl"]
+        and config["debridService"] in settings.STREMTHRU_AUTO_ENABLED_DEBRID_SERVICES
+    ):
         config["stremthruUrl"] = settings.STREMTHRU_DEFAULT_URL
diff --git a/comet/utils/models.py b/comet/utils/models.py
index 077e30a..8e4e241 100644
--- a/comet/utils/models.py
+++ b/comet/utils/models.py
@@ -42,7 +42,12 @@ class AppSettings(BaseSettings):
     PROXY_DEBRID_STREAM_DEBRID_DEFAULT_APIKEY: Optional[str] = None
     TITLE_MATCH_CHECK: Optional[bool] = True
     REMOVE_ADULT_CONTENT: Optional[bool] = False
-    STREMTHRU_DEFAULT_URL: Optional[str] = None
+    STREMTHRU_DEFAULT_URL: Optional[str] = "https://stremthru.elfhosted.com"
+    STREMTHRU_AUTO_ENABLED_DEBRID_SERVICES: List[str] = [
+        "realdebrid",
+        "alldebrid",
+        "debridlink",
+    ]
 
     @field_validator("DASHBOARD_ADMIN_PASSWORD")
     def set_dashboard_admin_password(cls, v, values):