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);
| }
| };
| };
|
|
