feat: 添加highlight调用脚本

2019-11-15 01:35:56 +08:00 · 2019-11-15 01:35:56 +08:00 · 58380da0ae
parent 9628d1b208
commit 58380da0ae
1 changed files with 107 additions and 0 deletions
--- a/wp-heightlight.js
+++ b/wp-heightlight.js
@ -0,0 +1,107 @@
+/*!
+ * WordPress highlight
+ * https://github.com/ineo6/wordpress-highlight
+ */
+
+(function ($) {
+    function initHighlight() {
+        function createBar(i) {
+            var attributes = {
+                'autocomplete': 'off',
+                'autocorrect': 'off',
+                'autocapitalize': 'off',
+                'spellcheck': 'false',
+                'contenteditable': 'false',
+            };
+            var codeCls = $('pre:eq(' + i + ')')[0].children[0].className;
+
+            var result = codeCls.match(/language-(\w+)/);
+
+            var lang = "text";
+
+            if (result && result.length) {
+                lang = result[1];
+            }
+
+            $('pre:eq(' + i + ')').addClass('highlight-wrap');
+            for (var t in attributes) {
+                $('pre:eq(' + i + ')').attr(t, attributes[t]);
+            }
+            $('pre:eq(' + i + ') code').attr('data-rel', lang.toUpperCase()).attr({id: "highlight-code-" + i})
+                .after('<a class="copy-code" data-clipboard-target="#highlight-code-' + i + '" title="拷贝代码">复制</a>');
+        }
+
+        function highlight_code_worker_function() {
+            importScripts('https://cdn.jsdelivr.net/gh/highlightjs/cdn-release@9.16.2/build/highlight.min.js');
+
+            onmessage = function (event) {
+                var data = JSON.parse(event.data);
+
+                var result = self.hljs.highlightAuto(data.code);
+
+                postMessage(JSON.stringify({result: result.value, index: data.index}));
+            };
+        }
+
+        // 使用 web worker
+        function highlightWorker() {
+            var codeBlocks = document.querySelectorAll('pre code');
+
+            if (window.theme_url) {
+                var worker = new Worker(window.theme_url + '/js/highlight-worker.js');
+            } else {
+                return false;
+            }
+
+            worker.onmessage = (event) => {
+                var highlightData = JSON.parse(event.data);
+
+                codeBlocks[highlightData.index].innerHTML = highlightData.result;
+                createBar(highlightData.index);
+
+                new ClipboardJS('.copy-code');
+
+                if (highlightData.index === codeBlocks.length - 1) {
+                    // 结束关闭worker
+                    worker.terminate();
+                    hljs.initLineNumbersOnLoad({
+                        singleLine: true
+                    });
+                }
+            };
+
+            codeBlocks.forEach(function (block, index) {
+                worker.postMessage(JSON.stringify({
+                    code: block.textContent,
+                    index,
+                    total: codeBlocks.length,
+                }));
+            })
+        }
+
+        function highlightNormal() {
+            $('pre code').each(function (i, block) {
+                hljs.highlightBlock(block);
+            });
+
+            for (var i = 0, len = $('pre').length; i < len; i++) {
+                createBar(i);
+            }
+
+            // 代码行数
+            hljs.initLineNumbersOnLoad({
+                singleLine: true
+            });
+        }
+
+        if (typeof (Worker) === undefined) {
+            highlightNormal();
+        } else {
+            highlightWorker();
+        }
+    }
+
+    $(document).ready(function () {
+        initHighlight();
+    });
+})(jQuery);