Index: chrome/browser/extensions/api/push_messaging/push_messaging_api.cc |
diff --git a/chrome/browser/extensions/api/push_messaging/push_messaging_api.cc b/chrome/browser/extensions/api/push_messaging/push_messaging_api.cc |
index 95684dd9cdd03156ecbccdd23daa3460e15c7c99..e1effd90e45ff4943df735a11cefb9ad8c8cefa1 100644 |
--- a/chrome/browser/extensions/api/push_messaging/push_messaging_api.cc |
+++ b/chrome/browser/extensions/api/push_messaging/push_messaging_api.cc |
@@ -28,6 +28,7 @@ |
#include "content/public/browser/notification_details.h" |
#include "content/public/browser/notification_source.h" |
#include "extensions/browser/event_router.h" |
+#include "extensions/browser/extension_registry.h" |
#include "extensions/browser/extension_system_provider.h" |
#include "extensions/browser/extensions_browser_client.h" |
#include "extensions/common/extension.h" |
@@ -75,8 +76,7 @@ void PushMessagingEventRouter::OnMessage(const std::string& extension_id, |
<< "' extension = '" << extension_id << "'"; |
scoped_ptr<base::ListValue> args(glue::OnMessage::Create(message)); |
- scoped_ptr<extensions::Event> event( |
- new extensions::Event(glue::OnMessage::kEventName, args.Pass())); |
+ scoped_ptr<Event> event(new Event(glue::OnMessage::kEventName, args.Pass())); |
event->restrict_to_browser_context = profile_; |
EventRouter::Get(profile_)->DispatchEventToExtension( |
extension_id, event.Pass()); |
@@ -287,14 +287,11 @@ void PushMessagingGetChannelIdFunction::OnObfuscatedGaiaIdFetchFailure( |
} |
PushMessagingAPI::PushMessagingAPI(content::BrowserContext* context) |
- : profile_(Profile::FromBrowserContext(context)) { |
+ : extension_registry_observer_(this), |
+ profile_(Profile::FromBrowserContext(context)) { |
+ extension_registry_observer_.Add(ExtensionRegistry::Get(profile_)); |
registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_INSTALLED, |
content::Source<Profile>(profile_->GetOriginalProfile())); |
- registrar_.Add(this, |
- chrome::NOTIFICATION_EXTENSION_LOADED_DEPRECATED, |
- content::Source<Profile>(profile_->GetOriginalProfile())); |
- registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_UNLOADED_DEPRECATED, |
- content::Source<Profile>(profile_->GetOriginalProfile())); |
} |
PushMessagingAPI::~PushMessagingAPI() { |
@@ -319,46 +316,56 @@ PushMessagingAPI::GetFactoryInstance() { |
return g_factory.Pointer(); |
} |
-void PushMessagingAPI::Observe(int type, |
- const content::NotificationSource& source, |
- const content::NotificationDetails& details) { |
+bool PushMessagingAPI::InitEventRouterAndHandler() { |
invalidation::InvalidationService* invalidation_service = |
invalidation::InvalidationServiceFactory::GetForProfile(profile_); |
if (!invalidation_service) |
- return; |
+ return false; |
if (!event_router_) |
event_router_.reset(new PushMessagingEventRouter(profile_)); |
if (!handler_) { |
- handler_.reset(new PushMessagingInvalidationHandler( |
- invalidation_service, event_router_.get())); |
+ handler_.reset(new PushMessagingInvalidationHandler(invalidation_service, |
+ event_router_.get())); |
} |
- switch (type) { |
- case chrome::NOTIFICATION_EXTENSION_INSTALLED: { |
- const Extension* extension = |
- content::Details<const InstalledExtensionInfo>(details)->extension; |
- if (extension->HasAPIPermission(APIPermission::kPushMessaging)) { |
- handler_->SuppressInitialInvalidationsForExtension(extension->id()); |
- } |
- break; |
- } |
- case chrome::NOTIFICATION_EXTENSION_LOADED_DEPRECATED: { |
- const Extension* extension = content::Details<Extension>(details).ptr(); |
- if (extension->HasAPIPermission(APIPermission::kPushMessaging)) { |
- handler_->RegisterExtension(extension->id()); |
- } |
- break; |
- } |
- case chrome::NOTIFICATION_EXTENSION_UNLOADED_DEPRECATED: { |
- const Extension* extension = |
- content::Details<UnloadedExtensionInfo>(details)->extension; |
- if (extension->HasAPIPermission(APIPermission::kPushMessaging)) { |
- handler_->UnregisterExtension(extension->id()); |
- } |
- break; |
- } |
- default: |
- NOTREACHED(); |
+ |
+ return true; |
+} |
+ |
+void PushMessagingAPI::OnExtensionLoaded( |
+ content::BrowserContext* browser_context, |
+ const Extension* extension) { |
+ if (!InitEventRouterAndHandler()) |
+ return; |
+ |
+ if (extension->HasAPIPermission(APIPermission::kPushMessaging)) { |
+ handler_->RegisterExtension(extension->id()); |
+ } |
+} |
+ |
+void PushMessagingAPI::OnExtensionUnloaded( |
+ content::BrowserContext* browser_context, |
+ const Extension* extension, |
+ UnloadedExtensionInfo::Reason reason) { |
+ if (!InitEventRouterAndHandler()) |
+ return; |
+ |
+ if (extension->HasAPIPermission(APIPermission::kPushMessaging)) { |
+ handler_->UnregisterExtension(extension->id()); |
+ } |
+} |
+ |
+void PushMessagingAPI::Observe(int type, |
+ const content::NotificationSource& source, |
+ const content::NotificationDetails& details) { |
+ DCHECK_EQ(type, chrome::NOTIFICATION_EXTENSION_INSTALLED); |
+ if (!InitEventRouterAndHandler()) |
+ return; |
+ |
+ const Extension* extension = |
+ content::Details<const InstalledExtensionInfo>(details)->extension; |
+ if (extension->HasAPIPermission(APIPermission::kPushMessaging)) { |
+ handler_->SuppressInitialInvalidationsForExtension(extension->id()); |
} |
} |