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