urllib
Version:
Help in opening URLs (mostly HTTP) in a complex world — basic and digest authentication, redirections, cookies and more. Base undici fetch API.
58 lines • 2.48 kB
JavaScript
var _HttpAgent_checkAddress;
import { __classPrivateFieldGet, __classPrivateFieldSet } from "tslib";
import dns from 'dns';
import { isIP } from 'net';
import { Agent, } from 'undici';
class IllegalAddressError extends Error {
constructor(hostname, ip, family) {
const message = 'illegal address';
super(message);
this.name = this.constructor.name;
this.hostname = hostname;
this.ip = ip;
this.family = family;
Error.captureStackTrace(this, this.constructor);
}
}
export class HttpAgent extends Agent {
constructor(options) {
var _a;
/* eslint node/prefer-promises/dns: off*/
const _lookup = (_a = options.lookup) !== null && _a !== void 0 ? _a : dns.lookup;
const lookup = (hostname, dnsOptions, callback) => {
_lookup(hostname, dnsOptions, (err, address, family) => {
if (err)
return callback(err, address, family);
if (options.checkAddress && !options.checkAddress(address, family)) {
err = new IllegalAddressError(hostname, address, family);
}
callback(err, address, family);
});
};
super({
connect: { ...options.connect, lookup },
});
_HttpAgent_checkAddress.set(this, void 0);
__classPrivateFieldSet(this, _HttpAgent_checkAddress, options.checkAddress, "f");
}
dispatch(options, handler) {
if (__classPrivateFieldGet(this, _HttpAgent_checkAddress, "f") && options.origin) {
const originUrl = typeof options.origin === 'string' ? new URL(options.origin) : options.origin;
let hostname = originUrl.hostname;
// [2001:db8:2de::e13] => 2001:db8:2de::e13
if (hostname.startsWith('[') && hostname.endsWith(']')) {
hostname = hostname.substring(1, hostname.length - 1);
}
const family = isIP(hostname);
if (family === 4 || family === 6) {
// if request hostname is ip, custom lookup won't excute
if (!__classPrivateFieldGet(this, _HttpAgent_checkAddress, "f").call(this, hostname, family)) {
throw new IllegalAddressError(hostname, hostname, family);
}
}
}
return super.dispatch(options, handler);
}
}
_HttpAgent_checkAddress = new WeakMap();
//# sourceMappingURL=HttpAgent.js.map