components/metrics/structured/key_data_provider_file.cc

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

#include "components/metrics/structured/key_data_provider_file.h"

#include "base/functional/bind.h"
#include "base/logging.h"
#include "components/metrics/structured/recorder.h"
#include "components/metrics/structured/structured_metrics_validator.h"

namespace metrics::structured {

KeyDataProviderFile::KeyDataProviderFile(const base::FilePath& file_path,
                                         base::TimeDelta write_delay)
    : file_path_(file_path), write_delay_(write_delay) {
  key_data_ =
      std::make_unique<KeyData>(file_path_, write_delay_,
                                base::BindOnce(&KeyDataProviderFile::OnKeyReady,
                                               weak_ptr_factory_.GetWeakPtr()));
}

KeyDataProviderFile::~KeyDataProviderFile() = default;

bool KeyDataProviderFile::IsReady() {
  return is_data_loaded_;
}

void KeyDataProviderFile::OnKeyReady() {
  is_data_loaded_ = true;
  NotifyKeyReady();
}

std::optional<uint64_t> KeyDataProviderFile::GetId(
    const std::string& project_name) {
  DCHECK(IsReady());

  // Validates the project. If valid, retrieve the metadata associated
  // with the event.
  auto maybe_project_validator =
      validator::Validators::Get()->GetProjectValidator(project_name);

  if (!maybe_project_validator.has_value()) {
    return std::nullopt;
  }
  const auto* project_validator = maybe_project_validator.value();
  return key_data_->Id(project_validator->project_hash(),
                       project_validator->key_rotation_period());
}

std::optional<uint64_t> KeyDataProviderFile::GetSecondaryId(
    const std::string& project_name) {
  return std::nullopt;
}

KeyData* KeyDataProviderFile::GetKeyData(const std::string& project_name) {
  DCHECK(IsReady());
  return key_data_.get();
}

// No-op.
void KeyDataProviderFile::OnProfileAdded(const base::FilePath& profile_path) {}

void KeyDataProviderFile::Purge() {
  if (IsReady()) {
    key_data_->Purge();
  }
}

}  // namespace metrics::structured