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
|
// Copyright 2017 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/browsing_data/core/counters/bookmark_counter.h"
#include "base/functional/bind.h"
#include "base/memory/raw_ptr.h"
#include "base/scoped_observation.h"
#include "components/bookmarks/browser/base_bookmark_model_observer.h"
#include "components/bookmarks/browser/bookmark_model.h"
#include "components/bookmarks/browser/bookmark_node.h"
namespace {
int CountBookmarksFromNode(const bookmarks::BookmarkNode* node,
base::Time period_start) {
int count = 0;
if (node->is_url()) {
if (node->date_added() >= period_start)
++count;
} else {
for (const auto& child : node->children())
count += CountBookmarksFromNode(child.get(), period_start);
}
return count;
}
using BookmarkModelCallback =
base::OnceCallback<void(const bookmarks::BookmarkModel*)>;
// This class waits for the |bookmark_model| to load, then executes |callback|
// and destroys itself afterwards.
class BookmarkModelHelper : public bookmarks::BaseBookmarkModelObserver {
public:
BookmarkModelHelper(bookmarks::BookmarkModel* bookmark_model,
BookmarkModelCallback callback)
: bookmark_model_(bookmark_model), callback_(std::move(callback)) {
DCHECK(!bookmark_model_->loaded());
bookmark_model_observation_.Observe(bookmark_model_);
}
void BookmarkModelLoaded(bool ids_reassigned) override {
std::move(callback_).Run(bookmark_model_);
delete this;
}
void BookmarkModelBeingDeleted() override {
// Don't leak this instance if the BookmarkModel never loads.
delete this;
}
void BookmarkModelChanged() override {}
private:
raw_ptr<bookmarks::BookmarkModel> bookmark_model_;
BookmarkModelCallback callback_;
base::ScopedObservation<bookmarks::BookmarkModel,
bookmarks::BaseBookmarkModelObserver>
bookmark_model_observation_{this};
};
} // namespace
namespace browsing_data {
const char BookmarkCounter::kPrefName[] =
"browser.clear_data.fake.pref.bookmarks";
BookmarkCounter::BookmarkCounter(bookmarks::BookmarkModel* bookmark_model)
: bookmark_model_(bookmark_model) {
DCHECK(bookmark_model);
}
BookmarkCounter::~BookmarkCounter() = default;
void BookmarkCounter::OnInitialized() {}
const char* BookmarkCounter::GetPrefName() const {
return kPrefName;
}
void BookmarkCounter::Count() {
if (bookmark_model_->loaded()) {
CountBookmarks(bookmark_model_);
} else {
new BookmarkModelHelper(bookmark_model_,
base::BindOnce(&BookmarkCounter::CountBookmarks,
weak_ptr_factory_.GetWeakPtr()));
}
}
void BookmarkCounter::CountBookmarks(
const bookmarks::BookmarkModel* bookmark_model) {
base::Time start = GetPeriodStart();
int count =
CountBookmarksFromNode(bookmark_model->bookmark_bar_node(), start) +
CountBookmarksFromNode(bookmark_model->other_node(), start) +
CountBookmarksFromNode(bookmark_model->mobile_node(), start);
ReportResult(count);
}
} // namespace browsing_data
|
