blob: ffb87e23b896668cce97c848de3b2b12931a6174 (
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
|
import cache from 'webext-storage-cache';
import elementReady from 'element-ready';
import {type RepositoryInfo} from 'github-url-detection';
import * as api from './api.js';
import {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 undefined;
}
// 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 branchPicker = await elementReady(branchSelector);
return branchPicker!.textContent!.trim();
}
async function fromAPI(repository: RepositoryInfo): Promise<string> {
const response = await api.v4(`
repository(owner: "${repository.owner}", name: "${repository.name}") {
defaultBranchRef {
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 getDefaultBranchOfRepo = cache.function('default-branch',
async (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 getDefaultBranchOfRepo(getRepo()!);
}
|
