Background service worker push message processing

chrome/browser/push_messaging/background_budget_service.cc

+// Copyright 2016 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/background_budget_service.h"
+
+#include "base/callback.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/common/pref_names.h"
+#include "components/pref_registry/pref_registry_syncable.h"
+#include "components/prefs/pref_service.h"
+#include "components/prefs/scoped_user_pref_update.h"
+
+namespace {
+
+std::string GetBudgetStringForOrigin(Profile* profile, const GURL& origin) {
+  DCHECK_EQ(origin.GetOrigin(), origin);
+  const base::DictionaryValue* map =
+      profile->GetPrefs()->GetDictionary(prefs::kBackgroundBudgetMap);
+
+  std::string map_string;
+  map->GetStringWithoutPathExpansion(origin.spec(), &map_string);
+  return map_string;
+}
+
+}  // namespace
+
+// static
+void BackgroundBudgetService::RegisterProfilePrefs(
+    user_prefs::PrefRegistrySyncable* registry) {
+  registry->RegisterDictionaryPref(prefs::kBackgroundBudgetMap);
+}
+
+// static
+std::string BackgroundBudgetService::GetBudget(Profile* profile,
+                                               const GURL& origin) {
+  return GetBudgetStringForOrigin(profile, origin);
+}
+
+// static
+void BackgroundBudgetService::StoreBudget(
+    Profile* profile,
+    const GURL& origin,
+    const std::string& notifications_shown) {
+  DictionaryPrefUpdate update(profile->GetPrefs(), prefs::kBackgroundBudgetMap);
+  base::DictionaryValue* map = update.Get();
+  map->SetStringWithoutPathExpansion(origin.spec(), notifications_shown);
+}