d2d9626164
This is imported from https://github.com/mozilla-services/screenshots/. It has been reviewed as patches landed, but also reviewed by Mossop and kmag. This also includes the patch from bug 1356394 MozReview-Commit-ID: FXIVw7WjxlN
81 lines
2.5 KiB
JavaScript
81 lines
2.5 KiB
JavaScript
/* globals browser, catcher, log */
|
|
|
|
"use strict";
|
|
|
|
this.communication = (function () {
|
|
let exports = {};
|
|
|
|
let registeredFunctions = {};
|
|
|
|
browser.runtime.onMessage.addListener(catcher.watchFunction((req, sender, sendResponse) => {
|
|
if (! (req.funcName in registeredFunctions)) {
|
|
log.error(`Received unknown internal message type ${req.funcName}`);
|
|
sendResponse({type: "error", name: "Unknown message type"});
|
|
return;
|
|
}
|
|
if (! Array.isArray(req.args)) {
|
|
log.error("Received message with no .args list");
|
|
sendResponse({type: "error", name: "No .args"});
|
|
return;
|
|
}
|
|
let func = registeredFunctions[req.funcName];
|
|
let result;
|
|
try {
|
|
req.args.unshift(sender);
|
|
result = func.apply(null, req.args);
|
|
} catch (e) {
|
|
log.error(`Error in ${req.funcName}:`, e, e.stack);
|
|
// FIXME: should consider using makeError from catcher here:
|
|
sendResponse({type: "error", message: e+""});
|
|
return;
|
|
}
|
|
if (result && result.then) {
|
|
result.then((concreteResult) => {
|
|
sendResponse({type: "success", value: concreteResult});
|
|
}).catch((errorResult) => {
|
|
log.error(`Promise error in ${req.funcName}:`, errorResult, errorResult && errorResult.stack);
|
|
sendResponse({type: "error", message: errorResult+""});
|
|
});
|
|
return true;
|
|
} else {
|
|
sendResponse({type: "success", value: result});
|
|
}
|
|
}));
|
|
|
|
exports.register = function (name, func) {
|
|
registeredFunctions[name] = func;
|
|
};
|
|
|
|
/** Send a message to bootstrap.js
|
|
Technically any worker can listen to this. If the bootstrap wrapper is not in place, then this
|
|
will *not* fail, and will return a value of exports.NO_BOOTSTRAP */
|
|
exports.sendToBootstrap = function (funcName, ...args) {
|
|
return browser.runtime.sendMessage({funcName, args}).then((result) => {
|
|
if (result.type === "success") {
|
|
return result.value;
|
|
} else {
|
|
throw new Error(`Error in ${funcName}: ${result.name || 'unknown'}`);
|
|
}
|
|
}, (error) => {
|
|
if (isBootstrapMissingError(error)) {
|
|
return exports.NO_BOOTSTRAP;
|
|
}
|
|
throw error;
|
|
});
|
|
};
|
|
|
|
function isBootstrapMissingError(error) {
|
|
if (! error) {
|
|
return false;
|
|
}
|
|
return error.errorCode === "NO_RECEIVING_END" ||
|
|
(! error.errorCode && error.message === "Could not establish connection. Receiving end does not exist.");
|
|
}
|
|
|
|
|
|
// A singleton/sentinal (with a name):
|
|
exports.NO_BOOTSTRAP = {name: "communication.NO_BOOTSTRAP"};
|
|
|
|
return exports;
|
|
})();
|