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

#ifndef COMPONENTS_SIGNIN_PUBLIC_IDENTITY_MANAGER_ACCOUNT_CAPABILITIES_TEST_MUTATOR_H_
#define COMPONENTS_SIGNIN_PUBLIC_IDENTITY_MANAGER_ACCOUNT_CAPABILITIES_TEST_MUTATOR_H_

#include <string_view>

#include "base/containers/span.h"
#include "base/memory/raw_ptr.h"
#include "build/build_config.h"
#include "components/signin/public/identity_manager/account_capabilities.h"

struct AccountInfo;

// Support class that allows callers to modify internal capability state
// mappings used for tests.
class AccountCapabilitiesTestMutator {
 public:
  explicit AccountCapabilitiesTestMutator(AccountCapabilities* capabilities);
  explicit AccountCapabilitiesTestMutator(AccountInfo* account_info);

  // Exposes the full list of supported capabilities for tests.
  static base::span<const std::string_view>
  GetSupportedAccountCapabilityNames();

  // Exposes setters for the supported capabilities.
  // keep-sorted start sticky_prefixes=#if,BUILDFLAG group_prefixes=#endif
  void set_can_fetch_family_member_info(bool value);
#if !BUILDFLAG(IS_IOS)
  void set_can_have_email_address_displayed(bool value);
#endif
#if !BUILDFLAG(IS_ANDROID)
  void set_can_make_chrome_search_engine_choice_screen_choice(bool value);
#endif
  void set_can_override_account_info(bool value);
#if !BUILDFLAG(IS_IOS)
  void set_can_run_chrome_privacy_sandbox_trials(bool value);
#endif
  void set_can_show_history_sync_opt_ins_without_minor_mode_restrictions(
      bool value);
#if BUILDFLAG(IS_IOS)
  void set_can_sign_in_to_chrome(bool value);
#endif
#if BUILDFLAG(IS_WIN) || BUILDFLAG(IS_MAC) || BUILDFLAG(IS_LINUX) || \
    BUILDFLAG(IS_IOS)
  void set_can_submit_feedback(bool value);
#endif
#if BUILDFLAG(IS_CHROMEOS)
  void set_can_toggle_auto_updates(bool value);
#endif
#if BUILDFLAG(IS_CHROMEOS)
  void set_can_use_chromeos_generative_ai(bool value);
#endif  // BUILDFLAG(IS_CHROMEOS)
#if !BUILDFLAG(IS_IOS)
  void set_can_use_devtools_generative_ai_features(bool value);
#endif
#if !BUILDFLAG(IS_IOS)
  void set_can_use_edu_features(bool value);
#endif
  void set_can_use_gemini_in_chrome(bool value);
#if BUILDFLAG(IS_CHROMEOS)
  void set_can_use_generative_ai_in_recorder_app(bool value);
#endif
#if BUILDFLAG(IS_CHROMEOS)
  void set_can_use_generative_ai_photo_editing(bool value);
#endif
  void set_can_use_manta_service(bool value);
  void set_can_use_model_execution_features(bool value);
  void set_can_use_speaker_label_in_recorder_app(bool value);
  void set_is_allowed_for_machine_learning(bool value);
  void set_is_opted_in_to_parental_supervision(bool value);
  void set_is_subject_to_account_level_enterprise_policies(bool value);
  void set_is_subject_to_chrome_privacy_sandbox_restricted_measurement_notice(
      bool value);
  void set_is_subject_to_enterprise_features(bool value);
  void set_is_subject_to_parental_controls(bool value);
  void set_is_subject_to_universal_opt_out(bool value);
#if BUILDFLAG(IS_IOS)
  void set_must_fetch_apple_age_range_in_chrome(bool value);
#endif
#if BUILDFLAG(IS_IOS)
  void set_must_skip_apple_age_range_in_chrome(bool value);
#endif
  void set_supports_wallet_private_passes_in_autofill(bool value);
  // keep-sorted end

  // Modifies all supported capabilities at once.
  void SetAllSupportedCapabilities(bool value);
  // Set capability with `name` to `value`.
  void SetCapability(const std::string& name, bool value);
  // Set override capability with `name` to `value`.
  void SetCapabilityOverride(std::string_view name,
                              std::optional<signin::Tribool> value);

 private:
  raw_ptr<AccountCapabilities> capabilities_;
};

#endif  // COMPONENTS_SIGNIN_PUBLIC_IDENTITY_MANAGER_ACCOUNT_CAPABILITIES_TEST_MUTATOR_H_