From 6c5078ec3fdaa50748934b882bc98a6b1c1a2cf4 Mon Sep 17 00:00:00 2001
From: brecert <11599528+Brecert@users.noreply.github.com>
Date: Tue, 9 Jul 2019 21:33:47 -0400
Subject: [PATCH] finished highlighting logic

---
 src/utils/messageFormatter.js | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

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 => `<div class="codeblock"><code>${hljs.highlightAuto(formatCode(unescapeHTML(text)).code).value}</code></div>`
+	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 `<div class="codeblock"><code lang="${highlighted.language}">${highlighted.value}</code></div>`
+	}
 })
 
 futoji.addTransformer({