46 lines
1.1 KiB
JavaScript
46 lines
1.1 KiB
JavaScript
import { request } from 'undici';
|
|
const redirectStatuses = new Set([301, 302, 303, 307, 308]);
|
|
|
|
export async function getRedirectingURL(url, dispatcher, userAgent) {
|
|
const location = await request(url, {
|
|
dispatcher,
|
|
method: 'HEAD',
|
|
headers: { 'user-agent': userAgent }
|
|
}).then(r => {
|
|
if (redirectStatuses.has(r.statusCode) && r.headers['location']) {
|
|
return r.headers['location'];
|
|
}
|
|
}).catch(() => null);
|
|
|
|
return location;
|
|
}
|
|
|
|
export function merge(a, b) {
|
|
for (const k of Object.keys(b)) {
|
|
if (Array.isArray(b[k])) {
|
|
a[k] = [...(a[k] ?? []), ...b[k]];
|
|
} else if (typeof b[k] === 'object') {
|
|
a[k] = merge(a[k], b[k]);
|
|
} else {
|
|
a[k] = b[k];
|
|
}
|
|
}
|
|
|
|
return a;
|
|
}
|
|
|
|
export function splitFilenameExtension(filename) {
|
|
const parts = filename.split('.');
|
|
const ext = parts.pop();
|
|
|
|
if (!parts.length) {
|
|
return [ ext, "" ]
|
|
} else {
|
|
return [ parts.join('.'), ext ]
|
|
}
|
|
}
|
|
|
|
export function zip(a, b) {
|
|
return a.map((value, i) => [ value, b[i] ]);
|
|
}
|