Refactor MessageCenterSettingsController.

chrome/browser/notifications/web_page_notifier_source.cc

Index: chrome/browser/notifications/web_page_notifier_source.cc
diff --git a/chrome/browser/notifications/web_page_notifier_source.cc b/chrome/browser/notifications/web_page_notifier_source.cc
new file mode 100644
index 0000000000000000000000000000000000000000..74b54cdc697f56a63d10ec477b7dda5260014628
--- /dev/null
+++ b/chrome/browser/notifications/web_page_notifier_source.cc
@@ -0,0 +1,144 @@
+// Copyright (c) 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/notifications/web_page_notifier_source.h"
+
+#include "base/strings/utf_string_conversions.h"
+#include "base/task/cancelable_task_tracker.h"
+#include "chrome/browser/content_settings/host_content_settings_map_factory.h"
+#include "chrome/browser/favicon/favicon_service_factory.h"
+#include "chrome/browser/notifications/desktop_notification_profile_util.h"
+#include "chrome/browser/notifications/notifier_state_tracker.h"
+#include "chrome/browser/notifications/notifier_state_tracker_factory.h"
+#include "components/content_settings/core/browser/host_content_settings_map.h"
+#include "components/content_settings/core/common/content_settings.h"
+#include "components/favicon/core/favicon_service.h"
+
+WebPageNotifiereSource::WebPageNotifiereSource(Observer* observer)
+    : observer_(observer) {}
+
+WebPageNotifiereSource::~WebPageNotifiereSource() {}
+
+std::vector<std::unique_ptr<message_center::Notifier>>
+WebPageNotifiereSource::GetNotifierList(Profile* profile) {
+  std::vector<std::unique_ptr<message_center::Notifier>> notifiers;
+
+  ContentSettingsForOneType settings;
+  DesktopNotificationProfileUtil::GetNotificationsSettings(profile, &settings);
+
+  favicon::FaviconService* const favicon_service =
+      FaviconServiceFactory::GetForProfile(profile,
+                                           ServiceAccessType::EXPLICIT_ACCESS);
+  favicon_tracker_.reset(new base::CancelableTaskTracker());
+  patterns_.clear();
+  for (ContentSettingsForOneType::const_iterator iter = settings.begin();
+       iter != settings.end(); ++iter) {
+    if (iter->primary_pattern == ContentSettingsPattern::Wildcard() &&
+        iter->secondary_pattern == ContentSettingsPattern::Wildcard() &&
+        iter->source != "preference") {
+      continue;
+    }
+
+    std::string url_pattern = iter->primary_pattern.ToString();
+    base::string16 name = base::UTF8ToUTF16(url_pattern);
+    GURL url(url_pattern);
+    message_center::NotifierId notifier_id(url);
+    NotifierStateTracker* const notifier_state_tracker =
+        NotifierStateTrackerFactory::GetForProfile(profile);
+    notifiers.emplace_back(new message_center::Notifier(
+        notifier_id, name,
+        notifier_state_tracker->IsNotifierEnabled(notifier_id)));
+    patterns_[name] = iter->primary_pattern;
+    // Note that favicon service obtains the favicon from history. This means
+    // that it will fail to obtain the image if there are no history data for
+    // that URL.
+    favicon_service->GetFaviconImageForPageURL(
+        url, base::Bind(&WebPageNotifiereSource::OnFaviconLoaded,
+                        base::Unretained(this), url),
+        favicon_tracker_.get());
+  }
+
+  return notifiers;
+}
+
+void WebPageNotifiereSource::SetNotifierEnabled(
+    Profile* profile,
+    const message_center::Notifier& notifier,
+    bool enabled) {
+  // WEB_PAGE notifier cannot handle in DesktopNotificationService
+  // since it has the exact URL pattern.
+  // TODO(mukai): fix this.
+  ContentSetting default_setting =
+      HostContentSettingsMapFactory::GetForProfile(profile)
+          ->GetDefaultContentSetting(CONTENT_SETTINGS_TYPE_NOTIFICATIONS, NULL);
+
+  DCHECK(default_setting == CONTENT_SETTING_ALLOW ||
+         default_setting == CONTENT_SETTING_BLOCK ||
+         default_setting == CONTENT_SETTING_ASK);
+
+  // The content setting for notifications needs to clear when it changes to
+  // the default value or get explicitly set when it differs from the
+  // default.
+  bool differs_from_default_value =
+      (default_setting != CONTENT_SETTING_ALLOW && enabled) ||
+      (default_setting == CONTENT_SETTING_ALLOW && !enabled);
+
+  if (differs_from_default_value) {
+    if (notifier.notifier_id.url.is_valid()) {
+      if (enabled) {
+        DesktopNotificationProfileUtil::GrantPermission(
+            profile, notifier.notifier_id.url);
+      } else {
+        DesktopNotificationProfileUtil::DenyPermission(
+            profile, notifier.notifier_id.url);
+      }
+    } else {
+      LOG(ERROR) << "Invalid url pattern: " << notifier.notifier_id.url.spec();
+    }
+  } else {
+    ContentSettingsPattern pattern;
+
+    const auto& iter = patterns_.find(notifier.name);
+    if (iter != patterns_.end()) {
+      pattern = iter->second;
+    } else if (notifier.notifier_id.url.is_valid()) {
+      pattern =
+          ContentSettingsPattern::FromURLNoWildcard(notifier.notifier_id.url);
+    } else {
+      LOG(ERROR) << "Invalid url pattern: " << notifier.notifier_id.url.spec();
+    }
+
+    if (pattern.IsValid()) {
+      // Note that we don't use
+      // DesktopNotificationProfileUtil::ClearSetting()
+      // here because pattern might be from user manual input and not match
+      // the default one used by ClearSetting().
+      HostContentSettingsMapFactory::GetForProfile(profile)
+          ->SetContentSettingCustomScope(
+              pattern, ContentSettingsPattern::Wildcard(),
+              CONTENT_SETTINGS_TYPE_NOTIFICATIONS,
+              content_settings::ResourceIdentifier(), CONTENT_SETTING_DEFAULT);
+    }
+  }
+
+  observer_->OnNotifierEnabledChanged(notifier.notifier_id, enabled);
+}
+
+void WebPageNotifiereSource::OnNotifierSettingsClosing() {
+  DCHECK(favicon_tracker_.get());
+  favicon_tracker_->TryCancelAll();
+  patterns_.clear();
+}
+
+message_center::NotifierId::NotifierType
+WebPageNotifiereSource::GetNotifierType() {
+  return message_center::NotifierId::WEB_PAGE;
+}
+
+void WebPageNotifiereSource::OnFaviconLoaded(
+    const GURL& url,
+    const favicon_base::FaviconImageResult& favicon_result) {
+  observer_->OnIconImageUpdated(message_center::NotifierId(url),
+                                favicon_result.image);
+}