diff options
Diffstat (limited to 'common/browser.js')
-rw-r--r-- | common/browser.js | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/common/browser.js b/common/browser.js index 4830774..582d2a4 100644 --- a/common/browser.js +++ b/common/browser.js @@ -3,7 +3,7 @@ * * Function: Export the browser API object. * - * Copyright (C) 2021 Wojtek Kosior + * Copyright (C) 2021, 2022 Wojtek Kosior * * This program is free software: you can redistribute it and/or modify * it under the terms of the CC0 1.0 Universal License as published by @@ -21,6 +21,26 @@ #ELIF CHROMIUM +/* Use promises for sendMessage(). */ +function response_cb(resolve, reject, response) { + if (arguments.length < 3) + reject(chrome.runtime.lastError); + else + resolve(response); +} + +function replacement_sendMessage(old_sendMessage, ...args) { + let callbacks, prom = new Promise((...cbs) => callbacks = cbs); + old_sendMessage(...args, resp => response_cb(...callbacks, resp)); + return prom; +} + +for (const api_object of [chrome.tabs, chrome.runtime].filter(a => a)) { + const old_sendMessage = api_object.sendMessage; + api_object.sendMessage = + (...args) => replacement_sendMessage(old_sendMessage, ...args); +} + #EXPORT chrome AS browser #ENDIF |