From d138c1e027c773e12f23a5eb9955514000d21308 Mon Sep 17 00:00:00 2001
From: Kris Maglione <maglione.k@gmail.com>
Date: Fri, 8 Oct 2010 18:18:36 -0400
Subject: [PATCH] Add :extupdate command.

---
 common/content/dactyl.js    | 70 +++++++++++++++++++++++++++++++++----
 common/locale/en-US/gui.xml | 16 +++++++++
 pentadactyl/NEWS            |  1 +
 3 files changed, 80 insertions(+), 7 deletions(-)

diff --git a/common/content/dactyl.js b/common/content/dactyl.js
index 6895a6a4..08b7a73b 100644
--- a/common/content/dactyl.js
+++ b/common/content/dactyl.js
@@ -1025,6 +1025,25 @@ const Dactyl = Module("dactyl", {
         }
     },
 
+    wrapCallback: function (callback, self) {
+        self = self || this;
+        let save = ["forceNewTab", "forceNewWindow"];
+        let saved = save.map(function (p) dactyl[p]);
+        return function wrappedCallback() {
+            let vals = save.map(function (p) dactyl[p]);
+            saved.forEach(function (p, i) dactyl[save[i]] = p);
+            try {
+                return callback.apply(self, arguments);
+            }
+            catch (e) {
+                dactyl.reportError(e, true);
+            }
+            finally {
+                vals.forEach(function (p, i) dactyl[save[i]] = p);
+            }
+        }
+    },
+
     /**
      * @property {Window[]} Returns an array of all the host application's
      *     open windows.
@@ -1415,6 +1434,34 @@ const Dactyl = Module("dactyl", {
             onInstallFailed:    listener("installation", "failed")
         };
 
+        const updateAddons = Class("UpgradeListener", {
+            init: function init(addons) {
+                this.remaining = addons;
+                this.upgrade = [];
+                dactyl.echomsg("Checking updates for addons: " + addons.map(function (a) a.name).join(", "));
+                for (let addon in values(addons))
+                    addon.findUpdates(this, AddonManager.UPDATE_WHEN_USER_REQUESTED, null, null);
+            },
+            addonListener: {
+                __proto__: addonListener,
+                onDownloadStarted: function () {},
+                onDownloadEnded: function () {}
+            },
+            onUpdateAvailable: function (addon, install) {
+                this.upgrade.push(addon);
+                install.addListener(this.addonListener);
+                install.install();
+            },
+            onUpdateFinished: function (addon, error) {
+                this.remaining = this.remaining.filter(function (a) a != addon);
+                if (!this.remaining.length)
+                    dactyl.echomsg(
+                        this.upgrade.length
+                            ? "Installing updates for addons: " + this.upgrade.map(function (i) i.name).join(", ")
+                            : "No addon updates found");
+            }
+        });
+
         ///////////////////////////////////////////////////////////////////////////
 
         function callResult(method) {
@@ -1470,6 +1517,13 @@ const Dactyl = Module("dactyl", {
                 action: function (addon) addon.userDisabled = true,
                 filter: function ({ item }) !item.userDisabled,
                 perm: "disable"
+            },
+            {
+                name: "extu[update]",
+                description: "Update an extension",
+                actions: updateAddons,
+                filter: function ({ item }) !item.userDisabled,
+                perm: "upgrade"
             }
         ].forEach(function (command) {
             let perm = AddonManager["PERM_CAN_" + command.perm.toUpperCase()];
@@ -1483,13 +1537,16 @@ const Dactyl = Module("dactyl", {
                     else
                         dactyl.assert(name, "E471: Argument required");
 
-                    AddonManager.getAddonsByTypes(["extension"], function (list) {
+                    AddonManager.getAddonsByTypes(["extension"], dactyl.wrapCallback(function (list) {
                         if (!args.bang)
                             list = list.filter(function (extension) extension.name == name);
                         if (!args.bang && !list.every(ok))
                             return dactyl.echoerr("Permission denied");
-                        list.forEach(command.action);
-                    });
+                        if (command.actions)
+                            command.actions(list);
+                        else
+                            list.forEach(command.action);
+                    }));
                 }, {
                     argCount: "?", // FIXME: should be "1"
                     bang: true,
@@ -1506,16 +1563,15 @@ const Dactyl = Module("dactyl", {
         commands.add(["exto[ptions]", "extp[references]"],
             "Open an extension's preference dialog",
             function (args) {
-                let tab = dactyl.forceNewTab;
-                AddonManager.getAddonsByTypes(["extension"], function (list) {
+                AddonManager.getAddonsByTypes(["extension"], dactyl.wrapCallback(function (list) {
                     list = list.filter(function (extension) extension.name == args[0]);
                     if (!list.length || !list[0].optionsURL)
                         dactyl.echoerr("E474: Invalid argument");
                     else if (args.bang)
                         window.openDialog(list[0].optionsURL, "_blank", "chrome");
                     else
-                        dactyl.open(list[0].optionsURL, { from: "extoptions", where: tab && dactyl.NEW_TAB });
-                });
+                        dactyl.open(list[0].optionsURL, { from: "extoptions" });
+                }));
             }, {
                 argCount: "1",
                 bang: true,
diff --git a/common/locale/en-US/gui.xml b/common/locale/en-US/gui.xml
index 2cb7a019..81d32a5a 100644
--- a/common/locale/en-US/gui.xml
+++ b/common/locale/en-US/gui.xml
@@ -81,6 +81,7 @@
 <item>
     <tags>:exta :extadd</tags>
     <spec>:exta<oa>dd</oa> <a>file|url</a></spec>
+    <strut/>
     <description>
         <p>
             Install an extension. <a>file|uri</a> must be the local file
@@ -93,6 +94,7 @@
     <tags>:extde :extdelete</tags>
     <spec>:extde<oa>lete</oa> <a>extension</a></spec>
     <spec>:extde<oa>lete</oa>!</spec>
+    <strut/>
     <description>
         <p>
             Uninstall an extension. <a>extension</a> is the extension's name. When <oa>!</oa> is given
@@ -105,6 +107,7 @@
     <tags>:extd :extdisable</tags>
     <spec>:extd<oa>isable</oa> <a>extension</a></spec>
     <spec>:extd<oa>isable</oa>!</spec>
+    <strut/>
     <description>
         <p>
             Disable an extension. <a>extension</a> is the extension's name. When <oa>!</oa> is given
@@ -117,6 +120,7 @@
     <tags>:exte :extenable</tags>
     <spec>:exte<oa>nable</oa> <a>extension</a></spec>
     <spec>:exte<oa>nable</oa>!</spec>
+    <strut/>
     <description>
         <p>
             Enable an extension. <a>extension</a> is the extension's name. When <oa>!</oa> is given all
@@ -129,6 +133,7 @@
     <tags>:exts :extens :extensions</tags>
     <spec>:extens<oa>ions</oa></spec>
     <spec>:exts</spec>
+    <strut/>
     <description>
         <p>List all installed extensions.</p>
     </description>
@@ -147,6 +152,17 @@
     </description>
 </item>
 
+<item>
+    <tags>:extu :extupdate</tags>
+    <spec>:extu<oa>pdate</oa><oa>!</oa> <a>extension</a></spec>
+    <description>
+        <p>
+            Update an extension. When <oa>!</oa> is given, update all
+            extensions.
+        </p>
+    </description>
+</item>
+
 <h2 tag="sidebar">Sidebar</h2>
 
 <item>
diff --git a/pentadactyl/NEWS b/pentadactyl/NEWS
index e6400105..20987801 100644
--- a/pentadactyl/NEWS
+++ b/pentadactyl/NEWS
@@ -58,6 +58,7 @@
     * Added BookmarkChange, BookmarkRemove autocommands.
     * Added -keyword, -tags, -title to :delbmarks.
     * Added "passwords" and "venkman" dialogs to :dialog.
+    * Added :extupdate command
     * Added 'hintkeys' option.
     * Added "transliterated" option to 'hintmatching'.
     * Replaced 'focuscontent' with 'strictfocus'.