forked from mirrors/gecko-dev
f386f9df93
MozReview-Commit-ID: 7lQp9hL9pNd --HG-- extra : rebase_source : 8e61f79fe131111e91cb5853354a0dcfabf5ccbf
72 lines
2.6 KiB
JavaScript
72 lines
2.6 KiB
JavaScript
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
"use strict";
|
|
|
|
XPCOMUtils.defineLazyModuleGetter(this, "ExtensionSettingsStore",
|
|
"resource://gre/modules/ExtensionSettingsStore.jsm");
|
|
|
|
XPCOMUtils.defineLazyServiceGetter(this, "aboutNewTabService",
|
|
"@mozilla.org/browser/aboutnewtab-service;1",
|
|
"nsIAboutNewTabService");
|
|
|
|
const STORE_TYPE = "url_overrides";
|
|
const NEW_TAB_SETTING_NAME = "newTabURL";
|
|
|
|
this.urlOverrides = class extends ExtensionAPI {
|
|
processNewTabSetting(action) {
|
|
let {extension} = this;
|
|
let item = ExtensionSettingsStore[action](extension, STORE_TYPE, NEW_TAB_SETTING_NAME);
|
|
if (item) {
|
|
aboutNewTabService.newTabURL = item.value || item.initialValue;
|
|
}
|
|
}
|
|
|
|
async onManifestEntry(entryName) {
|
|
let {extension} = this;
|
|
let {manifest} = extension;
|
|
|
|
await ExtensionSettingsStore.initialize();
|
|
|
|
if (manifest.chrome_url_overrides.newtab) {
|
|
// Set up the shutdown code for the setting.
|
|
extension.callOnClose({
|
|
close: () => {
|
|
switch (extension.shutdownReason) {
|
|
case "ADDON_DISABLE":
|
|
this.processNewTabSetting("disable");
|
|
break;
|
|
|
|
// We can remove the setting on upgrade or downgrade because it will be
|
|
// added back in when the manifest is re-read. This will cover the case
|
|
// where a new version of an add-on removes the manifest key.
|
|
case "ADDON_DOWNGRADE":
|
|
case "ADDON_UPGRADE":
|
|
case "ADDON_UNINSTALL":
|
|
this.processNewTabSetting("removeSetting");
|
|
break;
|
|
}
|
|
},
|
|
});
|
|
|
|
let url = extension.baseURI.resolve(manifest.chrome_url_overrides.newtab);
|
|
|
|
let item = await ExtensionSettingsStore.addSetting(
|
|
extension, STORE_TYPE, NEW_TAB_SETTING_NAME, url,
|
|
() => aboutNewTabService.newTabURL);
|
|
|
|
// If the extension was just re-enabled, change the setting to enabled.
|
|
// This is required because addSetting above is used for both add and update.
|
|
if (["ADDON_ENABLE", "ADDON_UPGRADE", "ADDON_DOWNGRADE"]
|
|
.includes(extension.startupReason)) {
|
|
item = ExtensionSettingsStore.enable(extension, STORE_TYPE, NEW_TAB_SETTING_NAME);
|
|
}
|
|
|
|
// Set the newTabURL to the current value of the setting.
|
|
if (item) {
|
|
aboutNewTabService.newTabURL = item.value || item.initialValue;
|
|
}
|
|
}
|
|
}
|
|
};
|