el.xwx.moe/lib/api/validateUrlSize.ts

44 lines
1.1 KiB
TypeScript
Raw Normal View History

2023-12-31 09:46:09 -06:00
import fetch from "node-fetch";
import https from "https";
2024-04-17 17:02:54 -05:00
import { SocksProxyAgent } from "socks-proxy-agent";
2023-12-31 09:46:09 -06:00
2023-11-25 02:19:02 -06:00
export default async function validateUrlSize(url: string) {
2024-04-17 17:02:54 -05:00
if (process.env.IGNORE_URL_SIZE_LIMIT === "true") return null;
2023-11-25 02:19:02 -06:00
try {
2023-12-31 09:46:09 -06:00
const httpsAgent = new https.Agent({
rejectUnauthorized:
process.env.IGNORE_UNAUTHORIZED_CA === "true" ? false : true,
});
2024-04-17 17:02:54 -05:00
let fetchOpts = {
2023-12-31 09:46:09 -06:00
method: "HEAD",
agent: httpsAgent,
2024-04-17 17:02:54 -05:00
};
if (process.env.PROXY) {
let proxy = new URL(process.env.PROXY);
if (process.env.PROXY_USERNAME) {
proxy.username = process.env.PROXY_USERNAME;
proxy.password = process.env.PROXY_PASSWORD || "";
}
fetchOpts = {
method: "HEAD",
agent: new SocksProxyAgent(proxy.toString()),
};
}
const response = await fetch(url, fetchOpts);
2023-11-25 02:19:02 -06:00
const totalSizeMB =
Number(response.headers.get("content-length")) / Math.pow(1024, 2);
if (totalSizeMB > (Number(process.env.NEXT_PUBLIC_MAX_FILE_SIZE) || 30))
return null;
2023-11-25 02:19:02 -06:00
else return response.headers;
} catch (err) {
console.log(err);
return null;
}
}