backup/pywb
1
0
Fork 0
mirror of https://github.com/webrecorder/pywb.git synced 2025-03-24 06:59:52 +01:00
Code Issues Projects Releases Activity

wombat: ensure document.write override handles elements that go into

Browse Source 
head as well as body
This commit is contained in:
Ilya Kreymer 2015-01-27 18:02:14 -08:00
parent 59630c08f6
commit 976decb3f1
1 changed files with 20 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
pywb/static/wombat.js
View File

@ -178,7 +178,7 @@ _WBWombat = (function() {
// if already rewriting url, must still check scheme // if already rewriting url, must still check scheme
if (starts_with(url, prefix + window.location.host + '/')) { if (starts_with(url, prefix + window.location.host + '/')) {
var curr_scheme = window.location.protocol + '//'; var curr_scheme = window.location.protocol + '//';
// replace scheme to ensure using the correct server scheme // replace scheme to ensure using the correct server scheme
if (starts_with(url, wb_orig_scheme) && (wb_orig_scheme != curr_scheme)) { if (starts_with(url, wb_orig_scheme) && (wb_orig_scheme != curr_scheme)) {
url = curr_scheme + url.substring(wb_orig_scheme.length); url = curr_scheme + url.substring(wb_orig_scheme.length);
@ -751,7 +751,6 @@ _WBWombat = (function() {
if (targetOrigin && targetOrigin != "*") { if (targetOrigin && targetOrigin != "*") {
targetOrigin = window.location.origin; targetOrigin = window.location.origin;
} }
return orig.call(this, message, targetOrigin, transfer); return orig.call(this, message, targetOrigin, transfer);
} }
@ -850,14 +849,30 @@ _WBWombat = (function() {
//============================================ //============================================
function init_write_override() function init_write_override()
{ {
var orig_doc_write = document.write;
document.write = function(string) { document.write = function(string) {
var doc = new DOMParser().parseFromString(string, "text/html"); var doc = new DOMParser().parseFromString(string, "text/html");
if (doc) { if (doc) {
var children = doc.body.children;
for (var i = 0; i < children.length; i++) { if (doc.head && document.head) {
document.body.appendChild(children[i]); var children = doc.head.children;
for (var i = 0; i < children.length; i++) {
//document.head.appendChild(children[i]);
// in head, must call original write to ensure same execution order..
rewrite_elem(children[i]);
orig_doc_write.call(this, children[i].outerHTML);
}
}
if (doc.body && document.body) {
var children = doc.body.children;
for (var i = 0; i < children.length; i++) {
document.body.appendChild(children[i]);
}
} }
} }
} }