diff --git a/dom/webidl/AddonManager.webidl b/dom/webidl/AddonManager.webidl index 845d1558de6c..33099a5c8f03 100644 --- a/dom/webidl/AddonManager.webidl +++ b/dom/webidl/AddonManager.webidl @@ -26,6 +26,7 @@ interface Addon { readonly attribute boolean isActive; Promise uninstall(); + Promise setEnabled(boolean value); }; [ChromeOnly, JSImplementation="dummy"] diff --git a/toolkit/mozapps/extensions/AddonManager.jsm b/toolkit/mozapps/extensions/AddonManager.jsm index 46805233d955..8ab4ee638fed 100644 --- a/toolkit/mozapps/extensions/AddonManager.jsm +++ b/toolkit/mozapps/extensions/AddonManager.jsm @@ -2930,6 +2930,18 @@ var AddonManagerInternal = { }); }, + addonSetEnabled(target, id, value) { + return new Promise((resolve, reject) => { + AddonManager.getAddonByID(id, addon => { + if (!addon) { + reject({message: `No such addon ${id}`}); + } + addon.userDisabled = !value; + resolve(); + }); + }); + }, + addonInstallDoInstall(target, id) { let state = this.installs.get(id); if (!state) { diff --git a/toolkit/mozapps/extensions/amWebAPI.js b/toolkit/mozapps/extensions/amWebAPI.js index d24303728853..a710875e1121 100644 --- a/toolkit/mozapps/extensions/amWebAPI.js +++ b/toolkit/mozapps/extensions/amWebAPI.js @@ -151,6 +151,10 @@ Addon.prototype = { uninstall: WebAPITask(function*() { return yield APIBroker.sendRequest("addonUninstall", this.id); }), + + setEnabled: WebAPITask(function* (value) { + return yield APIBroker.sendRequest("addonSetEnabled", this.id, value); + }), }; const INSTALL_EVENTS = [ diff --git a/toolkit/mozapps/extensions/test/browser/browser.ini b/toolkit/mozapps/extensions/test/browser/browser.ini index 740db1c348d5..da383c1aaf80 100644 --- a/toolkit/mozapps/extensions/test/browser/browser.ini +++ b/toolkit/mozapps/extensions/test/browser/browser.ini @@ -68,6 +68,7 @@ skip-if = require_signing [browser_webapi.js] [browser_webapi_access.js] [browser_webapi_addon_listener.js] +[browser_webapi_enable.js] [browser_webapi_install.js] [browser_webapi_uninstall.js] diff --git a/toolkit/mozapps/extensions/test/browser/browser_webapi_enable.js b/toolkit/mozapps/extensions/test/browser/browser_webapi_enable.js new file mode 100644 index 000000000000..27d8029c55f7 --- /dev/null +++ b/toolkit/mozapps/extensions/test/browser/browser_webapi_enable.js @@ -0,0 +1,62 @@ +const TESTPAGE = `${SECURE_TESTROOT}webapi_addon_listener.html`; + +Services.prefs.setBoolPref("extensions.webapi.testing", true); +registerCleanupFunction(() => { + Services.prefs.clearUserPref("extensions.webapi.testing"); +}); + +function* getListenerEvents(browser) { + let result = yield ContentTask.spawn(browser, null, function*() { + return content.document.getElementById("result").textContent; + }); + + return result.split('\n').map(JSON.parse); +} + +const ID = "test@tests.mozilla.org"; + +let provider = new MockProvider(false); +provider.createAddons([ + { + id: ID, + name: "Test add-on", + operationsRequiringRestart: AddonManager.OP_NEED_RESTART_NONE, + }, +]); + +// Test disable and enable from content +add_task(function* () { + yield BrowserTestUtils.withNewTab(TESTPAGE, function*(browser) { + let addon = yield promiseAddonByID(ID); + isnot(addon, null, "Test addon exists"); + is(addon.userDisabled, false, "addon is enabled"); + + // Disable the addon from content. + yield ContentTask.spawn(browser, null, function* () { + return content.navigator.mozAddonManager + .getAddonByID("test@tests.mozilla.org") + .then(addon => { addon.setEnabled(false); }); + }); + + let events = yield getListenerEvents(browser); + let expected = [ + {id: ID, needsRestart: false, event: "onDisabling"}, + {id: ID, needsRestart: false, event: "onDisabled"}, + ]; + Assert.deepEqual(events, expected, "Got expected disable events"); + + // Enable the addon from content. + yield ContentTask.spawn(browser, null, function* () { + return content.navigator.mozAddonManager + .getAddonByID("test@tests.mozilla.org") + .then(addon => { addon.setEnabled(true); }); + }); + + events = yield getListenerEvents(browser); + expected = expected.concat([ + {id: ID, needsRestart: false, event: "onEnabling"}, + {id: ID, needsRestart: false, event: "onEnabled"}, + ]); + Assert.deepEqual(events, expected, "Got expected enable events"); + }); +});