1 | (function () {
|
2 |
|
3 | if (typeof self === 'undefined' || !self.Prism || !self.document || !document.createRange) {
|
4 | return;
|
5 | }
|
6 |
|
7 | Prism.plugins.KeepMarkup = true;
|
8 |
|
9 | Prism.hooks.add('before-highlight', function (env) {
|
10 | if (!env.element.children.length) {
|
11 | return;
|
12 | }
|
13 |
|
14 | var pos = 0;
|
15 | var data = [];
|
16 | var f = function (elt, baseNode) {
|
17 | var o = {};
|
18 | if (!baseNode) {
|
19 |
|
20 | o.clone = elt.cloneNode(false);
|
21 | o.posOpen = pos;
|
22 | data.push(o);
|
23 | }
|
24 | for (var i = 0, l = elt.childNodes.length; i < l; i++) {
|
25 | var child = elt.childNodes[i];
|
26 | if (child.nodeType === 1) {
|
27 | f(child);
|
28 | } else if(child.nodeType === 3) {
|
29 | pos += child.data.length;
|
30 | }
|
31 | }
|
32 | if (!baseNode) {
|
33 | o.posClose = pos;
|
34 | }
|
35 | };
|
36 | f(env.element, true);
|
37 |
|
38 | if (data && data.length) {
|
39 |
|
40 | env.keepMarkup = data;
|
41 | }
|
42 | });
|
43 |
|
44 | Prism.hooks.add('after-highlight', function (env) {
|
45 | if(env.keepMarkup && env.keepMarkup.length) {
|
46 |
|
47 | var walk = function (elt, nodeState) {
|
48 | for (var i = 0, l = elt.childNodes.length; i < l; i++) {
|
49 |
|
50 | var child = elt.childNodes[i];
|
51 |
|
52 | if (child.nodeType === 1) {
|
53 | if (!walk(child, nodeState)) {
|
54 | return false;
|
55 | }
|
56 |
|
57 | } else if (child.nodeType === 3) {
|
58 | if(!nodeState.nodeStart && nodeState.pos + child.data.length > nodeState.node.posOpen) {
|
59 |
|
60 | nodeState.nodeStart = child;
|
61 | nodeState.nodeStartPos = nodeState.node.posOpen - nodeState.pos;
|
62 | }
|
63 | if(nodeState.nodeStart && nodeState.pos + child.data.length >= nodeState.node.posClose) {
|
64 |
|
65 | nodeState.nodeEnd = child;
|
66 | nodeState.nodeEndPos = nodeState.node.posClose - nodeState.pos;
|
67 | }
|
68 |
|
69 | nodeState.pos += child.data.length;
|
70 | }
|
71 |
|
72 | if (nodeState.nodeStart && nodeState.nodeEnd) {
|
73 |
|
74 | var range = document.createRange();
|
75 | range.setStart(nodeState.nodeStart, nodeState.nodeStartPos);
|
76 | range.setEnd(nodeState.nodeEnd, nodeState.nodeEndPos);
|
77 | nodeState.node.clone.appendChild(range.extractContents());
|
78 | range.insertNode(nodeState.node.clone);
|
79 | range.detach();
|
80 |
|
81 |
|
82 | return false;
|
83 | }
|
84 | }
|
85 | return true;
|
86 | };
|
87 |
|
88 |
|
89 | env.keepMarkup.forEach(function (node) {
|
90 | walk(env.element, {
|
91 | node: node,
|
92 | pos: 0
|
93 | });
|
94 | });
|
95 |
|
96 | env.highlightedCode = env.element.innerHTML;
|
97 | }
|
98 | });
|
99 | }());
|