Introduce ChromeZoomLevelPref, make zoom level prefs independent of profile prefs.

chrome/browser/ui/zoom/chrome_zoom_level_prefs.cc

Index: chrome/browser/ui/zoom/chrome_zoom_level_prefs.cc
diff --git a/chrome/browser/ui/zoom/chrome_zoom_level_prefs.cc b/chrome/browser/ui/zoom/chrome_zoom_level_prefs.cc
new file mode 100644
index 0000000000000000000000000000000000000000..5d740b012dfc77954035988e62b15dc81fbce3a2
--- /dev/null
+++ b/chrome/browser/ui/zoom/chrome_zoom_level_prefs.cc
@@ -0,0 +1,176 @@
+// Copyright (c) 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 "chrome/browser/ui/zoom/chrome_zoom_level_prefs.h"
+
+#include <sstream>
+
+#include "base/bind.h"
+#include "base/prefs/json_pref_store.h"
+#include "base/prefs/pref_filter.h"
+#include "base/prefs/pref_registry_simple.h"
+#include "base/prefs/pref_service_factory.h"
+#include "base/prefs/scoped_user_pref_update.h"
+#include "base/values.h"
+#include "chrome/common/chrome_constants.h"
+#include "chrome/common/pref_names.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/browser/host_zoom_map.h"
+#include "content/public/common/page_zoom.h"
+
+namespace chrome {
+
+ChromeZoomLevelPrefs::ChromeZoomLevelPrefs(PrefService* pref_service,
+                                           const base::FilePath& profile_path)
+    : pref_service_(pref_service),
+      profile_path_(profile_path),
+      host_zoom_map_(NULL) {
+  DCHECK(pref_service_);
+}
+
+ChromeZoomLevelPrefs::~ChromeZoomLevelPrefs() {
+  zoom_subscription_.reset();
+  pref_change_registrar_.RemoveAll();
+}
+
+void ChromeZoomLevelPrefs::InitPrefsAndCopyToHostZoomMap(
+    const base::FilePath& partition_path,
+    content::HostZoomMap* host_zoom_map) {
+  DCHECK(!partition_path.empty());
+  DCHECK((partition_path == profile_path_) ||
+         profile_path_.IsParent(partition_path));
+  DCHECK(host_zoom_map);
+  host_zoom_map_ = host_zoom_map;
+
+  pref_change_registrar_.Init(pref_service_);
+
+  // Create a partition_key string with no '.'s in it. For the default
+  // StoragePartition, this string will always be "0".
+  base::FilePath partition_relative_path;
+  profile_path_.AppendRelativePath(partition_path, &partition_relative_path);
+  size_t int_key =
+      BASE_HASH_NAMESPACE::hash<base::FilePath>()(partition_relative_path);
+  std::stringstream key_stream;
+  key_stream << int_key;

[Bernhard Bauer, 2014/10/01 15:09:14]

I don't think we use streams.

Could you use base::IntToString (from base/strings/string_number_conversions.h)?

[wjmaclean, 2014/10/01 20:33:00]

Done.

+  partition_key_ = key_stream.str();
+  default_zoom_level_path_ = prefs::kDefaultZoomLevel;
+  default_zoom_level_path_ += "." + partition_key_;
+
+  // Initialize the default zoom level.
+  double default_zoom_level = 0.0;
+  const base::DictionaryValue* default_zoom_level_dictionary =
+      pref_service_->GetDictionary(prefs::kDefaultZoomLevel);
+  if (!default_zoom_level_dictionary ||
+      !default_zoom_level_dictionary->GetDouble(partition_key_,
+                                                &default_zoom_level)) {
+    // We will only get here if the default zoom level for this partition
+    // has never been set before.
+    SetDefaultZoomLevel(default_zoom_level);
+  }
+  host_zoom_map_->SetDefaultZoomLevel(default_zoom_level);
+
+  // Only sign up for notifications if the default zoom level for this
+  // partition has been registered.
+  if (pref_service_->FindPreference(default_zoom_level_path_.c_str())) {
+    pref_change_registrar_.Add(
+        default_zoom_level_path_.c_str(),
+        base::Bind(&ChromeZoomLevelPrefs::OnDefaultZoomLevelChanged,
+                   base::Unretained(this)));
+  }
+
+  // Initialize the HostZoomMap with per-host zoom levels from the persisted
+  // zoom-level preference values.
+  const base::DictionaryValue* host_zoom_dictionaries =
+      pref_service_->GetDictionary(prefs::kPerHostZoomLevels);
+  const base::DictionaryValue* host_zoom_dictionary;
+  if (host_zoom_dictionaries &&
+      host_zoom_dictionaries->GetDictionary(partition_key_,
+                                            &host_zoom_dictionary)) {
+    std::vector<std::string> keys_to_remove;
+    for (base::DictionaryValue::Iterator i(*host_zoom_dictionary); !i.IsAtEnd();
+         i.Advance()) {
+      const std::string& host(i.key());
+      double zoom_level = 0;
+
+      bool success = i.value().GetAsDouble(&zoom_level);
+      DCHECK(success);
+
+      if (host.empty() ||
+          content::ZoomValuesEqual(zoom_level,
+                                   host_zoom_map_->GetDefaultZoomLevel())) {
+        keys_to_remove.push_back(host);
+        continue;
+      }
+
+      host_zoom_map_->SetZoomLevelForHost(host, zoom_level);
+    }
+
+    // Sanitize prefs to remove entries that match the default zoom level and/or
+    // have an empty host.
+    {
+      DictionaryPrefUpdate update(pref_service_, prefs::kPerHostZoomLevels);
+      base::DictionaryValue* host_zoom_dictionaries = update.Get();
+      base::DictionaryValue* host_zoom_dictionary;
+      if (!host_zoom_dictionaries->GetDictionary(partition_key_,
+                                                 &host_zoom_dictionary)) {
+        host_zoom_dictionary = new base::DictionaryValue();
+        host_zoom_dictionaries->Set(partition_key_, host_zoom_dictionary);
+      }
+
+      for (std::vector<std::string>::const_iterator it = keys_to_remove.begin();
+           it != keys_to_remove.end();
+           ++it) {
+        host_zoom_dictionary->RemoveWithoutPathExpansion(*it, NULL);
+      }
+    }
+  }
+  zoom_subscription_ = host_zoom_map_->AddZoomLevelChangedCallback(base::Bind(
+      &ChromeZoomLevelPrefs::OnZoomLevelChanged, base::Unretained(this)));
+}
+
+void ChromeZoomLevelPrefs::OnDefaultZoomLevelChanged(const std::string& key) {
+  // Push changes to the default zoom level to the HostZoomMap; all
+  // other consumers will get it from there.
+  if (key == default_zoom_level_path_) {
+    host_zoom_map_->SetDefaultZoomLevel(
+        pref_service_->GetDouble(default_zoom_level_path_.c_str()));
+  }
+}
+
+void ChromeZoomLevelPrefs::SetDefaultZoomLevel(double level) {
+  if (pref_service_->FindPreference(default_zoom_level_path_.c_str())) {
+    pref_service_->SetDouble(default_zoom_level_path_.c_str(), level);
+  } else {
+    DictionaryPrefUpdate update(pref_service_, prefs::kDefaultZoomLevel);
+    update->SetDouble(partition_key_, level);
+    // For unregistered paths, OnDefaultZoomLevelChanged won't be called, so
+    // set this manually.
+    host_zoom_map_->SetDefaultZoomLevel(level);
+  }
+}
+
+void ChromeZoomLevelPrefs::OnZoomLevelChanged(
+    const content::HostZoomMap::ZoomLevelChange& change) {
+  if (change.mode != content::HostZoomMap::ZOOM_CHANGED_FOR_HOST)
+    return;
+  double level = change.zoom_level;
+  // Note: the following will create the dictionary if it doesn't already exist.
+  DictionaryPrefUpdate update(pref_service_, prefs::kPerHostZoomLevels);
+  base::DictionaryValue* host_zoom_dictionaries = update.Get();
+  DCHECK(host_zoom_dictionaries);
+
+  base::DictionaryValue* host_zoom_dictionary;
+  if (!host_zoom_dictionaries->GetDictionary(partition_key_,
+                                             &host_zoom_dictionary)) {
+    host_zoom_dictionary = new base::DictionaryValue();
+    host_zoom_dictionaries->Set(partition_key_, host_zoom_dictionary);
+  }
+
+  if (content::ZoomValuesEqual(level, host_zoom_map_->GetDefaultZoomLevel()))
+    host_zoom_dictionary->RemoveWithoutPathExpansion(change.host, NULL);
+  else
+    host_zoom_dictionary->SetDoubleWithoutPathExpansion(change.host, level);
+}
+
+}  // namespace chrome