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

#include "third_party/blink/renderer/modules/screen_details/window_screen_details.h"

#include "third_party/blink/public/mojom/permissions/permission_status.mojom-blink.h"
#include "third_party/blink/renderer/bindings/core/v8/script_promise.h"
#include "third_party/blink/renderer/bindings/core/v8/script_promise_resolver.h"
#include "third_party/blink/renderer/bindings/core/v8/v8_throw_dom_exception.h"
#include "third_party/blink/renderer/core/execution_context/execution_context.h"
#include "third_party/blink/renderer/core/frame/local_dom_window.h"
#include "third_party/blink/renderer/core/frame/local_frame.h"
#include "third_party/blink/renderer/modules/permissions/permission_utils.h"
#include "third_party/blink/renderer/modules/screen_details/screen_details.h"
#include "third_party/blink/renderer/platform/bindings/exception_state.h"
#include "third_party/blink/renderer/platform/bindings/script_state.h"

namespace blink {

// static
const char WindowScreenDetails::kSupplementName[] = "WindowScreenDetails";

WindowScreenDetails::WindowScreenDetails(LocalDOMWindow* window)
    : ExecutionContextLifecycleObserver(window),
      Supplement<LocalDOMWindow>(*window),
      permission_service_(window) {}

// static
ScriptPromise<ScreenDetails> WindowScreenDetails::getScreenDetails(
    ScriptState* script_state,
    LocalDOMWindow& window,
    ExceptionState& exception_state) {
  return From(&window)->GetScreenDetails(script_state, exception_state);
}

void WindowScreenDetails::ContextDestroyed() {
  screen_details_.Clear();
}

void WindowScreenDetails::Trace(Visitor* visitor) const {
  visitor->Trace(screen_details_);
  visitor->Trace(permission_service_);
  ExecutionContextLifecycleObserver::Trace(visitor);
  Supplement<LocalDOMWindow>::Trace(visitor);
}

// static
WindowScreenDetails* WindowScreenDetails::From(LocalDOMWindow* window) {
  auto* supplement =
      Supplement<LocalDOMWindow>::From<WindowScreenDetails>(window);
  if (!supplement) {
    supplement = MakeGarbageCollected<WindowScreenDetails>(window);
    Supplement<LocalDOMWindow>::ProvideTo(*window, supplement);
  }
  return supplement;
}

ScriptPromise<ScreenDetails> WindowScreenDetails::GetScreenDetails(
    ScriptState* script_state,
    ExceptionState& exception_state) {
  if (!script_state->ContextIsValid()) {
    exception_state.ThrowDOMException(DOMExceptionCode::kInvalidStateError,
                                      "The execution context is not valid.");
    return EmptyPromise();
  }

  LocalDOMWindow* window = LocalDOMWindow::From(script_state);
  DCHECK(window->IsSecureContext());  // [SecureContext] in IDL.
  if (!permission_service_.is_bound()) {
    // See https://bit.ly/2S0zRAS for task types.
    ConnectToPermissionService(
        window, permission_service_.BindNewPipeAndPassReceiver(
                    window->GetTaskRunner(TaskType::kMiscPlatformAPI)));
  }

  auto permission_descriptor = CreatePermissionDescriptor(
      mojom::blink::PermissionName::WINDOW_MANAGEMENT);
  auto* resolver = MakeGarbageCollected<ScriptPromiseResolver<ScreenDetails>>(
      script_state, exception_state.GetContext());
  const bool has_transient_user_activation =
      LocalFrame::HasTransientUserActivation(GetSupplementable()->GetFrame());
  // Only request permission with transient activation, otherwise check quietly.
  // This lets sites with permission get screen details any time (e.g. on load),
  // but prevents sites from prompting users without a transient activation.
  auto callback =
      BindOnce(&WindowScreenDetails::OnPermissionInquiryComplete,
               WrapPersistent(this), WrapPersistent(resolver),
               /*permission_requested=*/has_transient_user_activation);
  if (has_transient_user_activation) {
    permission_service_->RequestPermission(std::move(permission_descriptor),
                                           std::move(callback));
  } else {
    permission_service_->HasPermission(std::move(permission_descriptor),
                                       std::move(callback));
  }

  return resolver->Promise();
}

void WindowScreenDetails::OnPermissionInquiryComplete(
    ScriptPromiseResolver<ScreenDetails>* resolver,
    bool permission_requested,
    mojom::blink::PermissionStatusWithDetailsPtr status) {
  if (!resolver->GetScriptState()->ContextIsValid())
    return;
  if (status->status != mojom::blink::PermissionStatus::GRANTED) {
    auto* const isolate = resolver->GetScriptState()->GetIsolate();
    ScriptState::Scope scope(resolver->GetScriptState());
    resolver->Reject(V8ThrowDOMException::CreateOrEmpty(
        isolate, DOMExceptionCode::kNotAllowedError,
        (status->status == mojom::blink::PermissionStatus::ASK &&
         !permission_requested)
            ? "Transient activation is required to request permission."
            : "Permission denied."));
    return;
  }

  if (!screen_details_)
    screen_details_ = MakeGarbageCollected<ScreenDetails>(GetSupplementable());
  resolver->Resolve(screen_details_);
}

}  // namespace blink