Implement WebPushProvider.unregister() in Chromium.

chrome/browser/services/gcm/push_messaging_service_impl.cc

 // 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/services/gcm/push_messaging_service_impl.h"
 
 #include <vector>
 
 #include "base/bind.h"
 #include "base/command_line.h"
@@ skipping 140 matching lines @@
   // }
   // TODO(johnme): Make sure this is clearly documented for developers.
   PushMessagingApplicationId application_id =
       PushMessagingApplicationId::Parse(app_id);
   DCHECK(application_id.IsValid());
   GCMClient::MessageData::const_iterator it = message.data.find("data");
   if (application_id.IsValid() && it != message.data.end()) {
     if (!HasPermission(application_id.origin)) {
       // The |origin| lost push permission. We need to unregister and drop this
       // message.
-      Unregister(application_id);
+      Unregister(application_id, UnregisterCallback());
       return;
     }
 
     const std::string& data = it->second;
     content::BrowserContext::DeliverPushMessage(
         profile_,
         application_id.origin,
         application_id.service_worker_registration_id,
         data,
         base::Bind(&PushMessagingServiceImpl::DeliverMessageCallback,
@@ skipping 18 matching lines @@
     const GCMClient::IncomingMessage& message,
     content::PushDeliveryStatus status) {
   // TODO(mvanouwerkerk): UMA logging.
   // TODO(mvanouwerkerk): Is there a way to recover from failure?
   switch (status) {
     case content::PUSH_DELIVERY_STATUS_SUCCESS:
     case content::PUSH_DELIVERY_STATUS_SERVICE_WORKER_ERROR:
     case content::PUSH_DELIVERY_STATUS_EVENT_WAITUNTIL_REJECTED:
       break;
     case content::PUSH_DELIVERY_STATUS_NO_SERVICE_WORKER:
-      Unregister(application_id);
+      Unregister(application_id, UnregisterCallback());
       break;
   }
 }
 
 void PushMessagingServiceImpl::OnMessagesDeleted(const std::string& app_id) {
   // TODO(mvanouwerkerk): Fire push error event on the Service Worker
   // corresponding to app_id.
 }
 
 void PushMessagingServiceImpl::OnSendError(
@@ skipping 196 matching lines @@
 
   gcm_profile_service_->driver()->Register(
       application_id.ToString(),
       sender_ids,
       base::Bind(&PushMessagingServiceImpl::DidRegister,
                  weak_factory_.GetWeakPtr(),
                  register_callback));
 }
 
 void PushMessagingServiceImpl::Unregister(
-    const PushMessagingApplicationId& application_id) {
+    const GURL& requesting_origin,
+    int64 service_worker_registration_id,
+    const content::PushMessagingService::UnregisterCallback& callback) {
+  DCHECK(gcm_profile_service_->driver());
+
+  PushMessagingApplicationId application_id = PushMessagingApplicationId(
+      requesting_origin, service_worker_registration_id);
+  DCHECK(application_id.IsValid());
+
+  Unregister(application_id, callback);
+}
+
+void PushMessagingServiceImpl::Unregister(
+    const PushMessagingApplicationId& application_id,
+    const content::PushMessagingService::UnregisterCallback& callback) {
   DCHECK(gcm_profile_service_->driver());
 
   gcm_profile_service_->driver()->Unregister(
       application_id.ToString(),
       base::Bind(&PushMessagingServiceImpl::DidUnregister,
-                 weak_factory_.GetWeakPtr()));
+                 weak_factory_.GetWeakPtr(), callback));
 }
 
-void PushMessagingServiceImpl::DidUnregister(GCMClient::Result result) {
-  if (result != GCMClient::SUCCESS) {
-    DVLOG(1) << "GCM unregistration failed.";
-    return;
+void PushMessagingServiceImpl::DidUnregister(
+    const content::PushMessagingService::UnregisterCallback& callback,
+    GCMClient::Result result) {
+  // Internal calls pass a null callback.
+  if (!callback.is_null()) {
+    switch (result) {
+    case GCMClient::SUCCESS:
+      callback.Run(content::PUSH_UNREGISTRATION_STATUS_SUCCESS_UNREGISTER);
+      break;
+    case GCMClient::NETWORK_ERROR:
+    case GCMClient::TTL_EXCEEDED:
+      callback.Run(content::PUSH_UNREGISTRATION_STATUS_NETWORK_ERROR);
+      break;
+    case GCMClient::SERVER_ERROR:
+    case GCMClient::INVALID_PARAMETER:
+    case GCMClient::GCM_DISABLED:
+    case GCMClient::NOT_SIGNED_IN:
+    case GCMClient::ASYNC_OPERATION_PENDING:
+    case GCMClient::UNKNOWN_ERROR:
+      callback.Run(content::PUSH_UNREGISTRATION_STATUS_UNKNOWN_ERROR);
+      break;
+    default:
+      NOTREACHED() << "Unexpected GCMClient::Result value.";
+      callback.Run(content::PUSH_UNREGISTRATION_STATUS_UNKNOWN_ERROR);
+      break;
+    }
   }
 
-  DecreasePushRegistrationCount(1);
+  if (result == GCMClient::SUCCESS)
+    DecreasePushRegistrationCount(1);
 }
 
 bool PushMessagingServiceImpl::HasPermission(const GURL& origin) {
   gcm::PushMessagingPermissionContext* permission_context =
       gcm::PushMessagingPermissionContextFactory::GetForProfile(profile_);
   DCHECK(permission_context);
 
   return permission_context->GetPermissionStatus(origin, origin) ==
       CONTENT_SETTING_ALLOW;
 }
 
 void PushMessagingServiceImpl::AddAppHandlerIfNecessary() {
   if (gcm_profile_service_->driver()->GetAppHandler(
           kPushMessagingApplicationIdPrefix) != this)
     gcm_profile_service_->driver()->AddAppHandler(
         kPushMessagingApplicationIdPrefix, this);
 }
 
 }  // namespace gcm