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

#ifndef COMPONENTS_ON_DEVICE_TRANSLATION_METRICS_H_
#define COMPONENTS_ON_DEVICE_TRANSLATION_METRICS_H_

#include <cstddef>
#include <string_view>

namespace on_device_translation {

// Record language code for a certain UMA. If the language code is not
// recognized it will be recorded as unknown.
void RecordLanguageUma(std::string_view uma_name,
                       std::string_view language_code);

// Record language code for a certain UMA.
// The source_lang and target_lang will be converted to the integer value
// of the LanguageCode enumeration and then concatenated to a single integer
// value.
void RecordLanguagePairUma(std::string_view uma_name,
                           std::string_view source_lang,
                           std::string_view target_lang);

// The following UMAs will be recorded:
// Translate.TranslatorApi.${action_name}.SourceLanguage
// Translate.TranslatorApi.${action_name}.TargetLanguage
// Translate.TranslatorApi.${action_name}.LanguagePair
void RecordTranslatorApiCallForLanguagePair(std::string_view action_name,
                                            std::string_view source_lang,
                                            std::string_view target_lang);
// The following UMAs will be recorded:
// Translate.OnDeviceTranslation.${action_name}.SourceLanguage
// Translate.OnDeviceTranslation.${action_name}.TargetLanguage
// Translate.OnDeviceTranslation.${action_name}.LanguagePair
void RecordOnDeviceTranslationCallForLanguagePair(std::string_view action_name,
                                                  std::string_view source_lang,
                                                  std::string_view target_lang);

// Record the character count UMA for:
// Translate.TranslatorApi.TranslationLength
// Translate.TranslatorApi.Source.${source_lang}.TranslationLength
// Translate.TranslatorApi.Target.${target_lang}.TranslationLength
void RecordTranslatorApiTranslationLength(std::string_view source_lang,
                                          std::string_view target_lang,
                                          size_t length);
// Record the character count UMA for:
// Translate.OnDeviceTranslation.TranslationLength
// Translate.OnDeviceTranslation.Source.${source_lang}.TranslationLength
// Translate.OnDeviceTranslation.Target.${target_lang}.TranslationLength
void RecordOnDeviceTranslationLength(std::string_view source_lang,
                                     std::string_view target_lang,
                                     size_t length);
}  // namespace on_device_translation

#endif  // COMPONENTS_ON_DEVICE_TRANSLATION_METRICS_H_