| // 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 "content/renderer/push_messaging/push_messaging_dispatcher.h" |
| |
| #include "base/strings/utf_string_conversions.h" |
| #include "content/child/service_worker/web_service_worker_registration_impl.h" |
| #include "content/common/push_messaging_messages.h" |
| #include "content/renderer/manifest/manifest_manager.h" |
| #include "content/renderer/render_frame_impl.h" |
| #include "ipc/ipc_message.h" |
| #include "third_party/WebKit/public/platform/WebServiceWorkerRegistration.h" |
| #include "third_party/WebKit/public/platform/WebString.h" |
| #include "third_party/WebKit/public/platform/modules/push_messaging/WebPushError.h" |
| #include "third_party/WebKit/public/platform/modules/push_messaging/WebPushSubscription.h" |
| #include "third_party/WebKit/public/platform/modules/push_messaging/WebPushSubscriptionOptions.h" |
| #include "third_party/WebKit/public/web/WebConsoleMessage.h" |
| #include "third_party/WebKit/public/web/WebLocalFrame.h" |
| #include "url/gurl.h" |
| |
| namespace content { |
| |
| PushMessagingDispatcher::PushMessagingDispatcher(RenderFrame* render_frame) |
| : RenderFrameObserver(render_frame) { |
| } |
| |
| PushMessagingDispatcher::~PushMessagingDispatcher() {} |
| |
| bool PushMessagingDispatcher::OnMessageReceived(const IPC::Message& message) { |
| bool handled = true; |
| IPC_BEGIN_MESSAGE_MAP(PushMessagingDispatcher, message) |
| IPC_MESSAGE_HANDLER(PushMessagingMsg_SubscribeFromDocumentSuccess, |
| OnSubscribeFromDocumentSuccess) |
| IPC_MESSAGE_HANDLER(PushMessagingMsg_SubscribeFromDocumentError, |
| OnSubscribeFromDocumentError) |
| IPC_MESSAGE_UNHANDLED(handled = false) |
| IPC_END_MESSAGE_MAP() |
| return handled; |
| } |
| |
| void PushMessagingDispatcher::subscribe( |
| blink::WebServiceWorkerRegistration* service_worker_registration, |
| const blink::WebPushSubscriptionOptions& options, |
| blink::WebPushSubscriptionCallbacks* callbacks) { |
| DCHECK(service_worker_registration); |
| DCHECK(callbacks); |
| RenderFrameImpl::FromRoutingID(routing_id()) |
| ->manifest_manager() |
| ->GetManifest(base::Bind(&PushMessagingDispatcher::DoSubscribe, |
| base::Unretained(this), |
| service_worker_registration, |
| options, |
| callbacks)); |
| } |
| |
| void PushMessagingDispatcher::DoSubscribe( |
| blink::WebServiceWorkerRegistration* service_worker_registration, |
| const blink::WebPushSubscriptionOptions& options, |
| blink::WebPushSubscriptionCallbacks* callbacks, |
| const Manifest& manifest) { |
| int request_id = subscription_callbacks_.Add(callbacks); |
| int64_t service_worker_registration_id = |
| static_cast<WebServiceWorkerRegistrationImpl*>( |
| service_worker_registration)->registration_id(); |
| |
| std::string sender_id = |
| manifest.gcm_sender_id.is_null() |
| ? std::string() |
| : base::UTF16ToUTF8(manifest.gcm_sender_id.string()); |
| if (sender_id.empty()) { |
| OnSubscribeFromDocumentError(request_id, |
| PUSH_REGISTRATION_STATUS_NO_SENDER_ID); |
| return; |
| } |
| |
| Send(new PushMessagingHostMsg_SubscribeFromDocument( |
| routing_id(), request_id, |
| manifest.gcm_sender_id.is_null() |
| ? std::string() |
| : base::UTF16ToUTF8(manifest.gcm_sender_id.string()), |
| options.userVisibleOnly, service_worker_registration_id)); |
| } |
| |
| void PushMessagingDispatcher::OnSubscribeFromDocumentSuccess( |
| int32_t request_id, |
| const GURL& endpoint) { |
| blink::WebPushSubscriptionCallbacks* callbacks = |
| subscription_callbacks_.Lookup(request_id); |
| |
| // TODO(peter): Change this back to a DCHECK before M45 branches. |
| CHECK(callbacks) << "Invalid request id received: " << request_id; |
| |
| scoped_ptr<blink::WebPushSubscription> subscription( |
| new blink::WebPushSubscription(endpoint)); |
| callbacks->onSuccess(subscription.release()); |
| |
| subscription_callbacks_.Remove(request_id); |
| } |
| |
| void PushMessagingDispatcher::OnSubscribeFromDocumentError( |
| int32_t request_id, |
| PushRegistrationStatus status) { |
| blink::WebPushSubscriptionCallbacks* callbacks = |
| subscription_callbacks_.Lookup(request_id); |
| |
| // TODO(peter): Change this back to a DCHECK before M45 branches. |
| CHECK(callbacks) << "Invalid request id received: " << request_id; |
| |
| scoped_ptr<blink::WebPushError> error(new blink::WebPushError( |
| blink::WebPushError::ErrorTypeAbort, |
| blink::WebString::fromUTF8(PushRegistrationStatusToString(status)))); |
| callbacks->onError(error.release()); |
| |
| subscription_callbacks_.Remove(request_id); |
| } |
| |
| } // namespace content |