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

#include "chromeos/ash/components/boca/babelorca/caption_bubble_settings_impl.h"

#include <memory>
#include <string>
#include <string_view>

#include "ash/constants/ash_features.h"
#include "base/functional/callback.h"
#include "base/i18n/language_tag.h"
#include "base/i18n/tag_converters.h"
#include "base/memory/weak_ptr.h"
#include "chromeos/ash/components/boca/babelorca/pref_names.h"
#include "components/prefs/pref_change_registrar.h"
#include "components/prefs/pref_service.h"

namespace ash::babelorca {

CaptionBubbleSettingsImpl::CaptionBubbleSettingsImpl(
    PrefService* profile_prefs,
    std::string_view caption_language_code,
    base::RepeatingClosure on_local_caption_closed_cb)
    : profile_prefs_(profile_prefs),
      caption_language_code_(caption_language_code),
      on_local_caption_closed_cb_(on_local_caption_closed_cb) {
  if (GetLiveTranslateTargetLanguageCode() ==
      base::i18n::GetKnownLanguageTag("und")) {
    std::optional<base::i18n::LanguageTag> tag =
        base::i18n::GetLanguageTagFromString(caption_language_code_);
    if (tag) {
      SetLiveTranslateTargetLanguageCode(*tag);
    }
  }
}

CaptionBubbleSettingsImpl::~CaptionBubbleSettingsImpl() = default;

void CaptionBubbleSettingsImpl::SetObserver(
    base::WeakPtr<::captions::CaptionBubbleSettings::Observer> observer) {
  pref_change_registrar_ = std::make_unique<PrefChangeRegistrar>();
  pref_change_registrar_->Init(profile_prefs_);
  pref_change_registrar_->Add(
      prefs::kTranslateTargetLanguageCode,
      base::BindRepeating(&CaptionBubbleSettings::Observer::
                              OnLiveTranslateTargetLanguageChanged,
                          observer));
  observer_ = observer;
}

void CaptionBubbleSettingsImpl::RemoveObserver() {
  pref_change_registrar_.reset();
  observer_.reset();
}

bool CaptionBubbleSettingsImpl::IsLiveTranslateFeatureEnabled() {
  return true;
}

bool CaptionBubbleSettingsImpl::GetLiveCaptionBubbleExpanded() {
  return profile_prefs_->GetBoolean(prefs::kCaptionBubbleExpanded);
}

bool CaptionBubbleSettingsImpl::GetLiveTranslateEnabled() {
  return translate_allowed_ && translate_enabled_;
}

std::string CaptionBubbleSettingsImpl::GetLiveCaptionLanguageCode() {
  return caption_language_code_;
}

base::i18n::LanguageTag
CaptionBubbleSettingsImpl::GetLiveTranslateTargetLanguageCode() {
  return profile_prefs_->GetLanguageTag(prefs::kTranslateTargetLanguageCode);
}

void CaptionBubbleSettingsImpl::SetLiveCaptionEnabled(bool enabled) {
  if (!enabled) {
    on_local_caption_closed_cb_.Run();
  }
}

void CaptionBubbleSettingsImpl::SetLiveCaptionBubbleExpanded(bool expanded) {
  profile_prefs_->SetBoolean(prefs::kCaptionBubbleExpanded, expanded);
}

void CaptionBubbleSettingsImpl::SetLiveTranslateTargetLanguageCode(
    const base::i18n::LanguageTag& language_tag) {
  profile_prefs_->SetLanguageTag(prefs::kTranslateTargetLanguageCode,
                                 language_tag);
}

bool CaptionBubbleSettingsImpl::ShouldAdjustPositionOnExpand() {
  return features::IsBocaAdjustCaptionBubbleOnExpandEnabled();
}

void CaptionBubbleSettingsImpl::SetLiveTranslateEnabled(bool enabled) {
  bool enabled_changed = translate_enabled_ != enabled;
  translate_enabled_ = enabled;
  if (!enabled_changed || !observer_) {
    return;
  }
  observer_->OnLiveTranslateEnabledChanged();
}

void CaptionBubbleSettingsImpl::SetTranslateAllowed(bool allowed) {
  bool allowed_changed = translate_allowed_ != allowed;
  translate_allowed_ = allowed;
  if (!allowed_changed || !observer_) {
    return;
  }
  observer_->OnLiveTranslateEnabledChanged();
}

bool CaptionBubbleSettingsImpl::GetTranslateAllowed() {
  return translate_allowed_;
}

}  // namespace ash::babelorca