components/feature_engagement/internal/feature_config_event_storage_validator.h

// 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.

#ifndef COMPONENTS_FEATURE_ENGAGEMENT_INTERNAL_FEATURE_CONFIG_EVENT_STORAGE_VALIDATOR_H_
#define COMPONENTS_FEATURE_ENGAGEMENT_INTERNAL_FEATURE_CONFIG_EVENT_STORAGE_VALIDATOR_H_

#include <string>
#include <unordered_map>
#include <unordered_set>

#include "base/macros.h"
#include "components/feature_engagement/internal/event_storage_validator.h"
#include "components/feature_engagement/public/feature_list.h"

namespace feature_engagement {
class Configuration;
struct EventConfig;
struct FeatureConfig;

// A EventStorageValidator that uses the FeatureConfig as the source of truth.
class FeatureConfigEventStorageValidator : public EventStorageValidator {
 public:
  FeatureConfigEventStorageValidator();
  ~FeatureConfigEventStorageValidator() override;

  // EventStorageValidator implementation.
  bool ShouldStore(const std::string& event_name) const override;
  bool ShouldKeep(const std::string& event_name,
                  uint32_t event_day,
                  uint32_t current_day) const override;

  // Set up internal configuration required for the given |features|.
  void InitializeFeatures(FeatureVector features,
                          const Configuration& configuration);

  // Resets the full state of this EventStorageValidator. After calling this
  // method it is valid to call InitializeFeatures() again.
  void ClearForTesting();

 private:
  // Updates the internal configuration with conditions from the given
  // |feature_config|.
  void InitializeFeatureConfig(const FeatureConfig& feature_config);

  // Updates the internal configuration with conditions from the given
  // |event_config|.
  void InitializeEventConfig(const EventConfig& event_config);

  // Contains an entry for each of the events that any EventConfig required to
  // be stored.
  std::unordered_set<std::string> should_store_event_names_;

  // Contains the longest time to store each event across all EventConfigs,
  // as a number of days.
  std::unordered_map<std::string, uint32_t> longest_storage_times_;

  DISALLOW_COPY_AND_ASSIGN(FeatureConfigEventStorageValidator);
};

}  // namespace feature_engagement

#endif  // COMPONENTS_FEATURE_ENGAGEMENT_INTERNAL_FEATURE_CONFIG_EVENT_STORAGE_VALIDATOR_H_