chromium/chrome/browser/ui/webui/ash/cloud_upload/cloud_upload_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_UI_WEBUI_ASH_CLOUD_UPLOAD_CLOUD_UPLOAD_UI_H_
#define CHROME_BROWSER_UI_WEBUI_ASH_CLOUD_UPLOAD_CLOUD_UPLOAD_UI_H_

#include "chrome/browser/ui/webui/ash/cloud_upload/cloud_upload.mojom-shared.h"
#include "chrome/browser/ui/webui/ash/cloud_upload/cloud_upload.mojom.h"
#include "chrome/browser/ui/webui/ash/cloud_upload/cloud_upload_page_handler.h"
#include "chrome/common/webui_url_constants.h"
#include "content/public/browser/webui_config.h"
#include "content/public/common/url_constants.h"
#include "ui/web_dialogs/web_dialog_ui.h"
#include "ui/webui/resources/cr_components/color_change_listener/color_change_listener.mojom.h"

namespace ui {
class ColorChangeHandler;
}

namespace ash::cloud_upload {

class CloudUploadUI;

// WebUIConfig for chrome://cloud-upload
class CloudUploadUIConfig : public content::DefaultWebUIConfig<CloudUploadUI> {
 public:
  CloudUploadUIConfig()
      : DefaultWebUIConfig(content::kChromeUIScheme,
                           chrome::kChromeUICloudUploadHost) {}

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

// The UI for chrome://cloud-upload, used for uploading files to the cloud.
class CloudUploadUI : public ui::MojoWebDialogUI,
                      public mojom::PageHandlerFactory {
 public:
  explicit CloudUploadUI(content::WebUI* web_ui);
  CloudUploadUI(const CloudUploadUI&) = delete;
  CloudUploadUI& operator=(const CloudUploadUI&) = delete;

  ~CloudUploadUI() override;

  void SetDialogArgs(mojom::DialogArgsPtr args);

  // Instantiates implementor of the mojom::PageHandlerFactory
  // mojo interface passing the pending receiver that will be internally bound.
  void BindInterface(
      mojo::PendingReceiver<mojom::PageHandlerFactory> pending_receiver);

  void BindInterface(
      mojo::PendingReceiver<color_change_listener::mojom::PageHandler>
          receiver);

  // mojom::PageHandlerFactory:
  void CreatePageHandler(
      mojo::PendingReceiver<mojom::PageHandler> pending_page_handler) override;

 private:
  void RespondWithUserActionAndCloseDialog(mojom::UserAction action);
  void RespondWithLocalTaskAndCloseDialog(int task_position);

  mojom::DialogArgsPtr dialog_args_;
  std::unique_ptr<CloudUploadPageHandler> page_handler_;
  mojo::Receiver<mojom::PageHandlerFactory> factory_receiver_{this};
  std::unique_ptr<ui::ColorChangeHandler> color_provider_handler_;

  WEB_UI_CONTROLLER_TYPE_DECL();
};

}  // namespace ash::cloud_upload

#endif  // CHROME_BROWSER_UI_WEBUI_ASH_CLOUD_UPLOAD_CLOUD_UPLOAD_UI_H_