// Copyright 2012 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/web_modal/web_contents_modal_dialog_manager.h"

#include <algorithm>
#include <utility>

#include "base/check.h"
#include "components/back_forward_cache/back_forward_cache_disable.h"
#include "components/web_modal/web_contents_modal_dialog_manager_delegate.h"
#include "content/public/browser/back_forward_cache.h"
#include "content/public/browser/navigation_handle.h"
#include "content/public/browser/visibility.h"
#include "content/public/browser/web_contents.h"
#include "net/base/registry_controlled_domains/registry_controlled_domain.h"

using content::WebContents;

namespace web_modal {

WebContentsModalDialogManager::~WebContentsModalDialogManager() {
  DCHECK(child_dialogs_.empty());
}

void WebContentsModalDialogManager::SetDelegate(
    WebContentsModalDialogManagerDelegate* d) {
  // If the delegate hasn't changed, return early avoid redundant
  // synchronization.
  if (delegate_ == d) {
    return;
  }
  delegate_ = d;
  UpdateDialogHost();

  // Synchronize the web contents blocked state with the new delegate.
  // This ensures the new window's TabModel correctly reflects an active
  // dialog after a tab is moved between windows.
  // Only call this if a dialog is active, to avoid redundant focus changes
  // when there are no dialogs.
  if (delegate_ && IsDialogActive()) {
    delegate_->SetWebContentsBlocked(web_contents(), true);
  }
}

// TODO(gbillock): Maybe "ShowBubbleWithManager"?
void WebContentsModalDialogManager::ShowDialogWithManager(
    gfx::NativeWindow dialog,
    std::unique_ptr<SingleWebContentsDialogManager> manager) {
  observer_list_.Notify(&Observer::OnWillShow);
  if (delegate_) {
    manager->HostChanged(
        delegate_->GetWebContentsModalDialogHost(web_contents()));
  }
  child_dialogs_.emplace_back(dialog, std::move(manager));

  if (child_dialogs_.size() == 1) {
    BlockWebContentsInteraction(true);
    if (delegate_ && delegate_->IsWebContentsVisible(web_contents())) {
      ShowNextDialog();
    }
  }
}

bool WebContentsModalDialogManager::IsDialogActive() const {
  return !child_dialogs_.empty();
}

void WebContentsModalDialogManager::FocusTopmostDialog() const {
  DCHECK(!child_dialogs_.empty());
  child_dialogs_.front().manager->Focus();
}

void WebContentsModalDialogManager::UpdateDialogHost() {
  for (const auto& dialog : child_dialogs_) {
    // Delegate can be null on Views/Win32 during tab drag.
    dialog.manager->HostChanged(
        delegate_ ? delegate_->GetWebContentsModalDialogHost(web_contents())
                  : nullptr);
  }
}

void WebContentsModalDialogManager::AddObserver(Observer* observer) {
  observer_list_.AddObserver(observer);
}

void WebContentsModalDialogManager::RemoveObserver(Observer* observer) {
  observer_list_.RemoveObserver(observer);
}

content::WebContents* WebContentsModalDialogManager::GetWebContents() const {
  return web_contents();
}

void WebContentsModalDialogManager::WillClose(gfx::NativeWindow dialog) {
  auto dlg = std::ranges::find(child_dialogs_, dialog, &DialogState::dialog);

  // The Views tab contents modal dialog calls WillClose twice.  Ignore the
  // second invocation.
  if (dlg == child_dialogs_.end()) {
    return;
  }

  bool removed_topmost_dialog = dlg == child_dialogs_.begin();
  child_dialogs_.erase(dlg);
  if (!closing_all_dialogs_ &&
      (!child_dialogs_.empty() && removed_topmost_dialog) &&
      (delegate_ && delegate_->IsWebContentsVisible(web_contents()))) {
    ShowNextDialog();
  }

  BlockWebContentsInteraction(!child_dialogs_.empty());
}

WebContentsModalDialogManager::WebContentsModalDialogManager(
    content::WebContents* web_contents)
    : content::WebContentsObserver(web_contents),
      content::WebContentsUserData<WebContentsModalDialogManager>(
          *web_contents),
      web_contents_visibility_(web_contents->GetVisibility()) {}

WebContentsModalDialogManager::DialogState::DialogState(
    gfx::NativeWindow dialog,
    std::unique_ptr<SingleWebContentsDialogManager> mgr)
    : dialog(dialog), manager(std::move(mgr)) {}

WebContentsModalDialogManager::DialogState::DialogState(DialogState&& state) =
    default;

WebContentsModalDialogManager::DialogState::~DialogState() = default;

// TODO(gbillock): Move this to Views impl within Show()? It would
// call WebContents* contents = native_delegate_->GetWebContents(); and
// then set the block state. Advantage: could restrict some of the
// WCMDM delegate methods, then, and pass them behind the scenes.
void WebContentsModalDialogManager::BlockWebContentsInteraction(bool blocked) {
  WebContents* contents = web_contents();
  if (!contents) {
    // The WebContents has already disconnected.
    return;
  }

  if (blocked) {
    scoped_ignore_input_events_ = contents->IgnoreInputEvents(std::nullopt);
  } else {
    scoped_ignore_input_events_.reset();
  }
  if (delegate_) {
    delegate_->SetWebContentsBlocked(contents, blocked);
  }
}

void WebContentsModalDialogManager::ShowNextDialog() {
  child_dialogs_.front().manager->Show();
}

void WebContentsModalDialogManager::CloseAllDialogs() {
  closing_all_dialogs_ = true;

  // Clear out any dialogs since we are leaving this page entirely.
  while (!child_dialogs_.empty()) {
    child_dialogs_.front().manager->Close();
  }

  closing_all_dialogs_ = false;
}

void WebContentsModalDialogManager::DidFinishNavigation(
    content::NavigationHandle* navigation_handle) {
  if (!navigation_handle->IsInPrimaryMainFrame() ||
      !navigation_handle->HasCommitted()) {
    return;
  }

  if (!child_dialogs_.empty()) {
    // Disable BFCache for the page which had any modal dialog open.
    // This prevents the page which has print, confirm form resubmission, http
    // password dialogs, etc. to go in to BFCache. We can't simply dismiss the
    // dialogs in the case, since they are requesting meaningful input from the
    // user that affects the loading or display of the content.
    content::BackForwardCache::DisableForRenderFrameHost(
        navigation_handle->GetPreviousRenderFrameHostId(),
        back_forward_cache::DisabledReason(
            back_forward_cache::DisabledReasonId::kModalDialog));
  }

  // Close constrained windows if necessary.
  if (!net::registry_controlled_domains::SameDomainOrHost(
          navigation_handle->GetPreviousPrimaryMainFrameURL(),
          navigation_handle->GetURL(),
          net::registry_controlled_domains::INCLUDE_PRIVATE_REGISTRIES)) {
    observer_list_.Notify(&Observer::OnWillCloseOnNavigation);
    CloseAllDialogs();
  }
}

void WebContentsModalDialogManager::DidGetIgnoredUIEvent() {
  if (!child_dialogs_.empty()) {
    child_dialogs_.front().manager->Focus();
  }
}

void WebContentsModalDialogManager::OnVisibilityChanged(
    content::Visibility visibility) {
  const content::Visibility previous_web_contents_visibility =
      web_contents_visibility_;
  web_contents_visibility_ = visibility;
  if (child_dialogs_.empty()) {
    return;
  }

  // Hide the dialog if the web contents are newly hidden.
  if (previous_web_contents_visibility != content::Visibility::HIDDEN &&
      web_contents_visibility_ == content::Visibility::HIDDEN) {
    child_dialogs_.front().manager->Hide();
    return;
  }

  // Show the dialog if it transitioned from HIDDEN to VISIBLE or OCCLUDED.
  if ((previous_web_contents_visibility == content::Visibility::HIDDEN &&
       web_contents_visibility_ != content::Visibility::HIDDEN) ||
      // Or from OCCLUDED to VISIBLE if the dialog is no longer active.
      (previous_web_contents_visibility == content::Visibility::OCCLUDED &&
       web_contents_visibility_ == content::Visibility::VISIBLE &&
       !child_dialogs_.front().manager->IsActive())) {
    // TODO(crbug.com/40283251): Add an interaction test for this.
    ShowNextDialog();
  }
}

void WebContentsModalDialogManager::WebContentsDestroyed() {
  // First cleanly close all child dialogs.
  // TODO(mpcomplete): handle case if MaybeCloseChildWindows() already asked
  // some of these to close.  CloseAllDialogs is async, so it might get called
  // twice before it runs.
  CloseAllDialogs();
}

WEB_CONTENTS_USER_DATA_KEY_IMPL(WebContentsModalDialogManager);

}  // namespace web_modal