/
host-rules.ts
91 lines (84 loc) · 2.45 KB
/
host-rules.ts
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
import {
GITHUB_API_USING_HOST_TYPES,
GITLAB_API_USING_HOST_TYPES,
PLATFORM_TYPE_GITHUB,
PLATFORM_TYPE_GITLAB,
} from '../../constants/platforms';
import { logger } from '../../logger';
import { hasProxy } from '../../proxy';
import type { HostRule } from '../../types';
import * as hostRules from '../host-rules';
import type { GotOptions } from './types';
function findMatchingRules(options: GotOptions, url: string): HostRule {
const { hostType } = options;
let res = hostRules.find({ hostType, url });
// Fallback to `github` hostType
if (
GITHUB_API_USING_HOST_TYPES.includes(hostType) &&
hostType !== PLATFORM_TYPE_GITHUB
) {
res = {
...hostRules.find({
hostType: PLATFORM_TYPE_GITHUB,
url,
}),
...res,
};
}
// Fallback to `gitlab` hostType
if (
GITLAB_API_USING_HOST_TYPES.includes(hostType) &&
hostType !== PLATFORM_TYPE_GITLAB
) {
res = {
...hostRules.find({
hostType: PLATFORM_TYPE_GITLAB,
url,
}),
...res,
};
}
return res;
}
// Apply host rules to requests
export function applyHostRules(url: string, inOptions: GotOptions): GotOptions {
const options: GotOptions = { ...inOptions };
const foundRules = findMatchingRules(options, url);
const { username, password, token, enabled, authType } = foundRules;
if (options.noAuth) {
logger.trace({ url }, `Authorization disabled`);
} else if (
options.headers?.authorization ||
options.password ||
options.token
) {
logger.trace({ url }, `Authorization already set`);
} else if (password !== undefined) {
logger.trace({ url }, `Applying Basic authentication`);
options.username = username;
options.password = password;
} else if (token) {
logger.trace({ url }, `Applying Bearer authentication`);
options.token = token;
options.context = { ...options.context, authType };
} else if (enabled === false) {
options.enabled = false;
}
// Apply optional params
['abortOnError', 'abortIgnoreStatusCodes', 'timeout'].forEach((param) => {
if (foundRules[param]) {
options[param] = foundRules[param];
}
});
if (!hasProxy() && foundRules.enableHttp2 === true) {
options.http2 = true;
}
return options;
}
export function getRequestLimit(url: string): number | null {
const hostRule = hostRules.find({
url,
});
const limit = hostRule.concurrentRequestLimit;
return typeof limit === 'number' && limit > 0 ? limit : null;
}