100 lines
2.6 KiB
JavaScript
100 lines
2.6 KiB
JavaScript
(function () {
|
|
|
|
if (typeof self === 'undefined' || !self.Prism || !self.document || !document.createRange) {
|
|
return;
|
|
}
|
|
|
|
Prism.plugins.KeepMarkup = true;
|
|
|
|
Prism.hooks.add('before-highlight', function (env) {
|
|
if (!env.element.children.length) {
|
|
return;
|
|
}
|
|
|
|
var pos = 0;
|
|
var data = [];
|
|
var f = function (elt, baseNode) {
|
|
var o = {};
|
|
if (!baseNode) {
|
|
// Clone the original tag to keep all attributes
|
|
o.clone = elt.cloneNode(false);
|
|
o.posOpen = pos;
|
|
data.push(o);
|
|
}
|
|
for (var i = 0, l = elt.childNodes.length; i < l; i++) {
|
|
var child = elt.childNodes[i];
|
|
if (child.nodeType === 1) { // element
|
|
f(child);
|
|
} else if(child.nodeType === 3) { // text
|
|
pos += child.data.length;
|
|
}
|
|
}
|
|
if (!baseNode) {
|
|
o.posClose = pos;
|
|
}
|
|
};
|
|
f(env.element, true);
|
|
|
|
if (data && data.length) {
|
|
// data is an array of all existing tags
|
|
env.keepMarkup = data;
|
|
}
|
|
});
|
|
|
|
Prism.hooks.add('after-highlight', function (env) {
|
|
if(env.keepMarkup && env.keepMarkup.length) {
|
|
|
|
var walk = function (elt, nodeState) {
|
|
for (var i = 0, l = elt.childNodes.length; i < l; i++) {
|
|
|
|
var child = elt.childNodes[i];
|
|
|
|
if (child.nodeType === 1) { // element
|
|
if (!walk(child, nodeState)) {
|
|
return false;
|
|
}
|
|
|
|
} else if (child.nodeType === 3) { // text
|
|
if(!nodeState.nodeStart && nodeState.pos + child.data.length > nodeState.node.posOpen) {
|
|
// We found the start position
|
|
nodeState.nodeStart = child;
|
|
nodeState.nodeStartPos = nodeState.node.posOpen - nodeState.pos;
|
|
}
|
|
if(nodeState.nodeStart && nodeState.pos + child.data.length >= nodeState.node.posClose) {
|
|
// We found the end position
|
|
nodeState.nodeEnd = child;
|
|
nodeState.nodeEndPos = nodeState.node.posClose - nodeState.pos;
|
|
}
|
|
|
|
nodeState.pos += child.data.length;
|
|
}
|
|
|
|
if (nodeState.nodeStart && nodeState.nodeEnd) {
|
|
// Select the range and wrap it with the clone
|
|
var range = document.createRange();
|
|
range.setStart(nodeState.nodeStart, nodeState.nodeStartPos);
|
|
range.setEnd(nodeState.nodeEnd, nodeState.nodeEndPos);
|
|
nodeState.node.clone.appendChild(range.extractContents());
|
|
range.insertNode(nodeState.node.clone);
|
|
range.detach();
|
|
|
|
// Process is over
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
};
|
|
|
|
// For each tag, we walk the DOM to reinsert it
|
|
env.keepMarkup.forEach(function (node) {
|
|
walk(env.element, {
|
|
node: node,
|
|
pos: 0
|
|
});
|
|
});
|
|
// Store new highlightedCode for later hooks calls
|
|
env.highlightedCode = env.element.innerHTML;
|
|
}
|
|
});
|
|
}());
|