// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROMEOS_ASH_COMPONENTS_BOCA_SPOTLIGHT_SPOTLIGHT_SESSION_MANAGER_H_
#define CHROMEOS_ASH_COMPONENTS_BOCA_SPOTLIGHT_SPOTLIGHT_SESSION_MANAGER_H_

#include <memory>
#include <optional>

#include "base/memory/raw_ref.h"
#include "base/memory/weak_ptr.h"
#include "base/sequence_checker.h"
#include "chromeos/ash/components/boca/boca_session_manager.h"
#include "chromeos/ash/components/boca/proto/session.pb.h"
#include "chromeos/ash/components/boca/spotlight/spotlight_crd_manager.h"
#include "chromeos/ash/components/boca/spotlight/spotlight_notification_handler.h"
#include "chromeos/ash/components/boca/spotlight/spotlight_service.h"

namespace ash::boca {

class SpotlightSessionManager : public boca::BocaSessionManager::Observer {
 public:
  // `boca_session_manager` must not be nullptr, and must outlive this.
  SpotlightSessionManager(
      BocaSessionManager* boca_session_manager,
      std::unique_ptr<SpotlightCrdManager> spotlight_crd_manager);
  // Constructor used for unit testing. By using this constructor we can rely on
  // a mock SpotlightService.
  SpotlightSessionManager(
      BocaSessionManager* boca_session_manager,
      std::unique_ptr<SpotlightNotificationHandler> notification_handler,
      std::unique_ptr<SpotlightCrdManager> spotlight_crd_manager,
      std::unique_ptr<SpotlightService> spotlight_service);
  SpotlightSessionManager(const SpotlightSessionManager&) = delete;
  SpotlightSessionManager& operator=(const SpotlightSessionManager&) = delete;
  ~SpotlightSessionManager() override;

  // BocaSessionManager::Observer:
  void OnSessionStarted(const std::string& session_id,
                        const ::boca::UserIdentity& producer) override;
  void OnSessionEnded(const std::string& session_id) override;
  void OnConsumerActivityUpdated(
      const std::map<std::string, ::boca::StudentStatus>& activities) override;

 private:
  SEQUENCE_CHECKER(sequence_checker_);

  void OnConnectionCodeReceived(const std::string& connection_code);
  void RegisterStudentScreen(const std::string& connection_code);
  void OnRegisterScreenRequestSent(
      base::expected<bool, google_apis::ApiErrorCode> result);
  // Handles showing the persistent notification after the Spotlight warning
  // countdown has ended.
  void OnCountdownEnded();

  const raw_ref<BocaSessionManager> boca_session_manager_;

  bool in_session_ = false;
  bool request_in_progress_ = false;
  std::string teacher_name_;
  std::string teacher_email_;
  const std::unique_ptr<SpotlightNotificationHandler> notification_handler_;
  const std::unique_ptr<SpotlightService> spotlight_service_;
  const std::unique_ptr<SpotlightCrdManager> spotlight_crd_manager_;

  base::WeakPtrFactory<SpotlightSessionManager> weak_ptr_factory_{this};
};
}  // namespace ash::boca

#endif  // CHROMEOS_ASH_COMPONENTS_BOCA_SPOTLIGHT_SPOTLIGHT_SESSION_MANAGER_H_