// 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_ASH_VIDEO_CONFERENCE_VIDEO_CONFERENCE_MANAGER_ASH_H_
#define CHROME_BROWSER_ASH_VIDEO_CONFERENCE_VIDEO_CONFERENCE_MANAGER_ASH_H_

#include <cstdint>
#include <map>
#include <string>
#include <vector>

#include "ash/system/video_conference/video_conference_common.h"
#include "base/functional/callback.h"
#include "base/memory/raw_ptr.h"

namespace base {
class UnguessableToken;
}  // namespace base

class CaptureModeVideoConferenceBrowserTests;

namespace ash {

class VideoConferenceTrayController;
struct VideoConferenceMediaState;

// VideoConferenceManagerAsh is the central hub responsible for:
// 1. Connecting VC clients to System UI components.
// 2. Being the single source of truth for VC parameters like
//    background blur, portrait relighting, apps using audio/camera, etc.
//    and providing this information to the UI as needed.
class VideoConferenceManagerAsh : public VideoConferenceManagerBase {
 public:
  // Gets the singleton instance.
  static VideoConferenceManagerAsh* Get();

  VideoConferenceManagerAsh();

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

  ~VideoConferenceManagerAsh() override;

  // VideoConferenceManagerBase overrides.
  void GetMediaApps(base::OnceCallback<void(MediaApps)>) override;
  void ReturnToApp(const base::UnguessableToken& id) override;
  void SetSystemMediaDeviceStatus(VideoConferenceMediaDevice device,
                                  bool enabled) override;
  void CreateBackgroundImage() override;

  // Registers a client. Clients need to manually call |UnregisterClient|,
  // e.g. inside their destructor.
  void RegisterCppClient(VideoConferenceManagerClient* client,
                         const base::UnguessableToken& client_id);

  void NotifyMediaUsageUpdate(VideoConferenceMediaUsageStatus status,
                              base::OnceCallback<void(bool)> callback);
  void NotifyDeviceUsedWhileDisabled(VideoConferenceMediaDevice device,
                                     const std::u16string& app_name,
                                     base::OnceCallback<void(bool)> callback);
  void NotifyClientUpdate(VideoConferenceClientUpdate update);

  // Removes entry corresponding to |client_id| from
  // |client_info_map_|. Called by the destructor of
  // cpp clients (ash browser, ARC++).
  void UnregisterClient(const base::UnguessableToken& client_id);

 protected:
  // Returns aggregated |VideoConferenceMediaState| from all clients. Allows
  // test code to get the manager's aggregated state.
  VideoConferenceMediaState GetAggregatedState();

  // Aggregates and sends updated state to VcUiController. Allows mock classes
  // extending |VideoConferenceManagerAsh| to make assertions on data sent to
  // VcUiController by overriding this method.
  virtual void SendUpdatedState();

  // Returns the `VideoConferenceTrayController`.
  VideoConferenceTrayController* GetTrayController();

 private:
  friend class ::CaptureModeVideoConferenceBrowserTests;
  friend class VideoConferenceAshfeatureClientTest;
  friend class VideoConferenceAppServiceClientTest;

  struct ClientInfo {
    raw_ptr<VideoConferenceManagerClient> client = nullptr;
    VideoConferenceMediaState state;
  };

  // A (client_id, client_info) entry is inserted into this map
  // whenever a new client is registered on the manager and deleted
  // upon destruction of the client.
  std::map<base::UnguessableToken, ClientInfo> client_info_map_;
};

}  // namespace ash

#endif  // CHROME_BROWSER_ASH_VIDEO_CONFERENCE_VIDEO_CONFERENCE_MANAGER_ASH_H_