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

#ifndef REMOTING_HOST_IPC_VIDEO_FRAME_CAPTURER_H_
#define REMOTING_HOST_IPC_VIDEO_FRAME_CAPTURER_H_

#include <memory>

#include "base/memory/raw_ptr.h"
#include "base/memory/scoped_refptr.h"
#include "base/memory/weak_ptr.h"
#include "mojo/public/cpp/bindings/receiver.h"
#include "mojo/public/cpp/bindings/remote.h"
#include "remoting/host/mojom/desktop_session.mojom.h"
#include "remoting/protocol/desktop_capturer.h"
#include "third_party/webrtc/modules/desktop_capture/desktop_capturer.h"

namespace remoting {

class DesktopSessionProxy;
class IpcSharedBufferCore;

// Routes webrtc::DesktopCapturer calls though the IPC channel to the desktop
// session agent running in the desktop integration process.
// GetSourceList() and SelectSource() functions are not implemented, they always
// return false.
class IpcVideoFrameCapturer : public DesktopCapturer,
                              public mojom::VideoCapturerEventHandler {
 public:
  explicit IpcVideoFrameCapturer(
      scoped_refptr<DesktopSessionProxy> desktop_session_proxy);

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

  ~IpcVideoFrameCapturer() override;

  // Sets the new Mojo endpoints for the capturer.
  void OnCreateVideoCapturerResult(mojom::CreateVideoCapturerResultPtr result);

  // Returns a WeakPtr to this capturer. Used by DesktopSessionProxy to set
  // the Mojo endpoints for this capturer (on initial creation, and after a
  // desktop detach/attach sequence. A WeakPtr is needed since the lifetime of
  // this capturer is bound to the VideoStream.
  base::WeakPtr<IpcVideoFrameCapturer> GetWeakPtr();

  // webrtc::DesktopCapturer interface.
  void Start(Callback* callback) override;
  bool GetSourceList(SourceList* sources) override;
  bool SelectSource(SourceId id) override;
  void SetComposeEnabled(bool enabled) override;
  void SetMaxFrameRate(uint32_t max_frame_rate) override;
  void Pause(bool pause) override;
  void BoostCaptureRate(base::TimeDelta capture_interval,
                        base::TimeDelta duration) override;

  // mojom::VideoCapturerEventHandler interface.
  void OnSharedMemoryRegionCreated(int id,
                                   base::ReadOnlySharedMemoryRegion region,
                                   uint32_t size) override;
  void OnSharedMemoryRegionReleased(int id) override;
  void OnFrameCaptureStart(base::TimeTicks start_time) override;
  void OnCaptureResult(mojom::CaptureResultPtr result) override;

 private:
  typedef std::map<int, scoped_refptr<IpcSharedBufferCore>> SharedBuffers;

  // Called when the Mojo endpoint is disconnected. Cleans up shared buffers,
  // and sends fake responses to `callback_` where needed to keep the frame
  // scheduler in sync.
  void OnDisconnect();

  // Returns a shared buffer from the list of known buffers.
  scoped_refptr<IpcSharedBufferCore> GetSharedBufferCore(int id);

  // Points to the callback passed to webrtc::DesktopCapturer::Start().
  raw_ptr<webrtc::DesktopCapturer::Callback> callback_ = nullptr;

  // Mojo endpoint for sending capturer commands to the Desktop process.
  mojo::Remote<mojom::VideoCapturer> capturer_control_;

  // Mojo endpoint for receiving capturer events from the Desktop process.
  mojo::Receiver<mojom::VideoCapturerEventHandler> event_handler_{this};

  // Shared memory buffers by Id. Each buffer is owned by the corresponding
  // frame.
  SharedBuffers shared_buffers_;

  // Used by SelectSource() in the single-stream case. Changing the display
  // requires creating a new video-capturer in the Desktop process, and
  // connecting the new Mojo endpoints.
  scoped_refptr<DesktopSessionProxy> desktop_session_proxy_;

  // Used to cancel tasks pending on the capturer when it is stopped.
  base::WeakPtrFactory<IpcVideoFrameCapturer> weak_factory_{this};
};

}  // namespace remoting

#endif  // REMOTING_HOST_IPC_VIDEO_FRAME_CAPTURER_H_