blob: 4855c465ab9de6c88b89ce32a8f4689e29fbf622 (
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
55
56
57
58
59
60
61
62
63
64
65
|
import cache from 'webext-storage-cache';
import select from 'select-dom';
import elementReady from 'element-ready';
import * as pageDetect from 'github-url-detection';
import * as api from './api';
import {getRepo, getCurrentBranchFromFeed} 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 = /([^ ]+)\.$/;
// DO NOT use optional arguments/defaults in "cached functions" because they can't be memoized effectively
// https://github.com/sindresorhus/eslint-plugin-unicorn/issues/1864
const _getDefaultBranch = cache.function(async function (repository: pageDetect.RepositoryInfo): Promise<string> {
if (arguments.length === 0 || JSON.stringify(repository) === JSON.stringify(getRepo())) {
if (pageDetect.isRepoHome()) {
const branchSelector = await elementReady('[data-hotkey="w"]');
if (branchSelector) {
return branchSelector.title === 'Switch branches or tags'
? branchSelector.textContent!.trim()
: branchSelector.title;
}
}
const defaultBranch = getCurrentBranchFromFeed();
if (defaultBranch) {
return defaultBranch;
}
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]) => 'default-branch:' + repository.nameWithOwner,
});
export default async function getDefaultBranch(repository: pageDetect.RepositoryInfo | undefined = getRepo()): Promise<string> {
if (!repository) {
throw new Error('getDefaultBranch was called on a non-repository page');
}
return _getDefaultBranch(repository);
}
|
