X-Git-Url: http://source.bookstackapp.com/bookstack/blobdiff_plain/b8c16b15a9f945b72d2ca4fe0c0172ba422199bc..refs/pull/4525/head:/resources/js/components/code-highlighter.js

diff --git a/resources/js/components/code-highlighter.js b/resources/js/components/code-highlighter.js
index db6112887..e12d77044 100644
--- a/resources/js/components/code-highlighter.js
+++ b/resources/js/components/code-highlighter.js
@@ -1,10 +1,16 @@
-import Code from "../services/code"
-class CodeHighlighter {
+import {Component} from './component';
 
-    constructor(elem) {
-        Code.highlightWithin(elem);
+export class CodeHighlighter extends Component {
+
+    setup() {
+        const container = this.$el;
+
+        const codeBlocks = container.querySelectorAll('pre');
+        if (codeBlocks.length > 0) {
+            window.importVersioned('code').then(Code => {
+                Code.highlightWithin(container);
+            });
+        }
     }
 
 }
-
-export default CodeHighlighter;
\ No newline at end of file