feat: filter by title

YurierDept · Jan 25, 2024 · 6235e0f · 6235e0f
1 parent 2580488
commit 6235e0f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/packages/get-bonus/src/scraper.ts b/packages/get-bonus/src/scraper.ts
@@ -8,10 +8,20 @@ export class Scraper {
   }
 
   async search(text: string, options: Partial<SearchOptions> = {}) {
+    const { onlySearchTitle = true } = options;
+
     const results = await Promise.all(
       this.providers.map(async (provider) => {
         try {
           const resp = await provider.search(text, options);
+
+          // 对搜索结果进行进一步过滤, 结果的标题必须包含所有搜索字符串
+          if (onlySearchTitle) {
+            const pieces = text.split(' ').filter(Boolean);
+            const filtered = resp.filter((result) => pieces.every((p) => result.title.includes(p)));
+            resp.splice(0, resp.length, ...filtered);
+          }
+
           return [provider.id, resp] as const;
         } catch (error) {
           console.log(`搜索 ${provider.id} 失败: ${text}`);
@@ -20,6 +30,7 @@ export class Scraper {
         }
       })
     );
+
     return Object.fromEntries(results);
   }
 

diff --git a/packages/get-bonus/src/types.ts b/packages/get-bonus/src/types.ts
@@ -1,4 +1,6 @@
 export interface SearchOptions {
+  onlySearchTitle: string;
+
   category: string;
   // TODO
 }