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

#include "chrome/browser/dictation/onboarding_manager.h"

#include "base/functional/bind.h"
#include "base/task/single_thread_task_runner.h"
#include "chrome/browser/dictation/dictation_keyed_service.h"
#include "chrome/browser/dictation/target.h"
#include "chrome/browser/ui/browser_window/public/browser_window_interface.h"
#include "chrome/browser/ui/views/dictation/onboarding_dialog_controller.h"
#include "chrome/common/pref_names.h"
#include "components/prefs/pref_service.h"
#include "components/tabs/public/tab_interface.h"

namespace dictation {

OnboardingManager::OnboardingManager(DictationKeyedService& service,
                                     PrefService& pref_service)
    : service_(service), pref_service_(pref_service) {}

OnboardingManager::~OnboardingManager() = default;

bool OnboardingManager::ShowOnboardingIfNeeded(
    tabs::TabInterface& tab,
    const TargetDetails& target_details,
    DictationSessionEntryPoint entry_point) {
  if (pref_service_->GetBoolean(prefs::kPrefDictationOnboardingCompleted)) {
    return false;
  }

  // If an FRE dialog is already active on another tab, close it before opening
  // a new FRE dialog on the current tab.
  if (dialog_controller_) {
    dialog_controller_.reset();
    pending_tab_.reset();
    pending_target_details_.reset();
    pending_entry_point_.reset();
  }

  // TODO(bokan): I think we can extract this from the dialog_controller_ rather
  // than explicitly holding a weak ptr here.
  pending_tab_ = tab.GetWeakPtr();
  pending_target_details_ = target_details;
  pending_entry_point_ = entry_point;

  dialog_controller_ = std::make_unique<OnboardingDialogController>(tab);
  dialog_controller_->Show(
      base::BindOnce(&OnboardingManager::OnOnboardingCompleted,
                     weak_ptr_factory_.GetWeakPtr()),
      base::BindOnce(&OnboardingManager::OnDialogClosed,
                     weak_ptr_factory_.GetWeakPtr()));

  if (!dialog_controller_->IsShowing()) {
    dialog_controller_.reset();
    pending_tab_.reset();
    pending_target_details_.reset();
    pending_entry_point_.reset();
    // TODO(b/527240600): Fails closed but this should report an error somehow.
  }

  return true;
}

void OnboardingManager::OnOnboardingCompleted() {
  pref_service_->SetBoolean(prefs::kPrefDictationOnboardingCompleted, true);
}

void OnboardingManager::OnDialogClosed() {
  if (pref_service_->GetBoolean(prefs::kPrefDictationOnboardingCompleted)) {
    if (pending_tab_) {
      CHECK(pending_target_details_);
      CHECK(pending_entry_point_);
      service_->DidCompleteOnboarding(*pending_tab_, *pending_target_details_,
                                      *pending_entry_point_);
    }
  }
  pending_tab_.reset();
  pending_target_details_.reset();
  pending_entry_point_.reset();
  dialog_controller_.reset();
}

}  // namespace dictation