mincomk/risuai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update : server.cjs

Browse Source
This commit is contained in:
shirosaki-hana
2025-04-17 17:31:46 +09:00
parent e0038749a4
commit 5168565b4f
3 changed files with 26 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
server/node/server.cjs
View File

@@ -153,26 +153,45 @@ async function hubProxyFunc(req, res) {
delete headersToSend.host;
delete headersToSend.connection;
const proxyReq = new Request(externalURL, {
const response = await fetch(externalURL, {
method: req.method,
headers: headersToSend,
body: req.method !== 'GET' && req.method !== 'HEAD' ? req : undefined,
duplex: 'half'
redirect: 'manual'
});
const response = await fetch(proxyReq);
for (const [key, value] of response.headers.entries()) {
res.setHeader(key, value);
}
res.status(response.status);
if (response.status >= 300 && response.status < 400) {
// Redirect handling (due to ‘/redirect/docs/lua’)
const redirectUrl = response.headers.get('location');
if (redirectUrl) {
if (redirectUrl.startsWith('http')) {
if (redirectUrl.startsWith(hubURL)) {
const newPath = redirectUrl.replace(hubURL, '/hub-proxy');
res.setHeader('location', newPath);
}
} else if (redirectUrl.startsWith('/')) {
res.setHeader('location', `/hub-proxy${redirectUrl}`);
}
}
return res.end();
}
await pipeline(response.body, res);
} catch (error) {
console.error("[Hub Proxy] Error:", error);
if (!res.headersSent) {
res.status(502).send({ error: 'Proxy request failed' });
res.status(502).send({ error: 'Proxy request failed: ' + error.message });
} else {
res.end();
}