Index: chrome/browser/browsing_data/browsing_data_important_sites_util.cc |
diff --git a/chrome/browser/browsing_data/browsing_data_important_sites_util.cc b/chrome/browser/browsing_data/browsing_data_important_sites_util.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..67cd278ff67091b3f090b9df22f550a21ba19228 |
--- /dev/null |
+++ b/chrome/browser/browsing_data/browsing_data_important_sites_util.cc |
@@ -0,0 +1,95 @@ |
+// Copyright 2017 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "chrome/browser/browsing_data/browsing_data_important_sites_util.h" |
+ |
+#include "base/scoped_observer.h" |
+#include "content/public/browser/browsing_data_filter_builder.h" |
+ |
+namespace { |
+ |
+// This object receives |task_count| calls from BrowsingDataRemover, calls |
+// |callback| when all tasks are finished and destroys itself. |
+class BrowsingDataTaskObserver : public content::BrowsingDataRemover::Observer { |
+ public: |
+ BrowsingDataTaskObserver(content::BrowsingDataRemover* remover, |
+ base::OnceClosure callback, |
+ int task_count); |
+ ~BrowsingDataTaskObserver() override; |
+ |
+ void OnBrowsingDataRemoverDone() override; |
+ |
+ private: |
+ base::OnceClosure callback_; |
+ ScopedObserver<content::BrowsingDataRemover, |
+ content::BrowsingDataRemover::Observer> |
+ remover_observer_; |
+ int task_count_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(BrowsingDataTaskObserver); |
+}; |
+ |
+BrowsingDataTaskObserver::BrowsingDataTaskObserver( |
+ content::BrowsingDataRemover* remover, |
+ base::OnceClosure callback, |
+ int task_count) |
+ : callback_(std::move(callback)), |
+ remover_observer_(this), |
+ task_count_(task_count) { |
+ remover_observer_.Add(remover); |
+} |
+ |
+BrowsingDataTaskObserver::~BrowsingDataTaskObserver() {} |
+ |
+void BrowsingDataTaskObserver::OnBrowsingDataRemoverDone() { |
+ DCHECK(task_count_); |
+ if (--task_count_) |
+ return; |
+ remover_observer_.RemoveAll(); |
+ std::move(callback_).Run(); |
+ delete this; |
+} |
+ |
+} // namespace |
+ |
+namespace browsing_data_important_sites_util { |
+ |
+void Remove(int remove_mask, |
+ int origin_mask, |
+ browsing_data::TimePeriod time_period, |
+ std::unique_ptr<content::BrowsingDataFilterBuilder> filter_builder, |
+ content::BrowsingDataRemover* remover, |
+ base::OnceClosure callback) { |
+ auto* observer = |
+ new BrowsingDataTaskObserver(remover, std::move(callback), 2); |
+ |
+ int filterable_mask = |
+ remove_mask & |
+ ChromeBrowsingDataRemoverDelegate::IMPORTANT_SITES_DATA_TYPES; |
+ int nonfilterable_mask = |
+ remove_mask & |
+ ~ChromeBrowsingDataRemoverDelegate::IMPORTANT_SITES_DATA_TYPES; |
+ |
+ browsing_data::RecordDeletionForPeriod(time_period); |
+ |
+ if (filterable_mask) { |
+ remover->RemoveWithFilterAndReply( |
+ browsing_data::CalculateBeginDeleteTime(time_period), |
+ browsing_data::CalculateEndDeleteTime(time_period), filterable_mask, |
+ origin_mask, std::move(filter_builder), observer); |
+ } else { |
+ observer->OnBrowsingDataRemoverDone(); |
+ } |
+ |
+ if (nonfilterable_mask) { |
+ remover->RemoveAndReply( |
+ browsing_data::CalculateBeginDeleteTime(time_period), |
+ browsing_data::CalculateEndDeleteTime(time_period), nonfilterable_mask, |
+ origin_mask, observer); |
+ } else { |
+ observer->OnBrowsingDataRemoverDone(); |
+ } |
+} |
+ |
+} // namespace browsing_data_important_sites_util |