gecko/mobile/components/AddonUpdateService.js
2011-04-27 11:52:27 -04:00

217 lines
7.3 KiB
JavaScript

/* ***** BEGIN LICENSE BLOCK *****
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
*
* The contents of this file are subject to the Mozilla Public License Version
* 1.1 (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
* http://www.mozilla.org/MPL/
*
* Software distributed under the License is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
* for the specific language governing rights and limitations under the
* License.
*
* The Original Code is Add-on Update Service.
*
* The Initial Developer of the Original Code is Mozilla Foundation.
* Portions created by the Initial Developer are Copyright (C) 2009
* the Initial Developer. All Rights Reserved.
*
* Contributor(s):
* Mark Finkle <mfinkle@mozilla.com>
*
* Alternatively, the contents of this file may be used under the terms of
* either the GNU General Public License Version 2 or later (the "GPL"), or
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
* in which case the provisions of the GPL or the LGPL are applicable instead
* of those above. If you wish to allow use of your version of this file only
* under the terms of either the GPL or the LGPL, and not to allow others to
* use your version of this file under the terms of the MPL, indicate your
* decision by deleting the provisions above and replace them with the notice
* and other provisions required by the GPL or the LGPL. If you do not delete
* the provisions above, a recipient may use your version of this file under
* the terms of any one of the MPL, the GPL or the LGPL.
*
* ***** END LICENSE BLOCK ***** */
const Cc = Components.classes;
const Ci = Components.interfaces;
const Cu = Components.utils;
Cu.import("resource://gre/modules/XPCOMUtils.jsm");
Cu.import("resource://gre/modules/Services.jsm");
XPCOMUtils.defineLazyGetter(this, "AddonManager", function() {
Components.utils.import("resource://gre/modules/AddonManager.jsm");
return AddonManager;
});
XPCOMUtils.defineLazyGetter(this, "AddonRepository", function() {
Components.utils.import("resource://gre/modules/AddonRepository.jsm");
return AddonRepository;
});
XPCOMUtils.defineLazyGetter(this, "NetUtil", function() {
Components.utils.import("resource://gre/modules/NetUtil.jsm");
return NetUtil;
});
function getPref(func, preference, defaultValue) {
try {
return Services.prefs[func](preference);
}
catch (e) {}
return defaultValue;
}
// -----------------------------------------------------------------------
// Add-on auto-update management service
// -----------------------------------------------------------------------
const PREF_ADDON_UPDATE_ENABLED = "extensions.autoupdate.enabled";
var gNeedsRestart = false;
function AddonUpdateService() {}
AddonUpdateService.prototype = {
classDescription: "Add-on auto-update management",
classID: Components.ID("{93c8824c-9b87-45ae-bc90-5b82a1e4d877}"),
QueryInterface: XPCOMUtils.generateQI([Ci.nsITimerCallback]),
notify: function aus_notify(aTimer) {
if (aTimer && !getPref("getBoolPref", PREF_ADDON_UPDATE_ENABLED, true))
return;
// If we already auto-upgraded and installed new versions, ignore this check
if (gNeedsRestart)
return;
Services.io.offline = false;
// Assume we are doing a periodic update check
let reason = AddonManager.UPDATE_WHEN_PERIODIC_UPDATE;
if (!aTimer)
reason = AddonManager.UPDATE_WHEN_USER_REQUESTED;
AddonManager.getAddonsByTypes(null, function(aAddonList) {
aAddonList.forEach(function(aAddon) {
if (aAddon.permissions & AddonManager.PERM_CAN_UPGRADE) {
let data = Cc["@mozilla.org/supports-string;1"].createInstance(Ci.nsISupportsString);
data.data = JSON.stringify({ id: aAddon.id, name: aAddon.name });
Services.obs.notifyObservers(data, "addon-update-started", null);
let listener = new UpdateCheckListener();
aAddon.findUpdates(listener, reason);
}
});
});
RecommendedSearchResults.search();
}
};
// -----------------------------------------------------------------------
// Add-on update listener. Starts a download for any add-on with a viable
// update waiting
// -----------------------------------------------------------------------
function UpdateCheckListener() {
this._status = null;
this._version = null;
}
UpdateCheckListener.prototype = {
onCompatibilityUpdateAvailable: function(aAddon) {
this._status = "compatibility";
},
onUpdateAvailable: function(aAddon, aInstall) {
this._status = "update";
this._version = aInstall.version;
aInstall.install();
},
onNoUpdateAvailable: function(aAddon) {
if (!this._status)
this._status = "no-update";
},
onUpdateFinished: function(aAddon, aError) {
let data = Cc["@mozilla.org/supports-string;1"].createInstance(Ci.nsISupportsString);
if (this._version)
data.data = JSON.stringify({ id: aAddon.id, name: aAddon.name, version: this._version });
else
data.data = JSON.stringify({ id: aAddon.id, name: aAddon.name });
if (aError)
this._status = "error";
Services.obs.notifyObservers(data, "addon-update-ended", this._status);
}
};
// -----------------------------------------------------------------------
// RecommendedSearchResults fetches add-on data and saves it to a cache
// -----------------------------------------------------------------------
var RecommendedSearchResults = {
_getFile: function() {
let dirService = Cc["@mozilla.org/file/directory_service;1"].getService(Ci.nsIProperties);
let file = dirService.get("ProfD", Ci.nsILocalFile);
file.append("recommended-addons.json");
return file;
},
_writeFile: function (aFile, aData) {
if (!aData)
return;
// Initialize the file output stream.
let ostream = Cc["@mozilla.org/network/safe-file-output-stream;1"].createInstance(Ci.nsIFileOutputStream);
ostream.init(aFile, 0x02 | 0x08 | 0x20, 0600, ostream.DEFER_OPEN);
// Obtain a converter to convert our data to a UTF-8 encoded input stream.
let converter = Cc["@mozilla.org/intl/scriptableunicodeconverter"].createInstance(Ci.nsIScriptableUnicodeConverter);
converter.charset = "UTF-8";
// Asynchronously copy the data to the file.
let istream = converter.convertToInputStream(aData);
NetUtil.asyncCopy(istream, ostream, function(rc) {
if (Components.isSuccessCode(rc))
Services.obs.notifyObservers(null, "recommended-addons-cache-updated", "");
});
},
searchSucceeded: function(aAddons, aAddonCount, aTotalResults) {
let json = {
addons: aAddons,
addonCount: aAddonCount,
totalResults: aTotalResults
};
// Avoid any NSS costs. Convert https to http.
json.addons.forEach(function(aAddon){
aAddon.iconURL = aAddon.iconURL.replace(/^https/, "http");
});
let file = this._getFile();
this._writeFile(file, JSON.stringify(json));
},
searchFailed: function searchFailed() { },
search: function() {
const kAddonsMaxDisplay = 2;
if (AddonRepository.isSearching)
AddonRepository.cancelSearch();
AddonRepository.retrieveRecommendedAddons(kAddonsMaxDisplay, RecommendedSearchResults);
}
}
const NSGetFactory = XPCOMUtils.generateNSGetFactory([AddonUpdateService]);