/* eslint-disable @typescript-eslint/no-explicit-any,no-console */ import { NextRequest, NextResponse } from 'next/server'; import { AdminConfig } from '@/lib/admin.types'; import { getAuthInfoFromCookie } from '@/lib/auth'; import { getAvailableApiSites, getConfig } from '@/lib/config'; import { searchFromApi } from '@/lib/downstream'; import { yellowWords } from '@/lib/yellow'; export const runtime = 'nodejs'; // 强制动态渲染 export const dynamic = 'force-dynamic'; export async function GET(request: NextRequest) { try { // 从 cookie 获取用户信息 const authInfo = getAuthInfoFromCookie(request); if (!authInfo || !authInfo.username) { return NextResponse.json({ error: 'Unauthorized' }, { status: 401 }); } const config = await getConfig(); const { searchParams } = new URL(request.url); const query = searchParams.get('q')?.trim(); if (!query) { return NextResponse.json({ suggestions: [] }); } // 生成建议 const suggestions = await generateSuggestions(config, query, authInfo.username); // 从配置中获取缓存时间,如果没有配置则使用默认值300秒(5分钟) const cacheTime = config.SiteConfig.SiteInterfaceCacheTime || 300; return NextResponse.json( { suggestions }, { headers: { 'Cache-Control': `public, max-age=${cacheTime}, s-maxage=${cacheTime}`, 'CDN-Cache-Control': `public, s-maxage=${cacheTime}`, 'Vercel-CDN-Cache-Control': `public, s-maxage=${cacheTime}`, 'Netlify-Vary': 'query', }, } ); } catch (error) { console.error('获取搜索建议失败', error); return NextResponse.json({ error: '获取搜索建议失败' }, { status: 500 }); } } async function generateSuggestions(config: AdminConfig, query: string, username: string): Promise< Array<{ text: string; type: 'exact' | 'related' | 'suggestion'; score: number; }> > { const queryLower = query.toLowerCase(); const apiSites = await getAvailableApiSites(username); let realKeywords: string[] = []; if (apiSites.length > 0) { // 取第一个可用的数据源进行搜索 const firstSite = apiSites[0]; const results = await searchFromApi(firstSite, query); 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(/[ -::·、-]/)) .filter( (w: string) => w.length > 1 && w.toLowerCase().includes(queryLower) ) ) ).slice(0, 8); } // 根据关键词与查询的匹配程度计算分数,并动态确定类型 const realSuggestions = realKeywords.map((word) => { const wordLower = word.toLowerCase(); const queryWords = queryLower.split(/[ -::·、-]/); // 计算匹配分数:完全匹配得分更高 let score = 1.0; if (wordLower === queryLower) { score = 2.0; // 完全匹配 } else if ( wordLower.startsWith(queryLower) || wordLower.endsWith(queryLower) ) { score = 1.8; // 前缀或后缀匹配 } else if (queryWords.some((qw) => wordLower.includes(qw))) { score = 1.5; // 包含查询词 } // 根据匹配程度确定类型 let type: 'exact' | 'related' | 'suggestion' = 'related'; if (score >= 2.0) { type = 'exact'; } else if (score >= 1.5) { type = 'related'; } else { type = 'suggestion'; } return { text: word, type, score, }; }); // 按分数降序排列,相同分数按类型优先级排列 const sortedSuggestions = realSuggestions.sort((a, b) => { if (a.score !== b.score) { return b.score - a.score; // 分数高的在前 } // 分数相同时,按类型优先级:exact > related > suggestion const typePriority = { exact: 3, related: 2, suggestion: 1 }; return typePriority[b.type] - typePriority[a.type]; }); return sortedSuggestions; }