feat: levenshtein distance match

Author: yjl9903

Diff for: package.json

@@ -24,6 +24,7 @@
     "class-variance-authority": "^0.7.0",
     "clipboard-polyfill": "^4.0.2",
     "clsx": "^2.1.0",
+    "fastest-levenshtein": "^1.0.16",
     "ipx": "^2.0.0",
     "radix-vue": "^1.4.0",
     "tailwind-merge": "^2.2.1",

Diff for: pnpm-lock.yaml

@@ -1,4 +1,5 @@
 import { BgmClient, type Search } from 'bgmc';
+import { distance } from 'fastest-levenshtein';
 
 const client = new BgmClient(fetch);
 
@@ -17,7 +18,7 @@ export default defineEventHandler(async (event) => {
 
   if (resp.list && resp.list.length > 0) {
     // 获取第一个搜索结果
-    const foundId = await inferSubject(resp.list);
+    const foundId = await inferSubject(input, resp.list);
 
     if (foundId) {
       const [subject, persons] = await Promise.all([
@@ -47,13 +48,40 @@ export default defineEventHandler(async (event) => {
   return { subject: null, persons: null };
 });
 
+/**
+ * 使用编辑距离 (levenshtein distance)，选择一个标题最相似的 subject
+ */
+function selectSubject(input: string, subjects: Search['list']) {
+  const result1 = closest(input, subjects!, (subject) => subject.name);
+  const result2 = closest(input, subjects!, (subject) => subject.name_cn);
+
+  return distance(input, result1.name ?? '') <= distance(input, result2.name_cn ?? '')
+    ? result1
+    : result2;
+
+  function closest<T>(str: string, arr: T[], key: (value: T) => string | null | undefined) {
+    let min_distance = Infinity;
+    let min_index = 0;
+    for (let i = 0; i < arr.length; i++) {
+      const name = key(arr[i]);
+      if (name === undefined || name === null) continue;
+      const dist = distance(str, name);
+      if (dist < min_distance) {
+        min_distance = dist;
+        min_index = i;
+      }
+    }
+    return arr[min_index];
+  }
+}
+
 /**
  * 1. 使用搜索结果的第一个
  * 2. 如果它是系列中的某一部，则使用系列
  */
-async function inferSubject(subjects: Search['list']) {
+async function inferSubject(input: string, subjects: Search['list']) {
   if (!subjects) return undefined;
-  const first = subjects[0];
+  const first = selectSubject(input, subjects);
   if (!first || !first.id) return undefined;
 
   const related = await client.subjectRelated(first.id);