feat: patch /play to use Bitflow API for YouTube, configurable API key via env

DragMusic/platforms/Youtube.py

@@ -7,7 +7,9 @@ import yt_dlp
 from pyrogram.enums import MessageEntityType
 from pyrogram.types import Message
 from youtubesearchpython.__future__ import VideosSearch
-
+import httpx
+import config
+import httpx
 from DragMusic.utils.database import is_on_off
 from DragMusic.utils.formatters import time_to_seconds
 
@@ -350,4 +352,21 @@ class YouTubeAPI:
         else:
             direct = True
             downloaded_file = await loop.run_in_executor(None, audio_dl)
-        return downloaded_file, direct
+        return downloaded_file, direct
+
+    async def bitflow_video(self, query: str, api_key: str = None):
+        """
+        Fetch video info from Bitflow API.
+        Args:
+            query (str): The search query or YouTube URL.
+            api_key (str, optional): Bitflow API key. Defaults to config.BITFLOW_API_KEY.
+        Returns:
+            dict: The JSON result from the Bitflow API.
+        """
+        api_url = "https://bitflow.in/api/youtube"
+        if api_key is None:
+            api_key = config.BITFLOW_API_KEY
+        params = {"query": query, "format": "video", "api_key": api_key}
+        async with httpx.AsyncClient(timeout=150) as client:
+            response = await client.get(api_url, params=params)
+            return response.json()

DragMusic/plugins/play/play.py

@@ -24,7 +24,7 @@ from DragMusic.utils.inline import (
 )
 from DragMusic.utils.logger import play_logs
 from DragMusic.utils.stream.stream import stream
-from config import BANNED_USERS, lyrical
+from config import BANNED_USERS, lyrical, BITFLOW_API_KEY
 
 
 @app.on_message(
@@ -179,15 +179,32 @@ async def play_commnd(
                 cap = _["play_9"]
             else:
                 try:
-                    details, track_id = await YouTube.track(url)
-                except:
-                    return await mystic.edit_text(_["play_3"])
-                streamtype = "youtube"
-                img = details["thumb"]
-                cap = _["play_10"].format(
-                    details["title"],
-                    details["duration_min"],
-                )
+                    bitflow_result = await YouTube.bitflow_video(url, api_key=BITFLOW_API_KEY)
+                    if bitflow_result and bitflow_result.get("status") == "success":
+                        details = {
+                            "title": bitflow_result.get("title"),
+                            "duration_min": bitflow_result.get("duration"),
+                            "thumb": bitflow_result.get("thumbnail"),
+                            "link": bitflow_result.get("url"),
+                        }
+                        track_id = bitflow_result.get("id")
+                        streamtype = "youtube"
+                        img = details["thumb"]
+                        cap = _["play_10"].format(details["title"], details["duration_min"])
+                    else:
+                        details, track_id = await YouTube.track(url)
+                        streamtype = "youtube"
+                        img = details["thumb"]
+                        cap = _["play_10"].format(details["title"], details["duration_min"])
+                except Exception as e:
+                    print(f"Bitflow API error: {e}")
+                    try:
+                        details, track_id = await YouTube.track(url)
+                        streamtype = "youtube"
+                        img = details["thumb"]
+                        cap = _["play_10"].format(details["title"], details["duration_min"])
+                    except:
+                        return await mystic.edit_text(_["play_3"])
         elif await Spotify.valid(url):
             spotify = True
             if not config.SPOTIFY_CLIENT_ID and not config.SPOTIFY_CLIENT_SECRET:
@@ -335,9 +352,23 @@ async def play_commnd(
         if "-v" in query:
             query = query.replace("-v", "")
         try:
-            details, track_id = await YouTube.track(query)
-        except:
-            return await mystic.edit_text(_["play_3"])
+            bitflow_result = await YouTube.bitflow_video(query, api_key=BITFLOW_API_KEY)
+            if bitflow_result and bitflow_result.get("status") == "success":
+                details = {
+                    "title": bitflow_result.get("title"),
+                    "duration_min": bitflow_result.get("duration"),
+                    "thumb": bitflow_result.get("thumbnail"),
+                    "link": bitflow_result.get("url"),
+                }
+                track_id = bitflow_result.get("id")
+            else:
+                details, track_id = await YouTube.track(query)
+        except Exception as e:
+            print(f"Bitflow API error: {e}")
+            try:
+                details, track_id = await YouTube.track(query)
+            except:
+                return await mystic.edit_text(_["play_3"])
         streamtype = "youtube"
     if str(playmode) == "Direct":
         if not plist_type:

config.py

@@ -31,6 +31,8 @@ LOGGER_ID = int(getenv("LOGGER_ID", ))
 # Get this value from @FallenxBot on Telegram by /id
 OWNER_ID = int(getenv("OWNER_ID", ))
 
+BITFLOW_API_KEY = getenv("BITFLOW_API_KEY", None)  # Bitflow API key for YouTube downloads
+
 ## Fill these variables if you're deploying on heroku.
 # Your heroku app name
 HEROKU_APP_NAME = getenv("HEROKU_APP_NAME")
@@ -98,6 +100,9 @@ SPOTIFY_ALBUM_IMG_URL = "https://te.legra.ph/file/b35fd1dfca73b950b1b05.jpg"
 SPOTIFY_PLAYLIST_IMG_URL = "https://te.legra.ph/file/95b3ca7993bbfaf993dcb.jpg"
 
 
+
+
+
 def time_to_seconds(time):
     stringt = str(time)
     return sum(int(x) * 60**i for i, x in enumerate(reversed(stringt.split(":"))))