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
|
// 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_UI_TABS_ORGANIZATION_TAB_SENSITIVITY_CACHE_H_
#define CHROME_BROWSER_UI_TABS_ORGANIZATION_TAB_SENSITIVITY_CACHE_H_
#include "base/memory/raw_ptr.h"
#include "chrome/browser/profiles/profile.h"
#include "components/page_content_annotations/core/page_content_annotations_service.h"
#include "url/gurl.h"
// In order to have GURL as a key in a hashmap, GURL hashing mechanism is
// needed.
struct GURLHash {
size_t operator()(const GURL& url) const {
return std::hash<std::string>()(url.spec());
}
};
// Adapts the sensitivity scores from the PageContentAnnotationService into a
// synchronously available form, for specifically the currently open tabs, by
// caching scores emitted to PageContentAnnotationsObservers.
class TabSensitivityCache final
: public page_content_annotations::PageContentAnnotationsService::
PageContentAnnotationsObserver {
public:
explicit TabSensitivityCache(Profile* profile);
~TabSensitivityCache() override;
// Returns the sensitivity score for `url`, if it currently has one. The score
// is a float from 0 to 1, where 1 is most likely to be sensitive.
std::optional<float> GetScore(const GURL& url) const;
// page_content_annotations
// ::PageContentAnnotationsService
// ::PageContentAnnotationsObserver
void OnPageContentAnnotated(
const GURL& url,
const page_content_annotations::PageContentAnnotationsResult& result)
override;
private:
void MaybeTrimCacheKeys();
raw_ptr<Profile> profile_;
std::unordered_map<GURL, float, GURLHash> sensitivy_scores_;
};
#endif // CHROME_BROWSER_UI_TABS_ORGANIZATION_TAB_SENSITIVITY_CACHE_H_
|
