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

#include "content/browser/conversions/conversion_internals_ui.h"

#include "content/browser/conversions/conversion_internals_handler_impl.h"
#include "content/browser/renderer_host/render_frame_host_impl.h"
#include "content/grit/dev_ui_content_resources.h"
#include "content/public/browser/render_frame_host.h"
#include "content/public/browser/web_contents.h"
#include "content/public/browser/web_ui.h"
#include "content/public/browser/web_ui_data_source.h"
#include "content/public/common/bindings_policy.h"
#include "content/public/common/url_constants.h"
#include "services/network/public/mojom/content_security_policy.mojom.h"

namespace content {

ConversionInternalsUI::ConversionInternalsUI(WebUI* web_ui)
    : WebUIController(web_ui) {
  // Initialize the UI with no bindings. Mojo bindings will be separately
  // granted to frames within this WebContents.
  web_ui->SetBindings(BINDINGS_POLICY_NONE);
  WebUIDataSource* source =
      WebUIDataSource::Create(kChromeUIConversionInternalsHost);

  source->AddResourcePath("conversion_internals.mojom-webui.js",
                          IDR_CONVERSION_INTERNALS_MOJOM_JS);
  source->AddResourcePath("conversion_internals.js",
                          IDR_CONVERSION_INTERNALS_JS);
  source->AddResourcePath("conversion_internals.css",
                          IDR_CONVERSION_INTERNALS_CSS);
  source->SetDefaultResource(IDR_CONVERSION_INTERNALS_HTML);
  source->OverrideContentSecurityPolicy(
      network::mojom::CSPDirectiveName::TrustedTypes,
      "trusted-types cr-ui-tree-js-static;");
  WebUIDataSource::Add(web_ui->GetWebContents()->GetBrowserContext(), source);
}

WEB_UI_CONTROLLER_TYPE_IMPL(ConversionInternalsUI)

ConversionInternalsUI::~ConversionInternalsUI() = default;

void ConversionInternalsUI::RenderFrameCreated(RenderFrameHost* rfh) {
  // Enable the JavaScript Mojo bindings in the renderer process, so the JS
  // code can call the Mojo APIs exposed by this WebUI.
  static_cast<RenderFrameHostImpl*>(rfh)->EnableMojoJsBindings();
}

void ConversionInternalsUI::BindInterface(
    mojo::PendingReceiver<::mojom::ConversionInternalsHandler> receiver) {
  ui_handler_ = std::make_unique<ConversionInternalsHandlerImpl>(
      web_ui(), std::move(receiver));
}

void ConversionInternalsUI::SetConversionManagerProviderForTesting(
    std::unique_ptr<ConversionManager::Provider> manager_provider) {
  ui_handler_->SetConversionManagerProviderForTesting(
      std::move(manager_provider));
}

}  // namespace content