content/renderer/push_messaging_dispatcher.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/renderer/push_messaging_dispatcher.h"
   6
   7 #include "base/strings/utf_string_conversions.h"
   8 #include "content/child/service_worker/web_service_worker_provider_impl.h"
   9 #include "content/common/push_messaging_messages.h"
  10 #include "content/renderer/manifest/manifest_manager.h"
  11 #include "content/renderer/render_frame_impl.h"
  12 #include "ipc/ipc_message.h"
  13 #include "third_party/WebKit/public/platform/WebPushError.h"
  14 #include "third_party/WebKit/public/platform/WebPushRegistration.h"
  15 #include "third_party/WebKit/public/platform/WebServiceWorkerProvider.h"
  16 #include "third_party/WebKit/public/platform/WebString.h"
  17 #include "third_party/WebKit/public/web/WebUserGestureIndicator.h"
  18 #include "url/gurl.h"
  19
  20 using blink::WebString;
  21
  22 namespace content {
  23
  24 PushMessagingDispatcher::PushMessagingDispatcher(RenderFrame* render_frame)
  25     : RenderFrameObserver(render_frame) {
  26 }
  27
  28 PushMessagingDispatcher::~PushMessagingDispatcher() {}
  29
  30 bool PushMessagingDispatcher::OnMessageReceived(const IPC::Message& message) {
  31   bool handled = true;
  32   IPC_BEGIN_MESSAGE_MAP(PushMessagingDispatcher, message)
  33     IPC_MESSAGE_HANDLER(PushMessagingMsg_RegisterSuccess, OnRegisterSuccess)
  34     IPC_MESSAGE_HANDLER(PushMessagingMsg_RegisterError, OnRegisterError)
  35     IPC_MESSAGE_UNHANDLED(handled = false)
  36   IPC_END_MESSAGE_MAP()
  37   return handled;
  38 }
  39
  40 void PushMessagingDispatcher::registerPushMessaging(
  41     blink::WebPushRegistrationCallbacks* callbacks,
  42     blink::WebServiceWorkerProvider* service_worker_provider) {
  43   RenderFrameImpl::FromRoutingID(routing_id())->manifest_manager()->GetManifest(
  44       base::Bind(&PushMessagingDispatcher::DoRegister,
  45                  base::Unretained(this),
  46                  callbacks,
  47                  service_worker_provider));
  48 }
  49
  50 void PushMessagingDispatcher::DoRegister(
  51     blink::WebPushRegistrationCallbacks* callbacks,
  52     blink::WebServiceWorkerProvider* service_worker_provider,
  53     const Manifest& manifest) {
  54   DCHECK(callbacks);
  55   int callbacks_id = registration_callbacks_.Add(callbacks);
  56   int service_worker_provider_id = static_cast<WebServiceWorkerProviderImpl*>(
  57                                        service_worker_provider)->provider_id();
  58   Send(new PushMessagingHostMsg_Register(
  59       routing_id(),
  60       callbacks_id,
  61       manifest.gcm_sender_id.is_null()
  62           ? std::string()
  63           : base::UTF16ToUTF8(manifest.gcm_sender_id.string()),
  64       blink::WebUserGestureIndicator::isProcessingUserGesture(),
  65       service_worker_provider_id));
  66 }
  67
  68 void PushMessagingDispatcher::OnRegisterSuccess(
  69     int32 callbacks_id,
  70     const GURL& endpoint,
  71     const std::string& registration_id) {
  72   blink::WebPushRegistrationCallbacks* callbacks =
  73       registration_callbacks_.Lookup(callbacks_id);
  74   CHECK(callbacks);
  75
  76   scoped_ptr<blink::WebPushRegistration> registration(
  77       new blink::WebPushRegistration(
  78           WebString::fromUTF8(endpoint.spec()),
  79           WebString::fromUTF8(registration_id)));
  80   callbacks->onSuccess(registration.release());
  81   registration_callbacks_.Remove(callbacks_id);
  82 }
  83
  84 void PushMessagingDispatcher::OnRegisterError(int32 callbacks_id,
  85                                               PushRegistrationStatus status) {
  86   blink::WebPushRegistrationCallbacks* callbacks =
  87       registration_callbacks_.Lookup(callbacks_id);
  88   CHECK(callbacks);
  89
  90   scoped_ptr<blink::WebPushError> error(new blink::WebPushError(
  91       blink::WebPushError::ErrorTypeAbort,
  92       WebString::fromUTF8(PushRegistrationStatusToString(status))));
  93   callbacks->onError(error.release());
  94   registration_callbacks_.Remove(callbacks_id);
  95 }
  96
  97 }  // namespace content