PushMessagingNotificationManager: utility function to get StoragePartition.

chrome/browser/push_messaging/push_messaging_notification_manager.cc

 // 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 "chrome/browser/push_messaging/push_messaging_notification_manager.h"
 
 #include <bitset>
 
 #include "base/metrics/histogram_macros.h"
 #include "base/prefs/pref_service.h"
@@ skipping 30 matching lines @@
 using content::BrowserThread;
 
 namespace {
 
 void RecordUserVisibleStatus(content::PushUserVisibleStatus status) {
   UMA_HISTOGRAM_ENUMERATION("PushMessaging.UserVisibleStatus",
                             status,
                             content::PUSH_USER_VISIBLE_STATUS_LAST + 1);
 }
 
+content::StoragePartition* GetStoragePartition(Profile* profile,
+                                               const GURL& origin) {
+  return content::BrowserContext::GetStoragePartitionForSite(profile, origin);
+}
+
 }  // namespace
 
 PushMessagingNotificationManager::PushMessagingNotificationManager(
     Profile* profile)
     : profile_(profile),
       weak_factory_(this) {}
 
 PushMessagingNotificationManager::~PushMessagingNotificationManager() {}
 
 void PushMessagingNotificationManager::EnforceUserVisibleOnlyRequirements(
     const GURL& requesting_origin, int64_t service_worker_registration_id,
     const base::Closure& message_handled_closure) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
   // TODO(johnme): Relax this heuristic slightly.
   scoped_refptr<content::PlatformNotificationContext> notification_context =
-      content::BrowserContext::GetStoragePartitionForSite(
-          profile_, requesting_origin)->GetPlatformNotificationContext();
+      GetStoragePartition(profile_, requesting_origin)
+          ->GetPlatformNotificationContext();
   BrowserThread::PostTask(
       BrowserThread::IO, FROM_HERE,
       base::Bind(
           &content::PlatformNotificationContext
                   ::ReadAllNotificationDataForServiceWorkerRegistration,
           notification_context,
           requesting_origin, service_worker_registration_id,
           base::Bind(
               &PushMessagingNotificationManager
                   ::DidGetNotificationsFromDatabaseIOProxy,
@@ skipping 96 matching lines @@
           notification_database_data.origin, false /* by_user */);
 
       break;
     }
   }
 
   // Don't track push messages that didn't show a notification but were exempt
   // from needing to do so.
   if (notification_shown || notification_needed) {
     content::ServiceWorkerContext* service_worker_context =
-        content::BrowserContext::GetStoragePartitionForSite(
-            profile_, requesting_origin)->GetServiceWorkerContext();
+        GetStoragePartition(profile_, requesting_origin)
+            ->GetServiceWorkerContext();
 
     content::PushMessagingService::GetNotificationsShownByLastFewPushes(
         service_worker_context, service_worker_registration_id,
         base::Bind(&PushMessagingNotificationManager
                        ::DidGetNotificationsShownAndNeeded,
                    weak_factory_.GetWeakPtr(),
                    requesting_origin, service_worker_registration_id,
                    notification_shown, notification_needed,
                    message_handled_closure));
   } else {
     RecordUserVisibleStatus(
         content::PUSH_USER_VISIBLE_STATUS_NOT_REQUIRED_AND_NOT_SHOWN);
     message_handled_closure.Run();
   }
 }
 
 static void IgnoreResult(bool unused) {
 }
 
 void PushMessagingNotificationManager::DidGetNotificationsShownAndNeeded(
     const GURL& requesting_origin, int64_t service_worker_registration_id,
     bool notification_shown, bool notification_needed,
     const base::Closure& message_handled_closure,
     const std::string& data, bool success, bool not_found) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
   content::ServiceWorkerContext* service_worker_context =
-      content::BrowserContext::GetStoragePartitionForSite(
-          profile_, requesting_origin)->GetServiceWorkerContext();
+      GetStoragePartition(profile_, requesting_origin)
+          ->GetServiceWorkerContext();
 
   // We remember whether the last (up to) 10 pushes showed notifications.
   const size_t MISSED_NOTIFICATIONS_LENGTH = 10;
   // data is a string like "0001000", where '0' means shown, and '1' means
   // needed but not shown. We manipulate it in bitset form.
   std::bitset<MISSED_NOTIFICATIONS_LENGTH> missed_notifications(data);
 
   DCHECK(notification_shown || notification_needed);  // Caller must ensure this
   bool needed_but_not_shown = notification_needed && !notification_shown;
 
@@ skipping 50 matching lines @@
   notification_data.icon = GURL();
   notification_data.silent = true;
 
   content::NotificationDatabaseData database_data;
   database_data.origin = requesting_origin;
   database_data.service_worker_registration_id =
       service_worker_registration_id;
   database_data.notification_data = notification_data;
 
   scoped_refptr<content::PlatformNotificationContext> notification_context =
-    content::BrowserContext::GetStoragePartitionForSite(
-        profile_, requesting_origin)->GetPlatformNotificationContext();
+      GetStoragePartition(profile_, requesting_origin)
+          ->GetPlatformNotificationContext();
   BrowserThread::PostTask(
       BrowserThread::IO, FROM_HERE,
       base::Bind(
           &content::PlatformNotificationContext::WriteNotificationData,
           notification_context,
           requesting_origin, database_data,
           base::Bind(&PushMessagingNotificationManager
                          ::DidWriteNotificationDataIOProxy,
                      weak_factory_.GetWeakPtr(),
                      requesting_origin, notification_data,
@@ skipping 30 matching lines @@
     return;
   }
   PlatformNotificationServiceImpl::GetInstance()->DisplayPersistentNotification(
       profile_,
       persistent_notification_id,
       requesting_origin,
       SkBitmap() /* icon */,
       notification_data);
   message_handled_closure.Run();
 }