chrome/browser/ash/system_extensions/system_extensions_ui.cc

// Copyright 2021 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 "chrome/browser/ash/system_extensions/system_extensions_ui.h"

#include "base/logging.h"
#include "chrome/browser/ash/system_extensions/system_extensions_data_source.h"
#include "chrome/browser/profiles/profile.h"
#include "content/public/common/url_constants.h"

SystemExtensionsWebUIConfig::SystemExtensionsWebUIConfig(
    const SystemExtension& system_extension)
    : WebUIConfig(content::kChromeUIUntrustedScheme,
                  system_extension.base_url.host()),
      system_extension_id_(system_extension.id),
      system_extension_base_url_(system_extension.base_url) {}

SystemExtensionsWebUIConfig::~SystemExtensionsWebUIConfig() = default;

std::unique_ptr<content::WebUIController>
SystemExtensionsWebUIConfig::CreateWebUIController(content::WebUI* web_ui) {
  return std::make_unique<SystemExtensionsWebUIController>(
      web_ui, system_extension_id_, system_extension_base_url_);
}

SystemExtensionsWebUIController::SystemExtensionsWebUIController(
    content::WebUI* web_ui,
    const SystemExtensionId& system_extension_id,
    const GURL& system_extension_base_url)
    : ui::UntrustedWebUIController(web_ui) {
  auto* profile = Profile::FromWebUI(web_ui);
  auto data_source = std::make_unique<SystemExtensionsDataSource>(
      profile, system_extension_id, system_extension_base_url);
  content::URLDataSource::Add(profile, std::move(data_source));
}