Make ClearBrowsingDataHandler only observe its own removal task

chrome/browser/ui/webui/settings/settings_clear_browsing_data_handler.cc

 // Copyright 2015 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/ui/webui/settings/settings_clear_browsing_data_handler.h"
 
 #include <stddef.h>
 
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
@@ skipping 24 matching lines @@
   browsing_data::prefs::kDeleteFormData,
   browsing_data::prefs::kDeleteHostedAppsData,
   browsing_data::prefs::kDeleteMediaLicenses,
   browsing_data::prefs::kDeletePasswords,
 };
 
 } // namespace
 
 namespace settings {
 
+// TaskObserver ----------------------------------------------------------------
+
+ClearBrowsingDataHandler::TaskObserver::TaskObserver(
+    BrowsingDataRemover* remover, const base::Closure& callback)
+    : callback_(callback),
+      remover_observer_(this) {
+  remover_observer_.Add(remover);
+}
+
+ClearBrowsingDataHandler::TaskObserver::~TaskObserver() {}
+
+void ClearBrowsingDataHandler::TaskObserver::OnBrowsingDataRemoverDone() {
+  remover_observer_.RemoveAll();
+  callback_.Run();
+}
+
+// ClearBrowsingDataHandler ----------------------------------------------------
+
 ClearBrowsingDataHandler::ClearBrowsingDataHandler(content::WebUI* webui)
     : profile_(Profile::FromWebUI(webui)),
       sync_service_(ProfileSyncServiceFactory::GetForProfile(profile_)),
       sync_service_observer_(this),
-      remover_(nullptr),
-      remover_observer_(this),
       show_history_footer_(false),
       show_history_deletion_dialog_(false),
       weak_ptr_factory_(this) {}
 
 ClearBrowsingDataHandler::~ClearBrowsingDataHandler() {
 }
 
 void ClearBrowsingDataHandler::RegisterMessages() {
   web_ui()->RegisterMessageCallback(
       "clearBrowsingData",
       base::Bind(&ClearBrowsingDataHandler::HandleClearBrowsingData,
                  base::Unretained(this)));
 
   web_ui()->RegisterMessageCallback(
       "initializeClearBrowsingData",
       base::Bind(&ClearBrowsingDataHandler::HandleInitialize,
                  base::Unretained(this)));
 }
 
 void ClearBrowsingDataHandler::OnJavascriptAllowed() {
   PrefService* prefs = profile_->GetPrefs();
   profile_pref_registrar_.Init(prefs);
   profile_pref_registrar_.Add(
       prefs::kAllowDeletingBrowserHistory,
       base::Bind(&ClearBrowsingDataHandler::OnBrowsingHistoryPrefChanged,
                  base::Unretained(this)));
 
   if (sync_service_)
     sync_service_observer_.Add(sync_service_);
-
-  if (!remover_)
-    remover_ = BrowsingDataRemoverFactory::GetForBrowserContext(profile_);
-
-  remover_observer_.Add(remover_);
 }
 
 void ClearBrowsingDataHandler::OnJavascriptDisallowed() {
-  remover_observer_.RemoveAll();
   profile_pref_registrar_.RemoveAll();
   sync_service_observer_.RemoveAll();
+  task_observer_.reset();
 }
 
 void ClearBrowsingDataHandler::HandleClearBrowsingData(
     const base::ListValue* args) {
-  // We should never be called when the previous clearing has not yet finished.
-  CHECK(!remover_->is_removing());
-  CHECK_EQ(1U, args->GetSize());
-  CHECK(webui_callback_id_.empty());
-  CHECK(args->GetString(0, &webui_callback_id_));
+  DCHECK(!task_observer_);
 
   PrefService* prefs = profile_->GetPrefs();
 
   int site_data_mask = BrowsingDataRemover::REMOVE_SITE_DATA;
   // Don't try to clear LSO data if it's not supported.
   if (!prefs->GetBoolean(prefs::kClearPluginLSODataEnabled))
     site_data_mask &= ~BrowsingDataRemover::REMOVE_PLUGIN_DATA;
 
   int remove_mask = 0;
   if (prefs->GetBoolean(prefs::kAllowDeletingBrowserHistory)) {
@@ skipping 56 matching lines @@
     int checked_other_types = std::count_if(
         other_types, other_types + num_other_types,
         [prefs](const std::string& pref) { return prefs->GetBoolean(pref); });
     UMA_HISTOGRAM_SPARSE_SLOWLY(
         "History.ClearBrowsingData.PasswordsDeletion.AdditionalDatatypesCount",
         checked_other_types);
   }
 
   int period_selected =
       prefs->GetInteger(browsing_data::prefs::kDeleteTimePeriod);
-  remover_->Remove(BrowsingDataRemover::Period(
-                       static_cast<browsing_data::TimePeriod>(period_selected)),
-                   remove_mask, origin_mask);
+
+  std::string webui_callback_id;
+  CHECK_EQ(1U, args->GetSize());
+  CHECK(args->GetString(0, &webui_callback_id));
+
+  BrowsingDataRemover* remover =
+      BrowsingDataRemoverFactory::GetForBrowserContext(profile_);
+  task_observer_ = base::MakeUnique<TaskObserver>(
+      remover,
+      base::Bind(&ClearBrowsingDataHandler::OnClearingTaskFinished,
+                 base::Unretained(this), webui_callback_id));
+  remover->RemoveAndReply(
+      BrowsingDataRemover::Period(
+          static_cast<browsing_data::TimePeriod>(period_selected)),
+      remove_mask, origin_mask, task_observer_.get());
 }
 
-void ClearBrowsingDataHandler::OnBrowsingDataRemoving(bool is_removing) {
-  CallJavascriptFunction("cr.webUIListenerCallback",
-                         base::StringValue("browsing-data-removing"),
-                         base::FundamentalValue(is_removing));
-
-  if (is_removing || webui_callback_id_.empty())
-    return;
-
+void ClearBrowsingDataHandler::OnClearingTaskFinished(
+    const std::string& webui_callback_id) {
   PrefService* prefs = profile_->GetPrefs();
   int notice_shown_times =
       prefs->GetInteger(prefs::kClearBrowsingDataHistoryNoticeShownTimes);
 
   // When the deletion is complete, we might show an additional dialog with
   // a notice about other forms of browsing history. This is the case if
   const bool show_notice =
       // 1. The dialog is relevant for the user.
       show_history_deletion_dialog_ &&
       // 2. The notice has been shown less than |kMaxTimesHistoryNoticeShown|.
       notice_shown_times < kMaxTimesHistoryNoticeShown &&
       // 3. The selected data types contained browsing history.
       prefs->GetBoolean(browsing_data::prefs::kDeleteBrowsingHistory);
 
   if (show_notice) {
     // Increment the preference.
     prefs->SetInteger(prefs::kClearBrowsingDataHistoryNoticeShownTimes,
                       notice_shown_times + 1);
   }
 
   UMA_HISTOGRAM_BOOLEAN(
       "History.ClearBrowsingData.ShownHistoryNoticeAfterClearing", show_notice);
 
   ResolveJavascriptCallback(
-      base::StringValue(webui_callback_id_),
+      base::StringValue(webui_callback_id),
       base::FundamentalValue(show_notice));
-  webui_callback_id_.clear();
+  task_observer_.reset();
 }
 
 void ClearBrowsingDataHandler::OnBrowsingHistoryPrefChanged() {
   CallJavascriptFunction(
       "cr.webUIListenerCallback",
       base::StringValue("browsing-history-pref-changed"),
       base::FundamentalValue(
           profile_->GetPrefs()->GetBoolean(
               prefs::kAllowDeletingBrowserHistory)));
 }
 
 void ClearBrowsingDataHandler::HandleInitialize(const base::ListValue* args) {
   AllowJavascript();
   const base::Value* callback_id;
   CHECK(args->Get(0, &callback_id));
 
+  task_observer_.reset();
   counters_.clear();
 
   for (const std::string& pref : kCounterPrefs) {
     AddCounter(
         BrowsingDataCounterFactory::GetForProfileAndPref(profile_, pref));
   }
 
   OnStateChanged();
   RefreshHistoryNotice();
 
-  ResolveJavascriptCallback(*callback_id,
-                            base::FundamentalValue(remover_->is_removing()));
+  ResolveJavascriptCallback(
+      *callback_id,
+      *base::Value::CreateNullValue() /* Promise<void> */);
 }
 
 void ClearBrowsingDataHandler::OnStateChanged() {
   CallJavascriptFunction(
       "cr.webUIListenerCallback",
       base::StringValue("update-footer"),
       base::FundamentalValue(sync_service_ && sync_service_->IsSyncActive()),
       base::FundamentalValue(show_history_footer_));
 }
 
@@ skipping 23 matching lines @@
 void ClearBrowsingDataHandler::UpdateHistoryNotice(bool show) {
   show_history_footer_ = show;
   OnStateChanged();
 
   UMA_HISTOGRAM_BOOLEAN(
       "History.ClearBrowsingData.HistoryNoticeShownInFooterWhenUpdated",
       show_history_footer_);
 }
 
 void ClearBrowsingDataHandler::UpdateHistoryDeletionDialog(bool show) {
-  // This is used by OnBrowsingDataRemoving (when the deletion finishes).
+  // This is used by OnClearingTaskFinished (when the deletion finishes).
   show_history_deletion_dialog_ = show;
 }
 
 void ClearBrowsingDataHandler::AddCounter(
     std::unique_ptr<browsing_data::BrowsingDataCounter> counter) {
   counter->Init(profile_->GetPrefs(),
                 base::Bind(&ClearBrowsingDataHandler::UpdateCounterText,
                            base::Unretained(this)));
   counter->Restart();
   counters_.push_back(std::move(counter));
 }
 
 void ClearBrowsingDataHandler::UpdateCounterText(
     std::unique_ptr<browsing_data::BrowsingDataCounter::Result> result) {
   CallJavascriptFunction(
       "cr.webUIListenerCallback",
       base::StringValue("update-counter-text"),
       base::StringValue(result->source()->GetPrefName()),
       base::StringValue(GetChromeCounterTextFromResult(result.get())));
 }
 
 }  // namespace settings