blob: a5f57875b3f794507ffc9823734179e84cd7d3a6 (
plain)
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
|
/**
* Check if the current block has a defined language.
* @param classList - A list of class.
* @returns {boolean} - True if a class starts with "language-".
*/
const isLanguageBlock = (classList: DOMTokenList) => {
const classes = Array.from(classList);
return classes.some((className) => /language-.*/.test(className));
};
/**
* Add automatically some classes and attributes for PrismJs.
*
* These classes and attributes are needed by Prism or to customize comments.
*/
export const addPrismClasses = () => {
const preTags = document.getElementsByTagName('pre');
Array.from(preTags).forEach((preTag) => {
if (!isLanguageBlock(preTag.classList)) return;
preTag.classList.add('match-braces');
if (preTag.classList.contains('filter-output')) {
preTag.setAttribute('data-filter-output', '#output#');
}
if (preTag.classList.contains('language-bash')) {
preTag.classList.add('command-line');
} else if (!preTag.classList.contains('language-diff')) {
preTag.classList.add('line-numbers');
}
});
};
|
