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

#include "extensions/browser/guest_view/extensions_guest_view.h"

#include <utility>

#include "base/memory/ptr_util.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/render_frame_host.h"
#include "extensions/browser/api/extensions_api_client.h"
#include "extensions/browser/guest_view/extensions_guest_view_manager_delegate.h"
#include "extensions/browser/guest_view/web_view/web_view_content_script_manager.h"
#include "extensions/browser/guest_view/web_view/web_view_renderer_state.h"
#include "mojo/public/cpp/bindings/self_owned_associated_receiver.h"

#if BUILDFLAG(ENABLE_EXTENSIONS)
#include "extensions/browser/guest_view/mime_handler_view/mime_handler_view_embedder.h"
#endif

namespace extensions {

ExtensionsGuestView::ExtensionsGuestView(
    const content::GlobalRenderFrameHostId& frame_id)
    : guest_view::GuestViewMessageHandler(frame_id) {}
ExtensionsGuestView::~ExtensionsGuestView() = default;

// static
void ExtensionsGuestView::CreateForComponents(
    const content::GlobalRenderFrameHostId& frame_id,
    mojo::PendingAssociatedReceiver<guest_view::mojom::GuestViewHost>
        receiver) {
  mojo::MakeSelfOwnedAssociatedReceiver(
      base::WrapUnique(new ExtensionsGuestView(frame_id)), std::move(receiver));
}

// static
void ExtensionsGuestView::CreateForExtensions(
    const content::GlobalRenderFrameHostId& frame_id,
    mojo::PendingAssociatedReceiver<extensions::mojom::GuestView> receiver) {
  mojo::MakeSelfOwnedAssociatedReceiver(
      base::WrapUnique(new ExtensionsGuestView(frame_id)), std::move(receiver));
}

std::unique_ptr<guest_view::GuestViewManagerDelegate>
ExtensionsGuestView::CreateGuestViewManagerDelegate() const {
  return ExtensionsAPIClient::Get()->CreateGuestViewManagerDelegate();
}

void ExtensionsGuestView::ReadyToCreateMimeHandlerView(bool success) {
#if BUILDFLAG(ENABLE_EXTENSIONS)
  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  auto* render_frame_host = content::RenderFrameHost::FromID(frame_id_);
  if (!render_frame_host) {
    return;
  }
  if (auto* mhve = MimeHandlerViewEmbedder::Get(
          render_frame_host->GetFrameTreeNodeId())) {
    mhve->ReadyToCreateMimeHandlerView(success);
  }
#endif
}

void ExtensionsGuestView::CanExecuteContentScript(
    const std::string& script_id,
    CanExecuteContentScriptCallback callback) {
  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
  WebViewRendererState::WebViewInfo info;
  const bool success = WebViewRendererState::GetInstance()->GetInfo(
      // TODO(crbug.com/379869738): Remove GetUnsafeValue
      render_process_id().GetUnsafeValue(), frame_id_.frame_routing_id, &info);
  // GetInfo can fail if the process id does not correspond to a WebView. Those
  // cases are just defaulted to false.
  if (!success) {
    std::move(callback).Run(false);
    return;
  }
  const bool can_execute = info.content_script_ids.contains(script_id);
  std::move(callback).Run(can_execute);
}

}  // namespace extensions