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

ash / metrics / feature_discovery_duration_reporter_impl.h [blame]

// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef ASH_METRICS_FEATURE_DISCOVERY_DURATION_REPORTER_IMPL_H_
#define ASH_METRICS_FEATURE_DISCOVERY_DURATION_REPORTER_IMPL_H_

#include <map>

#include "ash/public/cpp/feature_discovery_duration_reporter.h"
#include "ash/public/cpp/session/session_observer.h"
#include "ash/session/session_controller_impl.h"
#include "base/gtest_prod_util.h"
#include "base/scoped_observation.h"

class PrefRegistrySimple;
class PrefService;

namespace base {
class TimeTicks;
}

namespace ash {

class FeatureDiscoveryDurationReporterImpl
    : public FeatureDiscoveryDurationReporter,
      public SessionObserver {
 public:
  explicit FeatureDiscoveryDurationReporterImpl(
      SessionController* session_controller);
  FeatureDiscoveryDurationReporterImpl(
      const FeatureDiscoveryDurationReporterImpl&) = delete;
  FeatureDiscoveryDurationReporterImpl& operator=(
      const FeatureDiscoveryDurationReporterImpl&) = delete;
  ~FeatureDiscoveryDurationReporterImpl() override;

  static void RegisterProfilePrefs(PrefRegistrySimple* registry);

  // FeatureDiscoveryDurationReporter:
  void MaybeActivateObservation(
      feature_discovery::TrackableFeature feature) override;
  void MaybeFinishObservation(
      feature_discovery::TrackableFeature feature) override;
  void AddObserver(ReporterObserver* observer) override;
  void RemoveObserver(ReporterObserver* observer) override;

 private:
  friend class FeatureDiscoveryDurationReporterImplTest;

  FRIEND_TEST_ALL_PREFIXES(FeatureDiscoveryDurationReporterBrowserTest,
                           SaveCumulatedTimeWhenSignout);

  // Activates/deactivates the reporter.
  void SetActive(bool active);

  // Implements activation/deactivation.
  void Activate();
  void Deactivate();

  // SessionObserver:
  void OnSessionStateChanged(session_manager::SessionState state) override;
  void OnActiveUserPrefServiceChanged(PrefService* pref_service) override;

  bool is_active() const { return is_active_; }

  base::ObserverList<ReporterObserver> observers_;

  // The mappings from trackable feature enum types to observation start
  // timestamps. It gets cleared when the reporter is deactivated. New entries
  // are added when:
  // 1. a new observation starts; or
  // 2. unfinished observations resume when the reporter is activated.
  std::map<feature_discovery::TrackableFeature, base::TimeTicks>
      active_time_recordings_;

  // Specifies the pref service whose data is read/written by the reporter. It
  // is set when the active user pref service changes.
  // NOTE: `active_pref_service_` is not reset when signing out. Because the
  // reporter instance should be destroyed in this scenario.
  raw_ptr<PrefService> active_pref_service_ = nullptr;

  // If true, starting observations on feature discovery is allowed.
  bool is_active_ = false;

  base::ScopedObservation<SessionController, SessionObserver>
      session_controller_observation_{this};
};

}  // namespace ash

#endif  // ASH_METRICS_FEATURE_DISCOVERY_DURATION_REPORTER_IMPL_H_