| // Copyright 2015 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/public/browser/push_messaging_service.h" |
| |
| #include "base/bind.h" |
| #include "base/callback.h" |
| #include "base/task/post_task.h" |
| #include "content/browser/push_messaging/push_messaging_manager.h" |
| #include "content/browser/service_worker/service_worker_context_wrapper.h" |
| #include "content/public/browser/browser_context.h" |
| #include "content/public/browser/browser_task_traits.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "content/public/browser/storage_partition.h" |
| |
| namespace content { |
| |
| namespace { |
| |
| void CallStringCallbackFromIO( |
| const PushMessagingService::StringCallback& callback, |
| const std::vector<std::string>& data, |
| blink::ServiceWorkerStatusCode service_worker_status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| bool success = service_worker_status == blink::ServiceWorkerStatusCode::kOk; |
| bool not_found = |
| service_worker_status == blink::ServiceWorkerStatusCode::kErrorNotFound; |
| std::string result; |
| if (success) { |
| DCHECK_EQ(1u, data.size()); |
| result = data[0]; |
| } |
| base::PostTask(FROM_HERE, {BrowserThread::UI}, |
| base::BindOnce(callback, result, success, not_found)); |
| } |
| |
| void CallClosureFromIO(base::OnceClosure callback, |
| blink::ServiceWorkerStatusCode status) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| base::PostTask(FROM_HERE, {BrowserThread::UI}, std::move(callback)); |
| } |
| |
| void GetUserDataOnIO( |
| scoped_refptr<ServiceWorkerContextWrapper> service_worker_context_wrapper, |
| int64_t service_worker_registration_id, |
| const std::string& key, |
| const PushMessagingService::StringCallback& callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| service_worker_context_wrapper->GetRegistrationUserData( |
| service_worker_registration_id, {key}, |
| base::BindOnce(&CallStringCallbackFromIO, callback)); |
| } |
| |
| void ClearPushSubscriptionIdOnIO( |
| scoped_refptr<ServiceWorkerContextWrapper> service_worker_context, |
| int64_t service_worker_registration_id, |
| base::OnceClosure callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| |
| service_worker_context->ClearRegistrationUserData( |
| service_worker_registration_id, {kPushRegistrationIdServiceWorkerKey}, |
| base::BindOnce(&CallClosureFromIO, std::move(callback))); |
| } |
| |
| void StorePushSubscriptionOnIOForTesting( |
| scoped_refptr<ServiceWorkerContextWrapper> service_worker_context, |
| int64_t service_worker_registration_id, |
| const GURL& origin, |
| const std::string& subscription_id, |
| const std::string& sender_id, |
| base::OnceClosure callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::IO); |
| |
| service_worker_context->StoreRegistrationUserData( |
| service_worker_registration_id, origin, |
| {{kPushRegistrationIdServiceWorkerKey, subscription_id}, |
| {kPushSenderIdServiceWorkerKey, sender_id}}, |
| base::BindOnce(&CallClosureFromIO, std::move(callback))); |
| } |
| |
| scoped_refptr<ServiceWorkerContextWrapper> GetServiceWorkerContext( |
| BrowserContext* browser_context, const GURL& origin) { |
| StoragePartition* partition = |
| BrowserContext::GetStoragePartitionForSite(browser_context, origin); |
| return base::WrapRefCounted(static_cast<ServiceWorkerContextWrapper*>( |
| partition->GetServiceWorkerContext())); |
| } |
| |
| } // anonymous namespace |
| |
| // static |
| void PushMessagingService::GetSenderId(BrowserContext* browser_context, |
| const GURL& origin, |
| int64_t service_worker_registration_id, |
| const StringCallback& callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| base::PostTask( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(&GetUserDataOnIO, |
| GetServiceWorkerContext(browser_context, origin), |
| service_worker_registration_id, |
| kPushSenderIdServiceWorkerKey, callback)); |
| } |
| |
| // static |
| void PushMessagingService::ClearPushSubscriptionId( |
| BrowserContext* browser_context, |
| const GURL& origin, |
| int64_t service_worker_registration_id, |
| base::OnceClosure callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| base::PostTask( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(&ClearPushSubscriptionIdOnIO, |
| GetServiceWorkerContext(browser_context, origin), |
| service_worker_registration_id, std::move(callback))); |
| } |
| |
| // static |
| void PushMessagingService::StorePushSubscriptionForTesting( |
| BrowserContext* browser_context, |
| const GURL& origin, |
| int64_t service_worker_registration_id, |
| const std::string& subscription_id, |
| const std::string& sender_id, |
| const base::Closure& callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| base::PostTask( |
| FROM_HERE, {BrowserThread::IO}, |
| base::BindOnce(&StorePushSubscriptionOnIOForTesting, |
| GetServiceWorkerContext(browser_context, origin), |
| service_worker_registration_id, origin, subscription_id, |
| sender_id, callback)); |
| } |
| |
| } // namespace content |