34 lines
969 B
TypeScript
34 lines
969 B
TypeScript
import { NextRequest } from 'next/server';
|
|
import { getAuthInfoFromCookie } from './auth';
|
|
|
|
/**
|
|
* 获取用于代理的 token
|
|
* 优先级：全局 token > 用户 token > 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');
|
|
const userInfo = await db.getUserInfoV2(authInfo.username);
|
|
if (userInfo && userInfo.tvboxToken && !userInfo.banned) {
|
|
return userInfo.tvboxToken;
|
|
}
|
|
} catch (error) {
|
|
// 忽略错误，继续
|
|
}
|
|
}
|
|
}
|
|
|
|
// 3. 没有可用的 token
|
|
return null;
|
|
}
|