Count site data size for important sites

chrome/browser/engagement/important_sites_util.h

 // 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.
 
 #ifndef CHROME_BROWSER_ENGAGEMENT_IMPORTANT_SITES_UTIL_H_
 #define CHROME_BROWSER_ENGAGEMENT_IMPORTANT_SITES_UTIL_H_
 
 #include <string>
 #include <vector>
 
 #include "base/macros.h"
 #include "url/gurl.h"
 
 class Profile;
 
 namespace user_prefs {
 class PrefRegistrySyncable;
 }
 
+// Helper methods for important sites.
+// All methods should be used on the UI thread.
 class ImportantSitesUtil {
  public:
+#if defined(OS_ANDROID)
+  static const int kMaxImportantSites = 5;
+#else
+  static const int kMaxImportantSites = 10;
+#endif
+
   struct ImportantDomainInfo {
     std::string registerable_domain;
     GURL example_origin;
     double engagement_score = 0;
     int32_t reason_bitfield = 0;
+    // |usage| has to be initialized by ImportantSitesUsageCounter before it
+    // will contain the number of bytes used for quota and localstorage.
+    int64_t usage = 0;
   };
 
   // Do not change the values here, as they are used for UMA histograms.
   enum ImportantReason {
     ENGAGEMENT = 0,
     DURABLE = 1,
     BOOKMARKS = 2,
     HOME_SCREEN = 3,
     NOTIFICATIONS = 4,
     REASON_BOUNDARY
   };
 
+  static std::string GetRegisterableDomainOrIP(const GURL& url);

[Bernhard Bauer, 2017/05/04 14:16:45]

Nit: "registrable" is far more common than "registerable".

[dullweber, 2017/05/05 08:35:28]

"registerable" was already used everywhere in this file, I just moved this
function from the private namespace to the class. Is it ok to leave it this way?

+
+  static std::string GetRegisterableDomainOrIPFromHost(base::StringPiece host);
+
   static bool IsDialogDisabled(Profile* profile);
 
   static void RegisterProfilePrefs(user_prefs::PrefRegistrySyncable* registry);
 
   // This returns the top |<=max_results| important registrable domains. This
   // uses site engagement and notifications to generate the list. |max_results|
   // is assumed to be small.
   // See net/base/registry_controlled_domains/registry_controlled_domain.h for
   // more details on registrable domains and the current list of effective
   // eTLDs.
@@ skipping 18 matching lines @@
   // the score requirements for the Site Engagement Service, so ONLY call for
   // testing.
   static void MarkOriginAsImportantForTesting(Profile* profile,
                                               const GURL& origin);
 
  private:
   DISALLOW_IMPLICIT_CONSTRUCTORS(ImportantSitesUtil);
 };
 
 #endif  // CHROME_BROWSER_ENGAGEMENT_IMPORTANT_SITES_UTIL_H_