251 lines
7.4 KiB
TypeScript
251 lines
7.4 KiB
TypeScript
import { NextResponse } from 'next/server';
|
||
|
||
import { getConfig } from '@/lib/config';
|
||
|
||
export const runtime = 'nodejs';
|
||
|
||
/**
|
||
* M3U8 代理接口
|
||
* 用于外部播放器访问,会执行去广告逻辑并处理相对链接
|
||
* GET /api/proxy-m3u8?url=<原始m3u8地址>&source=<播放源>&token=<鉴权token>
|
||
*/
|
||
export async function GET(request: Request) {
|
||
try {
|
||
const { searchParams } = new URL(request.url);
|
||
const m3u8Url = searchParams.get('url');
|
||
const source = searchParams.get('source') || '';
|
||
const token = searchParams.get('token');
|
||
|
||
// Token 鉴权：如果环境变量设置了 token，则必须验证
|
||
const envToken = process.env.NEXT_PUBLIC_PROXY_M3U8_TOKEN;
|
||
if (envToken && envToken.trim() !== '') {
|
||
if (!token || token !== envToken) {
|
||
return NextResponse.json(
|
||
{ error: '无效的访问令牌' },
|
||
{ status: 401 }
|
||
);
|
||
}
|
||
}
|
||
|
||
if (!m3u8Url) {
|
||
return NextResponse.json(
|
||
{ error: '缺少必要参数: url' },
|
||
{ status: 400 }
|
||
);
|
||
}
|
||
|
||
// 获取当前请求的 origin
|
||
// 优先级：SITE_BASE 环境变量 > 从请求头构建
|
||
let origin = process.env.SITE_BASE;
|
||
if (!origin) {
|
||
const requestUrl = new URL(request.url);
|
||
origin = `${requestUrl.protocol}//${requestUrl.host}`;
|
||
}
|
||
|
||
// 获取原始 m3u8 内容
|
||
const m3u8UrlObj = new URL(m3u8Url);
|
||
const response = await fetch(m3u8Url, {
|
||
headers: {
|
||
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
|
||
'Accept': '*/*',
|
||
'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8',
|
||
'Accept-Encoding': 'gzip, deflate, br',
|
||
'Referer': `${m3u8UrlObj.protocol}//${m3u8UrlObj.host}/`,
|
||
},
|
||
});
|
||
|
||
if (!response.ok) {
|
||
return NextResponse.json(
|
||
{ error: '获取 m3u8 文件失败' },
|
||
{ status: response.status }
|
||
);
|
||
}
|
||
|
||
let m3u8Content = await response.text();
|
||
|
||
// 执行去广告逻辑
|
||
const config = await getConfig();
|
||
const customAdFilterCode = config.SiteConfig?.CustomAdFilterCode || '';
|
||
|
||
if (customAdFilterCode && customAdFilterCode.trim()) {
|
||
try {
|
||
// 移除 TypeScript 类型注解,转换为纯 JavaScript
|
||
const jsCode = customAdFilterCode
|
||
.replace(/(\w+)\s*:\s*(string|number|boolean|any|void|never|unknown|object)\s*([,)])/g, '$1$3')
|
||
.replace(/\)\s*:\s*(string|number|boolean|any|void|never|unknown|object)\s*\{/g, ') {')
|
||
.replace(/(const|let|var)\s+(\w+)\s*:\s*(string|number|boolean|any|void|never|unknown|object)\s*=/g, '$1 $2 =');
|
||
|
||
// 创建并执行自定义函数
|
||
const customFunction = new Function('type', 'm3u8Content',
|
||
jsCode + '\nreturn filterAdsFromM3U8(type, m3u8Content);'
|
||
);
|
||
m3u8Content = customFunction(source, m3u8Content);
|
||
} catch (err) {
|
||
console.error('执行自定义去广告代码失败,使用默认规则:', err);
|
||
// 继续使用默认规则
|
||
m3u8Content = filterAdsFromM3U8Default(source, m3u8Content);
|
||
}
|
||
} else {
|
||
// 使用默认去广告规则
|
||
m3u8Content = filterAdsFromM3U8Default(source, m3u8Content);
|
||
}
|
||
|
||
// 处理 m3u8 中的相对链接
|
||
m3u8Content = resolveM3u8Links(m3u8Content, m3u8Url, source, origin, token || '');
|
||
|
||
// 返回处理后的 m3u8 内容
|
||
return new NextResponse(m3u8Content, {
|
||
headers: {
|
||
'Content-Type': 'application/vnd.apple.mpegurl',
|
||
'Access-Control-Allow-Origin': '*',
|
||
'Cache-Control': 'no-cache',
|
||
},
|
||
});
|
||
} catch (error) {
|
||
console.error('代理 m3u8 失败:', error);
|
||
return NextResponse.json(
|
||
{ error: '代理失败', details: (error as Error).message },
|
||
{ status: 500 }
|
||
);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 默认去广告规则
|
||
*/
|
||
function filterAdsFromM3U8Default(type: string, m3u8Content: string): string {
|
||
if (!m3u8Content) return '';
|
||
|
||
// 广告关键字列表
|
||
const adKeywords = [
|
||
'sponsor',
|
||
'/ad/',
|
||
'/ads/',
|
||
'advert',
|
||
'advertisement',
|
||
'/adjump',
|
||
'redtraffic'
|
||
];
|
||
|
||
// 按行分割M3U8内容
|
||
const lines = m3u8Content.split('\n');
|
||
const filteredLines = [];
|
||
|
||
let i = 0;
|
||
while (i < lines.length) {
|
||
const line = lines[i];
|
||
|
||
// 跳过 #EXT-X-DISCONTINUITY 标识
|
||
if (line.includes('#EXT-X-DISCONTINUITY')) {
|
||
i++;
|
||
continue;
|
||
}
|
||
|
||
// 如果是 EXTINF 行，检查下一行 URL 是否包含广告关键字
|
||
if (line.includes('#EXTINF:')) {
|
||
// 检查下一行 URL 是否包含广告关键字
|
||
if (i + 1 < lines.length) {
|
||
const nextLine = lines[i + 1];
|
||
const containsAdKeyword = adKeywords.some(keyword =>
|
||
nextLine.toLowerCase().includes(keyword.toLowerCase())
|
||
);
|
||
|
||
if (containsAdKeyword) {
|
||
// 跳过 EXTINF 行和 URL 行
|
||
i += 2;
|
||
continue;
|
||
}
|
||
}
|
||
}
|
||
|
||
// 保留当前行
|
||
filteredLines.push(line);
|
||
i++;
|
||
}
|
||
|
||
return filteredLines.join('\n');
|
||
}
|
||
|
||
/**
|
||
* 将 m3u8 中的相对链接转换为绝对链接，并将子 m3u8 链接转为代理链接
|
||
*/
|
||
function resolveM3u8Links(m3u8Content: string, baseUrl: string, source: string, proxyOrigin: string, token: string): string {
|
||
const lines = m3u8Content.split('\n');
|
||
const resolvedLines = [];
|
||
|
||
// 解析基础URL
|
||
const base = new URL(baseUrl);
|
||
const baseDir = base.href.substring(0, base.href.lastIndexOf('/') + 1);
|
||
|
||
let isNextLineUrl = false;
|
||
|
||
for (let i = 0; i < lines.length; i++) {
|
||
let line = lines[i];
|
||
|
||
// 处理 EXT-X-KEY 标签中的 URI
|
||
if (line.startsWith('#EXT-X-KEY:')) {
|
||
// 提取 URI 部分
|
||
const uriMatch = line.match(/URI="([^"]+)"/);
|
||
if (uriMatch && uriMatch[1]) {
|
||
let keyUri = uriMatch[1];
|
||
|
||
// 转换为绝对路径
|
||
if (!keyUri.startsWith('http://') && !keyUri.startsWith('https://')) {
|
||
if (keyUri.startsWith('/')) {
|
||
keyUri = `${base.protocol}//${base.host}${keyUri}`;
|
||
} else {
|
||
keyUri = new URL(keyUri, baseDir).href;
|
||
}
|
||
|
||
// 替换原来的 URI
|
||
line = line.replace(/URI="[^"]+"/, `URI="${keyUri}"`);
|
||
}
|
||
}
|
||
resolvedLines.push(line);
|
||
continue;
|
||
}
|
||
|
||
// 注释行直接保留
|
||
if (line.startsWith('#')) {
|
||
resolvedLines.push(line);
|
||
// 检查是否是 EXT-X-STREAM-INF，下一行将是子 m3u8
|
||
if (line.startsWith('#EXT-X-STREAM-INF:')) {
|
||
isNextLineUrl = true;
|
||
}
|
||
continue;
|
||
}
|
||
|
||
// 空行直接保留
|
||
if (line.trim() === '') {
|
||
resolvedLines.push(line);
|
||
continue;
|
||
}
|
||
|
||
// 处理 URL 行
|
||
let url = line.trim();
|
||
|
||
// 1. 先转换为绝对 URL
|
||
if (!url.startsWith('http://') && !url.startsWith('https://')) {
|
||
if (url.startsWith('/')) {
|
||
// 以 / 开头，相对于域名根目录
|
||
url = `${base.protocol}//${base.host}${url}`;
|
||
} else {
|
||
// 相对于当前目录
|
||
url = new URL(url, baseDir).href;
|
||
}
|
||
}
|
||
|
||
// 2. 检查是否是子 m3u8，如果是，转换为代理链接
|
||
const isM3u8 = url.includes('.m3u8') || isNextLineUrl;
|
||
if (isM3u8) {
|
||
const tokenParam = token ? `&token=${encodeURIComponent(token)}` : '';
|
||
url = `${proxyOrigin}/api/proxy-m3u8?url=${encodeURIComponent(url)}${source ? `&source=${encodeURIComponent(source)}` : ''}${tokenParam}`;
|
||
}
|
||
|
||
resolvedLines.push(url);
|
||
isNextLineUrl = false;
|
||
}
|
||
|
||
return resolvedLines.join('\n');
|
||
}
|