64 lines
2.0 KiB
TypeScript
64 lines
2.0 KiB
TypeScript
import { NextRequest, NextResponse } from 'next/server';
|
||
|
||
const API_URL = process.env.API_URL || 'http://localhost:51515';
|
||
/**
|
||
* 通用代理函数
|
||
* @param request 原始请求
|
||
* @param targetHost 目标服务器地址（如 http://localhost:51515）
|
||
* @returns NextResponse 代理响应
|
||
*/
|
||
async function proxyRequest(request: NextRequest, targetHost: string): Promise<NextResponse> {
|
||
const { pathname } = request.nextUrl;
|
||
const targetUrl = new URL(pathname + request.nextUrl.search, targetHost);
|
||
|
||
try {
|
||
console.log(`Proxying request to: ${targetUrl.toString()}`);
|
||
const response = await fetch(targetUrl.toString(), {
|
||
method: request.method,
|
||
headers: {
|
||
// 复制原始请求的所有头部
|
||
...Object.fromEntries(request.headers.entries()),
|
||
// 确保转发 cookie
|
||
cookie: request.headers.get('cookie') || '',
|
||
},
|
||
body: request.method !== 'GET' && request.method !== 'HEAD' ? await request.text() : undefined,
|
||
// 包含 cookie 的请求
|
||
credentials: 'include',
|
||
});
|
||
|
||
// 创建响应，保留原始响应的所有头部（包括 Set-Cookie）
|
||
const responseHeaders = new Headers(response.headers);
|
||
|
||
return new NextResponse(response.body, {
|
||
status: response.status,
|
||
statusText: response.statusText,
|
||
headers: responseHeaders,
|
||
});
|
||
} catch (error) {
|
||
console.error('Proxy error:', error);
|
||
return NextResponse.json(
|
||
{ error: 'Proxy request failed' },
|
||
{ status: 502 }
|
||
);
|
||
}
|
||
}
|
||
|
||
export async function proxy(request: NextRequest) {
|
||
const { pathname } = request.nextUrl;
|
||
// 代理 /api 请求
|
||
if (pathname.startsWith('/api')) {
|
||
return proxyRequest(request, API_URL);
|
||
}
|
||
// 代理 /root/home 或 /root/home/ 请求 (第二个路径段后可以为空或以 / 结尾)
|
||
if (pathname.match(/^\/root\/\w+(\/)?$/)) {
|
||
return proxyRequest(request, API_URL);
|
||
}
|
||
|
||
return NextResponse.next();
|
||
}
|
||
|
||
export const config = {
|
||
matcher: ['/api/:path*', '/root/:segment/:path*'],
|
||
};
|
||
|
||
export default proxy; |