From e99481a7e3d877f0c4440396c7f183fe1b572089 Mon Sep 17 00:00:00 2001
From: Kris Maglione <maglione.k@gmail.com>
Date: Mon, 17 Feb 2014 13:50:15 -0800
Subject: [PATCH] Refactor storage observers to be less dodgy and stupid.

---
 common/modules/overlay.jsm |  4 ++-
 common/modules/storage.jsm | 67 ++++++++++++--------------------------
 2 files changed, 23 insertions(+), 48 deletions(-)

diff --git a/common/modules/overlay.jsm b/common/modules/overlay.jsm
index 4f3081e3..4a2470bc 100644
--- a/common/modules/overlay.jsm
+++ b/common/modules/overlay.jsm
@@ -185,8 +185,10 @@ var Overlay = Module("Overlay", XPCOM([Ci.nsIObserver, Ci.nsISupportsWeakReferen
 
     setData: function setData(obj, key, val) {
         let data = this.getData(obj);
+        if (val !== undefined)
+            return data[key] = val;
 
-        return data[key] = val;
+        delete data[key];
     },
 
     overlayWindow: function overlayWindow(url, fn) {
diff --git a/common/modules/storage.jsm b/common/modules/storage.jsm
index 21815418..4229408f 100644
--- a/common/modules/storage.jsm
+++ b/common/modules/storage.jsm
@@ -185,6 +185,7 @@ var Storage = Module("Storage", {
         if (!Services.dactylSession)
             Services.dactylSession = Cu.createObjectIn(sessionGlobal);
         this.session = Services.dactylSession;
+        this.windows = WeakMap();
     },
 
     cleanup: function () {
@@ -287,60 +288,33 @@ var Storage = Module("Storage", {
         return this.newObject(key, ArrayStore, update({ type: Array }, options));
     },
 
-    addObserver: function addObserver(key, callback, ref) {
-        if (ref) {
-            let refs = overlay.getData(ref, "storage-refs");
-            refs.push(callback);
-            var callbackRef = util.weakReference(callback);
-        }
-        else {
-            callbackRef = { get: function () callback };
-        }
+    get observerMaps() {
+        yield this.observers;
+        for (let window of overlay.windows)
+            yield overlay.getData(window, "storage-observers", Object);
+    },
 
-        this.removeDeadObservers();
+    addObserver: function addObserver(key, callback, window) {
+        var { observers } = this;
+        if (window)
+            observers = overlay.getData(window, "storage-observers", Object);
 
-        if (!(key in this.observers))
-            this.observers[key] = [];
+        if (!hasOwnProperty(observers, key))
+            observers[key] = RealSet();
 
-        if (!this.observers[key].some(o => o.callback.get() == callback))
-            this.observers[key].push({ ref: ref && Cu.getWeakReference(ref),
-                                       callback: callbackRef });
+        observers[key].add(callback);
     },
 
     removeObserver: function (key, callback) {
-        this.removeDeadObservers();
-
-        if (!(key in this.observers))
-            return;
-
-        this.observers[key] = this.observers[key].filter(elem => elem.callback.get() != callback);
-        if (this.observers[key].length == 0)
-            delete obsevers[key];
-    },
-
-    removeDeadObservers: function () {
-        function filter(o) {
-            if (!o.callback.get())
-                return false;
-
-            let ref = o.ref && o.ref.get();
-            return ref && !ref.closed && overlay.getData(ref, "storage-refs", null);
-        }
-
-        for (let [key, ary] in Iterator(this.observers)) {
-            this.observers[key] = ary = ary.filter(filter);
-            if (!ary.length)
-                delete this.observers[key];
-        }
+        for (let observers in this.observerMaps)
+            if (key in observers)
+                observers[key].remove(callback);
     },
 
     fireEvent: function fireEvent(key, event, arg) {
-        this.removeDeadObservers();
-
-        if (key in this.observers)
-            // Safe, since we have our own Array object here.
-            for each (let observer in this.observers[key])
-                observer.callback.get()(key, event, arg);
+        for (let observers in this.observerMaps)
+            for (let observer of observers[key] || [])
+                observer(key, event, arg);
 
         if (key in this.keys && this.keys[key].timer)
             this[key].timer.tell();
@@ -392,8 +366,7 @@ var Storage = Module("Storage", {
     }
 }, {
     cleanup: function (dactyl, modules, window) {
-        overlay.setData(window, "storage-refs", null);
-        this.removeDeadObservers();
+        overlay.setData(window, "storage-callbacks", undefined);
     }
 });