Switch to using the new equivalent of the pageMods API

1 files changed, 72 insertions, 0 deletions

diff --git a/lib/unload-2.js b/lib/unload-2.js
new file mode 100644
index 0000000..c4cbdcb
--- /dev/null
+++ b/lib/unload-2.js
@@ -0,0 +1,72 @@
+/* ***** 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 Jetpack.
+ *
+ * The Initial Developer of the Original Code is Mozilla.
+ * Portions created by the Initial Developer are Copyright (C) 2007
+ * the Initial Developer. All Rights Reserved.
+ *
+ * Contributor(s):
+ *   Atul Varma <atul@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 ***** */
+
+// TODO: Consider adding all this to the unload module.
+
+var unloaders = [];
+
+var addMethod = exports.addMethod = function addMethod(obj, unloader) {
+  var called = false;
+
+  function unloadWrapper() {
+    if (!called) {
+      called = true;
+      var index = unloaders.indexOf(unloadWrapper);
+      if (index == -1)
+        throw new Error("assertion failure");
+      unloaders.splice(index, 1);
+      unloader.apply(obj, []);
+    }
+  };
+
+  unloaders.push(unloadWrapper);
+  obj.unload = unloadWrapper;
+};
+
+var ensure = exports.ensure = function ensure(obj) {
+  if (!("unload" in obj))
+    throw new Error("object has no 'unload' property");
+
+  addMethod(obj, obj.unload);
+};
+
+require("unload").when(
+  function() {
+    unloaders.slice().forEach(
+      function(unloadWrapper) {
+        unloadWrapper();
+      });
+  });