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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/omnibox/browser/tab_group_provider.h"
#include <algorithm>
#include "base/i18n/case_conversion.h"
#include "base/strings/string_util.h"
#include "base/strings/utf_string_conversions.h"
#include "base/trace_event/trace_event.h"
#include "components/omnibox/common/omnibox_features.h"
#include "components/saved_tab_groups/public/saved_tab_group.h"
#include "components/saved_tab_groups/public/tab_group_sync_service.h"
#if BUILDFLAG(IS_ANDROID)
#include "components/browser_ui/util/android/url_constants.h"
#endif
#include "components/omnibox/browser/autocomplete_enums.h"
#include "components/omnibox/browser/autocomplete_input.h"
#include "components/omnibox/browser/autocomplete_match.h"
#include "components/omnibox/browser/autocomplete_match_classification.h"
#include "components/omnibox/browser/autocomplete_provider_client.h"
#include "components/omnibox/browser/in_memory_url_index_types.h"
#include "components/omnibox/browser/keyword_provider.h"
#include "components/omnibox/browser/match_compare.h"
#include "components/omnibox/browser/omnibox_field_trial.h"
#include "components/omnibox/browser/scoring_functor.h"
#include "components/omnibox/browser/tab_matcher.h"
#include "components/query_parser/query_parser.h"
#include "components/search_engines/template_url.h"
#include "components/strings/grit/components_strings.h"
#include "components/url_formatter/url_formatter.h"
#include "content/public/common/url_constants.h"
#include "third_party/omnibox_proto/groups.pb.h"
#include "ui/base/l10n/l10n_util.h"
#include "url/gurl.h"
namespace {
int Score(const AutocompleteInput& input,
const query_parser::QueryNodeVector& input_query_nodes,
const tab_groups::SavedTabGroup& group) {
TRACE_EVENT_BEGIN0("omnibox", "TabGroupProvider::Score");
// Extract query words from the title.
const std::u16string lower_title = base::i18n::ToLower(group.title());
query_parser::QueryWordVector title_words;
query_parser::QueryParser::ExtractQueryWords(lower_title, &title_words);
// Every input term must be included in either (or both) the title or URL.
query_parser::Snippet::MatchPositions title_matches;
query_parser::Snippet::MatchPositions url_matches;
if (!std::ranges::all_of(input_query_nodes, [&](const auto& query_node) {
// Using local vars so to not short circuit adding URL matches when
// title matches are found.
const bool has_title_match =
query_node->HasMatchIn(title_words, &title_matches);
return has_title_match;
})) {
return 0;
}
// Max score is based on these suggestions sharing a group with open tab
// matches.
const int kMaxScore = 1000;
const double title_factor =
for_each(title_matches.begin(), title_matches.end(),
ScoringFunctor(lower_title.length()))
.ScoringFactor();
const double normalized_factors =
std::min((title_factor) / (lower_title.length() + 10), 1.0);
TRACE_EVENT_END0("omnibox", "TabGroupProvider::Score");
return normalized_factors * kMaxScore;
}
} // namespace
TabGroupProvider::TabGroupProvider(AutocompleteProviderClient* client)
: AutocompleteProvider(AutocompleteProvider::TYPE_OPEN_TAB),
client_(client) {}
TabGroupProvider::~TabGroupProvider() = default;
// TODO(crbug.com/412433887): Make the TabGroupProvider async.
void TabGroupProvider::Start(const AutocompleteInput& input,
bool minimal_changes) {
Stop(AutocompleteStopReason::kClobbered);
if (input.current_page_classification() !=
::metrics::OmniboxEventProto::ANDROID_HUB) {
return;
}
TRACE_EVENT_BEGIN0("omnibox", "TabGroupProvider::Start");
// Remove the keyword from input if we're in keyword mode for a starter pack
// engine.
const auto& [adjusted_input, template_url] =
AdjustInputForStarterPackKeyword(input, client_->GetTemplateURLService());
// Preprocess the query into query nodes.
const auto adjusted_input_text = std::u16string(
base::TrimWhitespace(base::i18n::ToLower(adjusted_input.text()),
base::TrimPositions::TRIM_ALL));
query_parser::QueryNodeVector input_query_nodes;
query_parser::QueryParser::ParseQueryNodes(
adjusted_input_text,
query_parser::MatchingAlgorithm::ALWAYS_PREFIX_SEARCH,
&input_query_nodes);
for (auto& group : client_->GetTabGroupSyncService()->GetAllGroups()) {
int score = Score(input, input_query_nodes, group);
if (score > 0) {
matches_.push_back(CreateTabGroupMatch(input, group, score));
}
}
TRACE_EVENT_END0("omnibox", "TabGroupProvider::Start");
}
AutocompleteMatch TabGroupProvider::CreateTabGroupMatch(
const AutocompleteInput& input,
const tab_groups::SavedTabGroup& group,
int score) {
AutocompleteMatch match(this, score, /*deletable=*/false,
AutocompleteMatchType::TAB_GROUP);
match.contents = group.title();
auto contents_terms = FindTermMatches(input.text(), match.contents);
match.contents_class = ClassifyTermMatches(
contents_terms, match.contents.size(), ACMatchClassification::MATCH,
ACMatchClassification::NONE);
match.matching_tab_group_uuid = group.saved_guid();
match.suggestion_group_id = omnibox::GROUP_MOBILE_OPEN_TABS;
return match;
}
|
