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
|
import {CachedFunction} from 'webext-storage-cache';
import elementReady from 'element-ready';
import {type RepositoryInfo} from 'github-url-detection';
import api from './api.js';
import {extractCurrentBranchFromBranchPicker, getRepo} from './index.js';
import {branchSelector} from './selectors.js';
const isCurrentRepo = ({nameWithOwner}: RepositoryInfo): boolean => Boolean(getRepo()?.nameWithOwner === nameWithOwner);
// Do not make this function complicated. We're only optimizing for the repo root.
async function fromDOM(): Promise<string | undefined> {
if (!['', 'commits'].includes(getRepo()!.path)) {
return;
}
// We're on the default branch, so we can extract it from the current page. This exclusively happens on the exact pages:
// /user/repo
// /user/repo/commits (without further path)
const element = await elementReady(branchSelector);
if (!element) {
return;
}
return extractCurrentBranchFromBranchPicker(element);
}
async function fromAPI(repository: RepositoryInfo): Promise<string> {
const response = await api.v4(`
query getDefaultBranch($owner: String!, $name: String!) {
repository(owner: $owner, name: $name) {
defaultBranchRef {
name
}
}
}
`, {
variables: {
owner: repository.owner,
name: repository.name,
},
});
return response.repository.defaultBranchRef.name;
}
// 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
export const defaultBranchOfRepo = new CachedFunction('default-branch', {
async updater(repository: RepositoryInfo): Promise<string> {
if (!repository) {
throw new Error('getDefaultBranch was called on a non-repository page');
}
// eslint-disable-next-line @typescript-eslint/prefer-nullish-coalescing -- Wrong, the type is `false | undefined`
return (isCurrentRepo(repository) && await fromDOM()) || fromAPI(repository);
},
maxAge: {hours: 1},
staleWhileRevalidate: {days: 20},
cacheKey: ([repository]) => repository.nameWithOwner,
},
);
export default async function getDefaultBranch(): Promise<string> {
return defaultBranchOfRepo.get(getRepo()!);
}
|
