修复压缩节目单无法加载
This commit is contained in:
88
src/app/api/live/epg/download/route.ts
Normal file
88
src/app/api/live/epg/download/route.ts
Normal file
@@ -0,0 +1,88 @@
|
||||
import { NextRequest, NextResponse } from 'next/server';
|
||||
|
||||
export const runtime = 'nodejs';
|
||||
|
||||
export async function GET(request: NextRequest) {
|
||||
try {
|
||||
const { searchParams } = new URL(request.url);
|
||||
const epgUrl = searchParams.get('url');
|
||||
|
||||
if (!epgUrl) {
|
||||
return NextResponse.json({ error: '缺少EPG URL参数' }, { status: 400 });
|
||||
}
|
||||
|
||||
console.log('[EPG Download] Fetching:', epgUrl);
|
||||
|
||||
// 获取EPG文件
|
||||
const response = await fetch(epgUrl, {
|
||||
headers: {
|
||||
'User-Agent': 'Mozilla/5.0',
|
||||
},
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
return NextResponse.json({ error: 'EPG文件下载失败' }, { status: 500 });
|
||||
}
|
||||
|
||||
// 检查是否是gzip压缩
|
||||
const isGzip = epgUrl.endsWith('.gz') || response.headers.get('content-encoding') === 'gzip';
|
||||
|
||||
if (isGzip) {
|
||||
console.log('[EPG Download] Decompressing gzip...');
|
||||
|
||||
// 读取所有数据
|
||||
const reader = response.body?.getReader();
|
||||
if (!reader) {
|
||||
return NextResponse.json({ error: '无法读取响应' }, { status: 500 });
|
||||
}
|
||||
|
||||
const chunks: Uint8Array[] = [];
|
||||
while (true) {
|
||||
const { done, value } = await reader.read();
|
||||
if (done) break;
|
||||
chunks.push(value);
|
||||
}
|
||||
|
||||
// 合并所有chunks
|
||||
const totalLength = chunks.reduce((acc, chunk) => acc + chunk.length, 0);
|
||||
const allChunks = new Uint8Array(totalLength);
|
||||
let offset = 0;
|
||||
for (const chunk of chunks) {
|
||||
allChunks.set(chunk, offset);
|
||||
offset += chunk.length;
|
||||
}
|
||||
|
||||
console.log('[EPG Download] Compressed size:', totalLength, 'bytes');
|
||||
|
||||
// 解压
|
||||
const zlib = await import('zlib');
|
||||
const decompressed = zlib.gunzipSync(Buffer.from(allChunks));
|
||||
const decompressedText = decompressed.toString('utf-8');
|
||||
|
||||
console.log('[EPG Download] Decompressed size:', decompressedText.length, 'bytes');
|
||||
|
||||
// 返回解压后的XML
|
||||
return new NextResponse(decompressedText, {
|
||||
headers: {
|
||||
'Content-Type': 'application/xml; charset=utf-8',
|
||||
'Content-Disposition': 'attachment; filename="epg.xml"',
|
||||
},
|
||||
});
|
||||
} else {
|
||||
// 非压缩文件,直接返回
|
||||
const text = await response.text();
|
||||
return new NextResponse(text, {
|
||||
headers: {
|
||||
'Content-Type': 'application/xml; charset=utf-8',
|
||||
'Content-Disposition': 'attachment; filename="epg.xml"',
|
||||
},
|
||||
});
|
||||
}
|
||||
} catch (error) {
|
||||
console.error('[EPG Download] Error:', error);
|
||||
return NextResponse.json(
|
||||
{ error: '下载EPG文件失败' },
|
||||
{ status: 500 }
|
||||
);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user