watson/umami
1
0
Fork 0
mirror of https://github.com/umami-software/umami.git synced 2026-02-04 04:37:11 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

feat(geo): add redirect support for direct .mmdb downloads

Browse source
This commit is contained in:
Mintimate 2025-11-12 19:18:44 +08:00
parent 371ff47325
commit e13362bfec
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
scripts/build-geo.js
View file

@ -45,10 +45,16 @@ const downloadCompressed = url =>
}); });
// Download handler for direct .mmdb files // Download handler for direct .mmdb files
const downloadDirect = url => const downloadDirect = (url, originalUrl) =>
new Promise((resolve, reject) => { new Promise((resolve, reject) => {
https.get(url, res => { https.get(url, res => {
const filename = path.join(dest, path.basename(url)); // Follow redirects
if (res.statusCode === 301 || res.statusCode === 302) {
downloadDirect(res.headers.location, originalUrl || url).then(resolve).catch(reject);
return;
}
const filename = path.join(dest, path.basename(originalUrl || url));
const fileStream = fs.createWriteStream(filename); const fileStream = fs.createWriteStream(filename);
res.pipe(fileStream); res.pipe(fileStream);