alexveley
82 lines
2.3 KiB
JavaScript
82 lines
2.3 KiB
JavaScript
"use strict";
|
|
|
|
const HOST_NAME = "com.easygoingaming.firefox_agent_bridge";
|
|
const RECONNECT_MS = 2000;
|
|
|
|
const ALLOWED = {
|
|
"meta.ping": async () => ({ ok: true, extension: "firefox-agent-bridge" }),
|
|
"meta.methods": async () => Object.keys(ALLOWED).sort(),
|
|
"bookmarks.getTree": () => browser.bookmarks.getTree(),
|
|
"bookmarks.getSubTree": (id) => browser.bookmarks.getSubTree(id),
|
|
"bookmarks.getChildren": (id) => browser.bookmarks.getChildren(id),
|
|
"bookmarks.get": (idOrIds) => browser.bookmarks.get(idOrIds),
|
|
"bookmarks.search": (query) => browser.bookmarks.search(query),
|
|
"bookmarks.create": (details) => browser.bookmarks.create(details),
|
|
"bookmarks.update": (id, changes) => browser.bookmarks.update(id, changes),
|
|
"bookmarks.move": (id, destination) => browser.bookmarks.move(id, destination),
|
|
"bookmarks.remove": (id) => browser.bookmarks.remove(id),
|
|
"bookmarks.removeTree": (id) => browser.bookmarks.removeTree(id),
|
|
"bookmarks.getRecent": (numberOfItems) => browser.bookmarks.getRecent(numberOfItems),
|
|
};
|
|
|
|
let port = null;
|
|
let reconnectTimer = null;
|
|
|
|
function asArray(value) {
|
|
if (value === undefined || value === null) {
|
|
return [];
|
|
}
|
|
return Array.isArray(value) ? value : [value];
|
|
}
|
|
|
|
async function dispatch(message) {
|
|
const method = message && message.method;
|
|
const fn = ALLOWED[method];
|
|
if (!fn) {
|
|
throw new Error(`method not allowed: ${method || "(missing)"}`);
|
|
}
|
|
return fn(...asArray(message.args));
|
|
}
|
|
|
|
function attach(nextPort) {
|
|
port = nextPort;
|
|
port.onMessage.addListener(async (message) => {
|
|
const id = message && message.id;
|
|
try {
|
|
const result = await dispatch(message);
|
|
port.postMessage({ id, ok: true, result });
|
|
} catch (err) {
|
|
port.postMessage({
|
|
id,
|
|
ok: false,
|
|
error: err && err.message ? err.message : String(err),
|
|
});
|
|
}
|
|
});
|
|
port.onDisconnect.addListener(() => {
|
|
port = null;
|
|
scheduleReconnect();
|
|
});
|
|
}
|
|
|
|
function scheduleReconnect() {
|
|
if (reconnectTimer) {
|
|
return;
|
|
}
|
|
reconnectTimer = setTimeout(() => {
|
|
reconnectTimer = null;
|
|
connect();
|
|
}, RECONNECT_MS);
|
|
}
|
|
|
|
function connect() {
|
|
try {
|
|
attach(browser.runtime.connectNative(HOST_NAME));
|
|
} catch (err) {
|
|
console.error("native host connect failed", err);
|
|
scheduleReconnect();
|
|
}
|
|
}
|
|
|
|
connect();
|