支持配置歌曲下载源 #12

xiaomusic/config.py

@@ -85,6 +85,9 @@ class Config:
     hostname: str = os.getenv("XIAOMUSIC_HOSTNAME", "192.168.2.5")
     port: int = int(os.getenv("XIAOMUSIC_PORT", "8090"))
     proxy: str | None = os.getenv("XIAOMUSIC_PROXY", None)
+    search_prefix: str = os.getenv(
+        "XIAOMUSIC_SEARCH", "ytsearch:"
+    )  # "bilisearch:" or "ytsearch:"
 
     def __post_init__(self) -> None:
         if self.proxy:

xiaomusic/xiaomusic.py

@@ -56,6 +56,7 @@ class XiaoMusic:
         self.hostname = config.hostname
         self.port = config.port
         self.proxy = config.proxy
+        self.search_prefix = config.search_prefix
 
         # 下载对象
         self.download_proc = None
@@ -298,7 +299,7 @@ class XiaoMusic:
 
         sbp_args = (
             "yt-dlp",
-            f"ytsearch:{name}",
+            f"{self.search_prefix}{name}",
             "-x",
             "--audio-format",
             "mp3",