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));
    }
  });
}