From 74faf8085cb13a4c3e636f2dca6fd24e20d71a7e Mon Sep 17 00:00:00 2001
From: Mark Finkle <mfinkle@mozilla.com>
Date: Wed, 24 Mar 2010 14:34:49 -0400
Subject: [PATCH] Bug 547993 - Awesome Bar history is not immediately flushed
 when clicking on the clear private data button [r=vingtetun]

---
 mobile/components/AutoCompleteCache.js | 62 ++++++++++++++++++++++++--
 1 file changed, 59 insertions(+), 3 deletions(-)

diff --git a/mobile/components/AutoCompleteCache.js b/mobile/components/AutoCompleteCache.js
index de1c252b7f9..7e7e7fcf459 100644
--- a/mobile/components/AutoCompleteCache.js
+++ b/mobile/components/AutoCompleteCache.js
@@ -60,6 +60,10 @@ XPCOMUtils.defineLazyGetter(this, "PACS", function() {
                    .getService(Ci.nsIAutoCompleteSearch);
 });
 
+XPCOMUtils.defineLazyServiceGetter(this, "gObsSvc",
+                                   "@mozilla.org/observer-service;1",
+                                   "nsIObserverService");
+
 XPCOMUtils.defineLazyServiceGetter(this, "gDirSvc",
                                    "@mozilla.org/file/directory_service;1",
                                    "nsIProperties");
@@ -224,6 +228,8 @@ var AutoCompleteUtils = {
 
 function AutoCompleteCache() {
   AutoCompleteUtils.init();
+
+  gObsSvc.addObserver(this, "browser:purge-session-history", true);
 }
 
 AutoCompleteCache.prototype = {
@@ -231,7 +237,7 @@ AutoCompleteCache.prototype = {
   contractID: "@mozilla.org/autocomplete/search;1?name=history",
   classID: Components.ID("{a65f9dca-62ab-4b36-a870-972927c78b56}"),
 
-  QueryInterface: XPCOMUtils.generateQI([Ci.nsIAutoCompleteSearch]),
+  QueryInterface: XPCOMUtils.generateQI([Ci.nsIAutoCompleteSearch, Ci.nsIObserver, Ci.nsISupportsWeakReference]),
 
   startSearch: function(query, param, prev, listener) {
     let self = this;
@@ -257,9 +263,59 @@ AutoCompleteCache.prototype = {
   stopSearch: function() {
     // Stop any active queries
     AutoCompleteUtils.stop();
+  },
+
+  observe: function (aSubject, aTopic, aData) {
+    switch (aTopic) {
+      case "browser:purge-session-history":
+        AutoCompleteUtils.update();
+        break;
+    }
   }
 };
 
-function NSGetModule(aCompMgr, aFileSpec) {
-  return XPCOMUtils.generateModule([AutoCompleteCache]);
+// -----------------------------------------------------------------------
+// BookmarkObserver updates the cache when a bookmark is added
+// -----------------------------------------------------------------------
+function BookmarkObserver() {
+  this._batch = false;
+}
+
+BookmarkObserver.prototype = {
+  onBeginUpdateBatch: function() {
+    this._batch = true;
+  },
+  onEndUpdateBatch: function() {
+    this._batch = false;
+    AutoCompleteUtils.update();
+  },
+  onItemAdded: function(aItemId, aParentId, aIndex, aItemType) {
+    if (!this._batch)
+      AutoCompleteUtils.update();
+  },
+  onItemChanged: function () {
+    if (!this._batch)
+      AutoCompleteUtils.update();
+  },
+  onBeforeItemRemoved: function() {},
+  onItemRemoved: function() {
+    if (!this._batch)
+      AutoCompleteUtils.update();
+  },
+  onItemVisited: function() {},
+  onItemMoved: function() {},
+
+  classDescription: "Bookmark observer used to update autocomplete cache",
+  classID: Components.ID("f570982e-4f15-48ab-b6a0-ed851ac551b2"),
+  contractID: "@mozilla.org/browser/autocomplete-observer;1",
+
+  // Registering in this category makes us get initialized only when
+  // the listener would be notified.
+  _xpcom_categories: [{ category: "bookmark-observers" }],
+
+  QueryInterface: XPCOMUtils.generateQI([Ci.nsINavBookmarkObserver])
+};
+
+function NSGetModule(aCompMgr, aFileSpec) {
+  return XPCOMUtils.generateModule([AutoCompleteCache, BookmarkObserver]);
 }