From 2c3a95b98d7a69cacf1cb1b3fcfb858bc9c759a2 Mon Sep 17 00:00:00 2001
From: AisukaYuki <yubingruo@gmail.com>
Date: Sun, 29 Dec 2024 19:40:10 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=8A=A0=E5=9D=87=E8=A1=A1?=
 =?UTF-8?q?=E6=AD=8C=E6=9B=B2=E5=93=8D=E5=BA=A6=EF=BC=88=E5=8F=AF=E9=80=89?=
 =?UTF-8?q?=EF=BC=89=20(#338)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

可选下载和转换mp3时，是否均衡的歌曲响度，避免歌声忽大忽小。#336
---
 xiaomusic/config.py                        |  1 +
 xiaomusic/static/default/setting.html      |  3 +++
 xiaomusic/static/default_past/setting.html |  3 +++
 xiaomusic/utils.py                         | 12 ++++++++++++
 xiaomusic/xiaomusic.py                     |  3 +++
 5 files changed, 22 insertions(+)

diff --git a/xiaomusic/config.py b/xiaomusic/config.py
index 16ae1ee..562e271 100644
--- a/xiaomusic/config.py
+++ b/xiaomusic/config.py
@@ -97,6 +97,7 @@ class Config:
     port: int = int(os.getenv("XIAOMUSIC_PORT", "8090"))  # 监听端口
     public_port: int = int(os.getenv("XIAOMUSIC_PUBLIC_PORT", 0))  # 歌曲访问端口
     proxy: str = os.getenv("XIAOMUSIC_PROXY", None)
+    loudnorm: str = os.getenv("XIAOMUSIC_LOUDNORM", None) # 均衡音量参数
     search_prefix: str = os.getenv(
         "XIAOMUSIC_SEARCH", "bilisearch:"
     )  # "bilisearch:" or "ytsearch:"
diff --git a/xiaomusic/static/default/setting.html b/xiaomusic/static/default/setting.html
index 872bac1..92e5e7f 100644
--- a/xiaomusic/static/default/setting.html
+++ b/xiaomusic/static/default/setting.html
@@ -110,6 +110,9 @@ var vConsole = new window.VConsole();
         <label for="proxy">XIAOMUSIC_PROXY(ytsearch需要):</label>
         <input id="proxy" type="text" placeholder="http://192.168.2.5:8080" />
 
+        <label for="loudnorm">均衡歌曲音量大小(loudnorm滤镜):</label>
+        <input id="loudnorm" type="text" placeholder="loudnorm=I=-14:TP=-1.5:LRA=6" />
+
         <label for="remove_id3tag">去除MP3 ID3v2和填充:</label>
         <select id="remove_id3tag">
           <option value="true">true</option>
diff --git a/xiaomusic/static/default_past/setting.html b/xiaomusic/static/default_past/setting.html
index e9e9d9b..dd8e1b6 100644
--- a/xiaomusic/static/default_past/setting.html
+++ b/xiaomusic/static/default_past/setting.html
@@ -111,6 +111,9 @@ var vConsole = new window.VConsole();
         <label for="proxy">XIAOMUSIC_PROXY(ytsearch需要):</label>
         <input id="proxy" type="text" placeholder="http://192.168.2.5:8080" />
 
+        <label for="loudnorm">均衡歌曲音量大小(loudnorm滤镜):</label>
+        <input id="loudnorm" type="text" placeholder="loudnorm=I=-14:TP=-1.5:LRA=6" />
+
         <label for="remove_id3tag">去除MP3 ID3v2和填充:</label>
         <select id="remove_id3tag">
           <option value="true">true</option>
diff --git a/xiaomusic/utils.py b/xiaomusic/utils.py
index 346f517..a24e7ec 100644
--- a/xiaomusic/utils.py
+++ b/xiaomusic/utils.py
@@ -487,6 +487,11 @@ def convert_file_to_mp3(input_file: str, config) -> str:
         log.info(f"File {out_file_path} already exists. Skipping convert_file_to_mp3.")
         return relative_path
 
+    # 检查是否存在 loudnorm 参数
+    loudnorm_args = []
+    if config.loudnorm:
+        loudnorm_args = ["-af", config.loudnorm]
+
     command = [
         os.path.join(config.ffmpeg_location, "ffmpeg"),
         "-i",
@@ -495,6 +500,7 @@ def convert_file_to_mp3(input_file: str, config) -> str:
         "mp3",
         "-vn",
         "-y",
+        *loudnorm_args,
         out_file_path,
     ]
 
@@ -895,6 +901,9 @@ async def download_playlist(config, url, dirname):
 
     if config.enable_yt_dlp_cookies:
         sbp_args += ("--cookies", f"{config.yt_dlp_cookies_path}")
+    
+    if config.loudnorm:
+        sbp_args += ("--postprocessor-args", f"-af {config.loudnorm}")
 
     sbp_args += (url,)
 
@@ -930,6 +939,9 @@ async def download_one_music(config, url, name=""):
 
     if config.enable_yt_dlp_cookies:
         sbp_args += ("--cookies", f"{config.yt_dlp_cookies_path}")
+    
+    if config.loudnorm:
+        sbp_args += ("--postprocessor-args", f"-af {config.loudnorm}")
 
     sbp_args += (url,)
 
diff --git a/xiaomusic/xiaomusic.py b/xiaomusic/xiaomusic.py
index 62245e7..d09a4c5 100644
--- a/xiaomusic/xiaomusic.py
+++ b/xiaomusic/xiaomusic.py
@@ -1724,6 +1724,9 @@ class XiaoMusicDevice:
 
         if self.config.enable_yt_dlp_cookies:
             sbp_args += ("--cookies", f"{self.config.yt_dlp_cookies_path}")
+    
+        if self.config.loudnorm:
+            sbp_args += ("--postprocessor-args", f"-af {self.config.loudnorm}")
 
         cmd = " ".join(sbp_args)
         self.log.info(f"download cmd: {cmd}")