1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
// @flow
import http from 'http';
import https from 'https';
type AgentType = http.Agent | https.Agent;
// eslint-disable-next-line flowtype/no-weak-types
export default (originalMethod: Function, agent: AgentType, forceGlobalAgent: boolean) => {
// eslint-disable-next-line unicorn/prevent-abbreviations
return (...args: *) => {
let url;
let options;
let callback;
if (typeof args[0] === 'string' || args[0] instanceof URL) {
url = args[0];
if (typeof args[1] === 'function') {
options = {};
callback = args[1];
} else {
options = {
...args[1],
};
callback = args[2];
}
} else {
options = {
...args[0],
};
callback = args[1];
}
if (forceGlobalAgent) {
options.agent = agent;
} else {
if (!options.agent) {
options.agent = agent;
}
if (options.agent === http.globalAgent || options.agent === https.globalAgent) {
options.agent = agent;
}
}
if (url) {
// $FlowFixMe
return originalMethod(url, options, callback);
} else {
return originalMethod(options, callback);
}
};
};