Forward user IP to torbox for CDN selection (#384)

mhdzumair · Dec 1, 2024 · 3e3b120 · 3e3b120
1 parent 348ccb7
commit 3e3b120
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/streaming_providers/torbox/client.py b/streaming_providers/torbox/client.py
@@ -107,11 +107,11 @@ async def get_available_torrent(self, info_hash) -> dict[str, Any] | None:
                 return torrent
         return {}
 
-    async def create_download_link(self, torrent_id, filename):
+    async def create_download_link(self, torrent_id, filename, user_ip):
         response = await self._make_request(
             "GET",
             "/torrents/requestdl",
-            params={"token": self.token, "torrent_id": torrent_id, "file_id": filename},
+            params={"token": self.token, "torrent_id": torrent_id, "file_id": filename, "user_ip": user_ip},
             is_expected_to_fail=True,
         )
         if "successfully" in response.get("detail"):

diff --git a/streaming_providers/torbox/utils.py b/streaming_providers/torbox/utils.py
@@ -14,6 +14,7 @@ async def get_video_url_from_torbox(
     magnet_link: str,
     user_data: UserData,
     filename: str,
+    user_ip: str,
     episode: Optional[int] = None,
     **kwargs: Any,
 ) -> str:
@@ -31,6 +32,7 @@ async def get_video_url_from_torbox(
                 response = await torbox_client.create_download_link(
                     torrent_info.get("id", ""),
                     file_id,
+                    user_ip,
                 )
                 return response["data"]
         else:
@@ -47,6 +49,7 @@ async def get_video_url_from_torbox(
                     response = await torbox_client.create_download_link(
                         torrent_info.get("id", ""),
                         file_id,
+                        user_ip,
                     )
                     return response["data"]