chrome/browser/optimization_guide/optimization_guide_internals_ui.h

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

#ifndef CHROME_BROWSER_OPTIMIZATION_GUIDE_OPTIMIZATION_GUIDE_INTERNALS_UI_H_
#define CHROME_BROWSER_OPTIMIZATION_GUIDE_OPTIMIZATION_GUIDE_INTERNALS_UI_H_

#include "base/functional/callback.h"
#include "components/optimization_guide/optimization_guide_internals/webui/optimization_guide_internals.mojom.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "ui/base/webui/resource_path.h"
#include "ui/webui/mojo_web_ui_controller.h"

class OptimizationGuideInternalsPageHandlerImpl;
namespace content {
class WebUI;
}  // namespace content

// The WebUI controller for chrome://optimization-guide-internals.
class OptimizationGuideInternalsUI
    : public ui::MojoWebUIController,
      public optimization_guide_internals::mojom::PageHandlerFactory {
 public:
  using SetupWebUIDataSourceCallback =
      base::OnceCallback<void(base::span<const webui::ResourcePath> resources,
                              int default_resource)>;

  // Constructs and returns an instance of this class if
  // OptimizationGuideKeyedService is valid, else returns nullptr.
  static OptimizationGuideInternalsUI* MaybeCreateOptimizationGuideInternalsUI(
      content::WebUI* web_ui,
      SetupWebUIDataSourceCallback set_up_data_source_callback);

  explicit OptimizationGuideInternalsUI(
      content::WebUI* web_ui,
      SetupWebUIDataSourceCallback set_up_data_source_callback);
  ~OptimizationGuideInternalsUI() override;

  OptimizationGuideInternalsUI(const OptimizationGuideInternalsUI&) = delete;
  OptimizationGuideInternalsUI& operator=(const OptimizationGuideInternalsUI&) =
      delete;

  void BindInterface(
      mojo::PendingReceiver<
          optimization_guide_internals::mojom::PageHandlerFactory> receiver);

 private:
  // optimization_guide_internals::mojom::PageHandlerFactory impls.
  void CreatePageHandler(
      mojo::PendingRemote<optimization_guide_internals::mojom::Page> page)
      override;
  void RequestDownloadedModelsInfo(
      RequestDownloadedModelsInfoCallback callback) override;

  std::unique_ptr<OptimizationGuideInternalsPageHandlerImpl>
      optimization_guide_internals_page_handler_;
  mojo::Receiver<optimization_guide_internals::mojom::PageHandlerFactory>
      optimization_guide_internals_page_factory_receiver_{this};

  WEB_UI_CONTROLLER_TYPE_DECL();
};

#endif  // CHROME_BROWSER_OPTIMIZATION_GUIDE_OPTIMIZATION_GUIDE_INTERNALS_UI_H_