From f962fcaa992fecf7838a97cc7f8c1f491809dd94 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=B6=B5=E6=9B=A6?= <im.hanxi@gmail.com>
Date: Tue, 30 Apr 2024 12:47:57 +0000
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E6=9C=AC=E5=9C=B0=E9=9F=B3?=
 =?UTF-8?q?=E4=B9=90=E6=A8=A1=E7=B3=8A=E6=90=9C=E7=B4=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .gitignore                  |  1 +
 xiaomusic/httpserver.py     |  4 ++++
 xiaomusic/static/app.js     | 22 ++++++++++++++++++++++
 xiaomusic/static/index.html |  3 ++-
 xiaomusic/utils.py          |  4 ++++
 xiaomusic/xiaomusic.py      | 11 +++++++++++
 6 files changed, 44 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index 558856e..de33f45 100644
--- a/.gitignore
+++ b/.gitignore
@@ -163,3 +163,4 @@ cython_debug/
 
 ffmpeg
 music
+test.sh
diff --git a/xiaomusic/httpserver.py b/xiaomusic/httpserver.py
index 7750857..d525a81 100644
--- a/xiaomusic/httpserver.py
+++ b/xiaomusic/httpserver.py
@@ -28,6 +28,10 @@ def getvolume():
         "volume": xiaomusic.get_volume(),
     }
 
+@app.route("/searchmusic")
+def searchmusic():
+    name = request.args.get('name')
+    return xiaomusic.searchmusic(name)
 
 @app.route("/", methods=["GET"])
 def redirect_to_index():
diff --git a/xiaomusic/static/app.js b/xiaomusic/static/app.js
index ed818c3..f68a7f4 100644
--- a/xiaomusic/static/app.js
+++ b/xiaomusic/static/app.js
@@ -64,4 +64,26 @@ $(function(){
       }
     });
   }
+
+  // 监听输入框的输入事件
+  $("#music-name").on('input', function() {
+    var inputValue = $(this).val();
+    // 发送Ajax请求
+    $.ajax({
+      url: "searchmusic", // 服务器端处理脚本
+      type: "GET",
+      dataType: "json",
+      data: {
+        name: inputValue
+      },
+      success: function(data) {
+        // 清空datalist
+        $("#autocomplete-list").empty();
+        // 添加新的option元素
+        $.each(data, function(i, item) {
+          $('<option>').val(item).appendTo("#autocomplete-list");
+        });
+      }
+    });
+  });
 });
diff --git a/xiaomusic/static/index.html b/xiaomusic/static/index.html
index 455a965..e7a1f24 100644
--- a/xiaomusic/static/index.html
+++ b/xiaomusic/static/index.html
@@ -44,7 +44,8 @@
     </div>
     <hr>
     <div>
-      <input id="music-name" type="text" placeholder="请输入搜索关键词(如:MV高清版 周杰伦 七里香)"></input>
+      <datalist id="autocomplete-list"></datalist>
+      <input id="music-name" type="text" placeholder="请输入搜索关键词(如:MV高清版 周杰伦 七里香)" list="autocomplete-list"></input>
       <input id="music-filename" type="text" placeholder="请输入保存为的文件名称(如:周杰伦七里香)"></input>
       <button id="play">播放</button>
     </div>
diff --git a/xiaomusic/utils.py b/xiaomusic/utils.py
index 5e2c274..661c31c 100644
--- a/xiaomusic/utils.py
+++ b/xiaomusic/utils.py
@@ -7,6 +7,7 @@ import socket
 from http.cookies import SimpleCookie
 from typing import AsyncIterator
 from urllib.parse import urlparse
+import difflib
 
 from requests.utils import cookiejar_from_dict
 
@@ -61,3 +62,6 @@ def validate_proxy(proxy_str: str) -> bool:
 
     return True
 
+# 模糊搜索
+def fuzzyfinder(user_input, collection):
+    return difflib.get_close_matches(user_input, collection, 10, cutoff=0.1)
diff --git a/xiaomusic/xiaomusic.py b/xiaomusic/xiaomusic.py
index d846f40..fa7413f 100644
--- a/xiaomusic/xiaomusic.py
+++ b/xiaomusic/xiaomusic.py
@@ -30,6 +30,7 @@ from xiaomusic.config import (
 from xiaomusic.utils import (
     calculate_tts_elapse,
     parse_cookie_string,
+    fuzzyfinder,
 )
 
 EOF = object()
@@ -330,6 +331,7 @@ class XiaoMusic:
     # 本地是否存在歌曲
     def get_filename(self, name):
         if name not in self._all_music:
+            self.log.debug("get_filename not in. name:%s", name)
             return ""
         filename = self._all_music[name]
         self.log.debug("try get_filename. filename:%s", filename)
@@ -490,6 +492,9 @@ class XiaoMusic:
         if search_key == "" and name == "":
             await self.play_next()
             return
+        if name == "":
+            name = search_key
+        self.log.debug("play. search_key:%s name:%s", search_key, name)
         filename = self.get_filename(name)
 
         if len(filename) <= 0:
@@ -569,3 +574,9 @@ class XiaoMusic:
 
     def get_volume(self):
         return self._volume
+
+    # 搜索音乐
+    def searchmusic(self, name):
+        search_list = fuzzyfinder(name, self._play_list)
+        self.log.debug("searchmusic. name:%s search_list:%s", name, search_list)
+        return search_list