root/chrome/browser/extensions/extension_gcm_app_handler.cc

DEFINITIONS

1. IsGCMPermissionEnabled
2. GetFactoryInstance
3. weak_factory_
4. ShutdownHandler
5. OnMessage
6. OnMessagesDeleted
7. OnSendError
8. Observe
9. GetGCMProfileService
10. OnUnregisterCompleted

// Copyright 2014 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/extensions/extension_gcm_app_handler.h"

#include "base/bind.h"
#include "base/lazy_instance.h"
#include "base/location.h"
#include "chrome/browser/chrome_notification_types.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/services/gcm/gcm_profile_service.h"
#include "chrome/browser/services/gcm/gcm_profile_service_factory.h"
#include "content/public/browser/notification_details.h"
#include "content/public/browser/notification_source.h"
#include "extensions/browser/extension_registry.h"
#include "extensions/browser/extension_system.h"
#include "extensions/common/extension.h"
#include "extensions/common/permissions/permissions_data.h"

#if !defined(OS_ANDROID)
#include "chrome/browser/extensions/api/gcm/gcm_api.h"
#endif

namespace extensions {

namespace {

base::LazyInstance<BrowserContextKeyedAPIFactory<ExtensionGCMAppHandler> >
    g_factory = LAZY_INSTANCE_INITIALIZER;

bool IsGCMPermissionEnabled(const Extension* extension) {
  return PermissionsData::HasAPIPermission(extension, APIPermission::kGcm);
}

}  // namespace


// static
BrowserContextKeyedAPIFactory<ExtensionGCMAppHandler>*
ExtensionGCMAppHandler::GetFactoryInstance() {
  return g_factory.Pointer();
}

ExtensionGCMAppHandler::ExtensionGCMAppHandler(content::BrowserContext* context)
    : profile_(Profile::FromBrowserContext(context)),
      weak_factory_(this) {
  // Listen to various extension related notifications.
  registrar_.Add(this,
                 chrome::NOTIFICATION_EXTENSION_LOADED,
                 content::Source<Profile>(profile_));
  registrar_.Add(this,
                 chrome::NOTIFICATION_EXTENSION_UNLOADED_DEPRECATED,
                 content::Source<Profile>(profile_));
  registrar_.Add(this,
                 chrome::NOTIFICATION_EXTENSION_UNINSTALLED,
                 content::Source<Profile>(profile_));

#if !defined(OS_ANDROID)
  js_event_router_.reset(new extensions::GcmJsEventRouter(profile_));
#endif
}

ExtensionGCMAppHandler::~ExtensionGCMAppHandler() {
  const ExtensionSet& enabled_extensions =
      ExtensionRegistry::Get(profile_)->enabled_extensions();
  for (ExtensionSet::const_iterator extension = enabled_extensions.begin();
       extension != enabled_extensions.end();
       ++extension) {
    if (IsGCMPermissionEnabled(extension->get()))
      GetGCMProfileService()->RemoveAppHandler((*extension)->id());
  }
}

void ExtensionGCMAppHandler::ShutdownHandler() {
#if !defined(OS_ANDROID)
  js_event_router_.reset();
#endif
}

void ExtensionGCMAppHandler::OnMessage(
    const std::string& app_id,
    const gcm::GCMClient::IncomingMessage& message) {
#if !defined(OS_ANDROID)
  js_event_router_->OnMessage(app_id, message);
#endif
}

void ExtensionGCMAppHandler::OnMessagesDeleted(const std::string& app_id) {
#if !defined(OS_ANDROID)
  js_event_router_->OnMessagesDeleted(app_id);
#endif
}

void ExtensionGCMAppHandler::OnSendError(
    const std::string& app_id,
    const gcm::GCMClient::SendErrorDetails& send_error_details) {
#if !defined(OS_ANDROID)
  js_event_router_->OnSendError(app_id, send_error_details);
#endif
}

void ExtensionGCMAppHandler::Observe(
    int type,
    const content::NotificationSource& source,
    const content::NotificationDetails& details) {
  switch (type) {
    case chrome:: NOTIFICATION_EXTENSION_LOADED: {
      const Extension* extension = content::Details<Extension>(details).ptr();
      if (IsGCMPermissionEnabled(extension))
        GetGCMProfileService()->AddAppHandler(extension->id(), this);
      break;
    }
    case chrome:: NOTIFICATION_EXTENSION_UNLOADED_DEPRECATED: {
      const Extension* extension =
          content::Details<UnloadedExtensionInfo>(details)->extension;
      if (IsGCMPermissionEnabled(extension))
        GetGCMProfileService()->RemoveAppHandler(extension->id());
      break;
    }
    case chrome:: NOTIFICATION_EXTENSION_UNINSTALLED: {
      const Extension* extension = content::Details<Extension>(details).ptr();
      if (IsGCMPermissionEnabled(extension)) {
        GetGCMProfileService()->Unregister(
            extension->id(),
            base::Bind(&ExtensionGCMAppHandler::OnUnregisterCompleted,
                       weak_factory_.GetWeakPtr(),
                       extension->id()));
        GetGCMProfileService()->RemoveAppHandler(extension->id());
      }
      break;
    }
    default:
      NOTREACHED();
  }
}

gcm::GCMProfileService* ExtensionGCMAppHandler::GetGCMProfileService() const {
  return gcm::GCMProfileServiceFactory::GetForProfile(profile_);
}

void ExtensionGCMAppHandler::OnUnregisterCompleted(
    const std::string& app_id, gcm::GCMClient::Result result) {
  // Nothing to do.
}

}  // namespace extensions