Ma/LunaTV
1
0
Fork 0
mirror of https://github.com/MoonTechLab/LunaTV.git synced 2026-05-23 07:07:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: use yellow filter in search suggestions

Browse Source
This commit is contained in:
shinya
2025-08-23 00:28:41 +08:00
parent 3bab1a9a11
commit de66c9afa0
2 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/app/api/search/suggestions/route.ts
View File

@@ -5,6 +5,8 @@ import { NextRequest, NextResponse } from 'next/server';
import { getAuthInfoFromCookie } from '@/lib/auth';
import { getAvailableApiSites, getConfig } from '@/lib/config';
import { searchFromApi } from '@/lib/downstream';
import { AdminConfig } from '@/lib/admin.types';
import { yellowWords } from '@/lib/yellow';
export const runtime = 'nodejs';
@@ -25,7 +27,7 @@ export async function GET(request: NextRequest) {
}
// 生成建议
const suggestions = await generateSuggestions(query, authInfo.username);
const suggestions = await generateSuggestions(config, query, authInfo.username);
// 从配置中获取缓存时间，如果没有配置则使用默认值300秒（5分钟）
const cacheTime = config.SiteConfig.SiteInterfaceCacheTime || 300;
@@ -47,7 +49,7 @@ export async function GET(request: NextRequest) {
}
}
async function generateSuggestions(query: string, username: string): Promise<
async function generateSuggestions(config: AdminConfig, query: string, username: string): Promise<
Array<{
text: string;
type: 'exact' | 'related' | 'suggestion';
@@ -67,6 +69,7 @@ async function generateSuggestions(query: string, username: string): Promise<
realKeywords = Array.from(
new Set(
results
.filter((r: any) => config.SiteConfig.DisableYellowFilter || !yellowWords.some((word: string) => (r.type_name || '').includes(word)))
.map((r: any) => r.title)
.filter(Boolean)
.flatMap((title: string) => title.split(/[ -:：·、-]/))