blob: 70a7d49f8f6227f5729587d95aa56731f175fda9 (
plain) (
blame)
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
|
import cache from 'webext-storage-cache';
import select from 'select-dom';
import * as pageDetect from 'github-url-detection';
import * as api from './api';
import {getRepo, getCurrentBranch} from '.';
// This regex should match all of these combinations:
// "This branch is even with master."
// "This branch is 1 commit behind master."
// "This branch is 1 commit ahead of master."
// "This branch is 1 commit ahead, 27 commits behind master."
const branchInfoRegex = /([^ ]+)\.$/;
const getDefaultBranch = cache.function(async function (repository?: pageDetect.RepositoryInfo): Promise<string> {
if (arguments.length === 0) {
repository = getRepo();
}
if (!repository) {
throw new Error('getDefaultBranch was called on a non-repository page');
}
if (arguments.length === 0 || JSON.stringify(repository) === JSON.stringify(getRepo())) {
if (pageDetect.isRepoHome()) {
return getCurrentBranch()!;
}
if (!pageDetect.isForkedRepo()) {
// We can find the name in the infobar, available in folder views
const branchInfo = select('.branch-infobar')?.textContent!.trim();
const defaultBranch = branchInfoRegex.exec(branchInfo!)?.[1];
if (defaultBranch) {
return defaultBranch;
}
}
}
const response = await api.v4(`
repository(owner: "${repository.owner}", name: "${repository.name}") {
defaultBranchRef {
name
}
}
`);
return response.repository.defaultBranchRef.name;
}, {
maxAge: {hours: 1},
staleWhileRevalidate: {days: 20},
cacheKey: ([repository = getRepo()]) => `default-branch:${repository?.nameWithOwner!}`
});
export default getDefaultBranch;
|
