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

#include "components/security_interstitials/core/https_only_mode_metrics.h"

#include "base/metrics/histogram_functions.h"

namespace security_interstitials::https_only_mode {

const char kEventHistogram[] = "Security.HttpsFirstMode.NavigationEvent";
const char kEventHistogramWithEngagementHeuristic[] =
    "Security.HttpsFirstModeWithEngagementHeuristic.NavigationEvent";
const char kEventHistogramWithEsbPairing[] =
    "Security.HttpsFirstModeWithEsbPairing.NavigationEvent";

const char kNavigationRequestSecurityLevelHistogram[] =
    "Security.NavigationRequestSecurityLevel";

const char kSiteEngagementHeuristicStateHistogram[] =
    "Security.HttpsFirstModeWithEngagementHeuristic.State";

const char kSiteEngagementHeuristicHostCountHistogram[] =
    "Security.HttpsFirstModeWithEngagementHeuristic.HostCount";

const char kSiteEngagementHeuristicAccumulatedHostCountHistogram[] =
    "Security.HttpsFirstModeWithEngagementHeuristic.AccumulatedHostCount";

const char kSiteEngagementHeuristicEnforcementDurationHistogram[] =
    "Security.HttpsFirstModeWithEngagementHeuristic.Duration";

const char kInterstitialReasonHistogram[] =
    "Security.HttpsFirstMode.InterstitialReason";

// TODO(crbug.com/40248833): Rename these metrics now that they apply to both
// HTTPS-First Mode and HTTPS Upgrades.
void RecordHttpsFirstModeNavigation(
    Event event,
    const HttpInterstitialState& interstitial_state) {
  base::UmaHistogramEnumeration(kEventHistogram, event);

  if (!interstitial_state.enabled_by_pref &&
      !interstitial_state.enabled_in_balanced_mode &&
      interstitial_state.enabled_by_engagement_heuristic) {
    // Only record the engagement heuristic histogram if HTTPS-First Mode wasn't
    // enabled by the UI setting.
    base::UmaHistogramEnumeration(kEventHistogramWithEngagementHeuristic,
                                  event);
  }

  if (interstitial_state.enabled_by_esb_pairing) {
    base::UmaHistogramEnumeration(kEventHistogramWithEsbPairing, event);
  }
}

void RecordNavigationRequestSecurityLevel(
    NavigationRequestSecurityLevel level) {
  base::UmaHistogramEnumeration(kNavigationRequestSecurityLevelHistogram,
                                level);
}

void RecordSiteEngagementHeuristicState(SiteEngagementHeuristicState state) {
  base::UmaHistogramEnumeration(kSiteEngagementHeuristicStateHistogram, state);
}

void RecordSiteEngagementHeuristicCurrentHostCounts(size_t current_count,
                                                    size_t accumulated_count) {
  base::UmaHistogramCounts1000(kSiteEngagementHeuristicHostCountHistogram,
                               current_count);
  base::UmaHistogramCounts1000(
      kSiteEngagementHeuristicAccumulatedHostCountHistogram, accumulated_count);
}

void RecordSiteEngagementHeuristicEnforcementDuration(
    base::TimeDelta enforcement_duration) {
  base::UmaHistogramTimes(kSiteEngagementHeuristicEnforcementDurationHistogram,
                          enforcement_duration);
}

InterstitialReason GetInterstitialReason(
    const HttpInterstitialState& interstitial_state) {
  // Multiple interstitial flags might be true here, but we assign higher
  // priority to Site Engagement heuristic because we expect SE interstitials
  // to be rare. Advanced Protection locks the HTTPS-First Mode UI setting so
  // it's higher priority than the HFM string as well. The lowest priority is
  // given to HFM-in-Incognito, where we want to use different strings only if
  // the user is not opted in to HFM for any other reason.
  if (interstitial_state.enabled_by_advanced_protection) {
    return InterstitialReason::kAdvancedProtection;
  }
  if (interstitial_state.enabled_by_engagement_heuristic) {
    return InterstitialReason::kSiteEngagementHeuristic;
  }
  if (interstitial_state.enabled_by_typically_secure_browsing) {
    return InterstitialReason::kTypicallySecureUserHeuristic;
  }
  if (interstitial_state.enabled_by_pref) {
    return InterstitialReason::kPref;
  }
  if (interstitial_state.enabled_by_esb_pairing) {
    return InterstitialReason::kEsbPairing;
  }
  if (interstitial_state.enabled_in_balanced_mode) {
    return InterstitialReason::kBalanced;
  }
  if (interstitial_state.enabled_by_incognito) {
    return InterstitialReason::kIncognito;
  }
  return InterstitialReason::kUnknown;
}

void RecordInterstitialReason(const HttpInterstitialState& interstitial_state) {
  base::UmaHistogramEnumeration(kInterstitialReasonHistogram,
                                GetInterstitialReason(interstitial_state));
}

}  // namespace security_interstitials::https_only_mode