Notify DataUseTabModel of navigations and tab closures

chrome/browser/android/data_usage/data_use_ui_tab_model.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 CHROME_BROWSER_ANDROID_DATA_USAGE_DATA_USE_UI_TAB_MODEL_H_
 #define CHROME_BROWSER_ANDROID_DATA_USAGE_DATA_USE_UI_TAB_MODEL_H_
 
 #include <stdint.h>
 
 #include <unordered_map>
 
 #include "base/gtest_prod_util.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/weak_ptr.h"
 #include "base/threading/thread_checker.h"
+#include "chrome/browser/android/data_usage/data_use_tab_model.h"
 #include "components/keyed_service/core/keyed_service.h"
 #include "ui/base/page_transition_types.h"
 
 class GURL;
 
 namespace chrome {
 
 namespace android {
 
 // DataUseUITabModel tracks data use tracking start and end transitions on the
@@ skipping 26 matching lines @@
   void ReportTabClosure(int32_t tab_id);
 
   // Returns true if data use tracking has been started for the tab with id
   // |tab_id|. Calling this function resets the state of the tab.
   bool HasDataUseTrackingStarted(int32_t tab_id);
 
   // Returns true if data use tracking has ended for the tab with id |tab_id|.
   // Calling this function resets the state of the tab.
   bool HasDataUseTrackingEnded(int32_t tab_id);
 
+  void set_data_use_tab_model(
+      base::WeakPtr<DataUseTabModel> data_use_tab_model) {
+    data_use_tab_model_ = data_use_tab_model_;

[bengr, 2015/11/14 03:54:39]

bug? You are setting it to itself?

[tbansal1, 2015/11/16 17:49:08]

Done.

+  }
+
  private:
   FRIEND_TEST_ALL_PREFIXES(DataUseUITabModelTest, EntranceExitState);
+  FRIEND_TEST_ALL_PREFIXES(DataUseUITabModelTest, ConvertTransitionType);
 
   // DataUseTrackingEvent indicates the state of a tab.
   enum DataUseTrackingEvent {
     // Indicates that data use tracking has started.
     DATA_USE_TRACKING_STARTED,
 
     // Indicates that data use tracking has ended.
     DATA_USE_TRACKING_ENDED,
   };
 
   typedef std::unordered_map<int32_t, DataUseTrackingEvent> TabEvents;
 
   // DataUseTabModel::Observer implementation:
   // TODO(tbansal): Add override once DataUseTabModel is checked in.
   void OnTrackingStarted(int32_t tab_id);
   void OnTrackingEnded(int32_t tab_id);
 
   // Creates |event| for tab with id |tab_id| and value |event|, if there is no
   // existing entry for |tab_id|, and returns true. Otherwise, returns false
   // without modifying the entry.
   bool MaybeCreateTabEvent(int32_t tab_id, DataUseTrackingEvent event);
 
   // Removes event entry for |tab_id|, if the entry is equal to |event|, and
   // returns true. Otherwise, returns false without modifying the entry.
   bool RemoveTabEvent(int32_t tab_id, DataUseTrackingEvent event);
 
+  // Converts |page_transition| to DataUseTabModel::TransitionType enum.
+  // Returns true if conversion was successful, and updates |transition_type|.
+  // Otherwise, returns false, and |transition_type| is not changed.
+  // |transition_type| must not be null.
+  bool ConvertTransitionType(
+      ui::PageTransition page_transition,
+      DataUseTabModel::TransitionType* transition_type) const;
+
+  // |data_use_tab_model_| is notified of browser navigations and tab closures
+  // on IO thread.
+  base::WeakPtr<DataUseTabModel> data_use_tab_model_;
+
   // |tab_events_| stores tracking events of multiple tabs.
   TabEvents tab_events_;
 
   // |io_task_runner_| accesses DataUseTabModel members on IO thread.
   scoped_refptr<base::SingleThreadTaskRunner> io_task_runner_;
 
   base::ThreadChecker thread_checker_;
 
   DISALLOW_COPY_AND_ASSIGN(DataUseUITabModel);
 };
 
 }  // namespace android
 
 }  // namespace chrome
 
 #endif  // CHROME_BROWSER_ANDROID_DATA_USAGE_DATA_USE_UI_TAB_MODEL_H_