mirror of https://github.com/gorhill/uBlock.git
give specific message handlers priority over default one
This commit is contained in:
parent
2b957dc389
commit
338bf59507
|
@ -19,9 +19,9 @@
|
|||
Home: https://github.com/gorhill/uBlock
|
||||
*/
|
||||
|
||||
// For background page
|
||||
/* global self */
|
||||
|
||||
/* global SafariBrowserTab, Services, XPCOMUtils */
|
||||
// For background page
|
||||
|
||||
/******************************************************************************/
|
||||
|
||||
|
@ -44,142 +44,150 @@ vAPI.storage = chrome.storage.local;
|
|||
|
||||
/******************************************************************************/
|
||||
|
||||
vAPI.tabs = {
|
||||
registerListeners: function() {
|
||||
if (typeof this.onNavigation === 'function') {
|
||||
chrome.webNavigation.onCommitted.addListener(this.onNavigation);
|
||||
vAPI.tabs = {};
|
||||
|
||||
/******************************************************************************/
|
||||
|
||||
vAPI.tabs.registerListeners = function() {
|
||||
if ( typeof this.onNavigation === 'function' ) {
|
||||
chrome.webNavigation.onCommitted.addListener(this.onNavigation);
|
||||
}
|
||||
|
||||
if ( typeof this.onUpdated === 'function' ) {
|
||||
chrome.tabs.onUpdated.addListener(this.onUpdated);
|
||||
}
|
||||
|
||||
if ( typeof this.onClosed === 'function' ) {
|
||||
chrome.tabs.onRemoved.addListener(this.onClosed);
|
||||
}
|
||||
|
||||
if ( typeof this.onPopup === 'function' ) {
|
||||
chrome.webNavigation.onCreatedNavigationTarget.addListener(this.onPopup);
|
||||
}
|
||||
};
|
||||
|
||||
/******************************************************************************/
|
||||
|
||||
vAPI.tabs.get = function(tabId, callback) {
|
||||
if ( tabId !== null ) {
|
||||
chrome.tabs.get(tabId, callback);
|
||||
return;
|
||||
}
|
||||
var onTabReceived = function(tabs) {
|
||||
callback(tabs[0]);
|
||||
};
|
||||
chrome.tabs.query({ active: true, currentWindow: true }, onTabReceived);
|
||||
};
|
||||
|
||||
/******************************************************************************/
|
||||
|
||||
// properties of the details object:
|
||||
// url: 'URL', // the address that will be opened
|
||||
// tabId: 1, // the tab is used if set, instead of creating a new one
|
||||
// index: -1, // undefined: end of the list, -1: following tab, or after index
|
||||
// active: false, // opens the tab in background - true and undefined: foreground
|
||||
// select: true // if a tab is already opened with that url, then select it instead of opening a new one
|
||||
|
||||
vAPI.tabs.open = function(details) {
|
||||
var url = details.url;
|
||||
if ( typeof url !== 'string' || url === '' ) {
|
||||
return null;
|
||||
}
|
||||
// extension pages
|
||||
if ( /^[\w-]{2,}:/.test(url) !== true ) {
|
||||
url = vAPI.getURL(url);
|
||||
}
|
||||
|
||||
// dealing with Chrome's asynchronous API
|
||||
var wrapper = function() {
|
||||
if ( details.active === undefined ) {
|
||||
details.active = true;
|
||||
}
|
||||
|
||||
if (typeof this.onUpdated === 'function') {
|
||||
chrome.tabs.onUpdated.addListener(this.onUpdated);
|
||||
}
|
||||
|
||||
if (typeof this.onClosed === 'function') {
|
||||
chrome.tabs.onRemoved.addListener(this.onClosed);
|
||||
}
|
||||
|
||||
if (typeof this.onPopup === 'function') {
|
||||
chrome.webNavigation.onCreatedNavigationTarget.addListener(this.onPopup);
|
||||
}
|
||||
},
|
||||
|
||||
get: function(tabId, callback) {
|
||||
if (tabId === null) {
|
||||
chrome.tabs.query(
|
||||
{
|
||||
active: true,
|
||||
currentWindow: true
|
||||
},
|
||||
function(tabs) {
|
||||
callback(tabs[0]);
|
||||
}
|
||||
);
|
||||
}
|
||||
else {
|
||||
chrome.tabs.get(tabId, callback);
|
||||
}
|
||||
},
|
||||
/*open: function(details) {
|
||||
// to keep incognito context?
|
||||
chrome.windows.getCurrent(function(win) {
|
||||
details.windowId = win.windowId;
|
||||
chrome.tabs.create(details);
|
||||
});
|
||||
},*/
|
||||
open: function(details) {
|
||||
if (!details.url) {
|
||||
return null;
|
||||
}
|
||||
// extension pages
|
||||
else if (!details.url.match(/^\w{2,20}:/)) {
|
||||
details.url = vAPI.getURL(details.url);
|
||||
}
|
||||
|
||||
// dealing with Chrome's asynhronous API
|
||||
var wrapper = function() {
|
||||
if (details.active === undefined) {
|
||||
details.active = true;
|
||||
}
|
||||
|
||||
var subWrapper = function() {
|
||||
var _details = {
|
||||
url: details.url,
|
||||
active: !!details.active
|
||||
};
|
||||
|
||||
if (details.tabId) {
|
||||
// update doesn't accept index, must use move
|
||||
chrome.tabs.update(details.tabId, _details, function(tab) {
|
||||
// if the tab doesn't exist
|
||||
if ( vAPI.lastError() ) {
|
||||
chrome.tabs.create(_details);
|
||||
} else if ( details.index !== undefined ) {
|
||||
chrome.tabs.move(tab.id, {index: details.index});
|
||||
}
|
||||
});
|
||||
}
|
||||
else {
|
||||
if (details.index !== undefined) {
|
||||
_details.index = details.index;
|
||||
}
|
||||
|
||||
chrome.tabs.create(_details);
|
||||
}
|
||||
var subWrapper = function() {
|
||||
var _details = {
|
||||
url: details.url,
|
||||
active: !!details.active
|
||||
};
|
||||
|
||||
if (details.index === -1) {
|
||||
vAPI.tabs.get(null, function(tab) {
|
||||
if (tab) {
|
||||
details.index = tab.index + 1;
|
||||
if ( details.tabId ) {
|
||||
// update doesn't accept index, must use move
|
||||
chrome.tabs.update(details.tabId, _details, function(tab) {
|
||||
// if the tab doesn't exist
|
||||
if ( vAPI.lastError() ) {
|
||||
chrome.tabs.create(_details);
|
||||
} else if ( details.index !== undefined ) {
|
||||
chrome.tabs.move(tab.id, {index: details.index});
|
||||
}
|
||||
else {
|
||||
delete details.index;
|
||||
}
|
||||
|
||||
subWrapper();
|
||||
});
|
||||
}
|
||||
else {
|
||||
subWrapper();
|
||||
} else {
|
||||
if ( details.index !== undefined ) {
|
||||
_details.index = details.index;
|
||||
}
|
||||
|
||||
chrome.tabs.create(_details);
|
||||
}
|
||||
};
|
||||
|
||||
if (details.select) {
|
||||
// note that currentWindow may be even the window of Developer Tools
|
||||
// so, test with setTimeout...
|
||||
chrome.tabs.query({currentWindow: true}, function(tabs) {
|
||||
var url = details.url.replace(rgxHash, '');
|
||||
// this is questionable
|
||||
var rgxHash = /#.*/;
|
||||
|
||||
tabs = tabs.some(function(tab) {
|
||||
if (tab.url.replace(rgxHash, '') === url) {
|
||||
chrome.tabs.update(tab.id, {active: true});
|
||||
return true;
|
||||
}
|
||||
});
|
||||
|
||||
if (!tabs) {
|
||||
wrapper();
|
||||
if ( details.index === -1 ) {
|
||||
vAPI.tabs.get(null, function(tab) {
|
||||
if ( tab ) {
|
||||
details.index = tab.index + 1;
|
||||
} else {
|
||||
delete details.index;
|
||||
}
|
||||
|
||||
subWrapper();
|
||||
});
|
||||
}
|
||||
else {
|
||||
wrapper();
|
||||
}
|
||||
},
|
||||
close: chrome.tabs.remove.bind(chrome.tabs),
|
||||
injectScript: function(tabId, details, callback) {
|
||||
if (!callback) {
|
||||
callback = function(){};
|
||||
subWrapper();
|
||||
}
|
||||
};
|
||||
|
||||
if (tabId) {
|
||||
chrome.tabs.executeScript(tabId, details, callback);
|
||||
}
|
||||
else {
|
||||
chrome.tabs.executeScript(details, callback);
|
||||
if ( details.select ) {
|
||||
chrome.tabs.query({ currentWindow: true }, function(tabs) {
|
||||
var url = details.url.replace(rgxHash, '');
|
||||
// this is questionable
|
||||
var rgxHash = /#.*/;
|
||||
var selected = tabs.some(function(tab) {
|
||||
if ( tab.url.replace(rgxHash, '') === url ) {
|
||||
chrome.tabs.update(tab.id, { active: true });
|
||||
return true;
|
||||
}
|
||||
});
|
||||
|
||||
if ( selected.length === 0 ) {
|
||||
wrapper();
|
||||
}
|
||||
});
|
||||
}
|
||||
else {
|
||||
wrapper();
|
||||
}
|
||||
};
|
||||
|
||||
/******************************************************************************/
|
||||
|
||||
vAPI.tabs.remove = function(tabId) {
|
||||
var onTabRemoved = function() {
|
||||
if ( vAPI.lastError() ) {
|
||||
}
|
||||
};
|
||||
chrome.tabs.remove(tabId, onTabRemoved);
|
||||
};
|
||||
|
||||
/******************************************************************************/
|
||||
|
||||
vAPI.tabs.injectScript = function(tabId, details, callback) {
|
||||
if ( typeof callback !== 'function' ) {
|
||||
callback = function(){};
|
||||
}
|
||||
|
||||
if ( tabId ) {
|
||||
chrome.tabs.executeScript(tabId, details, callback);
|
||||
} else {
|
||||
chrome.tabs.executeScript(details, callback);
|
||||
}
|
||||
};
|
||||
|
||||
|
@ -197,9 +205,7 @@ vAPI.setIcon = function(tabId, img, badge) {
|
|||
if ( vAPI.lastError() ) {
|
||||
return;
|
||||
}
|
||||
|
||||
chrome.browserAction.setBadgeText({ tabId: tabId, text: badge });
|
||||
|
||||
if ( badge !== '' ) {
|
||||
chrome.browserAction.setBadgeBackgroundColor({ tabId: tabId, color: '#666' });
|
||||
}
|
||||
|
@ -212,71 +218,93 @@ vAPI.setIcon = function(tabId, img, badge) {
|
|||
vAPI.messaging = {
|
||||
ports: {},
|
||||
listeners: {},
|
||||
connector: null,
|
||||
defaultHandler: null,
|
||||
UNHANDLED: 'vAPI.messaging.notHandled'
|
||||
};
|
||||
|
||||
listen: function(listenerName, callback) {
|
||||
this.listeners[listenerName] = callback;
|
||||
},
|
||||
/******************************************************************************/
|
||||
|
||||
setup: function(connector) {
|
||||
if ( this.connector ) {
|
||||
vAPI.messaging.listen = function(listenerName, callback) {
|
||||
this.listeners[listenerName] = callback;
|
||||
};
|
||||
|
||||
/******************************************************************************/
|
||||
|
||||
vAPI.messaging.onConnect = function(port) {
|
||||
var onMessage = function(request) {
|
||||
var callback = function(response) {
|
||||
if ( vAPI.lastError() || response === undefined ) {
|
||||
return;
|
||||
}
|
||||
|
||||
if ( request.requestId ) {
|
||||
port.postMessage({
|
||||
requestId: request.requestId,
|
||||
portName: request.portName,
|
||||
msg: response
|
||||
});
|
||||
}
|
||||
};
|
||||
|
||||
// Specific handler
|
||||
var r;
|
||||
var listener = vAPI.messaging.listeners[request.portName];
|
||||
if ( typeof listener === 'function' ) {
|
||||
r = listener(request.msg, port.sender, callback);
|
||||
}
|
||||
if ( r !== vAPI.messaging.UNHANDLED ) {
|
||||
return;
|
||||
}
|
||||
|
||||
this.connector = function(port) {
|
||||
var onMessage = function(request) {
|
||||
var callback = function(response) {
|
||||
if ( vAPI.lastError() || response === undefined ) {
|
||||
return;
|
||||
}
|
||||
|
||||
if ( request.requestId ) {
|
||||
port.postMessage({
|
||||
requestId: request.requestId,
|
||||
portName: request.portName,
|
||||
msg: response
|
||||
});
|
||||
}
|
||||
};
|
||||
|
||||
// Default handler
|
||||
var listener = connector(request.msg, port.sender, callback);
|
||||
if ( listener !== null ) {
|
||||
return;
|
||||
}
|
||||
|
||||
// Specific handler
|
||||
listener = vAPI.messaging.listeners[request.portName];
|
||||
if ( typeof listener === 'function' ) {
|
||||
listener(request.msg, port.sender, callback);
|
||||
} else {
|
||||
console.error('µBlock> messaging > unknown request: %o', request);
|
||||
}
|
||||
};
|
||||
|
||||
var onDisconnect = function(port) {
|
||||
port.onDisconnect.removeListener(onDisconnect);
|
||||
port.onMessage.removeListener(onMessage);
|
||||
delete vAPI.messaging.ports[port.name];
|
||||
};
|
||||
|
||||
port.onDisconnect.addListener(onDisconnect);
|
||||
port.onMessage.addListener(onMessage);
|
||||
vAPI.messaging.ports[port.name] = port;
|
||||
};
|
||||
|
||||
chrome.runtime.onConnect.addListener(this.connector);
|
||||
},
|
||||
|
||||
broadcast: function(message) {
|
||||
message = {
|
||||
broadcast: true,
|
||||
msg: message
|
||||
};
|
||||
|
||||
for ( var portName in this.ports ) {
|
||||
this.ports[portName].postMessage(message);
|
||||
// Default handler
|
||||
r = vAPI.messaging.defaultHandler(request.msg, port.sender, callback);
|
||||
if ( r !== vAPI.messaging.UNHANDLED ) {
|
||||
return;
|
||||
}
|
||||
|
||||
console.error('µBlock> messaging > unknown request: %o', request);
|
||||
};
|
||||
|
||||
var onDisconnect = function(port) {
|
||||
port.onDisconnect.removeListener(onDisconnect);
|
||||
port.onMessage.removeListener(onMessage);
|
||||
delete vAPI.messaging.ports[port.name];
|
||||
};
|
||||
|
||||
port.onDisconnect.addListener(onDisconnect);
|
||||
port.onMessage.addListener(onMessage);
|
||||
vAPI.messaging.ports[port.name] = port;
|
||||
};
|
||||
|
||||
/******************************************************************************/
|
||||
|
||||
vAPI.messaging.setup = function(defaultHandler) {
|
||||
// Already setup?
|
||||
if ( this.defaultHandler !== null ) {
|
||||
return;
|
||||
}
|
||||
|
||||
if ( typeof defaultHandler !== 'function' ) {
|
||||
defaultHandler = function(){ return null; };
|
||||
};
|
||||
this.defaultHandler = defaultHandler;
|
||||
|
||||
chrome.runtime.onConnect.addListener(this.onConnect);
|
||||
};
|
||||
|
||||
/******************************************************************************/
|
||||
|
||||
vAPI.messaging.broadcast = function(message) {
|
||||
var messageWrapper = {
|
||||
broadcast: true,
|
||||
msg: message
|
||||
};
|
||||
|
||||
for ( var portName in this.ports ) {
|
||||
if ( this.ports.hasOwnProperty(portName) === false ) {
|
||||
continue;
|
||||
}
|
||||
this.ports[portName].postMessage(messageWrapper);
|
||||
}
|
||||
};
|
||||
|
||||
|
|
|
@ -471,7 +471,7 @@ vAPI.messaging = {
|
|||
|
||||
var listener = connector(request.message.msg, sender, callback);
|
||||
|
||||
if (listener === null) {
|
||||
if (listener === vAPI.messaging.UNHANDLED) {
|
||||
listener = vAPI.messaging.listeners[request.message.portName];
|
||||
|
||||
if (typeof listener === 'function') {
|
||||
|
|
|
@ -19,13 +19,15 @@
|
|||
Home: https://github.com/gorhill/uBlock
|
||||
*/
|
||||
|
||||
/* global µBlock, YaMD5 */
|
||||
/* global µBlock, vAPI, YaMD5 */
|
||||
|
||||
'use strict';
|
||||
|
||||
/******************************************************************************/
|
||||
/******************************************************************************/
|
||||
|
||||
// message router and default handler
|
||||
// Message router and default handler
|
||||
|
||||
vAPI.messaging.setup(function(request, sender, callback) {
|
||||
var µb = µBlock;
|
||||
|
||||
|
@ -45,10 +47,6 @@ vAPI.messaging.setup(function(request, sender, callback) {
|
|||
µb.contextMenuClientY = request.clientY;
|
||||
break;
|
||||
|
||||
/*case 'forceReloadTab':
|
||||
µb.forceReload(request.pageURL);
|
||||
break;*/
|
||||
|
||||
case 'getUserSettings':
|
||||
callback(µb.userSettings);
|
||||
break;
|
||||
|
@ -66,12 +64,10 @@ vAPI.messaging.setup(function(request, sender, callback) {
|
|||
break;
|
||||
|
||||
default:
|
||||
// if none of the above, then check the channel listeners
|
||||
return null;
|
||||
return vAPI.messaging.UNHANDLED;
|
||||
}
|
||||
});
|
||||
|
||||
|
||||
/******************************************************************************/
|
||||
/******************************************************************************/
|
||||
|
||||
|
@ -169,6 +165,9 @@ var onMessage = function(request, sender, callback) {
|
|||
response['.'] = getDynamicFilterResults(request.pageHostname);
|
||||
}
|
||||
break;
|
||||
|
||||
default:
|
||||
return vAPI.messaging.UNHANDLED;
|
||||
}
|
||||
|
||||
callback(response);
|
||||
|
@ -214,6 +213,9 @@ var onMessage = function(request, sender, callback) {
|
|||
response = µb.cosmeticFilteringEngine.retrieveDomainSelectors(request);
|
||||
}
|
||||
break;
|
||||
|
||||
default:
|
||||
return vAPI.messaging.UNHANDLED;
|
||||
}
|
||||
|
||||
callback(response);
|
||||
|
@ -340,6 +342,9 @@ var onMessage = function(details, sender, callback) {
|
|||
response = filterRequest(pageStore, details);
|
||||
}
|
||||
break;
|
||||
|
||||
default:
|
||||
return vAPI.messaging.UNHANDLED;
|
||||
}
|
||||
|
||||
callback(response);
|
||||
|
@ -399,6 +404,8 @@ var onMessage = function(request, sender, callback) {
|
|||
µb.appendUserFilters(request.filters);
|
||||
break;
|
||||
|
||||
default:
|
||||
return vAPI.messaging.UNHANDLED;
|
||||
}
|
||||
|
||||
callback(response);
|
||||
|
@ -471,6 +478,8 @@ var onMessage = function(request, sender, callback) {
|
|||
µb.assets.purge(request.path);
|
||||
break;
|
||||
|
||||
default:
|
||||
return vAPI.messaging.UNHANDLED;
|
||||
}
|
||||
|
||||
callback(response);
|
||||
|
@ -512,6 +521,8 @@ var onMessage = function(request, sender, callback) {
|
|||
var response;
|
||||
|
||||
switch ( request.what ) {
|
||||
default:
|
||||
return vAPI.messaging.UNHANDLED;
|
||||
}
|
||||
|
||||
callback(response);
|
||||
|
@ -556,6 +567,8 @@ var onMessage = function(request, sender, callback) {
|
|||
µb.saveWhitelist();
|
||||
break;
|
||||
|
||||
default:
|
||||
return vAPI.messaging.UNHANDLED;
|
||||
}
|
||||
|
||||
callback(response);
|
||||
|
@ -631,6 +644,10 @@ var getPageDetails = function(µb, tabId) {
|
|||
var onMessage = function(request, sender, callback) {
|
||||
// Async
|
||||
switch ( request.what ) {
|
||||
case 'getTabForStats':
|
||||
vAPI.tabs.get(request.tabId, callback);
|
||||
return;
|
||||
|
||||
default:
|
||||
break;
|
||||
}
|
||||
|
@ -647,9 +664,8 @@ var onMessage = function(request, sender, callback) {
|
|||
response = getPageDetails(µb, request.tabId);
|
||||
break;
|
||||
|
||||
case 'getTabForStats':
|
||||
vAPI.tabs.get(request.tabId, callback);
|
||||
return;
|
||||
default:
|
||||
return vAPI.messaging.UNHANDLED;
|
||||
}
|
||||
|
||||
callback(response);
|
||||
|
@ -745,6 +761,9 @@ var onMessage = function(request, sender, callback) {
|
|||
case 'resetUserData':
|
||||
resetUserData();
|
||||
break;
|
||||
|
||||
default:
|
||||
return vAPI.messaging.UNHANDLED;
|
||||
}
|
||||
|
||||
callback(response);
|
||||
|
|
|
@ -81,13 +81,14 @@ vAPI.tabs.onPopup = function(details) {
|
|||
// Blocked
|
||||
|
||||
// Safari blocks before the pop-up opens, so there is no window to remove.
|
||||
if (vAPI.safari) {
|
||||
// TODO: Can't this test be done within the platform-specific code?
|
||||
if ( vAPI.safari ) {
|
||||
return true;
|
||||
}
|
||||
|
||||
// It is a popup, block and remove the tab.
|
||||
µBlock.unbindTabFromPageStats(details.tabId);
|
||||
µBlock.XAL.destroyTab(details.tabId);
|
||||
vAPI.tabs.remove(details.tabId);
|
||||
};
|
||||
|
||||
vAPI.tabs.registerListeners();
|
||||
|
|
|
@ -207,7 +207,7 @@ var onBeforeSendHeaders = function(details) {
|
|||
|
||||
// It is a popup, block and remove the tab.
|
||||
µb.unbindTabFromPageStats(tabId);
|
||||
µb.XAL.destroyTab(tabId);
|
||||
vAPI.tabs.remove(tabId);
|
||||
|
||||
return { 'cancel': true };
|
||||
};
|
||||
|
|
|
@ -695,7 +695,7 @@ if (self.chrome) {
|
|||
|
||||
var listener = connector(request.message.msg, sender, callback);
|
||||
|
||||
if (listener === null) {
|
||||
if (listener === vAPI.messaging.UNHANDLED) {
|
||||
listener = vAPI.messaging.listeners[request.message.portName];
|
||||
|
||||
if (typeof listener === 'function') {
|
||||
|
|
|
@ -62,16 +62,6 @@ exports.restart = function() {
|
|||
|
||||
/******************************************************************************/
|
||||
|
||||
exports.destroyTab = function(tabId) {
|
||||
vAPI.tabs.remove(tabId, function() {
|
||||
// required by chrome API, or else warnings at console (also, mind jshint)
|
||||
if ( chrome.runtime.lastError ) {
|
||||
}
|
||||
});
|
||||
};
|
||||
|
||||
/******************************************************************************/
|
||||
|
||||
return exports;
|
||||
|
||||
/******************************************************************************/
|
||||
|
|
Loading…
Reference in New Issue