127 lines
3.8 KiB
TypeScript
127 lines
3.8 KiB
TypeScript
/* eslint-disable @typescript-eslint/no-explicit-any,no-console */
|
|
|
|
import { NextRequest, NextResponse } from 'next/server';
|
|
|
|
import { getAuthInfoFromCookie } from '@/lib/auth';
|
|
import { getAvailableApiSites, getConfig } from '@/lib/config';
|
|
import { searchFromApi } from '@/lib/downstream';
|
|
|
|
export const runtime = 'nodejs';
|
|
|
|
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(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(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
|
|
.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;
|
|
}
|