35 lines
1003 B
TypeScript
35 lines
1003 B
TypeScript
import { NextRequest } from 'next/server';
|
|
import { getAuthInfoFromCookie } from './auth';
|
|
|
|
/**
|
|
* 获取用于代理的 token
|
|
* 优先级：全局 token > 用户 token（从 cookie 获取）> null
|
|
*/
|
|
export async function getProxyToken(request?: NextRequest): Promise<string | null> {
|
|
// 1. 尝试获取全局 token
|
|
const globalToken = process.env.TVBOX_SUBSCRIBE_TOKEN;
|
|
if (globalToken) {
|
|
return globalToken;
|
|
}
|
|
|
|
// 2. 如果提供了 request，尝试从用户登录信息获取用户的 tvbox token
|
|
if (request) {
|
|
const authInfo = getAuthInfoFromCookie(request);
|
|
if (authInfo && authInfo.username) {
|
|
try {
|
|
const { db } = await import('./db');
|
|
// 通过用户名获取用户的 tvbox token
|
|
const userToken = await db.getTvboxSubscribeToken(authInfo.username);
|
|
if (userToken) {
|
|
return userToken;
|
|
}
|
|
} catch (error) {
|
|
// 忽略错误，继续
|
|
}
|
|
}
|
|
}
|
|
|
|
// 3. 没有可用的 token
|
|
return null;
|
|
}
|