Plumb site engagement to the renderer process.

chrome/browser/engagement/important_sites_util.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/engagement/important_sites_util.h"
 
 #include <algorithm>
 #include <map>
 #include <memory>
 #include <set>
 #include <utility>
 
 #include "base/containers/hash_tables.h"
 #include "base/memory/ptr_util.h"
 #include "base/metrics/histogram_macros.h"
 #include "base/stl_util.h"
 #include "base/time/time.h"
 #include "base/values.h"
 #include "chrome/browser/banners/app_banner_settings_helper.h"
 #include "chrome/browser/bookmarks/bookmark_model_factory.h"
 #include "chrome/browser/content_settings/host_content_settings_map_factory.h"
 #include "chrome/browser/engagement/site_engagement_score.h"
 #include "chrome/browser/engagement/site_engagement_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "components/bookmarks/browser/bookmark_model.h"
 #include "components/content_settings/core/browser/host_content_settings_map.h"
 #include "components/content_settings/core/common/content_settings.h"
 #include "net/base/registry_controlled_domains/registry_controlled_domain.h"
+#include "third_party/WebKit/public/platform/site_engagement.mojom.h"
 #include "url/gurl.h"
 
 namespace {
 using bookmarks::BookmarkModel;
 using ImportantDomainInfo = ImportantSitesUtil::ImportantDomainInfo;
 
 static const char kNumTimesIgnoredName[] = "NumTimesIgnored";
 static const int kTimesIgnoredForBlacklist = 3;
 
 // These are the maximum # of bookmarks we can use as signals. If the user has
@@ skipping 164 matching lines @@
       continue;
     }
 
     ignoring_domains.insert(origin.host());
   }
   return ignoring_domains;
 }
 
 void PopulateInfoMapWithSiteEngagement(
     Profile* profile,
-    SiteEngagementService::EngagementLevel minimum_engagement,
+    blink::mojom::EngagementLevel minimum_engagement,
     std::map<GURL, double>* engagement_map,
     base::hash_map<std::string, ImportantDomainInfo>* output) {
   SiteEngagementService* service = SiteEngagementService::Get(profile);
   *engagement_map = service->GetScoreMap();
   // We can have multiple origins for a single domain, so we record the one
   // with the highest engagement score.
   for (const auto& url_engagement_pair : *engagement_map) {
     if (!service->IsEngagementAtLeast(url_engagement_pair.first,
                                       minimum_engagement)) {
       continue;
@@ skipping 44 matching lines @@
   model->GetBookmarks(&untrimmed_bookmarks);
 
   // Process the bookmarks and optionally trim them if we have too many.
   std::vector<BookmarkModel::URLAndTitle> result_bookmarks;
   if (untrimmed_bookmarks.size() > kMaxBookmarks) {
     std::copy_if(untrimmed_bookmarks.begin(), untrimmed_bookmarks.end(),
                  std::back_inserter(result_bookmarks),
                  [service](const BookmarkModel::URLAndTitle& entry) {
                    return service->IsEngagementAtLeast(
                        entry.url.GetOrigin(),
-                       SiteEngagementService::ENGAGEMENT_LEVEL_LOW);
+                       blink::mojom::EngagementLevel::LOW);
                  });
     std::sort(result_bookmarks.begin(), result_bookmarks.end(),
               [&engagement_map](const BookmarkModel::URLAndTitle& a,
                                 const BookmarkModel::URLAndTitle& b) {
                 double a_score = engagement_map.at(a.url.GetOrigin());
                 double b_score = engagement_map.at(b.url.GetOrigin());
                 return a_score > b_score;
               });
     if (result_bookmarks.size() > kMaxBookmarks)
       result_bookmarks.resize(kMaxBookmarks);
@@ skipping 30 matching lines @@
 
 }  // namespace
 
 std::vector<ImportantDomainInfo>
 ImportantSitesUtil::GetImportantRegisterableDomains(Profile* profile,
                                                     size_t max_results) {
   base::hash_map<std::string, ImportantDomainInfo> important_info;
   std::map<GURL, double> engagement_map;
 
   PopulateInfoMapWithSiteEngagement(
-      profile, SiteEngagementService::ENGAGEMENT_LEVEL_MEDIUM, &engagement_map,
+      profile, blink::mojom::EngagementLevel::MEDIUM, &engagement_map,
       &important_info);
 
   PopulateInfoMapWithContentTypeAllowed(
       profile, CONTENT_SETTINGS_TYPE_NOTIFICATIONS,
       ImportantReason::NOTIFICATIONS, &important_info);
 
   PopulateInfoMapWithContentTypeAllowed(
       profile, CONTENT_SETTINGS_TYPE_DURABLE_STORAGE, ImportantReason::DURABLE,
       &important_info);
 
@@ skipping 91 matching lines @@
 
 void ImportantSitesUtil::MarkOriginAsImportantForTesting(Profile* profile,
                                                          const GURL& origin) {
   SiteEngagementScore::SetParamValuesForTesting();
   // First get data from site engagement.
   SiteEngagementService* site_engagement_service =
       SiteEngagementService::Get(profile);
   site_engagement_service->ResetScoreForURL(
       origin, SiteEngagementScore::GetMediumEngagementBoundary());
   DCHECK(site_engagement_service->IsEngagementAtLeast(
-      origin, SiteEngagementService::ENGAGEMENT_LEVEL_MEDIUM));
+      origin, blink::mojom::EngagementLevel::MEDIUM));
 }