45 lines
1.0 KiB
JavaScript
45 lines
1.0 KiB
JavaScript
import { censorFn } from "pretty-text/censored-words";
|
|
|
|
function recurse(tokens, apply) {
|
|
let i;
|
|
for (i = 0; i < tokens.length; i++) {
|
|
apply(tokens[i]);
|
|
if (tokens[i].children) {
|
|
recurse(tokens[i].children, apply);
|
|
}
|
|
}
|
|
}
|
|
|
|
function censorTree(state, censor) {
|
|
if (!state.tokens) {
|
|
return;
|
|
}
|
|
|
|
recurse(state.tokens, token => {
|
|
if (token.content) {
|
|
token.content = censor(token.content);
|
|
}
|
|
});
|
|
}
|
|
|
|
export function setup(helper) {
|
|
helper.registerOptions((opts, siteSettings) => {
|
|
opts.watchedWordsRegularExpressions =
|
|
siteSettings.watched_words_regular_expressions;
|
|
});
|
|
|
|
helper.registerPlugin(md => {
|
|
const words = md.options.discourse.censoredWords;
|
|
|
|
if (words && words.length > 0) {
|
|
const replacement = String.fromCharCode(9632);
|
|
const censor = censorFn(
|
|
words,
|
|
replacement,
|
|
md.options.discourse.watchedWordsRegularExpressions
|
|
);
|
|
md.core.ruler.push("censored", state => censorTree(state, censor));
|
|
}
|
|
});
|
|
}
|