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);
|
2023-12-13 16:32:01 -06:00
|
|
|
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;
|
|
|
|
}
|
|
|
|
}
|