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

wombat fixes: message listener fixes for multiple listeners

Browse Source 
- don't reject multiple listeners
- create new WrappedListener() obj for each listener
- extract_orig() add current scheme if url starts with '//'
This commit is contained in:
Ilya Kreymer 2017-03-15 11:14:04 -07:00
parent 93f26452e5
commit 1344907032
1 changed files with 48 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
pywb/static/wombat.js
View File

@ -18,7 +18,7 @@ This file is part of pywb, https://github.com/ikreymer/pywb
*/ */
//============================================ //============================================
// Wombat JS-Rewriting Library v2.21 // Wombat JS-Rewriting Library v2.25
//============================================ //============================================
@ -359,6 +359,10 @@ var _WBWombat = function($wbwindow, wbinfo) {
href = href.substr(wb_orig_origin.length); href = href.substr(wb_orig_origin.length);
} }
if (starts_with(href, REL_PREFIX)) {
href = wb_info.wombat_scheme + href;
}
return href; return href;
} }
@ -896,7 +900,7 @@ var _WBWombat = function($wbwindow, wbinfo) {
if (orig_getter) { if (orig_getter) {
var new_getter = function() { var new_getter = function() {
var res = orig_getter.call(this); var res = orig_getter.call(this);
return extract_orig(res, true); return extract_orig(res);
} }
def_prop(proto, prop, undefined, new_getter); def_prop(proto, prop, undefined, new_getter);
@ -1949,17 +1953,9 @@ var _WBWombat = function($wbwindow, wbinfo) {
$wbwindow.Window.prototype.postMessage = postmessage_rewritten; $wbwindow.Window.prototype.postMessage = postmessage_rewritten;
/* function WrappedListener(orig_listener, win) {
for (var i = 0; i < $wbwindow.frames.length; i++) {
try { function listen(event) {
//init_postmessage_override($wbwindow.frames[i]);
$wbwindow.frames[i].postMessage = postmessage_rewritten;
} catch (e) {
console.log(e);
}
}
*/
function WrappedListener(event, orig_listener, win) {
var ne = event; var ne = event;
if (event.data.from && event.data.message) { if (event.data.from && event.data.message) {
@ -1996,6 +1992,9 @@ var _WBWombat = function($wbwindow, wbinfo) {
return orig_listener(ne); return orig_listener(ne);
} }
return {"listen": listen};
}
var listen_map = {}; var listen_map = {};
// ADD // ADD
@ -2006,18 +2005,17 @@ var _WBWombat = function($wbwindow, wbinfo) {
var addEventListener_rewritten = function(type, listener, useCapture) { var addEventListener_rewritten = function(type, listener, useCapture) {
if (type == "message") { if (type == "message") {
var win = this; var wrapped_listener = new WrappedListener(listener, this);
var wrapped_listener = function(event) { return WrappedListener(event, listener, win); }
if (listen_map[listener]) { //if (listen_map[listener]) {
//console.log("Listener Already Added"); //console.log("Listener Already Added");
//_orig_removeEventListener.call(this, type, listen_map[listener], useCapture); //_orig_removeEventListener.call(this, type, listen_map[listener], useCapture);
return; //return;
} //}
listen_map[listener] = wrapped_listener; listen_map[listener] = wrapped_listener;
return _orig_addEventListener.call(this, type, wrapped_listener, useCapture); return _orig_addEventListener.call(this, type, wrapped_listener.listen, useCapture);
} else { } else {
return _orig_addEventListener.call(this, type, listener, useCapture); return _orig_addEventListener.call(this, type, listener, useCapture);
} }
@ -2029,10 +2027,10 @@ var _WBWombat = function($wbwindow, wbinfo) {
var removeEventListener_rewritten = function(type, listener, useCapture) { var removeEventListener_rewritten = function(type, listener, useCapture) {
if (type == "message") { if (type == "message") {
var mapped = listen_map[listener]; var wrapped_listener = listen_map[listener];
if (mapped) { if (wrapped_listener) {
delete listen_map[listener]; delete listen_map[listener];
return _orig_removeEventListener.call(this, type, mapped, useCapture); return _orig_removeEventListener.call(this, type, wrapped_listener.listen, useCapture);
} }
} else { } else {
return _orig_removeEventListener.call(this, type, listener, useCapture); return _orig_removeEventListener.call(this, type, listener, useCapture);