// Copyright 2024 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_ASH_MAGIC_BOOST_MAGIC_BOOST_CONTROLLER_ASH_H_
#define CHROME_BROWSER_ASH_MAGIC_BOOST_MAGIC_BOOST_CONTROLLER_ASH_H_

#include "base/memory/weak_ptr.h"
#include "chromeos/crosapi/mojom/magic_boost.mojom.h"
#include "ui/views/widget/unique_widget_ptr.h"

namespace ash {

using OptInFeatures = crosapi::mojom::MagicBoostController::OptInFeatures;

// `MagicBoostControllerAsh` is the central point to deal with the ChromeOS -
// Chrome browser communication. it is responsible for showing the disclaimer UI
// and connect with Orca services in ash.
class MagicBoostControllerAsh : public crosapi::mojom::MagicBoostController {
 public:
  static MagicBoostControllerAsh* Get();

  MagicBoostControllerAsh();

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

  ~MagicBoostControllerAsh() override;

  // crosapi::mojom::MagicBoostController:
  void ShowDisclaimerUi(
      int64_t display_id,
      crosapi::mojom::MagicBoostController::TransitionAction action,
      OptInFeatures opt_in_features) override;
  void CloseDisclaimerUi() override;

  views::Widget* disclaimer_widget_for_test() {
    return disclaimer_widget_.get();
  }

 private:
  friend class MagicBoostControllerAshTest;

  // Called when the disclaimer view's accept button is clicked. `display_id`
  // indicates the display where the disclaimer view shows. `action` specifies
  // the action to take after the opt-in flow.
  void OnDisclaimerAcceptButtonPressed(
      int64_t display_id,
      crosapi::mojom::MagicBoostController::TransitionAction action);

  // Called when the diclaimer view's declination button is clicked.
  void OnDisclaimerDeclineButtonPressed();
  void OnLinkPressed(const std::string& url);

  views::UniqueWidgetPtr disclaimer_widget_;

  OptInFeatures opt_in_features_;

  base::WeakPtrFactory<MagicBoostControllerAsh> weak_ptr_factory_{this};
};

}  // namespace ash

#endif  // CHROME_BROWSER_ASH_MAGIC_BOOST_MAGIC_BOOST_CONTROLLER_ASH_H_