diff --git a/src/utils/messageFormatter.js b/src/utils/messageFormatter.js index f0c7376..f44a7a5 100644 --- a/src/utils/messageFormatter.js +++ b/src/utils/messageFormatter.js @@ -84,7 +84,18 @@ futoji.addTransformer({ name: 'code-block', symbol: '```', recursive: false, - transformer: text => `
${hljs.highlightAuto(formatCode(unescapeHTML(text)).code).value}
` + transformer: text => { + let formatted = formatCode(unescapeHTML(text)) + + let highlighted + if(formatted.lang in hljs.listLanguages()) { + highlighted = hljs.highlight(formatted.lang, formatted.code, true) + } else { + highlighted = hljs.highlightAuto(formatted.code) + } + + return `
${highlighted.value}
` + } }) futoji.addTransformer({