From 8da5ae70e32ca3f039befff4a45cf99df1f2efa9 Mon Sep 17 00:00:00 2001
From: Cory LaViska <cory@abeautifulsite.net>
Date: Tue, 6 Oct 2020 08:17:33 -0400
Subject: [PATCH] Add module support to code blocks

---
 docs/assets/plugins/code-block/code-block.js | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/docs/assets/plugins/code-block/code-block.js b/docs/assets/plugins/code-block/code-block.js
index dad9e3be8..b6ed38d5c 100644
--- a/docs/assets/plugins/code-block/code-block.js
+++ b/docs/assets/plugins/code-block/code-block.js
@@ -7,13 +7,14 @@
 
   function runScript(script) {
     const newScript = document.createElement('script');
-    newScript.appendChild(
-      document.createTextNode(`
-      (() => {
-        ${script.innerHTML}
-      })();
-    `)
-    );
+
+    if (script.type === 'module') {
+      newScript.type = 'module';
+      newScript.textContent = script.innerHTML;
+    } else {
+      newScript.appendChild(document.createTextNode(`(() => { ${script.innerHTML} })();`));
+    }
+
     script.parentNode.replaceChild(newScript, script);
   }