From 13336cd02cb289e98bdc45d996595c81441a0f2d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=B6=B5=E6=9B=A6?= <im.hanxi@gmail.com>
Date: Fri, 2 Feb 2024 20:55:51 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=AF=E6=8C=81=E9=85=8D=E7=BD=AE=E6=AD=8C?=
 =?UTF-8?q?=E6=9B=B2=E4=B8=8B=E8=BD=BD=E6=BA=90=20#12?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 xiaomusic/config.py    | 3 +++
 xiaomusic/xiaomusic.py | 3 ++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/xiaomusic/config.py b/xiaomusic/config.py
index 3991bfb..38a9ddd 100644
--- a/xiaomusic/config.py
+++ b/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:
diff --git a/xiaomusic/xiaomusic.py b/xiaomusic/xiaomusic.py
index 88099d3..e4cf2e5 100644
--- a/xiaomusic/xiaomusic.py
+++ b/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",