chrome/browser/ui/webui/lens/lens_ui.cc

// 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 "chrome/browser/ui/webui/lens/lens_ui.h"

#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/webui/webui_util.h"
#include "chrome/common/webui_url_constants.h"
#include "chrome/grit/lens_resources.h"
#include "chrome/grit/lens_resources_map.h"
#include "content/public/browser/web_contents.h"
#include "content/public/browser/web_ui.h"
#include "content/public/browser/web_ui_data_source.h"

LensUI::LensUI(content::WebUI* web_ui) : WebUIController(web_ui) {
  // Set up the chrome://lens source.
  content::WebUIDataSource* html_source =
      content::WebUIDataSource::CreateAndAdd(Profile::FromWebUI(web_ui),
                                             chrome::kChromeUILensHost);
  webui::SetupWebUIDataSource(
      html_source, base::make_span(kLensResources, kLensResourcesSize),
      IDR_LENS_REGION_SEARCH_HTML);

  // Set up Content Security Policy (CSP) for chrome-untrusted://lens iframe.
  web_ui->AddRequestableScheme(content::kChromeUIUntrustedScheme);
  // Allow chrome-untrusted://lens page to load as an iframe in the page.
  std::string frame_src = base::StringPrintf("frame-src %s 'self';",
                                             chrome::kChromeUILensUntrustedURL);
  html_source->OverrideContentSecurityPolicy(
      network::mojom::CSPDirectiveName::FrameSrc, frame_src);
}

LensUI::~LensUI() = default;