File: profile_preload_candidate_selector.cc

package info (click to toggle)
chromium 138.0.7204.183-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 6,071,908 kB
  • sloc: cpp: 34,937,088; ansic: 7,176,967; javascript: 4,110,704; python: 1,419,953; asm: 946,768; xml: 739,971; pascal: 187,324; sh: 89,623; perl: 88,663; objc: 79,944; sql: 50,304; cs: 41,786; fortran: 24,137; makefile: 21,806; php: 13,980; tcl: 13,166; yacc: 8,925; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (100 lines) | stat: -rw-r--r-- 3,585 bytes parent folder | download | duplicates (5)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/ui/webui/top_chrome/profile_preload_candidate_selector.h"

#include <vector>

#include "base/feature_list.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/ui_features.h"
#include "chrome/browser/ui/webui/top_chrome/per_profile_webui_tracker.h"
#include "chrome/browser/ui/webui/top_chrome/top_chrome_webui_config.h"
#include "components/site_engagement/content/site_engagement_service.h"
#include "url/gurl.h"

using site_engagement::SiteEngagementService;

namespace webui {

ProfilePreloadCandidateSelector::ProfilePreloadCandidateSelector(
    PerProfileWebUITracker* webui_tracker)
    : webui_tracker_(webui_tracker) {
  no_preload_if_most_engaged_url_is_background_ = base::FeatureList::IsEnabled(
      features::kPreloadTopChromeWebUILessNavigations);
}

ProfilePreloadCandidateSelector::~ProfilePreloadCandidateSelector() = default;

void ProfilePreloadCandidateSelector::Init(
    const std::vector<GURL>& preloadable_urls) {
  preloadable_urls_ = preloadable_urls;
}

std::optional<GURL> ProfilePreloadCandidateSelector::GetURLToPreload(
    const PreloadContext& context) const {
  if (preloadable_urls_.empty()) {
    return std::nullopt;
  }

  // Sanity check that the context is a Profile.
  CHECK(context.IsProfile());
  Profile* profile = const_cast<Profile*>(context.GetProfile());
  SiteEngagementService* engagement_service =
      SiteEngagementService::Get(profile);

  // Sort the URLs by engagement score in descending order. This is O(nlogn)
  // complexity which is good enough under the assumption that the number of
  // preloadable Top Chrome WebUIs is small. We might revisit the algorithm when
  // necessary.
  std::vector<GURL> sorted_urls = preloadable_urls_;
  std::sort(sorted_urls.begin(), sorted_urls.end(),
            [&](const GURL& a, const GURL& b) {
              return engagement_service->GetScore(a) >
                     engagement_service->GetScore(b);
            });

  // Iterate over all preloable URLs to find the URL with the highest
  // engagement score and is not present under the profile.
  for (const GURL& url : sorted_urls) {
    // Skip if this URL is disabled. This is seen in History Clusters side panel
    // where some locales are not supported.
    auto* config = TopChromeWebUIConfig::From(profile, url);
    if (!config || !config->IsWebUIEnabled(profile)) {
      continue;
    }

    // Skip URLs that have low engagement score. Because the URLs are sorted by
    // engagement score, we can stop iterating once we find a URL with low
    // engagement score.
    blink::mojom::EngagementLevel engagement_level =
        engagement_service->GetEngagementLevel(url);
    if (engagement_level == blink::mojom::EngagementLevel::NONE ||
        engagement_level == blink::mojom::EngagementLevel::MINIMAL) {
      return std::nullopt;
    }

    if (IsUrlExcludedByFlag(url)) {
      continue;
    }

    // If the URL is in the background, don't preload at all.
    if (no_preload_if_most_engaged_url_is_background_ &&
         webui_tracker_->ProfileHasBackgroundWebUI(profile, url.spec())) {
      return std::nullopt;
    }

    // Skip URLs that are already present under the profile.
    if (webui_tracker_->ProfileHasWebUI(profile, url.spec())) {
      continue;
    }

    // We have found a candidate URL with the highest engagement score.
    return url;
  }

  return std::nullopt;
}

}  // namespace webui