XiaoMo/alist-proxy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

1111

Browse Source
This commit is contained in:
蒋小陌 2024-11-03 00:12:15 +08:00
parent 3a38632505
commit 7b1971da4e
2 changed files with 15 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
index.js
View File

File diff suppressed because one or more lines are too long

31
source.js
View File

@ -6,7 +6,7 @@ const fs = require('fs');
const pathModule = require('path');
const crypto = require('crypto');
const requestTimeout = 10000; // 10 seconds
const requestTimeout = 20000; // 10 seconds
const cacheDir = pathModule.join(__dirname, '.cache');
const args = process.argv.slice(2);
const pathIndex = {};
@ -62,11 +62,11 @@ const server = http.createServer(async (req, res) => {
req.url = req.url.replace(/\/{2,}/g, '/');
const parsedUrl = url.parse(req.url, true);
// 获取第一个路径
let reqPath = parsedUrl.pathname.split('/')[1];
// 获取第二路径为 token
let token = parsedUrl.pathname.split('/')[2];
// 取第一个路径以外的路径
let token = parsedUrl.pathname.split('/').slice(2).join('/');
// 处理根路径请求
if (reqPath === 'favicon.ico') {
@ -97,15 +97,15 @@ const server = http.createServer(async (req, res) => {
reqPath = 'go';
}
// 检查第一个路径只能是 attachment,avatar,endpoint,go,bbs,www
if (!['attachment', 'avatar', 'go', 'bbs', 'www', 'url'].includes(reqPath)) {
res.writeHead(404, { 'Content-Type': 'text/plain' });
// 检查第一个路径只能是 avatar,endpoint,go,bbs,www
if (!['avatar', 'go', 'bbs', 'www', 'url', 'thumb'].includes(reqPath)) {
res.writeHead(404, { 'Content-Type': 'text/plain;charset=UTF-8' });
res.end('Not Found');
return;
}
if (!token || reqPath === '') {
res.writeHead(400, { 'Content-Type': 'text/plain' });
res.writeHead(400, { 'Content-Type': 'text/plain;charset=UTF-8' });
res.end('Bad Request: Missing Token or path (' + reqPath + ')');
return;
}
@ -141,7 +141,6 @@ const server = http.createServer(async (req, res) => {
// 增加 API 调用次数
viewsInfo.apiCall++;
const apiData = await fetchApiData(reqPath, token);
if (apiData.code === 200 && apiData.data && apiData.data.url) {
const { url: realUrl, cloudtype, expiration, path, headers, uniqid } = apiData.data;
@ -163,11 +162,12 @@ const server = http.createServer(async (req, res) => {
fetchAndServe(data, tempCacheContentFile, cacheContentFile, cacheMetaFile, res);
}
} else {
res.writeHead(502, { 'Content-Type': 'text/plain' });
// utf8 解码
res.writeHead(502, { 'Content-Type': 'text/plain;charset=UTF-8;' });
res.end(apiData.message || 'Bad Gateway');
}
} catch (error) {
res.writeHead(502, { 'Content-Type': 'text/plain' });
res.writeHead(502, { 'Content-Type': 'text/plain;charset=UTF-8' });
res.end('Bad Gateway: Failed to decode JSON ' + error);
}
}
@ -224,7 +224,7 @@ const fetchApiData = (reqPath, token) => {
try {
resolve(JSON.parse(data));
} catch (error) {
reject(error);
reject(error + data);
}
});
});
@ -307,10 +307,7 @@ const serveFromCache = (cacheData, cacheContentFile, cacheMetaFile, res) => {
}
}
readStream.on('open', () => {
const defaultHeaders = {
'Cloud-Type': cacheData.cloudtype,
'Cloud-Expiration': cacheData.expiration,
@ -336,7 +333,7 @@ const serveFromCache = (cacheData, cacheContentFile, cacheMetaFile, res) => {
// 处理响应错误
const handleResponseError = (res, tempCacheContentFile, realUrl) => {
if (!res.headersSent) {
res.writeHead(502, { 'Content-Type': 'text/plain' });
res.writeHead(502, { 'Content-Type': 'text/plain;charset=UTF-8' });
res.end(`Bad Gateway: ${realUrl}`);
}
if (fs.existsSync(tempCacheContentFile)) {
@ -347,7 +344,7 @@ const handleResponseError = (res, tempCacheContentFile, realUrl) => {
// 处理缓存读取错误
const handleCacheReadError = (res) => {
if (!res.headersSent) {
res.writeHead(500, { 'Content-Type': 'text/plain' });
res.writeHead(500, { 'Content-Type': 'text/plain;charset=UTF-8' });
res.end('Internal Server Error: Unable to read cache content file');
}
};