-
-
Notifications
You must be signed in to change notification settings - Fork 23
/
index.js
91 lines (77 loc) · 2.14 KB
/
index.js
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 process from 'node:process';
import got from 'got';
const getRateLimit = headers => ({
limit: Number.parseInt(headers['x-ratelimit-limit'], 10),
remaining: Number.parseInt(headers['x-ratelimit-remaining'], 10),
reset: new Date(Number.parseInt(headers['x-ratelimit-reset'], 10) * 1000),
});
const create = () => got.extend({
prefixUrl: process.env.GITHUB_ENDPOINT || 'https://api.github.com',
headers: {
accept: 'application/vnd.github.v3+json',
'user-agent': 'https://github.com/sindresorhus/gh-got',
},
responseType: 'json',
context: {
token: process.env.GITHUB_TOKEN,
},
hooks: {
init: [
(raw, options) => {
// TODO: This should be fixed in Got.
// TODO: This doesn't seem to have any effect.
if (typeof options.url === 'string' && options.url.startsWith('/')) {
options.url = options.url.slice(1);
}
if ('token' in raw) {
options.context.token = raw.token;
delete raw.token;
}
},
],
},
handlers: [
(options, next) => {
// TODO: This should be fixed in Got
// TODO: This doesn't seem to have any effect.
if (typeof options.url === 'string' && options.url.startsWith('/')) {
options.url = options.url.slice(1);
}
// Authorization
const {token} = options.context;
if (token && !options.headers.authorization) {
options.headers.authorization = `token ${token}`;
}
// Don't touch streams
if (options.isStream) {
return next(options);
}
// Magic begins
return (async () => {
try {
const response = await next(options);
// Rate limit for the Response object
response.rateLimit = getRateLimit(response.headers);
return response;
} catch (error) {
const {response} = error;
// Nicer errors
if (response && response.body) {
error.name = 'GitHubError';
error.message = `${response.body.message} (${response.statusCode})`;
}
// Rate limit for errors
if (response) {
error.rateLimit = getRateLimit(response.headers);
}
throw error;
}
})();
},
],
});
const ghGot = create();
export default ghGot;
if (process.env.NODE_ENV === 'test') {
ghGot.recreate = create;
}