1 files changed, 42 insertions, 0 deletions
diff --git a/chrome/browser/automation/automation_extension_tracker.cc b/chrome/browser/automation/automation_extension_tracker.cc
new file mode 100644
index 0000000..8f09e1f
--- /dev/null
+++ b/chrome/browser/automation/automation_extension_tracker.cc
@@ -0,0 +1,42 @@
+// Copyright (c) 2010 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/automation/automation_extension_tracker.h"
+#include "chrome/browser/extensions/extensions_service.h"
+#include "chrome/browser/profile.h"
+#include "chrome/common/extensions/extension.h"
+#include "chrome/common/notification_service.h"
+
+AutomationExtensionTracker::AutomationExtensionTracker(
+    IPC::Message::Sender* automation)
+    : AutomationResourceTracker<Extension*>(automation) {
+  registrar_.Add(this, NotificationType::EXTENSION_UNLOADED,
+                 NotificationService::AllSources());
+  registrar_.Add(this, NotificationType::EXTENSION_UNLOADED_DISABLED,
+                 NotificationService::AllSources());
+}
+
+AutomationExtensionTracker::~AutomationExtensionTracker() {
+}
+
+void AutomationExtensionTracker::Observe(NotificationType type,
+                                         const NotificationSource& source,
+                                         const NotificationDetails& details) {
+  if (type != NotificationType::EXTENSION_UNLOADED &&
+      type != NotificationType::EXTENSION_UNLOADED_DISABLED)
+    return;
+
+  Extension* extension = Details<Extension>(details).ptr();
+  Profile* profile = Source<Profile>(source).ptr();
+  if (profile) {
+    ExtensionsService* service = profile->GetExtensionsService();
+    if (service) {
+      // Remove this extension only if it is uninstalled, not just disabled.
+      // If it is being uninstalled, the extension will not be in the regular
+      // or disabled list.
+      if (!service->GetExtensionById(extension->id(), true))
+        CloseResource(extension);
+    }
+  }
+}