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

#include "cryptohome/mount_encrypted/mount_encrypted_metrics.h"

#include <base/check_op.h>
#include <base/time/time.h>

namespace mount_encrypted {

namespace {
static MountEncryptedMetrics* g_metrics = nullptr;

constexpr char kSystemKeyStatus[] = "Platform.MountEncrypted.SystemKeyStatus";
constexpr char kEncryptionKeyStatus[] =
    "Platform.MountEncrypted.EncryptionKeyStatus";

constexpr char kTimeToTakeTpmOwnership[] =
    "Platform.MountEncrypted.TimeToTakeTpmOwnership";
constexpr base::TimeDelta kTimeToTakeTpmOwnershipMin =
    base::TimeDelta::FromMilliseconds(1);
constexpr base::TimeDelta kTimeToTakeTpmOwnershipMax =
    base::TimeDelta::FromMinutes(5);
constexpr int kTimeToTakeTpmOwnershipNumBuckets = 50;

}  // namespace

MountEncryptedMetrics::MountEncryptedMetrics(const std::string& output_file) {
  metrics_library_.SetOutputFile(output_file);
}

// static
void MountEncryptedMetrics::Initialize(const std::string& output_file) {
  DCHECK_EQ(g_metrics, nullptr);
  g_metrics = new MountEncryptedMetrics(output_file);
}

// static
MountEncryptedMetrics* MountEncryptedMetrics::Get() {
  DCHECK_NE(g_metrics, nullptr);
  return g_metrics;
}

// static
void MountEncryptedMetrics::Reset() {
  DCHECK_NE(g_metrics, nullptr);
  delete g_metrics;
  g_metrics = nullptr;
}

void MountEncryptedMetrics::ReportSystemKeyStatus(
    EncryptionKey::SystemKeyStatus status) {
  metrics_library_.SendEnumToUMA(
      kSystemKeyStatus, static_cast<int>(status),
      static_cast<int>(EncryptionKey::SystemKeyStatus::kCount));
}

void MountEncryptedMetrics::ReportEncryptionKeyStatus(
    EncryptionKey::EncryptionKeyStatus status) {
  metrics_library_.SendEnumToUMA(
      kEncryptionKeyStatus, static_cast<int>(status),
      static_cast<int>(EncryptionKey::EncryptionKeyStatus::kCount));
}

void MountEncryptedMetrics::ReportTimeToTakeTpmOwnership(
    base::TimeDelta elapsed_time) {
  metrics_library_.SendToUMA(kTimeToTakeTpmOwnership,
                             elapsed_time.InMilliseconds(),
                             kTimeToTakeTpmOwnershipMin.InMilliseconds(),
                             kTimeToTakeTpmOwnershipMax.InMilliseconds(),
                             kTimeToTakeTpmOwnershipNumBuckets);
}

}  // namespace mount_encrypted