| // 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/browser/push_messaging_router.h" |
| |
| #include "base/bind.h" |
| #include "content/browser/service_worker/service_worker_context_wrapper.h" |
| #include "content/browser/service_worker/service_worker_registration.h" |
| #include "content/browser/service_worker/service_worker_storage.h" |
| #include "content/public/browser/browser_context.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "content/public/browser/storage_partition.h" |
| |
| namespace content { |
| |
| // static |
| void PushMessagingRouter::DeliverMessage( |
| BrowserContext* browser_context, |
| const GURL& origin, |
| int64 service_worker_registration_id, |
| const std::string& data, |
| const DeliverMessageCallback& deliver_message_callback) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| StoragePartition* partition = |
| BrowserContext::GetStoragePartitionForSite(browser_context, origin); |
| scoped_refptr<ServiceWorkerContextWrapper> service_worker_context = |
| static_cast<ServiceWorkerContextWrapper*>( |
| partition->GetServiceWorkerContext()); |
| BrowserThread::PostTask( |
| BrowserThread::IO, |
| FROM_HERE, |
| base::Bind(&PushMessagingRouter::FindServiceWorkerRegistration, |
| origin, |
| service_worker_registration_id, |
| data, |
| deliver_message_callback, |
| service_worker_context)); |
| } |
| |
| // static |
| void PushMessagingRouter::FindServiceWorkerRegistration( |
| const GURL& origin, |
| int64 service_worker_registration_id, |
| const std::string& data, |
| const DeliverMessageCallback& deliver_message_callback, |
| scoped_refptr<ServiceWorkerContextWrapper> service_worker_context) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
| // Try to acquire the registration from storage. If it's already live we'll |
| // receive it right away. If not, it will be revived from storage. |
| service_worker_context->context()->storage()->FindRegistrationForId( |
| service_worker_registration_id, |
| origin, |
| base::Bind(&PushMessagingRouter::FindServiceWorkerRegistrationCallback, |
| data, |
| deliver_message_callback)); |
| } |
| |
| // static |
| void PushMessagingRouter::FindServiceWorkerRegistrationCallback( |
| const std::string& data, |
| const DeliverMessageCallback& deliver_message_callback, |
| ServiceWorkerStatusCode service_worker_status, |
| const scoped_refptr<ServiceWorkerRegistration>& |
| service_worker_registration) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
| if (service_worker_status == SERVICE_WORKER_OK) { |
| // Hold on to the service worker registration in the callback to keep it |
| // alive until the callback dies. Otherwise the registration could be |
| // released when this method returns - before the event is delivered to the |
| // service worker. |
| base::Callback<void(ServiceWorkerStatusCode)> dispatch_event_callback = |
| base::Bind(&PushMessagingRouter::DeliverMessageEnd, |
| deliver_message_callback, |
| service_worker_registration); |
| service_worker_registration->active_version()->DispatchPushEvent( |
| dispatch_event_callback, data); |
| } else { |
| // TODO(mvanouwerkerk): UMA logging. |
| BrowserThread::PostTask( |
| BrowserThread::UI, |
| FROM_HERE, |
| base::Bind( |
| deliver_message_callback, |
| PUSH_MESSAGING_STATUS_MESSAGE_DELIVERY_FAILED_NO_SERVICE_WORKER)); |
| } |
| } |
| |
| // static |
| void PushMessagingRouter::DeliverMessageEnd( |
| const DeliverMessageCallback& deliver_message_callback, |
| const scoped_refptr<ServiceWorkerRegistration>& service_worker_registration, |
| ServiceWorkerStatusCode service_worker_status) { |
| DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
| // TODO(mvanouwerkerk): UMA logging. |
| PushMessagingStatus push_messaging_status = |
| service_worker_status == SERVICE_WORKER_OK |
| ? PUSH_MESSAGING_STATUS_OK |
| : PUSH_MESSAGING_STATUS_MESSAGE_DELIVERY_FAILED_SERVICE_WORKER_ERROR; |
| BrowserThread::PostTask( |
| BrowserThread::UI, |
| FROM_HERE, |
| base::Bind(deliver_message_callback, push_messaging_status)); |
| } |
| |
| } // namespace content |