Add code highlighting

Author: tmclaugh

src/highlightCode.test.ts

@@ -0,0 +1,33 @@
+import hljs from 'highlight.js'
+import * as cheerio from 'cheerio'
+
+/*
+ * Highlight code blocks in page
+ *
+ * @param $page - Cheerio object
+ */
+export async function highlightCode($page: cheerio.CheerioAPI): Promise<void> {
+    const $elements = $page('pre code')
+    $elements.each((i, $e) => {
+        if(
+            !$e.attribs.class.startsWith('yaml') &&
+            !$e.attribs.class.startsWith('json')
+        ) {
+            return
+        }
+
+        console.log(`Highlighting code block: ${i}`)
+        console.log($page($e).text())
+        // Inserting the hljs class gets us the AWS CSS formatting we want.
+        const text = $page($e).addClass('hljs').text()
+
+        let highlightedCode = ''
+        if ($e.attribs.class.startsWith('json')) {
+            highlightedCode = hljs.highlight(text, { language: 'json' }).value
+        } else if ($e.attribs.class.startsWith('yaml')) {
+            highlightedCode = hljs.highlight(text, { language: 'yaml' }).value
+        }
+
+        $page($e).html(highlightedCode)
+    })
+}