Separate state of basic and advanced tab in CBD dialog

components/browsing_data/core/counters/browsing_data_counter.h

 // 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.
 
 #ifndef COMPONENTS_BROWSING_DATA_CORE_COUNTERS_BROWSING_DATA_COUNTER_H_
 #define COMPONENTS_BROWSING_DATA_CORE_COUNTERS_BROWSING_DATA_COUNTER_H_
 
 #include <stdint.h>
 
 #include <memory>
 #include <string>
 
 #include "base/callback.h"
 #include "base/macros.h"
 #include "base/timer/timer.h"
+#include "components/browsing_data/core/clear_browsing_data_tab.h"
 #include "components/prefs/pref_member.h"
 
 class PrefService;
 
 namespace browsing_data {
 
 class BrowsingDataCounter {
  public:
   typedef int64_t ResultInt;
 
@@ skipping 52 matching lines @@
     RESTARTED,
     SHOW_CALCULATING,
     REPORT_STAGED_RESULT,
     READY_TO_REPORT_RESULT,
   };
 
   BrowsingDataCounter();
   virtual ~BrowsingDataCounter();
 
   // Should be called once to initialize this class.
-  void Init(PrefService* pref_service, const Callback& callback);
+  void Init(PrefService* pref_service,
+            ClearBrowsingDataTab clear_browsing_data_tab,
+            const Callback& callback);
 
   // Name of the preference associated with this counter.
   virtual const char* GetPrefName() const = 0;
 
-  // PrefService that manages the preferences for the user profile
-  // associated with this counter.
-  PrefService* GetPrefs() const;
-
   // Restarts the counter. Will be called automatically if the counting needs
   // to be restarted, e.g. when the deletion preference changes state or when
   // we are notified of data changes.
   void Restart();
 
   // Returns the state transition of this counter since past restart.
   // Used only for testing.
   const std::vector<State>& GetStateTransitionsForTesting();
 
  protected:
   // Should be called from |Count| by any overriding class to indicate that
   // counting is finished and report |value| as the result.
   void ReportResult(ResultInt value);
 
   // A convenience overload of the previous method that allows subclasses to
   // provide a custom |result|.
   void ReportResult(std::unique_ptr<Result> result);
 
   // A synchronous implementation of ReportResult(). Called immediately in the
   // RESTARTED and READY_TO_REPORT_RESULT states, called later if the counter is
   // in the SHOW_CALCULATING stage. This method is made virtual to be overriden
   // in tests.
   virtual void DoReportResult(std::unique_ptr<Result> result);
 
   // Calculates the beginning of the counting period as |period_| before now.
   base::Time GetPeriodStart();
 
+  // Returns if this counter belongs to a preference on the default, basic or
+  // advanced CBD tab.
+  ClearBrowsingDataTab GetTab() const;
+
  private:
   // Called after the class is initialized by calling |Init|.
   virtual void OnInitialized();
 
   // Count the data.
   virtual void Count() = 0;
 
   // State transition methods.
   void TransitionToShowCalculating();
   void TransitionToReadyToReportResult();
 
-  // Pointer to the PrefService that manages the preferences for the user
-  // profile associated with this counter.
-  PrefService* pref_service_;
+  // Indicates if this counter belongs to a preference on the basic CBD tab.
+  ClearBrowsingDataTab clear_browsing_data_tab_;
 
   // The callback that will be called when the UI should be updated with a new
   // counter value.
   Callback callback_;
 
   // The boolean preference indicating whether this data type is to be deleted.
   // If false, we will not count it.
   BooleanPrefMember pref_;
 
   // The integer preference describing the time period for which this data type
@@ skipping 13 matching lines @@
   std::unique_ptr<Result> staged_result_;
 
   // A timer to time the RESTARTED->SHOW_CALCULATION and
   // SHOW_CALCULATION->READY_TO_REPORT_RESULT state transitions.
   base::OneShotTimer timer_;
 };
 
 }  // namespace browsing_data
 
 #endif  // COMPONENTS_BROWSING_DATA_CORE_COUNTERS_BROWSING_DATA_COUNTER_H_