// 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 "components/optimization_guide/optimization_guide_internals/webui/url_constants.h"
#include "content/public/browser/internal_webui_config.h"
#include "content/public/common/url_constants.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/receiver.h"
#include "ui/base/webui/resource_path.h"
#include "ui/webui/mojo_web_ui_controller.h"

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

class OptimizationGuideInternalsUIConfig
    : public content::DefaultInternalWebUIConfig<OptimizationGuideInternalsUI> {
 public:
  OptimizationGuideInternalsUIConfig()
      : DefaultInternalWebUIConfig(
            optimization_guide_internals::
                kChromeUIOptimizationGuideInternalsHost) {}

  // content::WebUIConfig:
  bool IsWebUIEnabled(content::BrowserContext* browser_context) override;
};

// The WebUI controller for chrome://optimization-guide-internals.
class OptimizationGuideInternalsUI
    : public ui::MojoWebUIController,
      public optimization_guide_internals::mojom::PageHandlerFactory {
 public:
  explicit OptimizationGuideInternalsUI(content::WebUI* web_ui);
  ~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;
  void RequestLoggedModelQualityClientIds(
      RequestLoggedModelQualityClientIdsCallback callback) override;
  void RequestMqlsLogs(RequestMqlsLogsCallback 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_