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
|
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_ANDROID_CUSTOMTABS_TEXT_FRAGMENT_LOOKUP_STATE_TRACKER_H_
#define CHROME_BROWSER_ANDROID_CUSTOMTABS_TEXT_FRAGMENT_LOOKUP_STATE_TRACKER_H_
#include "base/gtest_prod_util.h"
#include "content/public/browser/web_contents_observer.h"
#include "content/public/browser/web_contents_user_data.h"
namespace customtabs {
class TextFragmentLookupStateTracker
: public content::WebContentsObserver,
public content::WebContentsUserData<TextFragmentLookupStateTracker> {
public:
~TextFragmentLookupStateTracker() override;
// `state_key` is opaque id used by client to keep track of the request.
// `lookup_results` is the mapping between the text fragments that were looked
// up and whether they were found on the page or not.
using OnResultCallback = base::OnceCallback<void(
const std::string& state_key,
const std::vector<std::pair<std::string, bool>>& lookup_results)>;
// Looks up whether `text_directives` are present on the page or not. Once
// lookups are finished, `on_result_callback` is called with results.
// `state_key` is opaque id used by client to keep track of the request.
void LookupTextFragment(const std::string& state_key,
const std::vector<std::string>& text_directives,
OnResultCallback on_result_callback);
void FindScrollAndHighlight(const std::string& text_directive) const;
private:
friend class content::WebContentsUserData<TextFragmentLookupStateTracker>;
FRIEND_TEST_ALL_PREFIXES(TextFragmentLookupStateTrackerTest,
ExtractAllowedTextDirectives);
explicit TextFragmentLookupStateTracker(content::WebContents* web_contents);
// Extracts the first allowed number of text directives based on the remaining
// quota of lookups of the current page.
std::vector<std::string> ExtractAllowedTextDirectives(
const std::vector<std::string>& text_directives) const;
// Implements `content::WebContentsObserver`:
void PrimaryPageChanged(content::Page& page) override;
WEB_CONTENTS_USER_DATA_KEY_DECL();
size_t lookup_count_ = 0;
};
} // namespace customtabs
#endif // CHROME_BROWSER_ANDROID_CUSTOMTABS_TEXT_FRAGMENT_LOOKUP_STATE_TRACKER_H_
|
